D-Net

package eu.dnetlib.dli.resolver.model.serializer;

import java.io.ByteArrayOutputStream;

import java.io.StringWriter;

import java.time.LocalDateTime;

import com.ximpleware.AutoPilot;

import com.ximpleware.VTDGen;

import com.ximpleware.VTDNav;

import com.ximpleware.XMLModifier;

import eu.dnetlib.dli.resolver.model.DLIResolvedObject;

import eu.dnetlib.pid.resolver.model.ObjectType;

import org.antlr.stringtemplate.NoIndentWriter;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.beans.factory.annotation.Required;

/**

 * Created by sandro on 9/21/16.

 */

public class ResolverSerializer {

	private static final Log log = LogFactory.getLog(ResolverSerializer.class);

	private StringTemplate dmfTemplate;

	private StringTemplate pmfTemplate;

	private StringTemplate scholixTemplate;

    public String serializeToXML(final DLIResolvedObject object) {

        StringWriter writer = new StringWriter(16);

        NoIndentWriter out = new NoIndentWriter(writer);

		if (object.getType() == null)

			object.setType(ObjectType.unknown);

		switch (object.getType()) {

		case publication: {

            return serialize(pmfTemplate, object, writer, out);

        }

		case dataset: {

            return serialize(dmfTemplate, object, writer, out);

        }

		case unknown: {

            return serialize(scholixTemplate, object, writer, out);

        }

        }

		return null;

	}

    protected String serialize(final StringTemplate template, final DLIResolvedObject object, final StringWriter writer, final NoIndentWriter out) {

        try {

            template.removeAttribute("object");

            template.setAttribute("object", object);

            template.write(out);

            String result = writer.toString();

            writer.close();

            return result;

        } catch (Throwable e) {

            throw new RuntimeException(e);

        }

    }

    public String serializeReplacingXML(final String xml, final DLIResolvedObject object) {

		try {

			VTDGen vg = new VTDGen(); // Instantiate VTDGen

			XMLModifier xm = new XMLModifier(); //Instantiate XMLModifier

			vg.setDoc(xml.getBytes());

			vg.parse(true);

			VTDNav vn = vg.getNav();

			xm.bind(vn);

			final AutoPilot ap = new AutoPilot(vn);

            ap.selectXPath("//*[local-name()='objIdentifier']");

            if (ap.evalXPath() != -1) {

                LocalDateTime currentTime = LocalDateTime.now();

                xm.insertAfterElement(String.format("\n<dri:resolvedDate>%s</dri:resolvedDate>", currentTime));

            }

			ap.selectXPath("//*[local-name()='about']");

			if (ap.evalXPath() != -1) {

				xm.remove();

			}

			ap.selectXPath("//*[local-name()='metadata']");

			if (ap.evalXPath() != -1) {

				xm.insertBeforeElement(serializeToXML(object));

				xm.remove();

			}

			final ByteArrayOutputStream bas = new ByteArrayOutputStream();

			xm.output(bas);

			return new String(bas.toByteArray());

		} catch (Throwable e) {

			if (log.isDebugEnabled()) {

				log.debug(String.format("Error on replacing xml: %s", xml));

			}

			return null;

		}

	}

	public StringTemplate getDmfTemplate() {

		return dmfTemplate;

	}

	@Required

	public void setDmfTemplate(final StringTemplate dmfTemplate) {

		this.dmfTemplate = dmfTemplate;

	}

	public StringTemplate getPmfTemplate() {

		return pmfTemplate;

	}

	@Required

	public void setPmfTemplate(final StringTemplate pmfTemplate) {

		this.pmfTemplate = pmfTemplate;

	}

	public StringTemplate getScholixTemplate() {

		return scholixTemplate;

	}

	@Required

	public void setScholixTemplate(final StringTemplate scholixTemplate) {

		this.scholixTemplate = scholixTemplate;

	}

}