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

 import java.io.ByteArrayInputStream;

 import java.io.File;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.HashSet;

 import org.codehaus.plexus.util.xml.XmlStreamReader;

 import org.codehaus.plexus.util.xml.Xpp3Dom;

 import org.codehaus.plexus.util.xml.Xpp3DomBuilder;

 import org.codehaus.plexus.util.xml.pull.XmlPullParserException;

 /**

  * @author Jamie Townsend

  * 

  */

 public class ProjectAnalyzer {

   /**

    * gets a list of .esbp process files under a given base location.

    * 

    * @param _baseDir

    *          base location to search from

    * @return list of process filenames

    */

   public static File[] getProcessSourceFiles(final String _baseDir) {

     String[] processFileLocations = FileUtilities.getFileList(_baseDir, "**/*.esbp", null);

     File[] processFiles = new File[processFileLocations.length];

     for (int index = 0; index < processFileLocations.length; index++) {

       processFiles[index] = new File(_baseDir, processFileLocations[index]);

     }

     return processFiles;

   }

   /**

    * gets a list of process names located under a given base location.

    * 

    * @param _baseDir

    *          base location to search for process definitions from

    * @return a list of process names

    */

   public static HashSet<String> getProcessNames(final String _baseDir) {

     HashSet<String> processNames = new HashSet<String>();

     for (File processFile : getProcessSourceFiles(_baseDir)) {

       String currProcName = getProcessNameFromFile(processFile);

       processNames.add(currProcName);

     }

     return processNames;

   }

   /**

    * gets a list of .esbsvc Service Config files under a given base location.

    * 

    * @param _baseDir

    *          base location to search from

    * @return list of Service Config filenames

    */

   public static File[] getServiceConfigSourceFiles(final String _baseDir) {

     String[] serviceConfigFileLocations = FileUtilities.getFileList(_baseDir, "**/*.esbsvc", null);

     File[] serviceConfigFiles = new File[serviceConfigFileLocations.length];

     for (int index = 0; index < serviceConfigFileLocations.length; index++) {

       serviceConfigFiles[index] = new File(_baseDir, serviceConfigFileLocations[index]);

     }

     return serviceConfigFiles;

   }

   /**

    * gets a list of .esbstyp Service Type files under a given base location.

    * 

    * @param _baseDir

    *          base location to search from

    * @return list of Service Type filenames

    */

   public static File[] getServiceTypeSourceFiles(final String _baseDir) {

     String[] serviceTypeFileLocations = FileUtilities.getFileList(_baseDir, "**/*.esbstyp", null);

     File[] serviceTypeFiles = new File[serviceTypeFileLocations.length];

     for (int index = 0; index < serviceTypeFileLocations.length; index++) {

       serviceTypeFiles[index] = new File(_baseDir, serviceTypeFileLocations[index]);

     }

     return serviceTypeFiles;

   }

   /**

    * @param _baseDir

    *          base location to search from

    * @return list of sub process names referenced in the defined processes

    */

   public static HashSet<String> getSubProcessesOfProcesses(final File _baseDir) {

     HashSet<String> subProcesses = new HashSet<String>();

     ArrayList<String> subProcessesReferenced = new ArrayList<String>();

     for (File processFile : getProcessSourceFiles(_baseDir.getAbsolutePath())) {

       subProcesses.addAll(getSubProcessesOfProcess(processFile));

       subProcessesReferenced.addAll(getSubProcessesOfProcess(processFile));

     }

     return subProcesses;

   }

   /**

    * @param _processFile

    *          the process file to be searched for sub-process references.

    * @return a HashSet of sub-process names directly referenced in the _processFile

    */

   public static HashSet<String> getSubProcessesOfProcess(final File _processFile) {

     HashSet<String> hs = new HashSet<String>();

     try {

       XmlStreamReader reader = new XmlStreamReader(_processFile);

       Xpp3Dom processDom = Xpp3DomBuilder.build(reader);

       Xpp3Dom itinerary = processDom.getChild("xq:itinerary");

       Xpp3Dom[] steps = itinerary.getChildren();

       for (Xpp3Dom currStep : steps) {

         if ("xq:step".equals(currStep.getName()) && "PROCESS".equals(currStep.getAttribute("type"))) {

           hs.add(currStep.getAttribute("endpoint_ref"));

         }

       }

     } catch (IOException e) {

       System.out.println("[ERROR] Could not read process source file: " + _processFile);

       e.printStackTrace();

     } catch (XmlPullParserException e) {

       System.out.println("[ERROR] Could not parse process source file: " + _processFile);

       e.printStackTrace();

     }

     return hs;

   }

   public static HashSet<String> getServiceConfigsOfProcess(final String _processDefinition) {

     HashSet<String> hs = new HashSet<String>();

     try {

       XmlStreamReader reader = new XmlStreamReader(new ByteArrayInputStream(_processDefinition.getBytes()));

       Xpp3Dom processDom = Xpp3DomBuilder.build(reader);

       Xpp3Dom itinerary = processDom.getChild("xq:itinerary");

       Xpp3Dom[] steps = itinerary.getChildren();

       for (Xpp3Dom currStep : steps) {

         if ("xq:step".equals(currStep.getName()) && "SERVICE".equals(currStep.getAttribute("type"))) {

           hs.add(currStep.getAttribute("endpoint_ref"));

         }

       }

     } catch (IOException e) {

       System.out.println("[ERROR] Could not read process");

       e.printStackTrace();

     } catch (XmlPullParserException e) {

       System.out.println("[ERROR] Could not parse process");

       e.printStackTrace();

     }

     return hs;

   }

   /**

    * returns the name of a process from a given file name.

    * 

    * @param _processFile

    *          path to the file to be identified

    * @return the name of the process

    */

   public static String getProcessNameFromFile(final File _processFile) {

     String processName = null;

     // this could arguably be done be just return the filename with the file extension,

     // but actually parsing the file is a more secure way

     try {

       XmlStreamReader reader = new XmlStreamReader(_processFile);

       Xpp3Dom processDom = Xpp3DomBuilder.build(reader);

       processName = processDom.getAttribute("name");

     } catch (IOException e) {

       // getLog().error("Could not read process file: " + _processFile);

       System.out.println("[ERROR] Could not read process file: " + _processFile.getName());

     } catch (XmlPullParserException e) {

       // getLog().error("Could not parse process file: " + _processFile);

       System.out.println("[ERROR] Could not parse process file: " + _processFile.getName());

     }

     return processName;

   }

   /**

    * returns the name of a Service Configuration from a given file name.

    * 

    * @param _serviceConfigFile

    *          path to the file to be identified

    * @return the name of the Service Configuration

    */

   public static String getServiceConfigNameFromFile(final File _serviceConfigFile) {

     String serviceName = null;

     try {

       XmlStreamReader reader = new XmlStreamReader(_serviceConfigFile);

       Xpp3Dom serviceConfigDom = Xpp3DomBuilder.build(reader);

       serviceName = serviceConfigDom.getAttribute("name");

     } catch (IOException e) {

       System.out.println("[ERROR] Could not read Service Configuration file: " + _serviceConfigFile.getName());

     } catch (XmlPullParserException e) {

       System.out.println("[ERROR] Could not parse Service Configuration file: " + _serviceConfigFile.getName());

     }

     return serviceName;

   }

   /**

    * returns the name of a endpoint ref of a Service Configuration from a given file name.

    * 

    * @param _serviceConfigFile

    *          path to the file to be identified

    * @return the name of the endpoint ref of a Service Configuration

    */

   public static String getServiceConfigEntryRefFromFile(final File _serviceConfigFile) {

     String entryRefName = null;

     try {

       XmlStreamReader reader = new XmlStreamReader(_serviceConfigFile);

       Xpp3Dom serviceConfigDom = Xpp3DomBuilder.build(reader);

       Xpp3Dom entryRef = serviceConfigDom.getChild("xq:entry_ref");

       if (entryRef != null) {

               entryRefName = entryRef.getValue();

       }

     } catch (IOException e) {

       System.out.println("[ERROR] Could not read Service Configuration file: " + _serviceConfigFile.getName());

     } catch (XmlPullParserException e) {

       System.out.println("[ERROR] Could not parse Service Configuration file: " + _serviceConfigFile.getName());

     }

     return entryRefName;

   }

 }