package com.aurea.maven.plugins.sonic.utils;

 import java.io.BufferedReader;

 import java.io.File;

 import java.io.InputStreamReader;

 import java.util.Hashtable;

 import javax.management.ObjectName;

 import com.sonicsw.mf.jmx.client.JMSConnectorAddress;

 import com.sonicsw.mf.jmx.client.JMSConnectorClient;

 import com.sonicsw.mf.mgmtapi.runtime.IAgentProxy;

 import com.sonicsw.mf.mgmtapi.runtime.IDirectoryServiceProxy;

 import com.sonicsw.mf.mgmtapi.runtime.MFProxyFactory;

 import com.sonicsw.mf.mgmtapi.runtime.ProxyRuntimeException;

 import com.sonicsw.mq.mgmtapi.runtime.IBrokerProxy;

 import com.sonicsw.mq.mgmtapi.runtime.MQProxyFactory;

 /**

  *

  */

 public class SonicInvoker extends JavaInvoker {

   /**

    * @param _sonicDirectory

    *          _sonicDirectory

    * @throws Exception

    *           Exception

    */

   public SonicInvoker(final String _sonicDirectory) throws Exception {

     super("com.sonicsw.mf.framework.agent.ci.LaunchContainer");

     setWorkingDir(_sonicDirectory);

     configureSonicRuntime();

   }

   /**

    * @throws Exception

    *           Exception

    */

   public final void configureSonicRuntime() throws Exception {

     File sonicDir = new File(getWorkdir());

     checkDir(sonicDir);

     File sonicLib = new File(sonicDir, "lib");

     checkDir(sonicLib);

     addToBootClassPath(new File(sonicLib, "jmxri.jar").getAbsolutePath());

     addToClassPath(new File(sonicLib, "MFLaunch.jar").getAbsolutePath());

     addToClassPath(new File(sonicLib, "rsa_ssl.jar").getAbsolutePath());

     setProperty("java.protocol.handler.pkgs", "progress.message.net");

     setProperty("SSL_CA_CERTIFICATES_DIR", new File(sonicDir, "certs/CA").getAbsolutePath());

     setProperty("sonicsw.home", getWorkdir());

     setProperty("sonicsw.mf.allowRestart", "true");

     addArgument("container.xml");

     addArgument("ds.xml");

   }

   private void checkDir(final File _dir) throws Exception {

     if (!_dir.exists()) {

       throw new Exception("Directory " + _dir.getAbsolutePath() + " does not exist.");

     }

     if (!_dir.isDirectory()) {

       throw new Exception("Directory " + _dir.getAbsolutePath() + " is not a directory.");

     }

   }

   /**

    * @param _args

    *          command line arguments

    */

   @SuppressWarnings("deprecation")

 public static void main(final String[] _args) {

     try {

       System.out.println("Invoking");

       // SonicInvoker invoker = new

       // SonicInvoker("C:\\Playground\\Development\\ESBPlugin\\SonicBin\\MQ7.6");

       // invoker.invoke();

       ProcessBuilder process = new ProcessBuilder("C:\\Software\\Sonic7.6\\TestImage\\SonicData\\startCtManagement.bat");

       process.directory(new File("C:\\Software\\Sonic7.6\\TestImage\\SonicData"));

       Process p = process.start();

       BufferedReader br = new BufferedReader(new InputStreamReader(p.getInputStream()));

       String input;

       while (!(input = br.readLine()).toUpperCase().contains("STARTUP COMPLETE")) {

         System.out.println("Sonic: " + input);

       }

       System.out.println("Found: " + input + "\nContinuing");

       System.out.println("Done Invoking");

       br.close();

       System.out.println("Sleeping");

       Thread.sleep(10000);

       // performing the configuration

       System.out.println("Performing the configuration");

       OffLineDSConfigurator ods = new OffLineDSConfigurator();

       ods.go();

       System.out.println("Sleeping for shutdown");

       Thread.sleep(10000);

       System.out.println("Shutting Down");

       JMSConnectorClient connector = null;

       try {

         // get a JMX connector

         System.out.println("Get Connector");

         connector = getConnector("tcp://localhost:3506", "Administrator", "Administrator");

         // create a proxy to the broker component

         System.out.println("Agent Proxy");

         IAgentProxy agent = getAgentProxy(connector, "Domain1", "ctManagement");

         // shutdown the container

         System.out.println("shutdown");

         agent.shutdown();

         System.out.println("Shutdown Done");

       } catch (Throwable t) {

         t.printStackTrace();

         if (t instanceof ProxyRuntimeException) {

           System.err.println("Caused by:");

           ((ProxyRuntimeException) t).getTargetException().printStackTrace();

         }

       } finally {

         if (connector != null) {

           connector.disconnect();

         }

       }

     } catch (Exception e) {

       e.printStackTrace();

     }

   }

   /**

    * Gets a connected JMS/JMX Connector Client for management communications with components exposed as JMX MBeans. The

    * urls, user and pwd (password) parameters define connection parameters to the broker(s) through which management

    * communications will travel.

    * 

    * @param _urls

    *          urls

    * @param _user

    *          user

    * @param _pwd

    *          pwd

    * @return JMSConnectorClient

    * @throws Exception

    *           Exception

    */

   public static JMSConnectorClient getConnector(final String _urls, final String _user, final String _pwd)

     throws Exception {

     Hashtable<String, String> env = new Hashtable<String, String>();

     env.put("ConnectionURLs", _urls);

     env.put("DefaultUser", _user);

     env.put("DefaultPassword", _pwd);

     JMSConnectorAddress address = new JMSConnectorAddress(env);

     JMSConnectorClient connector = new JMSConnectorClient();

     connector.connect(address, 10000);

     return connector;

   }

   /**

    * Gets a management proxy for an AGENT component; the AGENT component is used to manage a container.

    * 

    * @param _connector

    *          _connector

    * @param _domain

    *          _domain

    * @param _container

    *          _container

    * @return IAgentProxy

    * @throws Exception

    *           Exception

    */

   public static IAgentProxy getAgentProxy(final JMSConnectorClient _connector, final String _domain,

     final String _container) throws Exception {

     ObjectName jmxName = new ObjectName(_domain + "." + _container + ":ID=AGENT");

     return MFProxyFactory.createAgentProxy(_connector, jmxName);

   }

   /**

    * Gets a management proxy for the Directory Service component; the DS component is used to manage the domain

    * configuration store.

    * 

    * @param _connector

    *          _connector

    * @param _domain

    *          _domain

    * @return IDirectoryServiceProxy

    * @throws Exception

    *           Exception

    */

   public static IDirectoryServiceProxy getDirectoryServiceProxy(final JMSConnectorClient _connector,

     final String _domain) throws Exception {

     ObjectName jmxName = new ObjectName(_domain + ".DIRECTORY SERVICE:ID=DIRECTORY SERVICE");

     return MFProxyFactory.createDirectoryServiceProxy(_connector, jmxName);

   }

   /**

    * Gets a management proxy for a SOnicMQ broker component.

    * 

    * @param _connector

    *          _connector

    * @param _domain

    *          _domain

    * @param _container

    *          _container

    * @param _id

    *          _id

    * @return IBrokerProxy

    * @throws Exception

    *           Exception

    */

   public static IBrokerProxy getBrokerProxy(final JMSConnectorClient _connector, final String _domain,

     final String _container, final String _id) throws Exception {

     ObjectName jmxName = new ObjectName(_domain + "." + _container + ":ID=" + _id);

     return MQProxyFactory.createBrokerProxy(_connector, jmxName);

   }

 }