D-Net

package eu.dnetlib.enabling.tools.registration;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.List;

import javax.annotation.PostConstruct;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import org.springframework.stereotype.Component;

import eu.dnetlib.clients.ISLookUpClient;

import eu.dnetlib.clients.ISRegistryClient;

import eu.dnetlib.conf.DnetGenericApplicationProperties;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.exceptions.ISLookUpException;

import eu.dnetlib.services.BaseService;

@Component

public class ServiceRegistrationManager {

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

	@Autowired

	private ISLookUpClient isLookup;

	@Autowired

	private ISRegistryClient isRegistry;

	private boolean disabled = false;

	@Autowired(required = false)

	private List<BaseService> services = new ArrayList<>();

	@Autowired

	private DnetGenericApplicationProperties containerConfiguration;

	@PostConstruct

	public void registerAllServices() {

		if (disabled) { return; }

		disabled = true;

		services.stream()

				.filter(this::filterService)

				.forEach(this::registerService);

	}

	/**

	 * check if we have already a registered service profile for this service.

	 *

	 * @throws

	 */

	private boolean filterService(final BaseService service) {

		try {

			final String query = "for $x in //RESOURCE_PROFILE[.//RESOURCE_URI/@value='" + containerConfiguration.getUrl() + "']"

					+ " where contains($x//RESOURCE_TYPE/@value/string(), 'Service') return $x//RESOURCE_IDENTIFIER/@value/string()";

			final List<String> profIds = isLookup.find(query);

			if (profIds.isEmpty()) {

				log.debug("there is no service registered for service: " + service.getServiceType());

				return true;

			} else {

				service.setProfileId(profIds.get(0));

				return false;

			}

		} catch (final Throwable e) {

			log.error("Error filtering endpoint", e);

			disabled = false;

			return false;

		}

	}

	private void registerService(final BaseService service) {

		try {

			final DnetServiceType type = service.getServiceType();

			ensureSchemaExists(type);

			final String st = IOUtils.toString(getClass().getResourceAsStream("ServiceProfileTemplate.st"));

			final StringTemplate templ = new StringTemplate(st);

			final String resourceType = type + "ResourceType";

			templ.setAttribute("resourceType", resourceType);

			templ.setAttribute("serviceName", type);

			templ.setAttribute("address", containerConfiguration.getUrl());

			templ.setAttribute("protocols", null);

			templ.setAttribute("properties", null);

			log.debug("template: " + templ.toString());

			final String profId = isRegistry.register(templ.toString());

			log.info("registered profile for " + resourceType + ": " + profId);

			service.setProfileId(profId);

		} catch (final Throwable e) {

			log.error("Error registering endpoint", e);

			disabled = false;

		}

	}

	private void ensureSchemaExists(final DnetServiceType type) throws Exception {

		final String xq = "//*[local-name() = 'complexType' and @name = 'RESOURCE_TYPEType']//*[local-name() = 'enumeration' and @value = '" + type

				+ "ResourceType']";

		try {

			isLookup.findOne(xq);

			log.info("schema for " + type + " appears to exist");

		} catch (final ISLookUpException e) {

			final InputStream schemaStream = getClass().getResourceAsStream("ServiceProfileSchemaTemplate.st");

			if (schemaStream == null) { throw new IllegalStateException("cannot find service profile schema template"); }

			final StringTemplate schema = new StringTemplate(IOUtils.toString(schemaStream));

			final String resourceType = type + "ResourceType";

			schema.setAttribute("resourceType", resourceType);

			isRegistry.registerSchema(resourceType, schema.toString());

			log.info("registered schema for " + type);

		}

	}

	public boolean isDisabled() {

		return disabled;

	}

}