D-Net

package eu.dnetlib.enabling.datasources;

import java.io.StringReader;

import java.text.ParseException;

import java.util.Date;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.io.SAXReader;

import org.quartz.CronExpression;

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

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

import org.springframework.core.io.ClassPathResource;

import org.springframework.core.io.Resource;

import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;

import com.google.common.collect.ImmutableMap;

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

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

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

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

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

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

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpDocumentNotFoundException;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;

import eu.dnetlib.enabling.is.registry.rmi.ISRegistryService;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

public class DatasourceManagerClients {

	private static final String REPOSITORY_SERVICE_RESOURCE_TYPE = "RepositoryServiceResourceType";

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

	private static final Resource isDefinedParamQuery = new ClassPathResource("/eu/dnetlib/enabling/datasources/queries/isDefinedParam.sql");

	private static final Resource dsQuery = new ClassPathResource("/eu/dnetlib/enabling/datasources/queries/getDatasource.sql");

	private static final Resource dsIdentitiesQuery = new ClassPathResource("/eu/dnetlib/enabling/datasources/queries/dsIdentitiesQuery.sql");

	private static final Resource dsOrganizationsQuery = new ClassPathResource("/eu/dnetlib/enabling/datasources/queries/dsOrganizationsQuery.sql");

	private static final Resource listApisByDsId = new ClassPathResource("/eu/dnetlib/enabling/datasources/queries/listApisByDsId.sql");

	private NamedParameterJdbcTemplate jdbcTemplate;

	public enum AfterSqlUpdate {

		DELETE_DS_PROFILE, UPDATE_DS_PROFILE, NONE

	}

	@Autowired

	private UniqueServiceLocator serviceLocator;

	public String findDatasourceId(final String profileId) throws DatasourceManagerException {

		try {

			return serviceLocator.getService(ISLookUpService.class).getResourceProfileByQuery(

					"/*[.//RESOURCE_IDENTIFIER/@value='" + profileId + "']//EXTRA_FIELDS/FIELD[./key='OpenAireDataSourceId']/value/text()");

		} catch (final Exception e) {

			log.error("Error finding dsId of profile " + profileId, e);

			throw new DatasourceManagerException(-1, "Error finding dsId of profile " + profileId, e);

		}

	}

	public String getDatasourceProfile(final String dsId) throws DatasourceManagerException {

		try {

			return serviceLocator.getService(ISLookUpService.class)

					.getResourceProfileByQuery(

							"collection('/db/DRIVER/RepositoryServiceResources/RepositoryServiceResourceType')/*[.//EXTRA_FIELDS/FIELD[./key='OpenAireDataSourceId']/value/text() = '"

									+ dsId + "']");

		} catch (final Exception e) {

			return null;

		}

	}

	public boolean deleteProfile(final String dsId) throws DatasourceManagerException {

		try {

			final SAXReader reader = new SAXReader();

			final String profile = getDatasourceProfile(dsId);

			if (profile != null) {

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

				final String profId = docOld.valueOf("//RESOURCE_IDENTIFIER/@value");

				serviceLocator.getService(ISRegistryService.class).deleteProfile(profId);

			}

			return true;

		} catch (final Exception e) {

			log.error("Error deleting profile", e);

			throw new DatasourceManagerException(-1, "Error deleting profile", e);

		}

	}

	public boolean regenerateProfile(final String dsId) throws DatasourceManagerException {

		final Datasource<Organization<?>, Identity> ds = getDatasourceById(dsId);

		final List<Api<ApiParam>> apis = getApis(dsId);

		try {

			final ISRegistryService registry = serviceLocator.getService(ISRegistryService.class);

			final String oldProfile = getDatasourceProfile(dsId);

			if (oldProfile != null) {

				final Document docOld = new SAXReader().read(new StringReader(oldProfile));

				final String profId = docOld.valueOf("//RESOURCE_IDENTIFIER/@value");

				final String profile = DatasourceFunctions.dsToProfile(ds, apis, profId);

				registry.updateProfile(profId, profile, REPOSITORY_SERVICE_RESOURCE_TYPE);

				log.info("Profile " + profId + " UPDATED for ds " + dsId);

			} else {

				final String profile = DatasourceFunctions.dsToProfile(ds, apis, "");

				final String profId = registry.registerProfile(profile);

				log.info("Valid Profile " + profId + " REGISTERED for ds " + dsId);

			}

			return true;

		} catch (final Exception e) {

			log.error("Error saving profile, id: " + dsId, e);

			throw new DatasourceManagerException(-1, "Error regenerating profile", e);

		}

	}

	public List<Map<String, Object>> searchSQL(final String sql, final Map<String, Object> sqlParams) throws DatasourceManagerException {

		try {

			log.debug("Executing SQL: " + sql);

			return jdbcTemplate.queryForList(sql, sqlParams);

		} catch (final Exception e) {

			log.error("Error executing sql", e);

			throw new DatasourceManagerException(-1, "Error obtaining datasources from db", e);

		}

	}

	public List<Map<String, Object>> searchSQL(final Resource sqlResource, final Map<String, Object> sqlParams) throws DatasourceManagerException {

		try {

			return searchSQL(IOUtils.toString(sqlResource.getInputStream()), sqlParams);

		} catch (final Exception e) {

			log.error("Error executing sql", e);

			throw new DatasourceManagerException(-1, "Error obtaining datasources from db", e);

		}

	}

	public void updateSQL(final String dsId, final String sql, final AfterSqlUpdate op, final Map<String, Object> sqlparams)

			throws DatasourceManagerException {

		log.debug("Executing query SQL: " + sql);

		jdbcTemplate.update(sql, sqlparams);

		switch (op) {

		case DELETE_DS_PROFILE:

			deleteProfile(dsId);

			break;

		case UPDATE_DS_PROFILE:

			regenerateProfile(dsId);

			break;

		default:

			break;

		}

	}

	public void updateSQL(final String dsId, final Resource sqlResource, final AfterSqlUpdate op, final Map<String, Object> sqlparams)

			throws DatasourceManagerException {

		try {

			updateSQL(dsId, IOUtils.toString(sqlResource.getInputStream()), op, sqlparams);

		} catch (final Exception e) {

			log.error("Error in updateSQL", e);

			throw new DatasourceManagerException(-1, "Error in updateSQL", e);

		}

	}

	public Datasource<Organization<?>, Identity> getDatasourceById(final String id) throws DatasourceManagerException {

		final List<Map<String, Object>> list = searchSQL(dsQuery, ImmutableMap.of("dsId", id));

		if (list.size() != 1) { throw new DatasourceManagerException("Invalid number of ds with id: " + id); }

		final Datasource<Organization<?>, Identity> ds = DatasourceFunctions.mapToDatasource(list.get(0));

		ds.setIdentities(searchSQL(dsIdentitiesQuery, ImmutableMap.of("dsId", id))

				.stream()

				.map(DatasourceFunctions::mapToDsIdentity)

				.collect(Collectors.toSet()));

		ds.setOrganizations(searchSQL(dsOrganizationsQuery, ImmutableMap.of("dsId", id))

				.stream()

				.map(DatasourceFunctions::mapToDsOrganization)

				.collect(Collectors.toSet()));

		return ds;

	}

	public List<Api<ApiParam>> getApis(final String dsId) throws DatasourceManagerException {

		return searchSQL(listApisByDsId, ImmutableMap.of("dsId", dsId))

				.stream()

				.map(DatasourceFunctions::mapToApi)

				.collect(Collectors.toList());

	}

	public boolean isDefinedParam(final String apiId, final String param) throws DatasourceManagerException {

		return !searchSQL(isDefinedParamQuery, ImmutableMap.of("apiId", apiId, "param", param)).isEmpty();

	}

	public Date findNextScheduledExecution(final String dsId, final String ifaceId) throws DatasourceManagerException {

		final String xquery = "/*[.//DATAPROVIDER/@interface='" + ifaceId + "' and .//SCHEDULING/@enabled='true']//CRON/text()";

		try {

			final String cronExpression = serviceLocator.getService(ISLookUpService.class).getResourceProfileByQuery(xquery);

			final CronExpression cron = new CronExpression(cronExpression);

			return cron.getNextValidTimeAfter(new Date());

		} catch (final ISLookUpDocumentNotFoundException e) {

			// When the value is not found a null value must be returned

			return null;

		} catch (final ISLookUpException e) {

			log.error("Error in xquery: " + xquery, e);

			throw new DatasourceManagerException(-1, "Error in xquery: " + xquery, e);

		} catch (final ParseException e) {

			log.error("Error parsing cron expression", e);

			throw new DatasourceManagerException(-1, "Error parsing cron expression", e);

		}

	}

	public UniqueServiceLocator getServiceLocator() {

		return serviceLocator;

	}

	@Required

	public void setServiceLocator(final UniqueServiceLocator serviceLocator) {

		this.serviceLocator = serviceLocator;

	}

	public NamedParameterJdbcTemplate getJdbcTemplate() {

		return jdbcTemplate;

	}

	@Required

	public void setJdbcTemplate(final NamedParameterJdbcTemplate jdbcTemplate) {

		this.jdbcTemplate = jdbcTemplate;

	}

}