NodeScriptDiscovery.java

package codeine.nodes;

import java.util.List;
import java.util.Map;

import javax.inject.Inject;

import org.apache.log4j.Logger;

import codeine.api.NodeInfo;
import codeine.configuration.PathHelper;
import codeine.jsons.nodes.NodeListJson;
import codeine.jsons.project.ProjectJson;
import codeine.model.Constants;
import codeine.utils.os.OsUtils;
import codeine.utils.os_process.ShellScript;

import com.google.common.collect.Maps;
import com.google.gson.Gson;
import com.google.gson.JsonSyntaxException;
import com.google.gson.reflect.TypeToken;

public class NodeScriptDiscovery {

	private static final Logger log = Logger.getLogger(NodeScriptDiscovery.class);
	@Inject private Gson gson;
	@Inject private PathHelper pathHelper;
	
	public NodeListJson get(ProjectJson projectJson) {
		if (projectJson.operating_system() != OsUtils.getHostOs()) {
			log.info("project " + projectJson.name() + " os is " + projectJson.operating_system() + " and host is " + OsUtils.getHostOs() + " so will not run it");
			return new NodeListJson();
		}
		String nodes_discovery_script = projectJson.nodes_discovery_script();
		String dir = pathHelper.getProjectDir(projectJson.name());
		log.info("running discovery from dir " + dir);
		Map<String, String> env = Maps.newHashMap();
		env.put(Constants.EXECUTION_ENV_PROJECT_NAME, projectJson.name());
		env.putAll(projectJson.environmentVariables());
		ShellScript shellScript = 
				new ShellScript("discovery_" + projectJson.name(), nodes_discovery_script, projectJson.operating_system(), null, dir, env, null);
		String result = shellScript.execute().outputFromFile();
		if (result.trim().isEmpty()){
			return new NodeListJson();
		}
		NodeListJson nodeListJson = null;
		try {
			if (result.trim().startsWith("{")) { //OLD FORMAT 
				//TODO remove after build 150, and all projects converted
				nodeListJson = gson.fromJson(result, NodeListJson.class);
			} else {
				List<NodeInfo> fromJson = gson.fromJson(result, new TypeToken<List<NodeInfo>>(){}.getType());
				nodeListJson = new NodeListJson(fromJson);
			}
		} catch (JsonSyntaxException e) {
			log.warn("json is " + result);
			log.warn("failed to parse nodes from discovery in project " + projectJson.name(), e);
		}
		if (null == nodeListJson){
			return new NodeListJson();
		}
		return nodeListJson;
	}

}