StormDeployAlternative.java

package dk.kaspergsm.stormdeploy;

import java.io.File;

import dk.kaspergsm.stormdeploy.userprovided.ConfigurationFactory;
import org.jclouds.compute.ComputeServiceContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import dk.kaspergsm.stormdeploy.commands.Attach;
import dk.kaspergsm.stormdeploy.commands.Deploy;
import dk.kaspergsm.stormdeploy.commands.Kill;
import dk.kaspergsm.stormdeploy.commands.ScaleOutCluster;
import dk.kaspergsm.stormdeploy.userprovided.Configuration;
import dk.kaspergsm.stormdeploy.userprovided.Credential;

/**
 * Main class for project
 * 
 * @author Kasper Grud Skat Madsen
 */
public class StormDeployAlternative {
	private static Logger log = LoggerFactory.getLogger(StormDeployAlternative.class);

	public static void main(String[] args) {
		if (args.length <= 1) {
			log.error("Wrong arguments provided, the following is supported:");
			log.error(" deploy CLUSTERNAME");
			log.error(" kill CLUSTERNAME");
			log.error(" attach CLUSTERNAME");
			log.error(" scaleout CLUSTERNAME #InstancesToAdd InstanceType");
			System.exit(0);
		}


		/**
		 * Parse
		 */
		String operation = args[0];
		String clusterName = args[1].toLowerCase();
		Configuration config = ConfigurationFactory.initialize(clusterName);
		Credential credentials = new Credential(new File(Tools.getWorkDir() + "conf" + File.separator + "credential.yaml"));


		/**
		 * Check selected cloud provider is supported
		 */
		if (!Tools.getAllProviders().contains("aws-ec2")) {
			log.error("aws-ec2 not in supported list: " + Tools.getAllProviders());
			System.exit(0);
		}


		/**
		 * Check if file ssh key pair private and public exist
		 */
		String sshKeyPath = System.getProperty("user.home") + File.separator + ".ssh" + File.separator;
		if (!new File(sshKeyPath + config.getSSHKeyName()).exists() || !new File(sshKeyPath + config.getSSHKeyName() + ".pub").exists()) {
			log.error("Missing rsa ssh keypair with name: " + config.getSSHKeyName() + ". Please generate keypair, without passphrase, by issuing: ssh-keygen -t rsa");
			System.exit(0);
		}


		/**
		 * Initialize connection to cloud provider
		 */
		ComputeServiceContext computeContext = Tools.initComputeServiceContext(config, credentials);
		log.info("Initialized cloud provider service");


		/**
		 * Execute specified operation now
		 */
		if (operation.trim().equalsIgnoreCase("deploy")) {

			Deploy.deploy(clusterName, credentials, config, computeContext);

		} else if (operation.trim().equalsIgnoreCase("scaleout")) {

			try {
				int newNodes = Integer.valueOf(args[2]);
				String instanceType = args[3];
				ScaleOutCluster.AddWorkers(newNodes, clusterName, instanceType, config, credentials, computeContext);
			} catch (Exception ex) {
				log.error("Error parsing arguments", ex);
				return;
			}

		} else if (operation.trim().equalsIgnoreCase("attach")) {

			Attach.attach(clusterName, config.getStormVersion(), computeContext);

		} else if (operation.trim().equalsIgnoreCase("kill")) {

			Kill.kill(clusterName, computeContext.getComputeService());

		} else {
			log.error("Unsupported operation " + operation);
		}
	}
}