D-Net

package eu.dnetlib.functionality.modular.ui.is;

import java.io.IOException;

import java.io.StringReader;

import java.net.HttpURLConnection;

import java.net.URL;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Date;

import java.util.List;

import java.util.Map;

import javax.annotation.Resource;

import javax.servlet.ServletOutputStream;

import javax.servlet.ServletResponse;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.stereotype.Controller;

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

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

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

import com.google.common.base.Splitter;

import com.google.common.collect.Iterables;

import com.google.common.collect.Maps;

import eu.dnetlib.enabling.is.rmi.InformationService;

import eu.dnetlib.enabling.is.rmi.objects.BlackboardMessage;

import eu.dnetlib.enabling.is.rmi.objects.DnetDataStructure;

import eu.dnetlib.enabling.is.rmi.objects.DnetResourceType;

import eu.dnetlib.enabling.is.rmi.objects.DnetService;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.functionality.modular.ui.AbstractAjaxController;

import eu.dnetlib.miscutils.datetime.DateUtils;

@Controller

@RequestMapping("/ui/ajax_is")

public class IsAjaxController extends AbstractAjaxController {

	@Resource

	private UniqueServiceLocator serviceLocator;

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

	@RequestMapping("query.do")

	public @ResponseBody

	List<String> query(@RequestParam(value = "query", required = true) final String query) throws Exception {

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

		return serviceLocator.getService(InformationService.class).searchSql(query);

	}

	@RequestMapping("listResourceTypes.do")

	public @ResponseBody

	List<DnetResourceType> listResourceTypes() throws Exception {

		return serviceLocator.getService(InformationService.class).listResourceTypes();

	}

	@RequestMapping("listDatastructures.do")

	public @ResponseBody

	List<DnetDataStructure> listDatastructures(@RequestParam(value = "type", required = true) final String type) throws Exception {

		return serviceLocator.getService(InformationService.class).listDatastructures(type);

	}

	@RequestMapping("getDs.do")

	public @ResponseBody

	DnetDataStructure getDs(@RequestParam(value = "id", required = true) final String id) throws Exception {

		return serviceLocator.getService(InformationService.class).getDsById(id);

	}

	@RequestMapping("registerDs.do")

	public @ResponseBody

	String registerDs(@RequestParam(value = "code", required = true) final String code,

			@RequestParam(value = "name", required = true) final String name,

			@RequestParam(value = "description", required = false) final String description,

			@RequestParam(value = "type", required = true) final String type,

			@RequestParam(value = "serviceId", required = false) final String serviceId,

			@RequestParam(value = "content", required = true) final String content) throws Exception {

		final DnetDataStructure ds = new DnetDataStructure();

		ds.setCode(code);

		ds.setName(name);

		ds.setDescription(description);

		ds.setType(type);

		ds.setServiceId(serviceId);

		ds.setContent(content);

		return serviceLocator.getService(InformationService.class).registerDs(ds);

	}

	@RequestMapping("updateDs.do")

	public @ResponseBody

	String updateDs(

			@RequestParam(value = "id", required = true) final String id,

			@RequestParam(value = "name", required = false) final String name,

			@RequestParam(value = "description", required = false) final String description,

			@RequestParam(value = "content", required = false) final String content) throws Exception {

		final InformationService is = serviceLocator.getService(InformationService.class);

		final DnetDataStructure ds = is.getDsById(id);

		if (name != null) {

			ds.setName(name);

		}

		if (description != null) {

			ds.setDescription(description);

		}

		if (content != null) {

			ds.setContent(content);

		}

		ds.setDate(new Date());

		return is.registerDs(ds);

	}

	@RequestMapping("deleteDs.do")

	public @ResponseBody

	boolean deleteDs(@RequestParam(value = "id", required = true) final String id) throws Exception {

		return serviceLocator.getService(InformationService.class).deleteDsById(id);

	}

	@RequestMapping("import.do")

	public @ResponseBody

	Map<String, Integer> importProfiles(

			@RequestParam(value = "path", required = true) final String path,

			@RequestParam(value = "types", required = true) final boolean types,

			@RequestParam(value = "resources", required = true) final boolean resources) throws Exception {

		log.info("importing types/resources from " + path);

		final Map<String, Integer> res = Maps.newHashMap();

		if (types) {

			// res.putAll(profileImporter.importSchemas(path + "/**/*.xsd"));

		}

		if (resources) {

			// res.putAll(profileImporter.importProfiles(path + "/**/*.xml"));

		}

		return res;

	}

	@RequestMapping("listBlackboards.do")

	public @ResponseBody

	List<BlackboardMessage> listBlackboards() throws Exception {

		return serviceLocator.getService(InformationService.class).listBlackboardMessages();

	}

	@RequestMapping("listServices.do")

	public @ResponseBody

	Collection<ServiceGrouper> listServices() throws Exception {

		final Map<String, ServiceGrouper> map = Maps.newHashMap();

		for (DnetService s : serviceLocator.getService(InformationService.class).listServices()) {

			for (String address : s.getProtocols().values()) {

				final URL url = new URL(address);

				final String host = url.getHost();

				final int port = url.getPort();

				final String context = Iterables.getFirst(Splitter.on("/").omitEmptyStrings().split(url.getPath()), "");

				final String tmpKey = host + "@@@" + port + "@@@" + context;

				if (!map.containsKey(tmpKey)) {

					map.put(tmpKey, new ServiceGrouper(host, port, context, new ArrayList<DnetService>()));

				}

				map.get(tmpKey).getServices().add(s);

			}

		}

		return map.values();

	}

	@RequestMapping("getService.do")

	public @ResponseBody

	DnetService getService(@RequestParam(value = "id", required = true) final String id) throws Exception {

		return serviceLocator.getService(InformationService.class).getService(id);

	}

	@RequestMapping("ping.do")

	public @ResponseBody

	long pingUrl(@RequestParam(value = "url", required = true) final String url, @RequestParam(value = "timeout", required = true) final int timeout)

			throws IOException {

		final long start = DateUtils.now();

		final HttpURLConnection urlConn = (HttpURLConnection) new URL(url).openConnection();

		urlConn.setConnectTimeout(timeout);

		urlConn.setReadTimeout(timeout);

		if (urlConn.getResponseCode() == HttpURLConnection.HTTP_OK) {

			return DateUtils.now() - start;

		} else {

			throw new IllegalArgumentException("Invalid Url");

		}

	}

	public void sendXML(final ServletResponse response, final String xml) throws IOException {

		response.setContentType("text/xml");

		final ServletOutputStream out = response.getOutputStream();

		IOUtils.copy(new StringReader(xml), out);

		out.flush();

		out.close();

	}

}