package eu.dnetlib.dhp.oa.provision.model;

import java.io.Serializable;
import java.util.Map;
import java.util.Optional;

import com.google.common.base.Objects;
import com.google.common.collect.ComparisonChain;
import com.google.common.collect.Maps;

import eu.dnetlib.dhp.schema.common.ModelConstants;
import eu.dnetlib.dhp.schema.oaf.Relation;

public class SortableRelationKey implements Comparable<SortableRelationKey>, Serializable {

	private static final Map<Relation.SUBRELTYPE, Integer> weights = Maps.newHashMap();

	static {
		//TODO Claudio check why we need to have SUBRELTYPE AND RELTYPE
		weights.put(Relation.SUBRELTYPE.participation, 0);
		weights.put(Relation.SUBRELTYPE.outcome, 1);
		weights.put(Relation.SUBRELTYPE.affiliation, 2);
		weights.put(Relation.SUBRELTYPE.dedup, 3);
		//TODO COMMENTED BUT SHOULD BE REPLACED WITH RELATIONSHIPS??
//		weights.put(ModelConstants.PUBLICATION_DATASET, 4);
		weights.put(Relation.SUBRELTYPE.supplement, 5);
		weights.put(Relation.SUBRELTYPE.review, 6);
		weights.put(Relation.SUBRELTYPE.relationship, 7);
		weights.put(Relation.SUBRELTYPE.part, 8);
		weights.put(Relation.SUBRELTYPE.provision, 9);
		weights.put(Relation.SUBRELTYPE.version, 10);
		weights.put(Relation.SUBRELTYPE.similarity, 11);
		weights.put(Relation.SUBRELTYPE.citation, 12);
	}

	private static final long serialVersionUID = 3232323;

	private String groupingKey;

	private Relation.SUBRELTYPE subRelType;

	public static SortableRelationKey create(Relation r, String groupingKey) {
		SortableRelationKey sr = new SortableRelationKey();
		sr.setGroupingKey(groupingKey);
		sr.setSubRelType(r.getSubRelType());
		return sr;
	}

	@Override
	public boolean equals(Object o) {
		if (this == o)
			return true;
		if (o == null || getClass() != o.getClass())
			return false;
		SortableRelationKey that = (SortableRelationKey) o;
		return getGroupingKey().equals(that.getGroupingKey());
	}

	@Override
	public int hashCode() {
		return Objects.hashCode(getGroupingKey());
	}

	@Override
	public int compareTo(SortableRelationKey o) {
		return ComparisonChain
			.start()
			.compare(getGroupingKey(), o.getGroupingKey())
			.compare(getWeight(this), getWeight(o))
			.result();
	}

	private Integer getWeight(SortableRelationKey o) {
		return Optional.ofNullable(weights.get(o.getSubRelType())).orElse(Integer.MAX_VALUE);
	}

	public Relation.SUBRELTYPE getSubRelType() {
		return subRelType;
	}

	public void setSubRelType(Relation.SUBRELTYPE subRelType) {
		this.subRelType = subRelType;
	}

	public String getGroupingKey() {
		return groupingKey;
	}

	public void setGroupingKey(String groupingKey) {
		this.groupingKey = groupingKey;
	}

}