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 java.io.File;

 import java.util.ArrayList;

 import java.util.List;

 import java.util.Set;

 import org.apache.maven.artifact.Artifact;

 import org.apache.maven.artifact.resolver.filter.ScopeArtifactFilter;

 import org.apache.maven.plugin.AbstractMojo;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.MojoFailureException;

 import org.apache.maven.plugin.ear.util.ArtifactTypeMappingService;

 import org.apache.maven.plugin.ear.util.JavaEEVersion;

 import org.apache.maven.plugins.annotations.Component;

 import org.apache.maven.plugins.annotations.Parameter;

 import org.apache.maven.project.MavenProject;

 import org.codehaus.plexus.configuration.PlexusConfiguration;

 import org.codehaus.plexus.configuration.PlexusConfigurationException;

 /**

  * A base class for EAR-processing related tasks.

  *

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

  * @version $Id: AbstractEarMojo.java 1359710 2012-07-10 14:53:53Z tchemit $

  */

 public abstract class AbstractEarMojo

     extends AbstractMojo

 {

     public static final String APPLICATION_XML_URI = "META-INF/application.xml";

     public static final String META_INF = "META-INF";

     public static final String UTF_8 = "UTF-8";

     /**

      * The version of the application.xml to generate. Valid values

      * are 1.3, 1.4, 5 and 6.

      */

     @Parameter( defaultValue = "1.3" )

     protected String version;

     /**

      * Character encoding for the auto-generated deployment file(s).

      */

     @Parameter( defaultValue = "UTF-8" )

     protected String encoding;

     /**

      * Directory where the deployment descriptor file(s) will be auto-generated.

      */

     @Parameter( defaultValue = "${project.build.directory}" )

     protected String generatedDescriptorLocation;

     /**

      * The maven project.

      */

     @Component

     protected MavenProject project;

     /**

      * The ear modules configuration.

      */

     @Parameter

     private EarModule[] modules;

     /**

      * The artifact type mappings.

      */

     @Parameter

     protected PlexusConfiguration artifactTypeMappings;

     /**

      * The default bundle dir for libraries.

      */

     @Parameter( alias = "defaultJavaBundleDir" )

     protected String defaultLibBundleDir;

     /**

      * Should libraries be added in application.xml

      */

     @Parameter( defaultValue = "false" )

     private Boolean includeLibInApplicationXml = Boolean.FALSE;

     /**

      * The file name mapping to use for all dependencies included

      * in the EAR file.

      */

     @Parameter

     private String fileNameMapping;

     /**

      * Directory that resources are copied to during the build.

      */

     @Parameter( defaultValue = "${project.build.directory}/${project.build.finalName}", required = true )

     private File workDirectory;

     /**

      * The JBoss specific configuration.

      *

      * @parameter

      */

     @Parameter

     private PlexusConfiguration jboss;

     /**

      * The id to use to define the main artifact (e.g. the artifact without

      * a classifier) when there is multiple candidates.

      *

      * @parameter

      */

     @Parameter

     private String mainArtifactId = "none";

     private List<EarModule> earModules;

     private List<EarModule> allModules;

     private JbossConfiguration jbossConfiguration;

     @SuppressWarnings( "unchecked" )

     public void execute()

         throws MojoExecutionException, MojoFailureException

     {

         final JavaEEVersion javaEEVersion = JavaEEVersion.getJavaEEVersion( version );

         getLog().debug( "Resolving artifact type mappings ..." );

         ArtifactTypeMappingService typeMappingService;

         try

         {

             typeMappingService = new ArtifactTypeMappingService();

             typeMappingService.configure( artifactTypeMappings );

         }

         catch ( EarPluginException e )

         {

             throw new MojoExecutionException( "Failed to initialize artifact type mappings", e );

         }

         catch ( PlexusConfigurationException e )

         {

             throw new MojoExecutionException( "Invalid artifact type mappings configuration", e );

         }

         getLog().debug( "Initializing JBoss configuration if necessary ..." );

         try

         {

             initializeJbossConfiguration();

         }

         catch ( EarPluginException e )

         {

             throw new MojoExecutionException( "Failed to initialize JBoss configuration", e );

         }

         getLog().debug( "Initializing ear execution context" );

         EarExecutionContext earExecutionContext =

             new EarExecutionContext( project, mainArtifactId, defaultLibBundleDir, jbossConfiguration, fileNameMapping,

                                      typeMappingService );

         getLog().debug( "Resolving ear modules ..." );

         allModules = new ArrayList<EarModule>();

         try

         {

             if ( modules != null && modules.length > 0 )

             {

                 // Let's validate user-defined modules

                 EarModule module = null;

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

                 {

                     module = modules[i];

                     getLog().debug( "Resolving ear module[" + module + "]" );

                     module.setEarExecutionContext( earExecutionContext );

                     module.resolveArtifact( project.getArtifacts() );

                     allModules.add( module );

                 }

             }

             // Let's add other modules

             Set<Artifact> artifacts = project.getArtifacts();

             for ( Artifact artifact : artifacts )

             {

                 // If the artifact's type is POM, ignore and continue

                 // since it's used for transitive deps only.

                 if ( "pom".equals( artifact.getType() ) )

                 {

                     continue;

                 }

                 // Artifact is not yet registered and it has neither test, nor a

                 // provided scope, not is it optional

                 ScopeArtifactFilter filter = new ScopeArtifactFilter( Artifact.SCOPE_RUNTIME );

                 if ( !isArtifactRegistered( artifact, allModules ) && !artifact.isOptional()

                     && filter.include( artifact ) )

                 {

                     EarModule module = EarModuleFactory.newEarModule( artifact, javaEEVersion, defaultLibBundleDir,

                                                                       includeLibInApplicationXml, typeMappingService );

                     module.setEarExecutionContext( earExecutionContext );

                     allModules.add( module );

                 }

             }

         }

         catch ( EarPluginException e )

         {

             throw new MojoExecutionException( "Failed to initialize ear modules", e );

         }

         // Now we have everything let's built modules which have not been excluded

         earModules = new ArrayList<EarModule>();

         for ( EarModule earModule :  allModules )

         {

             if ( earModule.isExcluded() )

             {

                 getLog().debug( "Skipping ear module[" + earModule + "]" );

             }

             else

             {

                 earModules.add( earModule );

             }

         }

     }

     protected List<EarModule> getModules()

     {

         if ( earModules == null )

         {

             throw new IllegalStateException( "Ear modules have not been initialized" );

         }

         return earModules;

     }

     protected MavenProject getProject()

     {

         return project;

     }

     protected File getWorkDirectory()

     {

         return workDirectory;

     }

     protected JbossConfiguration getJbossConfiguration()

     {

         return jbossConfiguration;

     }

     private static boolean isArtifactRegistered( Artifact a, List<EarModule> currentList )

     {

         for( EarModule em : currentList )

         {

             if ( em.getArtifact().equals( a ) )

             {

                 return true;

             }

         }

         return false;

     }

     /**

      * Initializes the JBoss configuration.

      *

      * @throws EarPluginException if the configuration is invalid

      */

     private void initializeJbossConfiguration()

         throws EarPluginException

     {

         if ( jboss == null )

         {

             jbossConfiguration = null;

         }

         else

         {

             try

             {

                 String version = jboss.getChild( JbossConfiguration.VERSION ).getValue();

                 if ( version == null )

                 {

                     getLog().info( "JBoss version not set, using JBoss 4 by default" );

                     version = JbossConfiguration.VERSION_4;

                 }

                 final String securityDomain = jboss.getChild( JbossConfiguration.SECURITY_DOMAIN ).getValue();

                 final String unauthenticatedPrincipal =

                     jboss.getChild( JbossConfiguration.UNAUHTHENTICTED_PRINCIPAL ).getValue();

                 final PlexusConfiguration loaderRepositoryEl = jboss.getChild( JbossConfiguration.LOADER_REPOSITORY );

                 final String loaderRepository = loaderRepositoryEl.getValue();

                 final String loaderRepositoryClass =

                     loaderRepositoryEl.getAttribute( JbossConfiguration.LOADER_REPOSITORY_CLASS_ATTRIBUTE );

                 final PlexusConfiguration loaderRepositoryConfigEl =

                     jboss.getChild( JbossConfiguration.LOADER_REPOSITORY_CONFIG );

                 final String loaderRepositoryConfig = loaderRepositoryConfigEl.getValue();

                 final String configParserClass =

                     loaderRepositoryConfigEl.getAttribute( JbossConfiguration.CONFIG_PARSER_CLASS_ATTRIBUTE );

                 final String jmxName = jboss.getChild( JbossConfiguration.JMX_NAME ).getValue();

                 final String moduleOrder = jboss.getChild( JbossConfiguration.MODULE_ORDER ).getValue();

                 final List<String> dataSources = new ArrayList<String>();

                 final PlexusConfiguration dataSourcesEl = jboss.getChild( JbossConfiguration.DATASOURCES );

                 if ( dataSourcesEl != null )

                 {

                     final PlexusConfiguration[] dataSourcesConfig =

                         dataSourcesEl.getChildren( JbossConfiguration.DATASOURCE );

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

                     {

                         PlexusConfiguration dataSourceConfig = dataSourcesConfig[i];

                         dataSources.add( dataSourceConfig.getValue() );

                     }

                 }

                 final String libraryDirectory = jboss.getChild( JbossConfiguration.LIBRARY_DIRECTORY ).getValue();

                 jbossConfiguration = new JbossConfiguration( version, securityDomain, unauthenticatedPrincipal, jmxName,

                                                              loaderRepository, moduleOrder, dataSources,

                                                              libraryDirectory, loaderRepositoryConfig,

                                                              loaderRepositoryClass, configParserClass );

             }

             catch ( PlexusConfigurationException e )

             {

                 throw new EarPluginException( "Invalid JBoss configuration", e );

             }

         }

     }

 }