D-Net

package eu.dnetlib.enabling.is.controller;

import java.io.StringReader;

import java.util.Collections;

import java.util.Date;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.UUID;

import org.apache.commons.lang3.StringUtils;

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

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.io.SAXReader;

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

import org.springframework.http.MediaType;

import org.springframework.web.bind.annotation.PathVariable;

import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestMethod;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.bind.annotation.RestController;

import com.google.common.collect.Lists;

import eu.dnetlib.enabling.annotations.DnetService;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.enabling.is.DnetInformationServiceException;

import eu.dnetlib.enabling.is.store.ISStore;

import eu.dnetlib.enabling.is.tools.DatabaseUtils;

import eu.dnetlib.enabling.is.tools.ProfileValidator;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.services.BaseService;

@RestController

@RequestMapping("/is")

@DnetService(DnetServiceType.is)

public class InformationServiceController extends BaseService {

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

	@Autowired

	private ISStore isStore;

	@Autowired

	private ProfileValidator profileValidator;

	@RequestMapping(value = "schema/{name}", method = RequestMethod.GET, produces = MediaType.APPLICATION_XML_VALUE)

	public String getSchema(@PathVariable("name") final String resourceType) throws DnetInformationServiceException {

		if (StringUtils.isBlank(resourceType)) { throw new DnetInformationServiceException("Invalid resourceType"); }

		return isStore.getXML(resourceType, DatabaseUtils.SCHEMAS_COLLECTION);

	}

	@RequestMapping("schemas")

	public List<String> listSchemas() throws DnetInformationServiceException {

		return isStore.getFileNames(DatabaseUtils.SCHEMAS_COLLECTION);

	}

	@RequestMapping(value = "schema/{name}", method = RequestMethod.POST)

	public boolean registerSchema(@PathVariable("name") final String resourceType, @RequestBody final String resourceSchema)

			throws DnetInformationServiceException {

		return isStore.insertXML(resourceType, DatabaseUtils.SCHEMAS_COLLECTION, resourceSchema);

	}

	@RequestMapping(value = "schema/{name}", method = RequestMethod.DELETE)

	public boolean deleteSchema(@PathVariable("name") final String resourceType) throws DnetInformationServiceException {

		return isStore.deleteXML(resourceType, DatabaseUtils.SCHEMAS_COLLECTION);

	}

	@RequestMapping("profiles/{kind}/{type}")

	public List<String> listProfiles(@PathVariable("kind") final String kind, @PathVariable("type") final String type) throws DnetInformationServiceException {

		return isStore.getFileNames(DatabaseUtils.getCollection(kind, type));

	}

	@RequestMapping(value = "profile", method = RequestMethod.POST)

	public String registerProfile(@RequestBody final String profile) throws DnetInformationServiceException {

		log.debug("registering profile");

		try {

			final Document doc = (new SAXReader()).read(new StringReader(profile));

			final String kind = doc.valueOf("/profile/kind").trim();

			final String type = doc.valueOf("/profile/type").trim();

			final String fileName = UUID.randomUUID().toString();

			final String newId = DatabaseUtils.composeResourceId(kind, type, fileName);

			final String date = new DateUtils(new Date()).getDateAsISO8601String();

			doc.selectSingleNode("/profile/id").setText(newId);

			doc.selectSingleNode("/profile/date").setText(date);

			final String finalDoc = doc.asXML();

			if (profileValidator.isValid(type, finalDoc)) {

				isStore.insertXML(fileName, DatabaseUtils.getCollection(kind, type), finalDoc);

				return newId;

			} else {

				throw new DnetInformationServiceException("Invalid profile");

			}

		} catch (final DocumentException e) {

			throw new DnetInformationServiceException(e);

		}

	}

	@RequestMapping("collections")

	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 : isStore.quickSearchXML(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;

	}

	@RequestMapping(value = "profile/{kind}/{type}/{fileName}", method = RequestMethod.GET, produces = MediaType.APPLICATION_XML_VALUE)

	public String getProfile(@PathVariable("kind") final String kind, @PathVariable("type") final String type, @PathVariable("fileName") final String fileName)

			throws DnetInformationServiceException {

		final String res = isStore.getXML(fileName, DatabaseUtils.getCollection(kind, type));

		if (StringUtils.isBlank(res)) {

			throw new DnetInformationServiceException("document " + DatabaseUtils.composeResourceId(kind, type, fileName) + " not found");

		} else {

			return res;

		}

	}

	@RequestMapping(value = "profile/{kind}/{type}/{fileName}", method = RequestMethod.POST)

	public boolean updateProfile(@PathVariable("kind") final String kind,

			@PathVariable("type") final String type,

			@PathVariable("fileName") final String fileName,

			@RequestBody final String profile) throws DnetInformationServiceException {

		try {

			final String resId = DatabaseUtils.composeResourceId(kind, type, fileName);

			final String coll = DatabaseUtils.getCollection(kind, type);

			final String oldProfileSrc = isStore.getXML(fileName, coll);

			if (oldProfileSrc == null) { throw new DnetInformationServiceException("cannot update a non existing profile " + resId); }

			final Document doc = (new SAXReader()).read(new StringReader(profile));

			final String date = new DateUtils(new Date()).getDateAsISO8601String();

			doc.selectSingleNode("/profile/id").setText(resId);

			doc.selectSingleNode("/profile/date").setText(date);

			final String finalDoc = doc.asXML();

			if (profileValidator.isValid(type, finalDoc)) {

				isStore.insertXML(fileName, coll, finalDoc);

				return true;

			} else {

				throw new DnetInformationServiceException("Invalid profile");

			}

		} catch (final DocumentException e) {

			throw new DnetInformationServiceException(e);

		}

	}

	@RequestMapping(value = "profile/{kind}/{type}/{fileName}", method = RequestMethod.DELETE)

	public boolean deleteProfile(@PathVariable("kind") final String kind,

			@PathVariable("type") final String type,

			@PathVariable("fileName") final String fileName) throws DnetInformationServiceException {

		final boolean res = isStore.deleteXML(fileName, DatabaseUtils.getCollection(kind, type));

		if (!res) { throw new DnetInformationServiceException("document " + DatabaseUtils.composeResourceId(kind, type, fileName) + " not found"); }

		return true;

	}

	@RequestMapping("xupdate")

	public boolean executeXUpdate(@RequestParam("q") final String xquery) throws DnetInformationServiceException {

		return isStore.executeXUpdate(xquery);

	}

	@RequestMapping("find")

	public List<String> find(@RequestParam("q") final String xquery) throws DnetInformationServiceException {

		return isStore.quickSearchXML(xquery);

	}

}