D-Net

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

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

import com.google.common.base.Splitter;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.enabling.datasources.common.Api;

import eu.dnetlib.enabling.datasources.common.Datasource;

import eu.dnetlib.enabling.datasources.common.LocalDatasourceManager;

import eu.dnetlib.msro.rmi.MSROException;

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

import eu.dnetlib.msro.workflows.util.WorkflowsConstants;

public class VerifyDatasourceJobNode extends SimpleJobNode {

	@Autowired

	private LocalDatasourceManager<Datasource<?, ?>, Api<?>> dsManager;

	private String expectedInterfaceTypologyPrefixes;

	private String expectedCompliancePrefixes;

	@Override

	protected String execute(final NodeToken token) throws Exception {

		final String dsId = token.getFullEnv().getAttribute(WorkflowsConstants.DATAPROVIDER_ID);

		final String ifaceId = token.getFullEnv().getAttribute(WorkflowsConstants.DATAPROVIDER_INTERFACE);

		final Api<?> api = dsManager.getApis(dsId)

			.stream()

			.filter(a -> a.getId().equals(ifaceId))

			.findFirst()

			.orElseThrow(() -> new MSROException("Intervace not found: " + ifaceId));;

		final String compliance = api.getCompatibility();

		final String typology = api.getTypology();

		verifyValue(compliance, expectedCompliancePrefixes);

		verifyValue(typology, expectedInterfaceTypologyPrefixes);

		token.getFullEnv().setAttribute(WorkflowsConstants.DATAPROVIDER_INTERFACE_COMPLIANCE, compliance);

		return Arc.DEFAULT_ARC;

	}

	private void verifyValue(final String value, final String expected) throws Exception {

		if (expected != null && !expected.isEmpty()) {

			for (final String s : Splitter.on(",").omitEmptyStrings().trimResults().split(expected)) {

				if (value.toLowerCase().startsWith(s.toLowerCase())) { return; }

			}

			throw new MSROException("Invalid value: " + value + ", Valid term prefixes are: [" + expected + "]");

		}

	}

	public String getExpectedInterfaceTypologyPrefixes() {

		return expectedInterfaceTypologyPrefixes;

	}

	public void setExpectedInterfaceTypologyPrefixes(final String expectedInterfaceTypologyPrefixes) {

		this.expectedInterfaceTypologyPrefixes = expectedInterfaceTypologyPrefixes;

	}

	public String getExpectedCompliancePrefixes() {

		return expectedCompliancePrefixes;

	}

	public void setExpectedCompliancePrefixes(final String expectedCompliancePrefixes) {

		this.expectedCompliancePrefixes = expectedCompliancePrefixes;

	}

}