D-Net

package eu.dnetlib.enabling.tools.registration;

import java.io.InputStream;

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 eu.dnetlib.clients.ISLookUpClient;

import eu.dnetlib.clients.ISRegistryClient;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.exceptions.ISLookUpException;

import eu.dnetlib.services.BaseService;

public class ServiceRegistrator {

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

	@Autowired

	private ISLookUpClient isLookup;

	@Autowired

	private ISRegistryClient isRegistry;

	/**

	 * component which finds an hnm profile.

	 *

	 * @param url

	 */

	public String registerService(final BaseService service, final String url) throws Exception {

		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", url);

		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);

		return profId;

	}

	/**

	 * Check that the service schema for this service already exists, and create it if it doesn't.

	 *

	 * @param type

	 *            service name

	 */

	protected 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) {

			registerServiceSchema(type);

		}

	}

	private void registerServiceSchema(final DnetServiceType type) throws Exception {

		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);

	}

}