D-Net

package eu.dnetlib.msro.openaireplus.workflows.index;

import java.io.IOException;

import java.io.InputStream;

import java.io.StringReader;

import java.io.StringWriter;

import javax.xml.stream.XMLStreamException;

import javax.xml.transform.Transformer;

import javax.xml.transform.TransformerException;

import javax.xml.transform.TransformerFactory;

import javax.xml.transform.stream.StreamResult;

import javax.xml.transform.stream.StreamSource;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;

import eu.dnetlib.functionality.index.solr.feed.StreamingInputDocumentFactory;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.miscutils.functional.xml.ApplyXslt;

import eu.dnetlib.miscutils.functional.xml.IndentXmlString;

import org.apache.commons.io.IOUtils;

import org.apache.solr.common.SolrInputDocument;

import org.junit.Test;

import org.springframework.core.io.ClassPathResource;

import static org.junit.Assert.assertFalse;

import static org.junit.Assert.assertNotNull;

public class OpenaireLayoutToRecordStylesheetTest {

	private static final String OPENAIRE_LAYOUT_TO_RECORD_STYLESHEET_XSL = "/eu/dnetlib/msro/openaireplus/workflows/index/openaireLayoutToRecordStylesheet.xsl";

	private static final String MDFORMAT_FIELDS = "/eu/dnetlib/msro/openaireplus/workflows/index/fields.xml";

	private static final String OAF_RECORD = "/eu/dnetlib/msro/openaireplus/workflows/index/oafRecord.xml";

	@Test

	public void test1() throws ISLookUpException, IOException, TransformerException, XMLStreamException {

		String xsl = prepareXslt("DMF");

		assertNotNull(xsl);

		assertFalse(xsl.isEmpty());

		System.out.println(xsl);

		ApplyXslt xslt = new ApplyXslt(xsl);

		String indexRecord = xslt.evaluate(readFromClasspath(OAF_RECORD));

		assertNotNull(indexRecord);

		assertFalse(indexRecord.isEmpty());

		System.out.println(IndentXmlString.apply(indexRecord));

		StreamingInputDocumentFactory factory = new StreamingInputDocumentFactory();

		SolrInputDocument doc = factory.parseDocument(DateUtils.now_ISO8601(), indexRecord, "dsId", "dnetResult");

		assertNotNull(doc);

		assertFalse(doc.isEmpty());

		// System.out.println(doc);

	}

	protected String prepareXslt(final String format) throws ISLookUpException, IOException, TransformerException {

		final TransformerFactory factory = TransformerFactory.newInstance();

		final Transformer layoutTransformer = factory.newTransformer(streamSource(OPENAIRE_LAYOUT_TO_RECORD_STYLESHEET_XSL));

		final StreamResult layoutToXsltXslt = new StreamResult(new StringWriter());

		layoutTransformer.setParameter("format", format);

		layoutTransformer.transform(streamSource(MDFORMAT_FIELDS), layoutToXsltXslt);

		return layoutToXsltXslt.getWriter().toString();

	}

	private StreamSource streamSource(final String s) throws IOException {

		return new StreamSource(new StringReader(readFromClasspath(s)));

	}

	private String readFromClasspath(final String s) throws IOException {

		ClassPathResource resource = new ClassPathResource(s);

		InputStream inputStream = resource.getInputStream();

		return IOUtils.toString(inputStream);

	}

}