D-Net

package eu.dnetlib.clients.is;

import java.util.Arrays;

import java.util.Collections;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.math.NumberUtils;

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 org.springframework.util.LinkedMultiValueMap;

import org.springframework.web.client.RestTemplate;

import com.google.common.collect.Lists;

import eu.dnetlib.clients.BaseServiceClient;

import eu.dnetlib.conf.DnetGenericApplicationProperties;

import eu.dnetlib.enabling.annotations.DnetServiceClient;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.exceptions.DnetGenericRuntimeException;

import eu.dnetlib.exceptions.InformationServiceException;

@Component

@DnetServiceClient(DnetServiceType.is)

public class InformationServiceClient extends BaseServiceClient {

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

	@Autowired

	private DnetGenericApplicationProperties props;

	public List<CollectionDesc> listCollections() throws Exception {

		final String xquery = "for $kind in xmldb:get-child-collections('/db/DRIVER') " +

				"for $type in xmldb:get-child-collections(concat('/db/DRIVER/', $kind)) " +

				"return concat ($kind, ' @@@ ', $type, ' @@@ ', count(xmldb:get-child-resources(concat('/db/DRIVER/', $kind, '/', $type))))";

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

		for (final String s : find(xquery)) {

			final String[] arr = s.split("@@@");

			final String kind = arr[0].trim();

			final String type = arr[1].trim();

			final int size = NumberUtils.toInt(arr[2].trim(), 0);

			if (!map.containsKey(kind)) {

				map.put(kind, new CollectionDesc(kind));

			}

			map.get(kind).addType(type, size);

		}

		final List<CollectionDesc> res = Lists.newArrayList(map.values());

		for (final CollectionDesc d : res) {

			Collections.sort(d.getTypes());

		}

		Collections.sort(res);

		return res;

	}

	public List<String> listSchemas() {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/schemas";

		return Arrays.asList(restTemplate.getForObject(url, String[].class));

	}

	public String getSchema(final String name) throws InformationServiceException {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/schema/" + name;

		return restTemplate.getForObject(url, String.class);

	}

	public void registerSchema(final String name, final String xsd) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/schema/" + name;

		restTemplate.postForObject(url, xsd, Boolean.class);

	}

	public void deleteSchema(final String name) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/schema/" + name;

		restTemplate.delete(url);

	}

	public String register(final String profile) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/profile";

		return restTemplate.postForObject(url, profile, String.class);

	}

	public String bulkRegister(final String profile) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/profile?bulk=true";

		return restTemplate.postForObject(url, profile, String.class);

	}

	public void deleteProfile(final String id) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/profile/" + id;

		restTemplate.delete(url);

	}

	public void updateProfile(final String id, final String profile) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/profile/" + id;

		restTemplate.postForObject(url, profile, String.class);

	}

	public void xupdate(final String query) {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/find";

		final LinkedMultiValueMap<String, String> params = new LinkedMultiValueMap<>();

		params.add("q", query);

		restTemplate.postForObject(url, params, String.class);

	}

	public String getProfile(final String id) throws InformationServiceException {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/profile/" + id;

		return restTemplate.getForObject(url, String.class);

	}

	public List<String> find(final String query) throws InformationServiceException {

		final RestTemplate restTemplate = new RestTemplate();

		final String url = props.getInformationServiceUrl() + "/is/find";

		final LinkedMultiValueMap<String, String> params = new LinkedMultiValueMap<>();

		params.add("q", query);

		return Arrays.asList(restTemplate.postForObject(url, params, String[].class));

	}

	public String findOne(final String query) throws InformationServiceException {

		final List<String> list = find(query);

		if (list.size() == 1) {

			return list.get(0);

		} else if (list.isEmpty()) {

			throw new InformationServiceException("Document not found, query: " + query);

		} else {

			throw new InformationServiceException("Too many results (" + list.size() + "), query: " + query);

		}

	}

	public void updateProfileNode(final String profileId, final String string, final String string2) throws InformationServiceException {

		// TODO Auto-generated method stub

		throw new DnetGenericRuntimeException("to be implemented");

	}

}