D-Net

package eu.dnetlib.data.mdstore.plugins.objects;

import java.io.StringReader;

import java.net.URI;

import java.util.concurrent.TimeUnit;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import javax.xml.bind.annotation.XmlAccessType;

import javax.xml.bind.annotation.XmlAccessorType;

import javax.xml.bind.annotation.XmlElement;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.http.impl.client.HttpClientBuilder;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.io.SAXReader;

import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;

import org.springframework.web.client.RestTemplate;

@XmlAccessorType(XmlAccessType.FIELD)

public class Project {

	@XmlElement(name = "infoId")

	private String infoId;

	@XmlElement(name = "openaireId")

	private String openaireId;

	@XmlElement(name = "code")

	private String code;

	@XmlElement(name = "name")

	private String name;

	@XmlElement(name = "acronym")

	private String acronym;

	@XmlElement(name = "funder")

	private String funder;

	@XmlElement(name = "program")

	private String program;

	@XmlElement(name = "jurisdiction")

	private String jurisdiction;

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

	private static final int MAX_NUMBER_OF_ATTEMPTS = 10;

	private static final int INTERVAL_MILLIS = 20000;

	private static final String PROJECT_REGEX = "info:eu-repo\\/grantAgreement\\/(.*)\\/(.*)\\/(.*)\\/(.*)\\/(.*)\\/(.*)";

	private static final HttpComponentsClientHttpRequestFactory httpRequestFactory = new HttpComponentsClientHttpRequestFactory(HttpClientBuilder

			.create()

			.setConnectionTimeToLive(0, TimeUnit.MILLISECONDS)

			.setMaxConnPerRoute(1)

			.setMaxConnTotal(1)

			.disableAutomaticRetries()

			.disableConnectionState()

			.build());

	public static Project newInstance(final URI url) {

		try {

			final SAXReader reader = new SAXReader();

			final Document doc = reader.read(new StringReader(fetchUrl(url, MAX_NUMBER_OF_ATTEMPTS)));

			final String openaireId = doc.valueOf("//*[local-name()='objIdentifier']");

			if (StringUtils.isBlank(openaireId)) {

				log.warn("Project not found using OpenAIRE API, url: " + url);

				return null;

			}

			final String code = doc.valueOf("//code");

			final String name = doc.valueOf("//title");

			final String acronym = doc.valueOf("//acronym");

			final String funder = doc.valueOf("//funder/shortname");

			final String program = doc.valueOf("//funding_level_0/name");

			final String jurisdiction = doc.valueOf("//funder/jurisdiction");

			return new Project(openaireId, code, name, acronym, funder, program, jurisdiction);

		} catch (final DocumentException e) {

			log.error("Error parsing document from url " + url);

			throw new RuntimeException(e);

		}

	}

	public static Project newInstance(final String infoId) {

		final Pattern pattern = Pattern.compile(PROJECT_REGEX);

		final Matcher matcher = pattern.matcher(infoId);

		if (matcher.find()) {

			final String openaireId = "";

			final String funder = matcher.group(1);

			final String program = matcher.group(2);

			final String code = matcher.group(3);

			final String jurisdiction = matcher.group(4);

			final String name = StringUtils.defaultIfBlank(matcher.group(5), funder + "/" + program + "/" + code);

			final String acronym = StringUtils.defaultIfBlank(matcher.group(6), name);

			if (StringUtils.isNotEmpty(code) && StringUtils.isNotEmpty(program)

					&& StringUtils.isNotEmpty(funder)) { return new Project(openaireId, code, name, acronym, funder, program, jurisdiction); }

		}

		log.warn("Invalid project ID: " + infoId);

		return null;

	}

	public static boolean isValid(final String infoId) {

		return Project.newInstance(infoId) != null;

	}

	private static String fetchUrl(final URI url, final int attempts) {

		if (attempts == 0) { throw new RuntimeException("Max number of attempts reached, downloading url: " + url); }

		/*

		 * try (CloseableHttpClient client = HttpClientBuilder.create().build()) { log.debug("Invoking url: " + url); final HttpGet request

		 * = new HttpGet(url); request.setHeader("Connection", "close"); request.setProtocolVersion(HttpVersion.HTTP_1_0); try (final

		 * CloseableHttpResponse response = client.execute(request)) { return EntityUtils.toString(response.getEntity()); }

		 */

		try {

			final RestTemplate restTemplate = new RestTemplate();

			restTemplate.setRequestFactory(httpRequestFactory);

			return restTemplate.getForObject(url, String.class);

		} catch (

		final Exception e) {

			try {

				log.error("Error downloading url: " + url + " - " + e.getMessage());

				Thread.sleep(INTERVAL_MILLIS);

				return fetchUrl(url, attempts - 1);

			} catch (final InterruptedException e1) {

				throw new RuntimeException(e1);

			}

		}

	}

	public Project() {}

	public Project(final String openaireId, final String code, final String name, final String acronym, final String funder, final String program,

			final String jurisdiction) {

		this.openaireId = openaireId;

		this.code = code;

		this.name = name;

		this.acronym = acronym;

		this.funder = funder;

		this.program = program;

		this.jurisdiction = jurisdiction;

	}

	public String getOpenaireId() {

		return openaireId;

	}

	public String getCode() {

		return code;

	}

	public String getName() {

		return name;

	}

	public String getAcronym() {

		return acronym;

	}

	public String getFunder() {

		return funder;

	}

	public String getProgram() {

		return program;

	}

	public String getJurisdiction() {

		return jurisdiction;

	}

	public boolean match(final String infoId) {

		final Project p = Project.newInstance(infoId);

		return ((p != null) && funder.equals(p.funder) && program.equals(p.program) && code.equals(p.code));

	}

	@Override

	public String toString() {

		return "Project [openaireId=" + openaireId + ", code=" + code + ", name=" + name + ", acronym=" + acronym + ", funder=" + funder + ", program="

				+ program + ", jurisdiction=" + jurisdiction + "]";

	}

	@Override

	public int hashCode() {

		final int prime = 31;

		int result = 1;

		result = (prime * result) + ((code == null) ? 0 : code.hashCode());

		result = (prime * result) + ((funder == null) ? 0 : funder.hashCode());

		result = (prime * result) + ((program == null) ? 0 : program.hashCode());

		return result;

	}

	@Override

	public boolean equals(final Object obj) {

		if (this == obj) { return true; }

		if (obj == null) { return false; }

		if (getClass() != obj.getClass()) { return false; }

		final Project other = (Project) obj;

		if (code == null) {

			if (other.code != null) { return false; }

		} else if (!code.equals(other.code)) { return false; }

		if (funder == null) {

			if (other.funder != null) { return false; }

		} else if (!funder.equals(other.funder)) { return false; }

		if (program == null) {

			if (other.program != null) { return false; }

		} else if (!program.equals(other.program)) { return false; }

		return true;

	}

	public String getInfoId() {

		return infoId;

	}

	public void setInfoId(final String infoId) {

		this.infoId = infoId;

	}

	public void setOpenaireId(final String openaireId) {

		this.openaireId = openaireId;

	}

	public void setCode(final String code) {

		this.code = code;

	}

	public void setName(final String name) {

		this.name = name;

	}

	public void setAcronym(final String acronym) {

		this.acronym = acronym;

	}

	public void setFunder(final String funder) {

		this.funder = funder;

	}

	public void setProgram(final String program) {

		this.program = program;

	}

	public void setJurisdiction(final String jurisdiction) {

		this.jurisdiction = jurisdiction;

	}

}