package org.apache.maven.plugin.eclipse.writers.myeclipse;

 import java.io.File;

 import java.io.FileFilter;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.Collection;

 import java.util.Collections;

 import java.util.Iterator;

 import java.util.Map;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.eclipse.Messages;

 import org.apache.maven.plugin.eclipse.writers.AbstractEclipseWriter;

 import org.codehaus.plexus.util.FileUtils;

 import org.codehaus.plexus.util.IOUtil;

 import org.codehaus.plexus.util.StringUtils;

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

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

 /**

  * MyEclipse .springBeans configuration file writer

  * 

  * @author Olivier Jacob

  */

 public class MyEclipseSpringBeansWriter

     extends AbstractEclipseWriter

 {

     private static final String MYECLIPSE_SPRING_CONFIGURATION_FILENAME = ".springBeans";

     private static final String MYECLIPSE_SPRING_BEANS_PROJECT_DESCRIPTION = "beansProjectDescription";

     private static final String MYECLIPSE_SPRING_CONFIG_EXTENSIONS = "configExtensions";

     private static final String MYECLIPSE_SPRING_CONFIG_EXTENSION = "configExtension";

     private static final String MYECLIPSE_SPRING_CONFIGS = "configs";

     private static final String MYECLIPSE_SPRING_CONFIG = "config";

     private static final String MYECLIPSE_SPRING_CONFIGSETS = "configSets";

     private static final String MYECLIPSE_SPRING_VERSION = "springVersion";

     /**

      * Spring configuration filenames (injected by the plugin)

      */

     private Map springConfig;

     /**

      * Allow injection of Spring configuration filenames through constructor

      * 

      * @param springConfig a map holding Spring configuration properties

      */

     public MyEclipseSpringBeansWriter( Map springConfig )

     {

         this.springConfig = springConfig;

     }

     /**

      * Write MyEclipse .springBeans configuration file

      * 

      * @throws MojoExecutionException

      */

     public void write()

         throws MojoExecutionException

     {

         FileWriter w;

         try

         {

             w =

                 new FileWriter( new File( config.getEclipseProjectDirectory(), MYECLIPSE_SPRING_CONFIGURATION_FILENAME ) );

         }

         catch ( IOException ex )

         {

             throw new MojoExecutionException( Messages.getString( "EclipsePlugin.erroropeningfile" ), ex ); //$NON-NLS-1$

         }

         XMLWriter writer = new PrettyPrintXMLWriter( w, "UTF-8", null );

         writer.startElement( MYECLIPSE_SPRING_BEANS_PROJECT_DESCRIPTION );

         // Configuration extension

         writer.startElement( MYECLIPSE_SPRING_CONFIG_EXTENSIONS );

         writer.startElement( MYECLIPSE_SPRING_CONFIG_EXTENSION );

         writer.writeText( "xml" );

         writer.endElement();

         writer.endElement();

         // Configuration files

         writer.startElement( MYECLIPSE_SPRING_CONFIGS );

         Iterator onConfigFiles =

             getConfigurationFilesList( (String) springConfig.get( "basedir" ),

                                        (String) springConfig.get( "file-pattern" ) ).iterator();

         while ( onConfigFiles.hasNext() )

         {

             writer.startElement( MYECLIPSE_SPRING_CONFIG );

             writer.writeText( StringUtils.replace( (String) onConfigFiles.next(), "\\", "/" ) );

             writer.endElement();

         }

         writer.endElement();

         // Configuration sets

         writer.startElement( MYECLIPSE_SPRING_CONFIGSETS );

         writer.endElement();

         // Spring version

         writer.startElement( MYECLIPSE_SPRING_VERSION );

         writer.writeText( (String) springConfig.get( "version" ) );

         writer.endElement();

         writer.endElement();

         IOUtil.close( w );

     }

     /**

      * Retrieve the list of Spring configuration files recursively from the <code>basedir</code> directory,

      * considering only filenames matching the <code>pattern</code> given

      * 

      * @param basedir the path to the base directory to search in

      * @param pattern file include pattern

      * @return the list of filenames matching the given pattern

      */

     private Collection getConfigurationFilesList( String basedir, String pattern )

     {

         ArrayList configFiles = new ArrayList();

         try

         {

             File directory = new File( basedir );

             File[] subdirs = directory.listFiles( new FileFilter()

             {

                 public boolean accept( File pathname )

                 {

                     return pathname.isDirectory();

                 }

             } );

             for ( int i = 0; i < subdirs.length; i++ )

             {

                 configFiles.addAll( getConfigurationFilesList( subdirs[i].getPath(), pattern ) );

             }

             configFiles.addAll( FileUtils.getFileNames( directory, pattern, null, true ) );

         }

         catch ( IOException ioe )

         {

             log.error( "Error while retrieving Spring configuration files. Returning list in current state" );

         }

         // Sort them to have something constant

         Collections.sort( configFiles );

         return configFiles;

     }

 }