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

 import java.util.ArrayList;

 import java.util.List;

 import com.sonicsw.deploy.IArtifact;

 import com.sonicsw.deploy.IArtifactTraverser;

 import com.sonicsw.deploy.artifact.ESBArtifact;

 import com.sonicsw.deploy.tools.common.ExportPropertiesArtifact;

 import com.sonicsw.deploy.traversal.AbstractTraverser;

 import com.sonicsw.deploy.traversal.TraverserContext;

 import com.sonicsw.deploy.traversal.TraverserFactory;

 import com.sonicsw.esb.mgmtapi.ESBAPI;

 import com.sonicsw.esb.mgmtapi.ESBAPIFactory;

 import com.sonicsw.esb.mgmtapi.config.IContainerConfig;

 import com.sonicsw.esb.mgmtapi.config.IProcessConfig;

 import com.sonicsw.esb.mgmtapi.config.IProcessConfigAPI;

 import com.sonicsw.esb.mgmtapi.config.IServiceConfig;

 import com.sonicsw.esb.mgmtapi.config.IServiceConfigAPI;

 import com.progress.sonic.utilities.mfutils.MFUtils;

 /**

  * 

  */

 public class ProcessDependencyFinder {

   // TODO Remove references to Toyota

   private String[] processName = { "testProcess", "ESBPluginTester", "Proc.Dealer_CarLookup",

     "Proc.RecreateToyotaEnvironment" };

   private MFUtils  mfutil;

   private boolean  failOnError = true;

   /**

    * @param _args

    *          command line arguments

    */

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

     // TODO Auto-generated method stub

     ProcessDependencyFinder pdf = new ProcessDependencyFinder();

     // pdf.connect();

     pdf.go();

     // pdf.disconnect();

   }

   private void disconnect() {

     mfutil.cleanup();

   }

   private void connect() {

     mfutil = new MFUtils("Domain1", "tcp://localhost:2506", "Administrator", "Administrator");

   }

   private void go() {

     connect();

     ExportPropertiesArtifact exportprop = new ExportPropertiesArtifact();

     for (String process : processName) {

       System.out.println("Searching for process: " + process);

       IArtifact artifact = new ESBArtifact(ESBArtifact.PROCESS, process);

       exportprop.addRoot(artifact);

     }

     IArtifact[] deptree = lookup(exportprop);

     String[][] depitems = filterContainerItems(deptree);

     depitems[0] = processName;

     listArtifacts(depitems);

     disconnect();

     addArtifactsToContainer(depitems);

   }

   private void addArtifactsToContainer(final String[][] _depitems) {

     System.out.println("addArtifactsToContainer");

     // Create connection

     ESBAPI eapi = null;

     try {

       System.out.println("Creating ESB API Factory");

       ESBAPIFactory efac = ESBAPIFactory.createESBAPIFactory();

       System.out.println("Factory API Created");

       System.out.println("Creating the API Handle");

       eapi = efac.createAPI("Domain1", "tcp://localhost:2506", "Administrator", "Administrator");

       System.out.println("API Handle Created");

       System.out.println("Getting the Container");

       IContainerConfig esbContainer = eapi.getContainerConfigAPI().getESBContainer("TestDomainESB");

       System.out.println("Adding the Services");

       for (String service : _depitems[2]) {

         System.out.println("Adding Service: " + service);

         IServiceConfigAPI svcconf = eapi.getServiceConfigAPI();

         IServiceConfig svc = svcconf.getServiceConfig(service);

         try {

           svc.addToContainer(esbContainer, 1);

           eapi.getContainerConfigAPI().saveESBContainer(esbContainer);

         } catch (Exception e) {

           System.out.println("MSG: " + e.getMessage());

         }

       }

       System.out.println("Adding the Main Processes");

       for (String process : _depitems[0]) {

         System.out.println("Adding Process: " + process);

         IProcessConfigAPI proconf = eapi.getProcessConfigAPI();

         IProcessConfig pro = proconf.getProcess(process);

         if (pro.getEntryEndpointConfig() == null) {

           System.out.println("No Endpoint Config found ... skipping the add to the container");

           continue;

         } else {

           if (pro.getEntryEndpointConfig().getName() == null || pro.getEntryEndpointConfig().getName().equals("")) {

             System.out.println("No Endpoint Name Found ... skipping the add to the container");

             continue;

           }

         }

         try {

           pro.addToContainer(esbContainer, 1);

           eapi.getContainerConfigAPI().saveESBContainer(esbContainer);

         } catch (Exception e) {

           System.out.println("MSG: " + e.getMessage());

         }

       }

       System.out.println("Adding the sub Processes");

       for (String process : _depitems[1]) {

         System.out.println("Adding Process: " + process);

         IProcessConfigAPI proconf = eapi.getProcessConfigAPI();

         IProcessConfig pro = proconf.getProcess(process);

         if (pro.getEntryEndpointConfig() == null) {

           System.out.println("No Endpoint Config found ... skipping the add to the container");

           continue;

         } else {

           if (pro.getEntryEndpointConfig().getName() == null || pro.getEntryEndpointConfig().getName().equals("")) {

             System.out.println("No Endpoint Name Found ... skipping the add to the container");

             continue;

           }

         }

         try {

           pro.addToContainer(esbContainer, 1);

           eapi.getContainerConfigAPI().saveESBContainer(esbContainer);

         } catch (Exception e) {

           System.out.println("MSG: " + e.getMessage());

         }

       }

     } catch (Exception e) {

       System.out.println("Error happened: " + e.getMessage());

     } finally {

       eapi.dispose();

     }

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

     // add the services

     // add the processes (if possible as they need an Entry Endpoint

   }

   private String[][] filterContainerItems(final IArtifact[] _deptree) {

     String[][] returnList = new String[3][];

     int procCounter = countProcesses(_deptree);

     returnList[1] = new String[procCounter];

     int svcCounter = countSvc(_deptree);

     returnList[2] = new String[svcCounter];

     int countsvc = 0;

     int countpro = 0;

     for (IArtifact a : _deptree) {

       if (a.getPath().startsWith("/Services/")) {

         returnList[2][countsvc] = a.getName();

         countsvc++;

       }

       if (a.getPath().startsWith("/Processes/")) {

         returnList[1][countpro] = a.getName();

         countpro++;

       }

     }

     return returnList;

   }

   private int countSvc(final IArtifact[] _deptree) {

     int count = 0;

     for (IArtifact a : _deptree) {

       if (a.getPath().startsWith("/Services/")) {

         count++;

       }

     }

     return count;

   }

   private int countProcesses(final IArtifact[] _deptree) {

     int count = 0;

     for (IArtifact a : _deptree) {

       if (a.getPath().startsWith("/Processes/")) {

         count++;

       }

     }

     return count;

   }

   private void listArtifacts(final String[][] _depitems) {

     System.out.println("Listing Main Processes");

     for (String art : _depitems[0]) {

       System.out.println("Process Name: " + art);

     }

     System.out.println("Listing Processes");

     for (String art : _depitems[1]) {

       System.out.println("Process Name: " + art);

     }

     System.out.println("Listing Services");

     for (String art : _depitems[2]) {

       System.out.println("Service Name: " + art);

     }

   }

   private IArtifact[] lookup(final ExportPropertiesArtifact _processes) {

     ExportPropertiesArtifact processes = _processes;

     if (processes == null) {

       processes = new ExportPropertiesArtifact();

     }

     List<IArtifact> ignores = filterIgnoreList(processes.getIgnoreList());

     processes.clearIgnoreList();

     processes.addIgnore(ignores.toArray(new IArtifact[] {}));

     TraverserContext ctxt;

     try {

       ctxt = new TraverserContext(mfutil.getDSArtifactStorage(), processes.getIgnoreList());

       ctxt.setTraverseCompressed(false);

       for (int i = 0; i < processes.getRootList().length; i++) {

         IArtifactTraverser traverser = TraverserFactory.createTraverser(processes.getRootList()[i]);

         ((AbstractTraverser) traverser)._traverse(ctxt);

       }

       IArtifact[] errored = ctxt.getErrored();

       if (errored.length > 0) {

         StringBuffer buf = new StringBuffer();

         buf.append("The following Artifacts could not be traversed : [\n");

         for (IArtifact error : errored) {

           buf.append("  " + error.getDisplayType() + ":" + error.getPath() + "\n");

         }

         buf.append("]\n");

         if (failOnError) {

           throw new Exception(buf.toString());

         }

       }

       IArtifact[] result = ctxt.completeTraversal();

       return result;

     } catch (Exception e) {

       // TODO Auto-generated catch block

       e.printStackTrace();

     }

     return null;

   }

   /**

    * This method calculates the default ignore list for exporting a xar file from a directory service. We will use the

    * Sonic default ignore list EXCEPT for the connections and sonicfs:///System roots.

    * 

    * @return The ignore list for generating the XAR archive.

    */

   private List<IArtifact> filterIgnoreList(final IArtifact[] _ignores) {

     List<IArtifact> result = new ArrayList<IArtifact>();

     List<String> invalidIgnores = new ArrayList<String>();

     invalidIgnores.add("/Connections/http_defaultConnection");

     invalidIgnores.add("/Connections/jms_defaultConnection");

     invalidIgnores.add("/System/");

     for (IArtifact a : _ignores) {

       if (!invalidIgnores.contains(a.getPath())) {

         result.add(a);

       }

     }

     return result;

   }

 }