D-Net

package eu.dnetlib.parthenos.registry;

import java.util.Map;

import com.google.common.collect.Maps;

import eu.dnetlib.parthenos.publisher.ParthenosPublisherException;

import eu.dnetlib.parthenos.publisher.SaxonHelper;

import net.sf.saxon.s9api.SaxonApiException;

import net.sf.saxon.s9api.Serializer;

import net.sf.saxon.s9api.XPathSelector;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.gcube.informationsystem.resourceregistry.client.ResourceRegistryClient;

import org.gcube.informationsystem.resourceregistry.client.ResourceRegistryClientFactory;

import org.gcube.informationsystem.resourceregistry.publisher.ResourceRegistryPublisher;

import org.gcube.informationsystem.resourceregistry.publisher.ResourceRegistryPublisherFactory;

/**

 * Created by Alessia Bardi on 26/09/2017.

 *

 * @author Alessia Bardi

 */

public class RegistryClient {

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

	private static final String OAI_NAMESPACE_URI = "http://www.openarchives.org/OAI/2.0/";

	private static final String DRI_NAMESPACE_URI = "http://www.driver-repository.eu/namespace/dri";

	private static final String RDF_NAMESPACE_URI = "http://www.w3.org/1999/02/22-rdf-syntax-ns#";

	private GCubeResourceRegistrator gCubeResourceRegistrator;

	private SaxonHelper saxonHelper;

	private XPathSelector xpathSelectorObjIdentifier;

	//	private XPathSelector xpathSelectorCollectionDate;

	//	private XPathSelector xpathSelectorTransformationDate;

	//	private XPathSelector xpathSelectorDatasourceName;

	//	private XPathSelector xpathSelectorDatasourceApi;

	private XPathSelector xpathSelectorRDF;

	private ResourceRegistryPublisher resourceRegistryPublisher;

	private ResourceRegistryClient resourceRegistryClient;

	private String registryBaseURL;

	private String defaultBaseURI;

	protected RegistryClient(final String registryBaseURL,

			final SaxonHelper saxonHelper,

			final String defaultBaseURI,

			final GCubeResourceRegistrator gCubeResourceRegistrator)

			throws ParthenosPublisherException {

		this.registryBaseURL = registryBaseURL;

		this.saxonHelper = saxonHelper;

		this.defaultBaseURI = defaultBaseURI;

		this.resourceRegistryPublisher = ResourceRegistryPublisherFactory.create();

		this.resourceRegistryClient = ResourceRegistryClientFactory.create();

		this.gCubeResourceRegistrator = gCubeResourceRegistrator;

		try {

			prepareXpathSelectors();

		} catch (SaxonApiException e) {

			throw new ParthenosPublisherException(e);

		}

	}

	public int unregister(final String datasourceInterface) {

		return gCubeResourceRegistrator.unregister(datasourceInterface);

	}

	public void register(final String record) throws ParthenosPublisherException {

		try {

			if (StringUtils.isBlank(record)) {

				log.warn("Got empty record");

			}

			String objIdentifier = extractFromRecord(record, xpathSelectorObjIdentifier);

			if (StringUtils.isBlank(objIdentifier)) {

				log.warn("Got record with no objIdentifier -- skipping");

			}

			String recordRDF = extractFromRecord(record, xpathSelectorRDF);

			gCubeResourceRegistrator.register(recordRDF, objIdentifier);

		} catch (Throwable e) {

			log.error(e.getMessage());

			throw new ParthenosPublisherException(e);

		}

	}

	private void prepareXpathSelectors() throws SaxonApiException {

		Map<String, String> namespaces = Maps.newHashMap();

		namespaces.put("oai", OAI_NAMESPACE_URI);

		namespaces.put("dri", DRI_NAMESPACE_URI);

		namespaces.put("rdf", RDF_NAMESPACE_URI);

		xpathSelectorObjIdentifier = this.saxonHelper.help().prepareXPathSelector("//oai:header/dri:objIdentifier/text()", namespaces);

		//xpathSelectorCollectionDate = this.saxonHelper.help().prepareXPathSelector("//oai:header/dri:dateOfCollection/text()", namespaces);

		//xpathSelectorTransformationDate = this.saxonHelper.help().prepareXPathSelector("//oai:header/dri:dateOfTransformation/text()", namespaces);

		//xpathSelectorDatasourceName = this.saxonHelper.help().prepareXPathSelector("//oai:header/dri:datasourcename/text()", namespaces);

		//xpathSelectorDatasourceApi = this.saxonHelper.help().prepareXPathSelector("//oai:header/dri:datasourceapi/text()", namespaces);

		xpathSelectorRDF = this.saxonHelper.help().prepareXPathSelector("//oai:metadata/rdf:RDF", namespaces);

	}

	private String extractFromRecord(final String record, final XPathSelector xPathSelector) {

		try {

			return this.saxonHelper.help().setSerializerProperty(Serializer.Property.OMIT_XML_DECLARATION, "yes").evaluateSingleAsString(record, xPathSelector);

		} catch (SaxonApiException e) {

			throw new RuntimeException("Cannot extract content ", e);

		}

	}

	public void setSaxonHelper(final SaxonHelper saxonHelper) {

		this.saxonHelper = saxonHelper;

	}

	public String getRegistryBaseURL() {

		return registryBaseURL;

	}

	public void setRegistryBaseURL(final String registryBaseURL) {

		this.registryBaseURL = registryBaseURL;

	}

	public void setDefaultBaseURI(final String defaultBaseURI) {

		this.defaultBaseURI = defaultBaseURI;

	}

	public SaxonHelper getSaxonHelper() {

		return saxonHelper;

	}

	public String getDefaultBaseURI() {

		return defaultBaseURI;

	}

	public GCubeResourceRegistrator getgCubeResourceRegistrator() {

		return gCubeResourceRegistrator;

	}

	public void setgCubeResourceRegistrator(final GCubeResourceRegistrator gCubeResourceRegistrator) {

		this.gCubeResourceRegistrator = gCubeResourceRegistrator;

	}

	public ResourceRegistryPublisher getResourceRegistryPublisher() {

		return resourceRegistryPublisher;

	}

	public void setResourceRegistryPublisher(final ResourceRegistryPublisher resourceRegistryPublisher) {

		this.resourceRegistryPublisher = resourceRegistryPublisher;

	}

	public ResourceRegistryClient getResourceRegistryClient() {

		return resourceRegistryClient;

	}

	public void setResourceRegistryClient(final ResourceRegistryClient resourceRegistryClient) {

		this.resourceRegistryClient = resourceRegistryClient;

	}

}