D-Net

package eu.dnetlib.resolver.model;

import java.util.List;

import eu.dnetlib.dli.DLIUtils;

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

import org.apache.commons.lang3.StringEscapeUtils;

/**

 * The Class DLIRelation.

 */

public class ObjectRelation {

	/**

	 * The source doi.

	 */

	private String sourceRecordId;

	/**

	 * The source doi.

	 */

	private String sourcePid;

	/**

	 * The target type.

	 */

	private ObjectType targetType;

	/**

	 * The target pid.

	 */

	private PID targetPID;

	/**

	 * The target title.

	 */

	private String targetTitle;

	/**

	 * The authors.

	 */

	private List<String> authors;

	/**

	 * The relation semantics.

	 */

	private String relationSemantics;

	/**

	 * The relation provenance.

	 */

	private List<ObjectProvenance> relationProvenance;

	/**

	 * The status.

	 */

	private String completionStatus = "incomplete";

	private ResolvedObject extraInfo;

	/**

	 * Gets the source do

	 *

	 * @return the source doi

	 */

	public String getSourceRecordId() {

		return sourceRecordId;

	}

	/**

	 * Sets the source doi.

	 *

	 * @param sourceRecordId the new source record id

	 */

	public void setSourceRecordId(final String sourceRecordId) {

		if (sourceRecordId != null) {

			this.sourceRecordId = sourceRecordId;

		}

	}

	/**

	 * Gets the escaped xmldoi.

	 *

	 * @return the escaped xmldoi

	 */

	public String getEscapedXMLDOI() {

		return StringEscapeUtils.escapeXml(sourceRecordId);

	}

	/**

	 * Gets the target type.

	 *

	 * @return the target type

	 */

	public ObjectType getTargetType() {

		return targetType;

	}

	/**

	 * Sets the target type.

	 *

	 * @param targetType the new target type

	 */

	public void setTargetType(final ObjectType targetType) {

		this.targetType = targetType;

	}

	/**

	 * Gets the ID relation.

	 *

	 * @return the ID relation

	 */

	public String getIDRelation() {

		if ((this.getSourceRecordId() == null) || (this.getTargetPID() == null) || (this.getTargetPID().getId() == null))

			return null;

		return String.format("%s::%s", this.getSourceRecordId(), DnetXsltFunctions.md5(getTargetPID().getId().toLowerCase().trim()));

	}

	public String getIDRecord() {

		return (DnetXsltFunctions.md5(String.format("%s::%s", this.getTargetPID().getId(), this.getTargetPID().getType())));

	}

	/**

	 * Gets the target pid.

	 *

	 * @return the target pid

	 */

	public PID getTargetPID() {

		return targetPID;

	}

	/**

	 * Sets the target pid.

	 *

	 * @param targetPID the new target pid

	 */

	public void setTargetPID(final PID targetPID) {

		if (targetPID.getType() != null && "doi".equals(targetPID.getType().trim().toLowerCase()))

			targetPID.setId(DLIUtils.fixPID(targetPID.getId()));

		this.targetPID = targetPID;

	}

	/**

	 * Gets the target title.

	 *

	 * @return the target title

	 */

	public String getTargetTitle() {

		if (targetTitle == null) return null;

		String bf = targetTitle;

		bf = bf.replaceAll("&", "&");

		bf = bf.replaceAll("\"", """);

		bf = bf.replaceAll("'", "'");

		bf = bf.replaceAll("<", "&lt;");

		bf = bf.replaceAll(">", ">");

		return bf;

	}

	/**

	 * Sets the target title.

	 *

	 * @param targetTitle the new target title

	 */

	public void setTargetTitle(final String targetTitle) {

		this.targetTitle = targetTitle;

	}

	/**

	 * Gets the relation semantics.

	 *

	 * @return the relation semantics

	 */

	public String getRelationSemantics() {

		return relationSemantics;

	}

	/**

	 * Sets the relation semantics.

	 *

	 * @param relationSemantics the new relation semantics

	 */

	public void setRelationSemantics(final String relationSemantics) {

		this.relationSemantics = relationSemantics;

	}

	/**

	 * Gets the relation provenance.

	 *

	 * @return the relationProvenance

	 */

	public List<ObjectProvenance> getRelationProvenance() {

		return relationProvenance;

	}

	/**

	 * Sets the relation provenance.

	 *

	 * @param relationProvenance the relationProvenance to set

	 */

	public void setRelationProvenance(final List<ObjectProvenance> relationProvenance) {

		this.relationProvenance = relationProvenance;

	}

	/**

	 * Gets the completion status.

	 *

	 * @return the completion status

	 */

	public String getCompletionStatus() {

		return this.completionStatus;

	}

	/**

	 * Sets the completion status.

	 *

	 * @param completionStatus the new completion status

	 */

	public void setCompletionStatus(final String completionStatus) {

		this.completionStatus = completionStatus;

	}

	/**

	 * Gets the source pid.

	 *

	 * @return the source pid

	 */

	public String getSourcePid() {

		return sourcePid;

	}

	/**

	 * Sets the source pid.

	 *

	 * @param sourcePid the new source pid

	 */

	public void setSourcePid(final String sourcePid) {

		this.sourcePid = sourcePid;

	}

	/**

	 * {@inheritDoc}

	 *

	 * @see Object#toString()

	 */

	@Override

	public String toString() {

		String format =

				"Relation: \n\tId:%s \n\tSourcePid%s \n\tTargetPid:%s \n\tTargetPidType:%s \n\ttitle:%s \n\tCompletionStatus:%s \n\t Relation Semantic:%s \n\tProvenances:%s";

		return String.format(format, this.getIDRelation(), this.getSourcePid(), this.getTargetPID() == null ? "" : this.getTargetPID(),

				this.getTargetType(), this.getTargetTitle(), this.getCompletionStatus(), this.getRelationSemantics(),

				this.getRelationProvenance());

	}

	/**

	 * Gets the authors.

	 *

	 * @return the authors

	 */

	public List<String> getAuthors() {

		return authors;

	}

	/**

	 * Sets the authors.

	 *

	 * @param authors the new authors

	 */

	public void setAuthors(final List<String> authors) {

		this.authors = authors;

	}

	/**

	 * Gets the escaped xml authors.

	 *

	 * @return the escaped xml authors

	 */

	public List<String> getEscapedXMLAuthors() {

		if (authors != null)

			return ResolvedObject.escapedArray(authors);

		return null;

	}

	public ResolvedObject getExtraInfo() {

		return extraInfo;

	}

	public void setExtraInfo(final ResolvedObject extraInfo) {

		this.extraInfo = extraInfo;

	}

}