D-Net

package eu.dnetlib.resolver;

import java.net.URLEncoder;

import com.google.common.collect.Lists;

import com.google.gson.JsonArray;

import com.google.gson.JsonElement;

import com.google.gson.JsonObject;

import com.google.gson.JsonParser;

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

import eu.dnetlib.resolver.model.ObjectProvenance;

import eu.dnetlib.resolver.model.ResolvedObject;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.core.io.ClassPathResource;

public class OpenaireResolver extends AbstractPIDResolver {

	private final static String templateURL = "objidentifier:\"%s\"";

	private final static String baseURL = "http://solr.openaire.eu:8983/solr/DMF-index-openaire_shard1_replica1/select?q=%s&wt=json";

	private static final Log log = LogFactory.getLog(OpenaireResolver.class); // NOPMD by marko on 11/24/08 5:02 PM

	private ApplyXslt xsltMapper = new ApplyXslt(new ClassPathResource("eu/dnetlib/resolver/xslt/openaire2Dli.xslt"));

	private OpenAireParser parser = new OpenAireParser();

	@Override

	protected boolean canResolvePid(final String pidType) {

		return (pidType != null && pidType.toLowerCase().equals("openaire"));

	}

	@Override

	protected ResolvedObject resolve(final String pid, final String pidType) {

		try {

			final String currentQuery = String.format(baseURL, URLEncoder.encode(String.format(templateURL, pid), "UTF-8"));

			ResolvedObject obj = parsingResponse(requestURL(currentQuery));

			if (obj == null)

				return null;

			obj.setPidType(pidType);

			obj.setPid(pid);

			ObjectProvenance provenance = new ObjectProvenance("dli::openaireResolver", "resolved", "complete", null, null, true);

			obj.fixContribution(provenance);

			obj.setDatasourceProvenance(Lists.newArrayList(provenance));

			return obj;

		} catch (Throwable e) {

			log.error("Error on resolve pid " + pid, e);

		}

		return null;

	}

	private ResolvedObject parsingResponse(final String response) {

		if (response == null) return null;

		JsonElement jElement = new JsonParser().parse(response);

		JsonObject jobject = jElement.getAsJsonObject();

		if (jobject.has("response")) {

			long total = jobject.get("response").getAsJsonObject().get("numFound").getAsLong();

			if (total == 0) return null;

			JsonArray hits = ((JsonObject) jobject.get("response")).get("docs").getAsJsonArray();

			JsonElement s = ((JsonObject) hits.get(0)).get("__result").getAsJsonArray().get(0);

			return parser.parseObject(s.getAsString());

		}

		return null;

	}

}