D-Net

package eu.dnetlib.msro.workflows.nodes.repohi;

import java.io.StringReader;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.stream.Collectors;

import org.dom4j.Document;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

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

import org.springframework.context.annotation.Scope;

import org.springframework.stereotype.Component;

import eu.dnetlib.clients.dsManager.DsManagerClient;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.clients.locators.ServiceClientFactory;

import eu.dnetlib.msro.annotations.ProcessNode;

import eu.dnetlib.msro.workflows.Arc;

import eu.dnetlib.msro.workflows.nodes.SimpleParallelProcessNode;

@Component

@Scope("prototype")

@ProcessNode("RemoveApiExtraFields")

public class RemoveApiExtraFieldsJobNode extends SimpleParallelProcessNode {

	private String datasourceId;

	private String datasourceInterface;

	private List<String> fields;

	@Autowired

	private ServiceClientFactory clientFactory;

	@Override

	protected String execute() throws Exception {

		if (fields != null) {

			final Set<String> invalidFields = fields.stream().map(String::toLowerCase).collect(Collectors.toSet());

			final Map<String, String> map = calculateValidExtraFields(datasourceId, datasourceInterface, invalidFields);

			clientFactory.getClient(DsManagerClient.class).bulkUpdateApiExtraFields(datasourceId, datasourceInterface, map);

		}

		return Arc.DEFAULT_ARC;

	}

	private Map<String, String> calculateValidExtraFields(final String repoId, final String ifaceId, final Set<String> invalidFields) throws Exception {

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

		final String profile = clientFactory.getClient(InformationServiceClient.class).getProfile(repoId);

		final SAXReader reader = new SAXReader();

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

		final Node ifcNode = doc.selectSingleNode("//INTERFACE[@id='" + ifaceId + "']");

		if (ifcNode != null) {

			for (final Object o : ifcNode.selectNodes("./INTERFACE_EXTRA_FIELD")) {

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

				if (!invalidFields.contains(name.toLowerCase())) {

					res.put(name, ((Node) o).getText());

				}

			}

		}

		return res;

	}

	public List<String> getFields() {

		return fields;

	}

	public void setFields(final List<String> fields) {

		this.fields = fields;

	}

	public String getDatasourceId() {

		return datasourceId;

	}

	public void setDatasourceId(final String datasourceId) {

		this.datasourceId = datasourceId;

	}

	public String getDatasourceInterface() {

		return datasourceInterface;

	}

	public void setDatasourceInterface(final String datasourceInterface) {

		this.datasourceInterface = datasourceInterface;

	}

}