D-Net

package eu.dnetlib.openaire.exporter.funders.context;

import java.io.StringReader;

import java.util.List;

import java.util.Queue;

import java.util.stream.Collectors;

import eu.dnetlib.openaire.exporter.funders.model.FunderDetails;

import org.apache.commons.lang3.BooleanUtils;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.Element;

import org.dom4j.io.SAXReader;

public class MappingUtils {

	public static Context parseContext(final String s, final Queue<Throwable> errors) {

		final SAXReader reader = new SAXReader();

		final Document doc;

		try {

			doc = reader.read(new StringReader(s));

		} catch (DocumentException e) {

			errors.add(e);

			return new Context();

		}

		try {

			final Element eContext = (Element) doc.selectSingleNode("/RESOURCE_PROFILE/BODY/CONFIGURATION/context");

			return new Context()

					.setId(eContext.attributeValue("id"))

					.setLabel(eContext.attributeValue("label"))

					.setType(eContext.attributeValue("type"))

					.setParams(parseParams(eContext))

					.setCategories(parseCategories(eContext));

		} catch (Throwable e) {

			errors.add(e);

			return new Context();

		}

	}

	private static List<Category> parseCategories(final Element eContext) {

		final List<Element> eCategory = (List<Element>) eContext.selectNodes("//category");

		return eCategory.stream()

				.map(eCat -> new Category()

						.setClaim(BooleanUtils.toBooleanObject(eCat.attributeValue("claim")))

						.setId(eCat.attributeValue("id"))

						.setLabel(eCat.attributeValue("label"))

						.setParams(parseParams(eCat))

						.setConcepts(parseConcepts(eCat)))

				.collect(Collectors.toList());

	}

	private static List<Concept> parseConcepts(final Element eCategory) {

		final List<Element> eConcepts = (List<Element>) eCategory.selectNodes("./concept");

		return eConcepts.stream()

				.map(eCon -> new Concept()

						.setClaim(BooleanUtils.toBooleanObject(eCon.attributeValue("claim")))

						.setId(eCon.attributeValue("id"))

						.setLabel(eCon.attributeValue("label"))

						.setParams(parseParams(eCon))

						.setConcepts(parseConcepts(eCon)))

				.collect(Collectors.toList());

	}

	private static List<Param> parseParams(final Element e) {

		final List<Element> params = (List<Element>) e.selectNodes("./param");

		return params.stream()

				.map(p -> new Param()

						.setName(p.attributeValue("name"))

						.setValue(p.getTextTrim()))

				.collect(Collectors.toList());

	}

	public static FunderDetails asFunderDetails(final Context c) {

		return new FunderDetails()

				.setId(c.getId())

				.setName(c.getLabel())

				.setShortname(c.getId());

	}

}