D-Net

package eu.dnetlib.miscutils.templates;

import java.io.StringReader;

import java.util.LinkedHashMap;

import java.util.Map;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.exceptions.DnetGenericException;

@Component

public class TemplateUtil {

	@Autowired

	private InformationServiceClient isClient;

	private static final Log log = LogFactory.getLog(TemplateUtil.class);

	public Map<String, String> processTemplateProfile(final String profileId, final Map<String, Object> map) throws DnetGenericException {

		final String profile = isClient.getProfile(profileId);

		try {

			final Document doc = (new SAXReader()).read(new StringReader(profile));

			final Map<String, String> res = new LinkedHashMap<>();

			for (final Object o : doc.selectNodes("//TEMPLATE")) {

				final String type = ((Node) o).valueOf("@type");

				final String language = ((Node) o).valueOf("@language");

				final String template = ((Node) o).getText();

				switch (language) {

				case "StringTemplate":

					final StringTemplate st = new StringTemplate(template);

					st.setAttributes(map);

					res.put(type, st.toString());

					break;

				default:

					log.error("Unknown template language: " + language);

					throw new DnetGenericException("Unknown template language: " + language);

				}

			}

			if (res.isEmpty()) {

				log.error("Empty result for template: " + profileId);

				throw new DnetGenericException("Empty result for template: " + profileId);

			}

			return res;

		} catch (final DocumentException e) {

			log.error("Error processing template: " + profileId, e);

			throw new DnetGenericException("Error processing template: " + profileId, e);

		}

	}

}