/**

  * 

  */

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

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.FileNotFoundException;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.util.Collections;

 import java.util.Enumeration;

 import java.util.Properties;

 import java.util.Vector;

 import org.apache.maven.artifact.handler.ArtifactHandler;

 import org.apache.maven.execution.MavenSession;

 import org.apache.maven.plugin.BuildPluginManager;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.MojoFailureException;

 import org.codehaus.plexus.archiver.ArchiverException;

 import org.codehaus.plexus.archiver.zip.ZipArchiver;

 import org.codehaus.plexus.util.FileUtils;

 import org.twdata.maven.mojoexecutor.MojoExecutor;

 import com.aurea.maven.plugins.sonic.utils.FileUtilities;

 /**

  * @author Jamie Townsend

  * 

  * @goal connect-package

  */

 public class ConnectPackageMojo extends EsbPackageMojo {

   /**

    * The Maven Session Object (only required for unpacking dependencies with the maven-unpack-plugin).

    * 

    * @parameter default-value="${session}"

    * @required

    * @readonly

    */

   private MavenSession    session;

   /**

    * The Maven PluginManager Object (only required for unpacking dependencies with the maven-unpack-plugin).

    * 

    * @component

    * @required

    */

   protected BuildPluginManager pluginManager;

   /**

    * @component

    */

   ArtifactHandler         handler;

   /**

    * the spring.xml configuration file for Sonic Connect

    * 

    * @parameter default-value="config/spring.xml"

    */

   protected String        connectSpringLocation;

   /**

    * the project.properties configuration file for Sonic Connect

    * 

    * @parameter default-value="${project.artifactId}.properties"

    */

   protected String        connectPropertiesLocation;

   /**

    * the .services folder for Sonic Connect services (expose/invoke)

    * 

    * @parameter default-value=".services"

    */

   protected String        connectServicesFolder;

   // /**

   // * borrowed from Eclipse sonicconnect plugin /resrc/defaultArchiveFilter.properties

   // *

   // * - added the Maven target and Sonic deploy directory.

   // *

   // * @parameter expression="${sonicconnect.excludeDirs}" default-value=".settings,bin,classes,target,deploy"

   // */

   // private String excludeDirs;

   //

   // /**

   // * borrowed from Eclipse sonicconnect plugin /resrc/defaultArchiveFilter.properties

   // *

   // * @parameter expression="${sonicconnect.excludeFileExtensions}" default-value=

   // * ".classpath,.project,.class,.esbp,.esbstyp,.java,.properties,.csar"

   // */

   // private String excludeFileExtensions;

   /**

    * {@inheritDoc}

    */

   @Override

   protected void doExecute() throws MojoExecutionException, MojoFailureException {

     getLog().info("Packaging Sonic Connect artifact");

     project.getArtifact().setArtifactHandler(handler);

     // getLog().info(project.getArtifact().getArtifactHandler().getLanguage());

     try {

       createConnectFiles();

       copyLibFilesToPackageDir();

       super.doExecute();

     } catch (final MojoExecutionException e) {

       getLog().error("Error packaging Sonic Connect artifact : " + e.getMessage(), e);

       throw new MojoExecutionException("Error packaging Sonic Connect artifact : ", e);

     }

   }

   private void copyLibFilesToPackageDir() throws MojoExecutionException {

     final String targetLibDir = getPackageXarDir() + "/SonicFS/workspace/" + getProjectName() + "/lib";

     try {

       FileUtilities.copyFiles(project.getBasedir() + "/lib", targetLibDir);

     } catch (final Exception e) {

       // this should never happen

       e.printStackTrace();

       throw new MojoExecutionException("Error copy files from lib directory: " + e.getMessage(), e);

     }

   }

   /**

    * creates the .csar and .properties files associated with Sonic Connect projects.

    * 

    * @throws MojoExecutionException

    *           if something goes wrong

    */

   protected void createConnectFiles() throws MojoExecutionException {

     final String packagedConnectFileTemplate = project.getBasedir() + "/lib/" + getProjectName();

     final File connectCsarFile = new File(packagedConnectFileTemplate + ".csar");

     final File connectPropertiesFile = new File(packagedConnectFileTemplate + ".properties");

     final String baseDir = project.getBasedir().getAbsolutePath();

     // create the directory if it doesn't already exist

     connectCsarFile.getParentFile().mkdirs();

     // delete the files if they exist to ensure they get re-generated

     connectCsarFile.delete();

     connectPropertiesFile.delete();

     refreshDependenciesInLib();

     // create the minimal csar file

     try {

       final ZipArchiver za = new ZipArchiver();

       za.setDestFile(connectCsarFile);

       // add spring.xml

       za.addFile(new File(baseDir, connectSpringLocation), "config/spring.xml");

       // add .esbcamel files

       final String servicesDir = new File(baseDir, connectServicesFolder).getPath();

       final String[] esbCamelFiles = FileUtilities.getFileList(servicesDir, "**/*.esbcamel", null);

       for (String fileName : esbCamelFiles) {

               final File file = new File(servicesDir, fileName);

               za.addFile(file, ".services/" + fileName);

       }

       // add all .wsdl files (note that this will pick up all .wsdl files, regardless of location, including

       // target/**/.wsdl. also added the .xsd files as wsdl may have dependencies on the local xsd NVR)

       final String[] wsdlFileNames = FileUtilities.getFileList(baseDir,

           "**/*.wsdl,**/*.xsd", null);

       for (final String fileName : wsdlFileNames) {

         za.addFile(new File(baseDir, fileName), fileName);

       }

       // add the artifact jar (if one exists)

       final String jarFileName = project.getArtifactId() + "-" + project.getVersion() + ".jar";

       za.addFile(new File(project.getBuild().getDirectory(), jarFileName), jarFileName);

       // add any dependent jar's

       final String[] dependentJars = {

         "*.jar"

       };

       za.addDirectory(new File(baseDir, "lib"), dependentJars, null);

       za.createArchive();

     } catch (final ArchiverException e) {

             throw new MojoExecutionException(e.getMessage(), e);

     } catch (final IOException e) {

             throw new MojoExecutionException(e.getMessage(), e);

       }

     // create the csar's properties file

     final SortedProperties completePropertyList = new SortedProperties();

     try {

       // add the ${artifactId}.properties from the root directory

       final String connectProperties = connectPropertiesLocation != null ? connectPropertiesLocation :

               project.getArtifactId() + ".properties";

       final String[] topLevelPropertiesFileNames = FileUtilities.getFileList(baseDir,

                       connectProperties, null);

       for (final String fileName : topLevelPropertiesFileNames) {

         final Properties props = new Properties();

         props.load(new FileInputStream(new File(baseDir, fileName)));

         completePropertyList.putAll(props);

       }

       // add *.properties from the .services directory

       final String[] servicesPropertiesFileNames = FileUtilities.getFileList(

           new File(baseDir, connectServicesFolder).getAbsolutePath(), "**/*.properties", null);

       for (final String fileName : servicesPropertiesFileNames) {

         final Properties props = new Properties();

         props.load(new FileInputStream(new File(new File(baseDir, connectServicesFolder), fileName)));

         completePropertyList.putAll(props);

       }

       // persiste the complete .properties file into /lib

       completePropertyList.store(new FileOutputStream(new File(new File(baseDir, "lib"), project

           .getArtifactId()

           + ".properties")), "Generated by maven-sonicesb-plugin:connect-package");

     } catch (final FileNotFoundException e) {

       throw new MojoExecutionException(e.getMessage(), e);

     } catch (final IOException e) {

       throw new MojoExecutionException(e.getMessage(), e);

     }

   }

   /**

    * cleans out dependencies (in fact all files!) from the /lib directory and copies runtime dependencies back into

    * /lib.

    * 

    * @throws MojoExecutionException

    *           if something goes wrong

    */

   private void refreshDependenciesInLib() throws MojoExecutionException {

     final String libDirName = project.getBasedir() + "/lib/";

     final File markersDirectory = new File(libDirName, "markers");

     // delete any other files in the lib directory - perhaps the maven-clean-plugin should do this instead?

     FileUtilities.deleteFilesInDir(project.getBasedir() + "/lib/", "*", "");

     MojoExecutor.executeMojo(MojoExecutor.plugin(MojoExecutor.groupId("org.apache.maven.plugins"), MojoExecutor

         .artifactId("maven-dependency-plugin"), MojoExecutor.version("2.7")), MojoExecutor.goal("copy-dependencies"),

         MojoExecutor.configuration(new MojoExecutor.Element[] {

           MojoExecutor.element(MojoExecutor.name("outputDirectory"), project.getBasedir() + "/lib"),

           MojoExecutor.element(MojoExecutor.name("includeScope"), "runtime"),

           MojoExecutor.element(MojoExecutor.name("markersDirectory"), markersDirectory.getAbsolutePath()),

           MojoExecutor.element(MojoExecutor.name("includeTypes"), "jar"),

           MojoExecutor.element(MojoExecutor.name("useRepositoryLayout"), "false"),

           MojoExecutor.element(MojoExecutor.name("useSubDirectoryPerType"), "false")

         }), MojoExecutor.executionEnvironment(project, session, pluginManager));

     try {

       FileUtils.deleteDirectory(markersDirectory);

     } catch (final IOException e) {

       // This should never happen and if it does, it shouldn't really matter

       getLog().error(e);

     }

   }

   // /**

   // * DeploymentArtifactsGenerator requires a Sonic Connect Archive Filter File. In the Sonic Workbench, this file is

   // * %SONIC_INSTALL%\Workbench8.0\eclipse \plugins\com.sonicsw.tools.sonicconnect_8

   // * .0.0\resrc\defaultArchiveFilter.properties since we don't have access to this file from Maven, we'll just make

   // our

   // * own. This also gives us the advantage of users being able to define additional exclusions should the need arise.

   // *

   // * @return the ArchiveFilterFile to be used

   // * @throws MojoExecutionException

   // * if something goes wrong

   // */

   // private File getArchiveFilterFile() throws MojoExecutionException {

   //

   // File archiveFilterFile;

   // try {

   // archiveFilterFile = File.createTempFile("sonicConnectArchiveFilter", "properties");

   // archiveFilterFile.deleteOnExit();

   //

   // Properties props = new Properties();

   // props.setProperty("excludeDirs", excludeDirs);

   // props.setProperty("excludeFileExtensions", excludeFileExtensions);

   // props.store(new FileOutputStream(archiveFilterFile),

   // "Temporary sonicConnectArchiveFilter.properties file used by the maven-sonicesb-plugin");

   //

   // } catch (IOException e) {

   // throw new MojoExecutionException("Error creating temporary file", e);

   // }

   // return archiveFilterFile;

   //

   // }

   @SuppressWarnings("serial")

   class SortedProperties extends Properties {

     @Override

     @SuppressWarnings({ "unchecked", "rawtypes" })

     public Enumeration keys() {

       final Enumeration keysEnum = super.keys();

       final Vector keyList = new Vector();

       while (keysEnum.hasMoreElements()) {

         keyList.add(keysEnum.nextElement());

       }

       Collections.sort(keyList);

       return keyList.elements();

     }

   }

 }