D-Net

package eu.dnetlib.services.hcm;

import java.io.IOException;

import java.lang.management.ManagementFactory;

import java.lang.management.RuntimeMXBean;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.TreeMap;

import java.util.UUID;

import java.util.stream.Collectors;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.maven.model.Model;

import org.apache.maven.model.Parent;

import org.apache.maven.model.io.xpp3.MavenXpp3Reader;

import org.dom4j.DocumentHelper;

import org.dom4j.Element;

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

import org.springframework.core.env.AbstractEnvironment;

import org.springframework.core.env.Environment;

import org.springframework.core.env.MapPropertySource;

import org.springframework.core.env.PropertySource;

import org.springframework.core.io.Resource;

import org.springframework.core.io.ResourceLoader;

import org.springframework.core.io.support.ResourcePatternUtils;

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

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

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

import com.google.common.collect.Maps;

import eu.dnetlib.conf.DnetGenericApplicationProperties;

import eu.dnetlib.enabling.annotations.DnetService;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.miscutils.datetime.HumanTime;

import eu.dnetlib.miscutils.streams.DnetStreamSupport;

import eu.dnetlib.services.BaseService;

@RestController

@RequestMapping(value = "/hcm", method = RequestMethod.GET)

@DnetService(DnetServiceType.hcm)

public class HCM extends BaseService {

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

	@Autowired

	private Environment env;

	@Autowired

	private DnetGenericApplicationProperties containerConfiguration;

	@Autowired

	private ResourceLoader resourceLoader;

	@RequestMapping(value = "properties", method = RequestMethod.GET)

	public Map<String, Object> listProperties() {

		final Iterator<PropertySource<?>> iter = ((AbstractEnvironment) env).getPropertySources().iterator();

		return DnetStreamSupport.generateStreamFromIterator(iter)

				.filter(ps -> ps instanceof MapPropertySource)

				.map(ps -> (MapPropertySource) ps)

				.collect(Collectors.toMap(MapPropertySource::getName, MapPropertySource::getSource));

	}

	@RequestMapping(value = "info", method = RequestMethod.GET)

	public Map<String, Object> info() {

		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, Object> genInfo = Maps.newLinkedHashMap();

		genInfo.put("Hostname", containerConfiguration.getHost());

		genInfo.put("Port", containerConfiguration.getPort());

		genInfo.put("Uptime", HumanTime.exactly(mxbean.getUptime()));

		genInfo.put("Start Time", DateUtils.calculate_ISO8601(mxbean.getStartTime()));

		return genInfo;

	}

	@RequestMapping(value = "jvm", method = RequestMethod.GET)

	public Map<String, String> jvmInfo() {

		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, String> jvmInfo = Maps.newLinkedHashMap();

		jvmInfo.put("JVM Name", mxbean.getVmName());

		jvmInfo.put("JVM Vendor", mxbean.getVmVendor());

		jvmInfo.put("JVM Version", mxbean.getVmVersion());

		jvmInfo.put("JVM Spec Name", mxbean.getSpecName());

		jvmInfo.put("JVM Spec Vendor", mxbean.getSpecVendor());

		jvmInfo.put("JVM Spec Version", mxbean.getSpecVersion());

		jvmInfo.put("Running JVM Name", mxbean.getName());

		jvmInfo.put("Management Spec Version", mxbean.getManagementSpecVersion());

		return jvmInfo;

	}

	@RequestMapping(value = "libs", method = RequestMethod.GET)

	public Map<String, String> libInfo() {

		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, String> libInfo = Maps.newLinkedHashMap();

		libInfo.put("Classpath", mxbean.getClassPath().replaceAll(":", " : "));

		libInfo.put("Boot ClassPath", mxbean.getBootClassPath().replaceAll(":", " : "));

		libInfo.put("Input arguments", mxbean.getInputArguments().toString());

		libInfo.put("Library Path", mxbean.getLibraryPath().replaceAll(":", " : "));

		return libInfo;

	}

	@RequestMapping(value = "maven", method = RequestMethod.GET)

	private Map<String, Map<String, Map<String, List<String>>>> mavenModules() throws IOException {

		final Map<String, Map<String, Map<String, List<String>>>> modules = new TreeMap<>();

		final MavenXpp3Reader reader = new MavenXpp3Reader();

		for (final Resource res : ResourcePatternUtils.getResourcePatternResolver(resourceLoader).getResources("classpath*:/META-INF/**/pom.xml")) {

			try {

				final Model model = reader.read(res.getInputStream());

				final String name = model.getArtifactId();

				String groupId = model.getGroupId();

				for (Parent parent = model.getParent(); (groupId == null) && (model.getParent() != null); parent = model.getParent()) {

					groupId = parent.getGroupId();

				}

				String version = model.getVersion();

				for (Parent parent = model.getParent(); (version == null) && (model.getParent() != null); parent = model.getParent()) {

					version = parent.getVersion();

				}

				if (!modules.containsKey(groupId)) {

					modules.put(groupId, new TreeMap<>());

				}

				if (!modules.get(groupId).containsKey(name)) {

					modules.get(groupId).put(name, new TreeMap<>());

				}

				if (!modules.get(groupId).get(name).containsKey(version)) {

					modules.get(groupId).get(name).put(version, new ArrayList<>());

				}

				modules.get(groupId).get(name).get(version).add(res.getURI().toString());

			} catch (final Exception e) {

				log.warn("Error evaluating pom: " + res.getURI());

				log.debug("-- ERROR --", e);

			}

		}

		return modules;

	}

	@Override

	public List<Element> geXmlProfileSections() {

		final Element elem = DocumentHelper.createElement("HCM_SESSION_ID");

		elem.addAttribute("value", "session-" + UUID.randomUUID());

		return Arrays.asList(elem);

	}

}