package org.apache.maven.plugin.ear;

 /*

  * Licensed to the Apache Software Foundation (ASF) under one

  * or more contributor license agreements.  See the NOTICE file

  * distributed with this work for additional information

  * regarding copyright ownership.  The ASF licenses this file

  * to you under the Apache License, Version 2.0 (the

  * "License"); you may not use this file except in compliance

  * with the License.  You may obtain a copy of the License at

  *

  *  http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing,

  * software distributed under the License is distributed on an

  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

  * KIND, either express or implied.  See the License for the

  * specific language governing permissions and limitations

  * under the License.

  */

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.MojoFailureException;

 import org.codehaus.plexus.configuration.PlexusConfiguration;

 import org.codehaus.plexus.configuration.PlexusConfigurationException;

 import org.codehaus.plexus.util.FileUtils;

 import java.io.File;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.List;

 /**

  * A Mojo that generates the EAR deployment descriptor file(s).

  *

  * @author <a href="snicoll@apache.org">Stephane Nicoll</a>

  * @version $Id: GenerateApplicationXmlMojo.java 942855 2010-05-10 19:13:52Z krosenvold $

  * @goal generate-application-xml

  * @phase generate-resources

  * @threadSafe

  * @requiresDependencyResolution test

  */

 public class GenerateApplicationXmlMojo

     extends AbstractEarMojo

 {

     /**

      * Whether the application.xml should be generated or not.

      *

      * @parameter

      */

     private Boolean generateApplicationXml = Boolean.TRUE;

     /**

      * Display name of the application to be used when application.xml

      * file is autogenerated.

      *

      * @parameter expression="${project.artifactId}"

      */

     private String displayName;

     /**

      * Description of the application to be used when application.xml

      * file is autogenerated.

      *

      * @parameter expression="${project.description}"

      */

     private String description;

     /**

      * The security-roles to be added to the auto-generated

      * application.xml file.

      *

      * @parameter

      */

     private PlexusConfiguration security;

     public void execute()

         throws MojoExecutionException, MojoFailureException

     {

         // Initializes ear modules

         super.execute();

         // Handle application.xml

         if ( !generateApplicationXml.booleanValue() )

         {

             getLog().debug( "Generation of application.xml is disabled" );

         }

         else

         {

             // Check version

             if ( !version.equals( VERSION_1_3 ) && !version.equals( VERSION_1_4 ) && 

                  !version.equals( VERSION_5 ) && !version.equals( VERSION_6 ) )

             {

                 throw new MojoExecutionException( "Invalid version[" + version + "]" );

             }

             // Generate deployment descriptor and copy it to the build directory

             getLog().info( "Generating application.xml" );

             try

             {

                 generateStandardDeploymentDescriptor();

             }

             catch ( EarPluginException e )

             {

                 throw new MojoExecutionException( "Failed to generate application.xml", e );

             }

             try

             {

                 FileUtils.copyFileToDirectory( new File( generatedDescriptorLocation, "application.xml" ),

                                                new File( getWorkDirectory(), "META-INF" ) );

             }

             catch ( IOException e )

             {

                 throw new MojoExecutionException( "Unable to copy application.xml to final destination", e );

             }

         }

         // Handle jboss-app.xml

         if ( getJbossConfiguration() == null )

         {

             getLog().debug( "Generation of jboss-app.xml is disabled" );

             return;

         }

         else

         {

             // Generate deployment descriptor and copy it to the build directory

             getLog().info( "Generating jboss-app.xml" );

             try

             {

                 generateJbossDeploymentDescriptor();

             }

             catch ( EarPluginException e )

             {

                 throw new MojoExecutionException( "Failed to generate jboss-app.xml", e );

             }

             try

             {

                 FileUtils.copyFileToDirectory( new File( generatedDescriptorLocation, "jboss-app.xml" ),

                                                new File( getWorkDirectory(), "META-INF" ) );

             }

             catch ( IOException e )

             {

                 throw new MojoExecutionException( "Unable to copy jboss-app.xml to final destination", e );

             }

         }

     }

     /**

      * Generates the deployment descriptor.

      */

     protected void generateStandardDeploymentDescriptor()

         throws EarPluginException

     {

         File outputDir = new File( generatedDescriptorLocation );

         if ( !outputDir.exists() )

         {

             outputDir.mkdirs();

         }

         File descriptor = new File( outputDir, "application.xml" );

         final ApplicationXmlWriter writer = new ApplicationXmlWriter( version, encoding );

         final ApplicationXmlWriterContext context = new ApplicationXmlWriterContext(descriptor, getModules(), 

                 buildSecurityRoles(), displayName, description, defaultLibBundleDir);

         writer.write( context );

     }

     /**

      * Generates the jboss deployment descriptor.

      */

     protected void generateJbossDeploymentDescriptor()

         throws EarPluginException

     {

         File outputDir = new File( generatedDescriptorLocation );

         if ( !outputDir.exists() )

         {

             outputDir.mkdirs();

         }

         File descriptor = new File( outputDir, "jboss-app.xml" );

         JbossAppXmlWriter writer = new JbossAppXmlWriter( encoding );

         writer.write( descriptor, getJbossConfiguration(), getModules() );

     }

     /**

      * Builds the security roles based on the configuration.

      *

      * @return a list of SecurityRole object(s)

      * @throws EarPluginException if the configuration is invalid

      */

     private List buildSecurityRoles()

         throws EarPluginException

     {

         final List result = new ArrayList();

         if ( security == null )

         {

             return result;

         }

         try

         {

             final PlexusConfiguration[] securityRoles = security.getChildren( SecurityRole.SECURITY_ROLE );

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

             {

                 PlexusConfiguration securityRole = securityRoles[i];

                 final String id = securityRole.getAttribute( SecurityRole.ID_ATTRIBUTE );

                 final String roleName = securityRole.getChild( SecurityRole.ROLE_NAME ).getValue();

                 final String roleNameId =

                     securityRole.getChild( SecurityRole.ROLE_NAME ).getAttribute( SecurityRole.ID_ATTRIBUTE );

                 final String description = securityRole.getChild( SecurityRole.DESCRIPTION ).getValue();

                 final String descriptionId =

                     securityRole.getChild( SecurityRole.DESCRIPTION ).getAttribute( SecurityRole.ID_ATTRIBUTE );

                 if ( roleName == null )

                 {

                     throw new EarPluginException( "Invalid security-role configuration, role-name could not be null." );

                 }

                 else

                 {

                     result.add( new SecurityRole( roleName, roleNameId, id, description, descriptionId ) );

                 }

             }

             return result;

         }

         catch ( PlexusConfigurationException e )

         {

             throw new EarPluginException( "Invalid security-role configuration", e );

         }

     }

 }