D-Net

package eu.dnetlib.parthenos.catalogue;

import java.io.BufferedOutputStream;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.net.URI;

import java.util.Iterator;

import javax.annotation.PostConstruct;

import com.fasterxml.jackson.core.JsonEncoding;

import com.fasterxml.jackson.core.JsonFactory;

import com.fasterxml.jackson.core.JsonGenerator;

import com.google.common.collect.Lists;

import eu.dnetlib.parthenos.CRM;

import eu.dnetlib.parthenos.CRMdig;

import eu.dnetlib.parthenos.CRMpe;

import eu.dnetlib.parthenos.publisher.ParthenosPublisherException;

import eu.dnetlib.parthenos.rdf.ResourceReader;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.http.client.utils.URIBuilder;

import org.apache.jena.ontology.OntModel;

import org.apache.jena.ontology.OntModelSpec;

import org.apache.jena.rdf.model.ModelFactory;

import org.apache.jena.rdf.model.Resource;

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

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

import org.springframework.http.*;

import org.springframework.stereotype.Component;

import org.springframework.web.client.RestTemplate;

import static eu.dnetlib.parthenos.CRM.*;

/**

 * Created by Alessia Bardi on 21/11/2017.

 *

 * @author Alessia Bardi

 */

@Component

public class CatalogueRegistrator {

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

	@Value("${gcube.catalogue.baseurl}")

	private String baseURL;

	@Value("${gcube.registry.application.token}")

	private String applicationToken;

	@Autowired

	private ResourceReader resourceReader;

	@Autowired

	private RestTemplate jrrRestTemplate;

	public String register(final Resource resource, final Resource type) throws IOException, ParthenosPublisherException {

		String resURI = resource.getURI();

		log.debug(String.format("Catalogue --> Processing resource : %s with type: %s", resURI, type.getLocalName()));

		String resCatName = resURI.substring(resURI.lastIndexOf("/") + 1);

		String json;

		switch(type.getLocalName()){

		case "E7_Activity":

			json = getJsonForActivity(resource, resCatName);

			break;

		case "E39_Actor":

			json = getJsonForActor(resource, resCatName);

			break;

		case "E70_Thing":

			json = getJsonForThing(resource, resCatName);

			break;

			//TODO: handle more tyopes, see JRRPublisher

		default:

			throw new IllegalArgumentException(String.format("Type "+type.getLocalName()+" not supported"));

		}

		String uuid = doRegister(json, resCatName);

		log.debug(String.format("%s registered on the catalogue with uuid: %s", resURI, uuid));

		return uuid;

	}

	protected String getJsonForActivity(final Resource res, final String resNameForCatalogue) throws IOException {

		JsonFactory jsonFactory = new JsonFactory();

		final ByteArrayOutputStream out = new ByteArrayOutputStream();

		BufferedOutputStream bos = new BufferedOutputStream(out);

		JsonGenerator jg = jsonFactory.createGenerator(bos, JsonEncoding.UTF8);

		jg.writeStartObject();

		writeCommonFields(jg, res, resNameForCatalogue);

		//TODO: from which linked E39_Actor?

		jg.writeStringField("maintainer", "");

		jg.writeStringField("maintainer_email", "");

		//the names of all superclasses of the entity

		jg.writeArrayFieldStart("tags");

		Iterator<String> classNames = resourceReader.getRDFClassNames(res);

		while (classNames.hasNext()) {

			jg.writeStartObject();

			jg.writeStringField("name", classNames.next());

			jg.writeEndObject();

		}

		jg.writeEndArray();

		//RI from which the entity has been collected, the source from which the RI collected the entity (if available).

		//TODO: from which linked E39_Actors?

		jg.writeArrayFieldStart("groups");

		jg.writeStartObject();

		jg.writeStringField("name", "");

		jg.writeEndObject();

		jg.writeStartObject();

		jg.writeStringField("name", "");

		jg.writeEndObject();

		jg.writeEndArray();

		jg.writeArrayFieldStart("extras");

		addExtra(jg, "system:type", E7_Activity.getLocalName());

		//specific class

		addExtra(jg, "instance of", resourceReader.findSpecificType(res, E7_Activity).getLocalName());

		if (res.getURI().startsWith("http://parthenos.d4science.org"))

			addExtra(jg, "Parthenos URL", res.getURI());

		addExtra(jg, "competence", res.getURI());

		//TODO: from which linked E39_Actors? Should be the same as for groups above

		addExtra(jg, "provided by", "");

		//TODO: get begin/end of operation from PP42_has_declarative_time

		addExtra(jg, "declared begin/end of operation", "");

		//TODO: where to get it?

		addExtra(jg, "last confirmation", "");

		//TODO: where to get it?

		addExtra(jg, "date of registration", "");

		addExtra(jg, "availability", resourceReader.getAvailability(res));

		addExtra(jg, "condition of use", resourceReader.getConditionOfUse(res));

		//TODO: where to get the contact person

		//Resource contactPerson = null;

		//jg.writeStringField("contact person", resourceReader.getTitle(contactPerson));

		//TODO: where to get it?

		addExtra(jg, "activity type", "");

		// can't add multiple fields with same name, hence skipping

		//jg.writeStringField("hosts", "");

		//TODO: where to get it?

		addExtra(jg, "online access point", "");

		//TODO: where to get it?

		addExtra(jg, "authorization", "");

		//TODO: where to get it?

		addExtra(jg, "protocol", "");

		//can't add multiple fields with same name, hence skipping

		//addExtra(jg,"curates", "");

		//TODO: where to get it?

		addExtra(jg, "runs on request", "");

		//TODO: where to get it?

		addExtra(jg, "delivers on request", "");

		jg.writeEndArray(); //end extras

		jg.writeEndObject();

		jg.close();

		return out.toString("UTF-8");

	}

	protected String getJsonForActor(final Resource res, final String resNameForCatalogue) throws IOException {

		JsonFactory jsonFactory = new JsonFactory();

		final ByteArrayOutputStream out = new ByteArrayOutputStream();

		BufferedOutputStream bos = new BufferedOutputStream(out);

		JsonGenerator jg = jsonFactory.createGenerator(bos, JsonEncoding.UTF8);

		jg.writeStartObject();

		writeCommonFields(jg, res, resNameForCatalogue);

		jg.writeArrayFieldStart("extras");

		addExtra(jg, "system:type", E39_Actor.getLocalName());

		//specific class

		addExtra(jg, "instance of", resourceReader.findSpecificType(res, E39_Actor).getLocalName());

		if (res.getURI().startsWith("http://parthenos.d4science.org"))

			addExtra(jg, "Parthenos URL", res.getURI());

		jg.writeEndArray();

		jg.writeEndObject();

		jg.close();

		return out.toString("UTF-8");

	}

	protected String getJsonForThing(final Resource res, final String resNameForCatalogue) throws IOException {

		JsonFactory jsonFactory = new JsonFactory();

		final ByteArrayOutputStream out = new ByteArrayOutputStream();

		BufferedOutputStream bos = new BufferedOutputStream(out);

		JsonGenerator jg = jsonFactory.createGenerator(bos, JsonEncoding.UTF8);

		jg.writeStartObject();

		writeCommonFields(jg, res, resNameForCatalogue);

		jg.writeArrayFieldStart("extras");

		addExtra(jg, "system:type", E70_Thing.getLocalName());

		//specific class

		addExtra(jg, "instance of", resourceReader.findSpecificType(res, E70_Thing).getLocalName());

		if (res.getURI().startsWith("http://parthenos.d4science.org")) {

			addExtra(jg, "Parthenos URL", res.getURI());

		}

		//TODO: things include digital objects, software, datasets, schema. Guess we should know what to add here.

		jg.writeEndArray();

		jg.writeEndObject();

		jg.close();

		return out.toString("UTF-8");

	}

	protected void addExtra(final JsonGenerator jg, final String key, final String value) throws IOException {

		jg.writeStartObject();

		jg.writeStringField("key", key);

		jg.writeStringField("value", value);

		jg.writeEndObject();

	}

	protected void writeCommonFields(final JsonGenerator jg, final Resource res, final String resNameForCatalogue) throws IOException {

		//end of URI

		jg.writeStringField("name", resNameForCatalogue);

		//default license

		jg.writeStringField("license_id", "notspecified");

		String title = resourceReader.getTitle(res);

		if (StringUtils.isBlank(title))

			title = resNameForCatalogue;

		jg.writeStringField("title", title);

		//description

		jg.writeStringField("notes", resourceReader.getDescription(res));

	}

	/**

	 * @param json

	 * @param resCatName

	 * @return the uuid assigned by the catalogue to the registered resource

	 * @throws ParthenosPublisherException

	 */

	private String doRegister(final String json, final String resCatName) throws ParthenosPublisherException {

		//TODO: implement me

		log.debug(String.format("Catalogue --> Registering %s : %s", resCatName, json));

		//if resCatName is already registered --> delete it and then saves it again

		HttpEntity<String> entity = new HttpEntity<String>(json, getHeaders());

		try {

			URI uri = new URIBuilder(getBaseURL()).addParameter("gcube-token", getApplicationToken()).build();

			ResponseEntity<String> res = jrrRestTemplate.<String>exchange(uri, HttpMethod.POST, entity, String.class);

			//TODO: I guess response is a json, need to check the format, parse it to get the uuid of the registered resource

			String uuid = res.getBody();

			return uuid;

		} catch (Throwable t) {

			throw new ParthenosPublisherException(t);

		}

	}

	private HttpHeaders getHeaders() {

		HttpHeaders headers = new HttpHeaders();

		headers.setContentType(MediaType.APPLICATION_JSON);

		headers.setAccept(Lists.newArrayList(MediaType.APPLICATION_JSON));

		//headers.set("gcube-token", getApplicationToken() );

		return headers;

	}

	public String getBaseURL() {

		return baseURL;

	}

	public void setBaseURL(final String baseURL) {

		this.baseURL = baseURL;

	}

	public String getApplicationToken() {

		return applicationToken;

	}

	public void setApplicationToken(final String applicationToken) {

		this.applicationToken = applicationToken;

	}

	public ResourceReader getResourceReader() {

		return resourceReader;

	}

	public void setResourceReader(final ResourceReader resourceReader) {

		this.resourceReader = resourceReader;

	}

}