contrail

/*

 * To change this template, choose Tools | Templates

 * and open the template in the editor.

 */

package vep;

import java.io.*;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.SocketException;

import java.net.SocketTimeoutException;

import java.sql.ResultSet;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.HashSet;

import java.util.LinkedList;

import java.util.List;

import java.util.Map;

import java.util.Set;

import javax.xml.parsers.ParserConfigurationException;

import javax.xml.parsers.SAXParser;

import javax.xml.parsers.SAXParserFactory;

import org.apache.log4j.Logger;

import org.opennebula.client.Client;

import org.opennebula.client.OneResponse;

import org.opennebula.client.host.HostPool;

import org.xml.sax.Attributes;

import org.xml.sax.InputSource;

import org.xml.sax.SAXException;

import org.xml.sax.helpers.DefaultHandler;

/**

 *

 * @author piyush, florian

 */

public class CLIServer extends DefaultHandler implements Runnable //CLI Server is a blocking server, handles only one client at a time

{

    private int port;

    private boolean continueFlag;

    private Thread t;

    private Logger logger;

    private ServerSocket server;

    private Socket client;

    private dbHandler dbhandle;

    private PrintWriter out;

    private BufferedReader in;

    private String vepProperties;

    private LinkedList<ONEHost> rowList;

    private ONEHost tempHost;

    private String tempVal;

    private ClusterStats runningStat;

    private List<String> nodes;

    public CLIServer(int sPort, dbHandler dhandle, String vepProp) //make sure CLI server is started only once DB operations are enabled

    {

        port = sPort;

        dbhandle = dhandle;

        vepProperties = vepProp;

        continueFlag = true;

        logger = Logger.getLogger("VEP.CLIServer");

        client = null;

        try

        {

            server = new ServerSocket(port);

            t = new Thread(this);

        }

        catch(IOException ex)

        {

            t = null;

            server = null;

            logger.debug("Error while trying to create CLI server socket, maybe the port is not free: " + ex.getMessage());

            ex.printStackTrace(System.err);

        }

    }

    public void startCLIserver()

    {

        if(t != null)

        {

            continueFlag = true;

            t.start();

        }

    }

    public void stopCLIserver()

    {

        continueFlag = false;

        //just send a dummy connect if server is blocking on client incoming request

        try

        {

            if(client != null && client.isConnected()) client.close();

            Socket sock = new Socket("127.0.0.1", port);

            sock.close();

            server.close();

            t.join(1000);

        }

        catch(Exception ex)

        {

        }

    }

    public void run()

    {

        logger.debug("CLI Server has started ...");

        while(continueFlag && (server != null))

        {

            try

            {

                client = server.accept();

                client.setSoTimeout(1000*60*2); //2 minutes idle time

                out = new PrintWriter(client.getOutputStream(), true);

                in = new BufferedReader(new InputStreamReader(client.getInputStream()));

                boolean authPhase = true;

                boolean passPhase = false;

                boolean isAuthenticated = false;

                boolean forceQuit = false;

                String username = "";

                String password = "";

                String output = "Welcome to Contrail VEP Command Line Interface!\n";

                output += "You must be authenticated before you are granted access to modify the execution of VEP.\n";

                output += "------------------------------------\n";

                output += "Administrator Username: ";

                out.println(output);

                String userInput  = in.readLine();

                while(!forceQuit && !client.isClosed() && userInput != null && !userInput.equalsIgnoreCase("bye") && !userInput.equalsIgnoreCase("exit"))

                {

                    output = "";

                    logger.debug("CLI incoming message: " + userInput);

                    if(authPhase)

                    {

                        username = userInput.trim();

                        authPhase = false;

                        passPhase = true;

                        output = "------------------------------------\nAdministrator " + username + "'s password: ";

                    }

                    else if(passPhase)

                    {

                        password = userInput.trim();

                        passPhase = false;

                        isAuthenticated = doAdminAuth(username, password, out, in);

                        username = "";

                        password = "";

                        if(isAuthenticated)

                            output = "------------------------------------\n"

                                    + "Authentication successful! List of commands:\n"

                                    + "\t show configuration\n"

                                    + "\t edit configuration\n"

                                    + "\t list hosts\n"

                                    + "\t edit hosts\n"

                                    + "\t add fedadmin\n"

                                    + "\t add datacenter\n"

                                    + "\t add cluster\n"

                                    + "\t add rack\n"

                                    + "\t help\n"

                                    + "\t exit\n"

                                    + "------------------------------------";

                    }

                    else if(isAuthenticated)

                    {

                        if(userInput.equalsIgnoreCase("help"))

                            output = showHelp();

                        else if(userInput.equalsIgnoreCase("show configuration"))

                        {

                            output = "------------------------------------\n"

                                    + readConfiguration()

                                    + "------------------------------------";

                        }

                        else if(userInput.equalsIgnoreCase("edit configuration"))

                        {

                            editConfiguration(out, in);

                        }

                        else if(userInput.equalsIgnoreCase("add datacenter"))

                        {

                            addDatacenter(out, in);

                        }

                        else if(userInput.equalsIgnoreCase("add cluster"))

                        {

                            addCluster(out, in);

                        }

                        else if(userInput.equalsIgnoreCase("add rack"))

                        {

                            addRack(out, in);

                        }

                        else if(userInput.equalsIgnoreCase("list hosts"))

                        {

                            listHost(out, in);

                        }

                        else if(userInput.equalsIgnoreCase("add host"))

                        {

                            addHost(out, in);

                        }

                        else if(userInput.equalsIgnoreCase("add fedadmin"))

                        {

                            addFedAdmin(out, in);

                        }

                        else

                        {

                            output = "------------------------------------\n"

                                    + "\tNot yet implemented! You may type help or exit ...\n"

                                    + "------------------------------------";

                        }

                    }

                    else

                    {

                        output = "You do not have the proper authorization to continue.";

                        forceQuit = true;

                    }

                    out.println(output);

                    if(!forceQuit)

                        userInput  = in.readLine();

                }

                client.close();

            }

            catch(SocketException sockex)

            {

                logger.warn("CLI client read error, closing connection: " + sockex.getMessage());

                try

                {

                    if(client != null && client.isConnected())

                    {

                        out.println("VEP server shutting down. Closing connection now ...");

                        client.close();

                    }

                }

                catch(Exception ex)

                {

                    //ignore this exception

                }

            }

            catch(SocketTimeoutException timeout)

            {

                logger.warn("CLI client timeout, closing connection: " + timeout.getMessage());

                try

                {

                    if(client != null && client.isConnected())

                    {

                        out.println("You have been inactive for more than 2 minutes. For security reasons this connection will be closed now.");

                        client.close();

                    }

                }

                catch(Exception ex)

                {

                    //ignore this exception

                }

            }

            catch(Exception ex)

            {

                logger.warn("CLI server encountered an exception: " + ex.getMessage());

                ex.printStackTrace(System.err);

            }

        }

        logger.debug("CLI Server exitting now ...");

    }

    private String showHelp()

    {

        String output = "------------------------------------\nList of commands:\n"

            + "\t show configuration\n"

            + "\t edit configuration\n"

            + "\t list hosts\n"

            + "\t edit hosts\n"

            + "\t add fedadmin\n"

            + "\t add datacenter\n"

            + "\t add cluster\n"

            + "\t add rack\n"

            + "\t help\n"

            + "\t exit\n"

            + "------------------------------------";

        return output;

    }

    private String readConfiguration()

    {

        String output = "";

        try

        {

//            BufferedReader fileReader = new BufferedReader(new FileReader("vep.properties"));

            BufferedReader fileReader = new BufferedReader(new FileReader(vepProperties));

            String strLine = "";

            while ((strLine = fileReader.readLine()) != null)

            {

                output += strLine + "\n";

            }

            fileReader.close();

        }

        catch(Exception ex)

        {

            logger.warn("Caught exception in CLI module while retrieving the VEP properties: " + ex.getMessage());

            ex.printStackTrace(System.err);

        }

        return output;

    }

    private boolean doAdminAuth(String user, String pass, PrintWriter out, BufferedReader in)

    {

        try

        {

            String userInput = "";

            String adminUser = "";

            String adminPass = "";

            ResultSet rs = dbhandle.query("select", "count(*)", "vepadmin", "");

            if(rs.next())

            {

                int count = rs.getInt(1);

                if(count < 1)

                {

                    rs.close();

                    out.println("It seems no local administrator has been setup! Would you like to setup one now (yes/no)?");

                    userInput  = in.readLine();

                    if(userInput.equalsIgnoreCase("yes"))

                    {

                        out.println("New administrator username:");

                        userInput  = in.readLine().trim();

                        if(!userInput.contains(" \t!@#$%^&*()+-=~`;:'\",<>?/") && userInput.length() > 0)

                        {

                            adminUser = userInput;

                            out.println("Provide a password for the new administrator [" + adminUser + "]'s account: ");

                            userInput  = in.readLine().trim();

                            adminPass = VEPHelperMethods.makeSHA1Hash(userInput);

                            return dbhandle.insert("vepadmin", "('" + adminUser + "', '" + adminPass + "')");

                        }

                    }

                    else

                    {

                        return false;

                    }

                }

                else

                {

                    rs.close();

                }

            }

            //now proceed with the argument's validation

            rs = dbhandle.query("select", "*", "vepadmin", "where username='" + user + "'");

            if(rs.next())

            {

                adminPass = VEPHelperMethods.makeSHA1Hash(pass);

                if(rs.getString("password").equalsIgnoreCase(adminPass))

                {

                    rs.close();

                    logger.trace("Admin Authentication successful.");

                    return true;

                }

                else

                {

                    rs.close();

                    logger.trace("Admin Authentication failed.");

                    return false;

                }

            }

            else

            {

                rs.close();

                return false;

            }

        }

        catch(Exception ex)

        {

            logger.warn("Caught exception while performing admin authentication inside the CLI process: " + ex.getMessage());

            ex.printStackTrace(System.err);

            return false;

        }

    }

    private void editConfiguration(PrintWriter out, BufferedReader in)

    {

        try

        {

            String editChoice = "";

            String userInput = "";

            out.println("\nChoose an option to edit: ");

            while(!editChoice.equalsIgnoreCase("q")){

                out.println("\n[d]b configuration, [l]og configuration, [o]pen nebula configuration, [c]luster configuration, [r]est configuration, [q]uit");

                editChoice = in.readLine();

                if(editChoice.equalsIgnoreCase("d")){

                    out.println("\nChoose DB type: [m]ysql, [s]qlite. Current choice is "+VEPHelperMethods.getProperty("vepdb.choice", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if (userInput.equalsIgnoreCase("m"))

                        VEPHelperMethods.addProperty("vepdb.choice", "mysql", logger, vepProperties);

                    else if (userInput.equalsIgnoreCase("s"))

                        VEPHelperMethods.addProperty("vepdb.choice", "sqlite", logger, vepProperties);

                    out.println("\nConfiguration of the chosen database");

                    if (VEPHelperMethods.getProperty("vepdb.choice", logger, vepProperties).equals("sqlite")){

                        out.println("\nConfigure path to the database: "+VEPHelperMethods.getProperty("sqlite.db", logger, vepProperties)+".");

                        userInput = in.readLine();

                        if(!userInput.equals("")){

                            VEPHelperMethods.addProperty("sqlite.db", userInput, logger, vepProperties);

                        }

                    }

                    else if (VEPHelperMethods.getProperty("vepdb.choice", logger, vepProperties).equals("mysql")){

                        out.println("\nConfigure IP of the database: "+VEPHelperMethods.getProperty("mysql.ip", logger, vepProperties)+".");

                        userInput = in.readLine();

                        if(!userInput.equals("")){

                            VEPHelperMethods.addProperty("mysql.ip", userInput, logger, vepProperties);

                        }

                        out.println("\nConfigure port of the database: "+VEPHelperMethods.getProperty("mysql.port", logger, vepProperties)+".");

                        userInput = in.readLine();

                        if(!userInput.equals("")){

                            VEPHelperMethods.addProperty("mysql.port", userInput, logger, vepProperties);

                        }

                        out.println("\nConfigure chosen login of the database: "+VEPHelperMethods.getProperty("mysql.user", logger, vepProperties)+".");

                        userInput = in.readLine();

                        if(!userInput.equals("")){

                            VEPHelperMethods.addProperty("mysql.user", userInput, logger, vepProperties);

                        }

                        out.println("\nConfigure password of the chosen user of the database: "+VEPHelperMethods.getProperty("mysql.pass", logger, vepProperties)+".");

                        userInput = in.readLine();

                        if(!userInput.equals("")){

                            VEPHelperMethods.addProperty("mysql.pass", userInput, logger, vepProperties);

                        }

                    }

                    out.println("\nDB configured, going back to the first menu");

                }

                else if(editChoice.equalsIgnoreCase("l")){

                    out.println("\nConfigure path to the log file: "+VEPHelperMethods.getProperty("veplog.file", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("veplog.file", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure max size to the log file: "+VEPHelperMethods.getProperty("veplog.size", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("veplog.size", userInput, logger, vepProperties);

                    }

                    out.println("\nLog file configured, going back to the first menu");

                }

                else if(editChoice.equalsIgnoreCase("o")){

                    out.println("\nConfigure OpenNebula server's ip: "+VEPHelperMethods.getProperty("one.ip", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("one.ip", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure OpenNebula server's port: "+VEPHelperMethods.getProperty("one.port", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("one.port", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure OpenNebula user: "+VEPHelperMethods.getProperty("one.user", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("one.user", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure OpenNebula chosen user's password: "+VEPHelperMethods.getProperty("one.pass", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("one.pass", userInput, logger, vepProperties);

                    }

                    out.println("\nOpenNebula configured, going back to the first menu");

                }

                else if(editChoice.equalsIgnoreCase("c")){

                    out.println("\nConfigure OpenNebula cluster id for contrail: "+VEPHelperMethods.getProperty("contrail.cluster", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("contrail.cluser", userInput, logger, vepProperties);

                    }

                    out.println("\nCluster id configured, going back to the first menu");

                }

                else if(editChoice.equalsIgnoreCase("r")){

                    out.println("\nConfigure REST keystore: "+VEPHelperMethods.getProperty("rest.keystore", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("rest.keystore", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure REST keystore password: "+VEPHelperMethods.getProperty("rest.keystorepass", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("rest.keystorepass", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure REST certificate's password: "+VEPHelperMethods.getProperty("rest.keypass", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("rest.keypass", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure REST HTTP Port: "+VEPHelperMethods.getProperty("rest.restHTTPPort", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("rest.restHTTPPort", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure REST HTTPS Port: "+VEPHelperMethods.getProperty("rest.restHTTPSPort", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("rest.restHTTPSPort", userInput, logger, vepProperties);

                    }

                    out.println("\nConfigure REST mandatory authentication: "+VEPHelperMethods.getProperty("rest.clientAuthSelected", logger, vepProperties)+".");

                    userInput = in.readLine();

                    if(!userInput.equals("")){

                        VEPHelperMethods.addProperty("rest.clientAuthSelected", userInput, logger, vepProperties);

                    }

                    out.println("\nREST Server configured, going back to the first menu");

                }

                else

                    out.println("\nExiting configuration's edit...");

            }

            logger.info("Configured local properties file");

        }

        catch(Exception ex){

            logger.warn("Caught exception while editing configuration inside the CLI process: " + ex.getMessage());

            ex.printStackTrace(System.err);

        }

    }

    private void addDatacenter(PrintWriter out, BufferedReader in)

    {

        boolean exit = false;

        String userInput = "";

        String[] codes = new String[] { "AFGHANISTAN ,AF", "ÅLAND ISLANDS ,AX", "ALBANIA ,AL", "ALGERIA ,DZ", "AMERICAN SAMOA ,AS", "ANDORRA ,AD", "ANGOLA ,AO", "ANGUILLA ,AI", "ANTARCTICA ,AQ", "ANTIGUA AND BARBUDA ,AG", "ARGENTINA ,AR", "ARMENIA ,AM", "ARUBA ,AW", "AUSTRALIA ,AU", "AUSTRIA ,AT", "AZERBAIJAN ,AZ ",

                    "BAHAMAS ,BS", "BAHRAIN ,BH", "BANGLADESH ,BD", "BARBADOS ,BB", "BELARUS ,BY", "BELGIUM ,BE", "BELIZE ,BZ", "BENIN ,BJ", "BERMUDA ,BM", "BHUTAN ,BT", "BOLIVIA; PLURINATIONAL STATE OF ,BO", "BONAIRE; SINT EUSTATIUS AND SABA ,BQ", "BOSNIA AND HERZEGOVINA ,BA", "BOTSWANA ,BW", "BOUVET ISLAND ,BV", "BRAZIL ,BR", "BRITISH INDIAN OCEAN TERRITORY ,IO", "BRUNEI DARUSSALAM ,BN", "BULGARIA ,BG",

                    "BURKINA FASO ,BF", "BURUNDI ,BI ", "CAMBODIA ,KH", "CAMEROON ,CM", "CANADA ,CA", "CAPE VERDE ,CV", "CAYMAN ISLANDS ,KY", "CENTRAL AFRICAN REPUBLIC ,CF","CHAD ,TD", "CHILE ,CL", "CHINA ,CN", "CHRISTMAS ISLAND ,CX", "COCOS (KEELING) ISLANDS ,CC", "COLOMBIA ,CO", "COMOROS ,KM", "CONGO ,CG", "CONGO; THE DEMOCRATIC REPUBLIC OF THE ,CD", "COOK ISLANDS ,CK", "COSTA RICA ,CR", "CÔTE D'IVOIRE ,CI",

                    "CROATIA ,HR", "CUBA ,CU", "CURAÇAO ,CW", "CYPRUS ,CY", "CZECH REPUBLIC ,CZ ", "DENMARK ,DK", "DJIBOUTI ,DJ", "DOMINICA ,DM", "DOMINICAN REPUBLIC ,DO ", "ECUADOR ,EC", "EGYPT ,EG", "EL SALVADOR ,SV", "EQUATORIAL GUINEA ,GQ", "ERITREA ,ER", "ESTONIA ,EE", "ETHIOPIA ,ET ", "FALKLAND ISLANDS (MALVINAS) ,FK","FAROE ISLANDS ,FO", "FIJI ,FJ", "FINLAND ,FI", "FRANCE ,FR", "FRENCH GUIANA ,GF",

                    "FRENCH POLYNESIA ,PF", "FRENCH SOUTHERN TERRITORIES ,TF ", "GABON ,GA", "GAMBIA ,GM", "GEORGIA ,GE", "GERMANY ,DE", "GHANA ,GH", "GIBRALTAR ,GI", "GREECE ,GR", "GREENLAND ,GL", "GRENADA ,GD", "GUADELOUPE ,GP", "GUAM ,GU", "GUATEMALA ,GT", "GUERNSEY ,GG", "GUINEA ,GN", "GUINEA-BISSAU ,GW", "GUYANA ,GY ", "HAITI ,HT", "HEARD ISLAND AND MCDONALD ISLANDS ,HM", "HOLY SEE (VATICAN CITY STATE) ,VA",

                    "HONDURAS ,HN", "HONG KONG ,HK", "HUNGARY ,HU ", "ICELAND ,IS", "INDIA ,IN", "INDONESIA ,ID", "IRAN; ISLAMIC REPUBLIC OF ,IR", "IRAQ ,IQ", "IRELAND ,IE", "ISLE OF MAN ,IM", "ISRAEL ,IL", "ITALY ,IT ", "JAMAICA ,JM", "JAPAN ,JP", "JERSEY ,JE", "JORDAN ,JO ", "KAZAKHSTAN ,KZ", "KENYA ,KE", "KIRIBATI ,KI", "KOREA; DEMOCRATIC PEOPLE'S REPUBLIC OF ,KP", "KOREA; REPUBLIC OF ,KR", "KUWAIT ,KW", "KYRGYZSTAN ,KG ",

                    "LAO PEOPLE'S DEMOCRATIC REPUBLIC ,LA", "LATVIA ,LV", "LEBANON ,LB", "LESOTHO ,LS", "LIBERIA ,LR", "LIBYAN ARAB JAMAHIRIYA ,LY", "LIECHTENSTEIN ,LI", "LITHUANIA ,LT", "LUXEMBOURG ,LU ", "MACAO ,MO", "MACEDONIA; THE FORMER YUGOSLAV REPUBLIC OF ,MK", "MADAGASCAR ,MG", "MALAWI ,MW", "MALAYSIA ,MY", "MALDIVES ,MV", "MALI ,ML", "MALTA ,MT", "MARSHALL ISLANDS ,MH", "MARTINIQUE ,MQ", "MAURITANIA ,MR", "MAURITIUS ,MU",

                    "MAYOTTE ,YT", "MEXICO ,MX", "MICRONESIA; FEDERATED STATES OF ,FM", "MOLDOVA; REPUBLIC OF ,MD", "MONACO ,MC", "MONGOLIA ,MN", "MONTENEGRO ,ME", "MONTSERRAT ,MS", "MOROCCO ,MA", "MOZAMBIQUE ,MZ", "MYANMAR ,MM ", "NAMIBIA ,NA", "NAURU ,NR", "NEPAL ,NP", "NETHERLANDS ,NL", "NEW CALEDONIA ,NC", "NEW ZEALAND ,NZ", "NICARAGUA ,NI", "NIGER ,NE", "NIGERIA ,NG", "NIUE ,NU", "NORFOLK ISLAND ,NF", "NORTHERN MARIANA ISLANDS ,MP",

                    "NORWAY ,NO ", "OMAN ,OM ", "PAKISTAN ,PK", "PALAU ,PW", "PALESTINIAN TERRITORY; OCCUPIED ,PS", "PANAMA ,PA", "PAPUA NEW GUINEA ,PG", "PARAGUAY ,PY", "PERU ,PE", "PHILIPPINES ,PH", "PITCAIRN ,PN", "POLAND ,PL", "PORTUGAL ,PT", "PUERTO RICO ,PR ", "QATAR ,QA ", "RÉUNION ,RE", "ROMANIA ,RO", "RUSSIAN FEDERATION ,RU", "RWANDA ,RW ", "SAINT BARTHÉLEMY ,BL", "SAINT HELENA; ASCENSION AND TRISTAN DA CUNHA ,SH",

                    "SAINT KITTS AND NEVIS ,KN", "SAINT LUCIA ,LC", "SAINT MARTIN (FRENCH PART) ,MF", "SAINT PIERRE AND MIQUELON ,PM", "SAINT VINCENT AND THE GRENADINES ,VC", "SAMOA ,WS", "SAN MARINO ,SM", "SAO TOME AND PRINCIPE ,ST", "SAUDI ARABIA ,SA", "SENEGAL ,SN", "SERBIA ,RS", "SEYCHELLES ,SC", "SIERRA LEONE ,SL", "SINGAPORE ,SG", "SINT MAARTEN (DUTCH PART) ,SX", "SLOVAKIA ,SK", "SLOVENIA ,SI", "SOLOMON ISLANDS ,SB",

                    "SOMALIA ,SO", "SOUTH AFRICA ,ZA", "SOUTH GEORGIA AND THE SOUTH SANDWICH ISLANDS ,GS", "SPAIN ,ES", "SRI LANKA ,LK", "SUDAN ,SD", "SURINAME ,SR", "SVALBARD AND JAN MAYEN ,SJ ", "SWAZILAND ,SZ", "SWEDEN ,SE", "SWITZERLAND ,CH", "SYRIAN ARAB REPUBLIC ,SY ", "TAIWAN; PROVINCE OF CHINA ,TW", "TAJIKISTAN ,TJ", "TANZANIA; UNITED REPUBLIC OF ,TZ", "THAILAND ,TH", "TIMOR-LESTE ,TL", "TOGO ,TG", "TOKELAU ,TK", "TONGA ,TO",

                    "TRINIDAD AND TOBAGO ,TT", "TUNISIA ,TN", "TURKEY ,TR", "TURKMENISTAN ,TM", "TURKS AND CAICOS ISLANDS ,TC", "TUVALU ,TV ", "UGANDA ,UG", "UKRAINE ,UA", "UNITED ARAB EMIRATES ,AE", "UNITED KINGDOM ,GB", "UNITED STATES ,US", "UNITED STATES MINOR OUTLYING ISLANDS ,UM", "URUGUAY ,UY",

                    "UZBEKISTAN ,UZ ", "VANUATU ,VU ", "VENEZUELA; BOLIVARIAN REPUBLIC OF ,VE", "VIET NAM ,VN", "VIRGIN ISLANDS; BRITISH ,VG", "VIRGIN ISLANDS; U.S. ,VI ", "WALLIS AND FUTUNA ,WF", "WESTERN SAHARA ,EH ", "YEMEN ,YE ", "ZAMBIA ,ZM", "ZIMBABWE ,ZW " };

        Set<String> locCodes = new HashSet<String>(Arrays.asList(new String[] {"AF", "AX", "AL", "DZ", "AS", "AD", "AO", "AI", "AQ", "AG", "AR", "AM", "AW", "AU", "AT", "AZ", "BS", "BH", "BD", "BB", "BY", "BE", "BZ", "BJ", "BM", "BT", "BO", "BQ", "BA", "BW", "BV", "BR", "IO", "BN", "BG", "BF", "BI", "KH", "CM", "CA", "CV", "KY", "CF", "TD", "CL", "CN", "CX", "CC", "CO", "KM", "CG", "CD", "CK", "CR", "CI", "HR", "CU", "CW", "CY", "CZ", "DK", "DJ", "DM", "DO", "EC", "EG",

            "SV", "GQ", "ER", "EE", "ET", "FK", "FO", "FJ", "FI", "FR", "GF", "PF", "TF", "GA", "GM", "GE", "DE", "GH", "GI", "GR", "GL", "GD", "GP", "GU", "GT", "GG", "GN", "GW", "GY", "HT", "HM", "VA", "HN", "HK", "HU", "IS", "IN", "ID", "IR", "IQ", "IE", "IM", "IL", "IT", "JM", "JP", "JE", "JO", "KZ", "KE", "KI", "KP", "KR", "KW", "KG", "LA", "LV", "LB", "LS", "LR", "LY", "LI", "LT", "LU", "MO", "MK", "MG", "MW", "MY", "MV", "ML",

            "MT", "MH", "MQ", "MR", "MU", "YT", "MX", "FM", "MD", "MC", "MN", "ME", "MS", "MA", "MZ", "MM", "NA", "NR", "NP", "NL", "NC", "NZ", "NI", "NE", "NG", "NU", "NF", "MP", "NO", "OM", "PK", "PW", "PS", "PA", "PG", "PY", "PE", "PH", "PN", "PL", "PT", "PR", "QA", "RE", "RO", "RU", "RW", "BL", "SH", "KN", "LC", "MF", "PM", "VC", "WS", "SM", "ST", "SA", "SN", "RS", "SC", "SL", "SG", "SX", "SK", "SI", "SB", "SO", "ZA", "GS", "ES",

            "LK", "SD", "SR", "SJ", "SZ", "SE", "CH", "SY", "TW", "TJ", "TZ", "TH", "TL", "TG", "TK", "TO", "TT", "TN", "TR", "TM", "TC", "TV", "UG", "UA", "AE", "GB", "US", "UM", "UY", "UZ", "VU", "VE", "VN", "VG", "VI", "WF", "EH", "YE", "ZM", "ZW"}));

        try {

            while (!exit)

            {

                String dname = "";

                String dloc = "";

                String ddesc = "";

                out.println("\nAdd a new datacenter:");

                out.println("\nEnter a name - enter nothing to exit");

                userInput = in.readLine();

                if(userInput.equals(""))

                    exit = true;

                else

                {

                    out.println("\nAdding a new datacenter called "+userInput);

                    dname = userInput;

                    while (dloc.equals("") && !exit)

                    {

                        out.println("\nEnter the code of its location (2 letters): Enter c to get a list of codes, otherwise enter the chosen code (2 letters)");

                        userInput = in.readLine();

                        if (userInput.equals("")){

                           exit = true;

                           out.println("\nExiting... ");

                        }

                        else if(userInput.equalsIgnoreCase("c"))

                           out.println(Arrays.toString(codes));

                       else if(userInput.length() != 2)

                           out.println("\nCode should be 2 letters' length");

                       else if(!locCodes.contains(userInput.toUpperCase()))

                           out.println("\nthis code is not in the authorized list");

                       else

                       {

                           out.println("\nChosen code: "+userInput);

                           dloc = userInput;

                           out.println("\nEnter a description on one line (facultative):");

                           ddesc = in.readLine();

                           out.println("\nAdding a new datacenter with these values:\nName: "+dname+"\nLoc: "+dloc+"\nDesc:"+ddesc);

                           userInput = "";

                           boolean recheck = true;

                           while ((userInput == null)||recheck){

                               out.println("\nConfirm? (y)es/(n)o");

                               userInput = in.readLine();

                               if(userInput.equalsIgnoreCase("y") || userInput.equalsIgnoreCase("n"))

                                   recheck = false;

                           }

                           if(userInput.equalsIgnoreCase("y")){

                                ResultSet rs = dbhandle.query("select", "max(did)", "datacenter", "");

                                int did = 0;

                                if(rs.next())

                                    did = rs.getInt(1) + 1;

                                logger.debug("New datacenter id to be assigned: " + did);

                                boolean status = dbhandle.insert("datacenter", "(" + did + ", '" + dloc.toUpperCase() + "', '" + dname + "', '" + ddesc + "')");

                                if(status)

                                {

                                    out.println("\nDatacenter was added successfully with datacenter ID: " + did + "\n");

                                    logger.debug("Datacenter was added successfully with datacenter ID: " + did);

                                }

                                else

                                {

                                        logger.error("Datacenter " + did + " addition failed. Error during insert operation into datacenter table.");

                                        exit = true;

                                }

                           }

                           else

                           {

                               exit = true;

                               out.println("\nExiting...");

                           }

                       }

                    }

               }

            }

        } catch (Exception ex) {

            logger.error("Caught exception while adding a datacenter inside the CLI process: " + ex.getMessage());

            ex.printStackTrace(System.err);

        }

    }

    private void addCluster(PrintWriter out, BufferedReader in) {

        boolean exit = false;

        String userInput = "";

        String[] codes = new String[] { "000, no selection", "001, Gigabit Ethernet", "002, Gigabit Ethernet (Level 5 NICs)", "003, 10 Gigabit Ethernet", "004, Infiniband", "005, Infinipath", "006, Myrinet", "007, QsNet (Quadrics)", "008, SCI (Dolphin)" };

        Set<String> nics = new HashSet(Arrays.asList(new String[]{"000","001","002","003","004","005","006","007","008"}));

        Map<String, String> dcenters = new HashMap<String,String>();

        try {

            int count = 0;

            ResultSet rs = dbhandle.query("SELECT", "did, name", "datacenter", "");

            while(rs.next())

            {

                dcenters.put(Integer.toString(rs.getInt("did")), rs.getString("name"));

                count++;

            }

            if(count == 0)

            {

                logger.warn("No datacenter was found.");

                out.println("\nThere must be at least 1 datacenter to connect this cluster to, please add one.");

                exit = true;

            }

            while(!exit){

                String cname = "";

                String cnic = "";

                String did = "";

                String cdesc = "";

                out.println("\nAdd a new cluster:");

                out.println("\nEnter a name - enter nothing to exit");

                userInput = in.readLine();

                if(userInput.equals(""))

                    exit = true;

                else

                {

                    out.println("\nAdding a new cluster called "+userInput);

                    cname = userInput;

                    while(cnic.equals("") && !exit){

                        out.println("\nChoose interconnection:  Enter c to get a list of codes, otherwise enter the chosen code (3 digits)");

                        userInput = in.readLine();

                        if (userInput.equals("")){

                           exit = true;

                           out.println("\nExiting... ");

                        }

                        else if(userInput.equalsIgnoreCase("c"))

                           out.println(Arrays.toString(codes));

                       else if(userInput.length() != 3)

                           out.println("\nCode should be 3 digits' length");

                       else if(!nics.contains(userInput.toUpperCase()))

                           out.println("\nthis code is not in the authorized list");

                       else

                       {

                           out.println("\nChosen code: "+userInput);

                           cnic = userInput;

                           while(did.equals("") && !exit){

                               out.println("\nChoose the datacenter this cluster belongs to: Enter c to get a list of codes, otherwise enter the chosen code (id)");

                               userInput = in.readLine();

                               if (userInput.equals("")){

                                  exit = true;

                                  out.println("\nExiting... ");

                               }

                               else if(userInput.equalsIgnoreCase("c"))

                              {

                                  out.println("\nid : name");

                                  for (Map.Entry<String, String> e : dcenters.entrySet())

                                      out.println(e.getKey()+" : "+e.getValue());

                              }

                              else if(!dcenters.containsKey(userInput))