package eu.dnetlib.dhp.transformation.functions;

import java.util.Map;
import java.util.Optional;

import eu.dnetlib.dhp.transformation.vocabulary.Term;
import eu.dnetlib.dhp.transformation.vocabulary.Vocabulary;
import net.sf.saxon.s9api.*;
import scala.Serializable;

public class Cleaner implements ExtensionFunction, Serializable {

	private final Map<String, Vocabulary> vocabularies;

	public Cleaner(Map<String, Vocabulary> vocabularies) {
		this.vocabularies = vocabularies;
	}

	@Override
	public QName getName() {
		return new QName("http://eu/dnetlib/trasform/extension", "clean");
	}

	@Override
	public SequenceType getResultType() {
		return SequenceType.makeSequenceType(ItemType.STRING, OccurrenceIndicator.ONE_OR_MORE);
	}

	@Override
	public SequenceType[] getArgumentTypes() {
		return new SequenceType[] {
			SequenceType.makeSequenceType(ItemType.STRING, OccurrenceIndicator.ONE),
			SequenceType.makeSequenceType(ItemType.STRING, OccurrenceIndicator.ONE)
		};
	}

	@Override
	public XdmValue call(XdmValue[] xdmValues) throws SaxonApiException {
		final String currentValue = xdmValues[0].itemAt(0).getStringValue();
		final String vocabularyName = xdmValues[1].itemAt(0).getStringValue();
		Optional<Term> cleanedValue = vocabularies
			.get(vocabularyName)
			.getTerms()
			.stream()
			.filter(it -> it.getNativeName().equalsIgnoreCase(currentValue))
			.findAny();

		return new XdmAtomicValue(
			cleanedValue.isPresent() ? cleanedValue.get().getCode() : currentValue);
	}
}