com.smash.revolance.ui.cmdline.Commands.java Source code

Introduction

Here is the source code for com.smash.revolance.ui.cmdline.Commands.java

Source

package com.smash.revolance.ui.cmdline;

/*
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * Revolance-UI-Cmdline
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * Copyright (C) 2012 - 2013 RevoLance
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public 
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.conn.HttpHostConnectException;
import org.apache.http.impl.client.DefaultHttpClient;
import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.nio.SelectChannelConnector;
import org.eclipse.jetty.webapp.WebAppContext;

import java.io.File;
import java.io.IOException;

/**
 * User: wsmash
 * Date: 28/04/13
 * Time: 22:10
 */
public enum Commands {
    STOP("stop", "port"), STATUS("status", "port"), START("start", "port", "war");

    private String[] opts = new String[] {};
    private final String cmd;

    private Commands(String cmd, String... opts) {
        this.cmd = cmd;
        this.opts = opts;
    }

    public String getCmd() {
        return cmd;
    }

    public int getOptsCount() {
        return this.opts == null ? 0 : this.opts.length;
    }

    private String getOpts() {
        String optsString = "";
        for (int optIdx = 0; optIdx < getOptsCount(); optIdx++) {
            optsString += "-D" + opts[optIdx] + " ";
        }
        return optsString.trim();
    }

    public String getUsage() {
        return String.format("%s %s", getCmd(), getOpts());
    }

    public static String getUsages() {
        String usage = "";
        for (Commands cmd : Commands.values()) {
            usage += String.format("Usage: %s\n", cmd.getUsage());
        }
        return usage;
    }

    public static Commands toCommand(String cmd) {
        for (Commands command : Commands.values()) {
            if (command.getCmd().contentEquals(cmd)) {
                return command;
            }
        }
        return null;
    }

    public static boolean allParametersDefined(Commands cmd) {
        for (String parameter : cmd.opts) {
            if (System.getProperty(parameter) == null) {
                System.out.println("Undefined parameter: '" + parameter + "' for command: '" + cmd.getCmd() + "");
                return false;
            }
        }
        return true;
    }

    public int exec() throws Exception {
        int execStatus = -1;
        File workingDir = new File(".").getAbsoluteFile();
        switch (this) {

        case START:

            execStatus = execStartCmd(workingDir, opts);
            break;

        case STOP:

            execStatus = execStopCmd(workingDir, opts);
            break;

        case STATUS:

            execStatus = execStatusCmd(workingDir, opts);
            break;
        }

        return execStatus;
    }

    private int execStatusCmd(File workingDir, String[] opts) {
        try {
            System.out.println("Retrieving server status");
            DefaultHttpClient client = new DefaultHttpClient();
            HttpGet request = new HttpGet(
                    "http://localhost:" + Integer.getInteger("port").intValue() + "/ui-monitoring-server/status");
            client.execute(request); // throw an IOException when
            System.out.println("server is started");
            return 0;
        } catch (HttpHostConnectException e) {
            System.out.println("server is stopped");
            return 1;
        } catch (Exception e) {
            return 2;
        }
    }

    private int execStartCmd(final File workingDir, final String[] opts) {
        new Thread(new Runnable() {

            @Override
            public void run() {
                Server server = new Server();

                Connector connector = new SelectChannelConnector();
                connector.setPort(Integer.getInteger("port").intValue());
                server.setConnectors(new Connector[] { connector });

                String war = workingDir + "/" + System.getProperty("war");

                WebAppContext webapp = new WebAppContext(war, "/ui-monitoring-server");
                server.setHandler(webapp);

                try {
                    System.out.println("Starting server");
                    server.start();
                    System.out.println("Starting server [Done]");
                    server.join();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }).run();

        return 0;
    }

    private int execStopCmd(File workingDir, String[] opts) {
        try {
            System.out.println("Stopping server");
            DefaultHttpClient client = new DefaultHttpClient();
            HttpGet request = new HttpGet(
                    "http://localhost:" + Integer.getInteger("port").intValue() + "/ui-monitoring-server/shutdown");
            HttpResponse response = client.execute(request);
            if (response.getStatusLine().getStatusCode() != 200) {
                System.out.println("Stopping server [Done]");
                return 0;
            } else {
                System.err.println("Stopping server [Failed]");
                return 1;
            }
        } catch (IOException e) {
            return 2;
        }
    }

}