package eu.dnetlib.dhp.oa.dedup;

import static org.apache.commons.lang3.StringUtils.substringAfter;
import static org.apache.commons.lang3.StringUtils.substringBefore;

import java.io.Serializable;
import java.util.List;

import eu.dnetlib.dhp.oa.dedup.model.Identifier;
import eu.dnetlib.dhp.schema.oaf.OafEntity;
import eu.dnetlib.dhp.schema.oaf.utils.PidType;

public class IdGenerator implements Serializable {

	// pick the best pid from the list (consider date and pidtype)
	public static <T extends OafEntity> String generate(List<Identifier<T>> pids, String defaultID) {
		if (pids == null || pids.size() == 0)
			return defaultID;

		Identifier<T> bp = pids
			.stream()
			.min(Identifier::compareTo)
			.get();

		String prefix = substringBefore(bp.getOriginalID(), "|");
		String ns = substringBefore(substringAfter(bp.getOriginalID(), "|"), "::");
		String suffix = substringAfter(bp.getOriginalID(), "::");

		final String pidType = substringBefore(ns, "_");
		if (PidType.isValid(pidType)) {
			return prefix + "|" + dedupify(ns) + "::" + suffix;
		} else {
			return prefix + "|dedup_wf_001::" + suffix;
		}
	}

	private static String dedupify(String ns) {
		StringBuilder prefix = new StringBuilder(substringBefore(ns, "_")).append("_dedup");
		while (prefix.length() < 12) {
			prefix.append("_");
		}
		return prefix.substring(0, 12);
	}

}