dnet-core/dnet-modular-ui/src/main/java/eu/dnetlib/functionality/modular/ui/InfoController.java

package eu.dnetlib.functionality.modular.ui;

import java.io.IOException;
import java.lang.management.ManagementFactory;
import java.lang.management.RuntimeMXBean;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

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.springframework.beans.factory.annotation.Required;
import org.springframework.context.ResourceLoaderAware;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.springframework.core.io.support.ResourcePatternUtils;
import org.springframework.ui.ModelMap;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

import eu.dnetlib.miscutils.datetime.DateUtils;
import eu.dnetlib.miscutils.datetime.HumanTime;

public class InfoController extends ModuleEntryPoint implements ResourceLoaderAware {

	private String hostname;
	private String port;
	private String context;

	private ResourceLoader resourceLoader;

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

	@Override
	protected void initialize(final ModelMap map, final HttpServletRequest request, final HttpServletResponse response) throws Exception {
		final RuntimeMXBean mxbean = ManagementFactory.getRuntimeMXBean();

		final Map<String, Map<String, String>> info = Maps.newLinkedHashMap();
		info.put("General", getGeneralInfo(mxbean));
		info.put("JVM", getJvmInfo(mxbean));
		info.put("Libraries and arguments", getLibInfo(mxbean));
		info.put("System properties", getSysInfo(mxbean));

		map.addAttribute("info", info);
		map.addAttribute("modules", getModules());

	}

	@SuppressWarnings("unchecked")
	private List<Map<String, Object>> getModules() throws IOException {
		final Map<String, Map<String, Map<String, Object>>> modules = Maps.newLinkedHashMap();

		final MavenXpp3Reader reader = new MavenXpp3Reader();
		for (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 HashMap<String, Map<String, Object>>());
				}
				if (!modules.get(groupId).containsKey(name)) {
					final Map<String, Object> map = Maps.newHashMap();
					map.put("group", groupId);
					map.put("name", name);
					map.put("files", new ArrayList<String>());
					map.put("versions", new ArrayList<String>());
					modules.get(groupId).put(name, map);
				} else {
					// Artifact already found
					modules.get(groupId).get(name).put("warning", "1");
				}
				((List<String>) modules.get(groupId).get(name).get("versions")).add(version);
				((List<String>) modules.get(groupId).get(name).get("files")).add(res.getURI().toString());
			} catch (Exception e) {
				log.error("Error evaluating pom: " + res.getURI());
				log.debug("-- ERROR --", e);
			}
		}

		final List<Map<String, Object>> list = Lists.newArrayList();
		for (Entry<String, Map<String, Map<String, Object>>> e : modules.entrySet()) {
			for (Entry<String, Map<String, Object>> e1 : e.getValue().entrySet()) {
				list.add(e1.getValue());
			}
		}

		Collections.sort(list, new Comparator<Map<String, Object>>() {

			@Override
			public int compare(final Map<String, Object> o1, final Map<String, Object> o2) {
				if (o1.get("group").equals(o2.get("group"))) {
					return o1.get("name").toString().compareTo(o2.get("name").toString());
				} else {
					return o1.get("group").toString().compareTo(o2.get("group").toString());
				}
			}
		});

		return list;
	}

	private Map<String, String> getSysInfo(final RuntimeMXBean mxbean) {
		return mxbean.getSystemProperties();
	}

	private Map<String, String> getGeneralInfo(final RuntimeMXBean mxbean) {
		final Map<String, String> genInfo = Maps.newLinkedHashMap();
		genInfo.put("Hostname", hostname);
		genInfo.put("Port", port);
		genInfo.put("Context", context);
		genInfo.put("Uptime", HumanTime.exactly(mxbean.getUptime()));
		genInfo.put("Start Time", DateUtils.calculate_ISO8601(mxbean.getStartTime()));
		return genInfo;
	}

	private Map<String, String> getJvmInfo(final RuntimeMXBean mxbean) {
		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;
	}

	private Map<String, String> getLibInfo(final RuntimeMXBean mxbean) {
		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;
	}

	public String getHostname() {
		return hostname;
	}

	@Required
	public void setHostname(final String hostname) {
		this.hostname = hostname;
	}

	public String getPort() {
		return port;
	}

	@Required
	public void setPort(final String port) {
		this.port = port;
	}

	public String getContext() {
		return context;
	}

	@Required
	public void setContext(final String context) {
		this.context = context;
	}

	@Override
	public void setResourceLoader(final ResourceLoader resourceLoader) {
		this.resourceLoader = resourceLoader;
	}
}