package org.apache.maven.archetype.generator;

 /*

  * 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.archetype.old.OldArchetype;

 import org.apache.maven.archetype.ArchetypeGenerationRequest;

 import org.apache.maven.archetype.ArchetypeGenerationResult;

 import org.apache.maven.archetype.common.ArchetypeArtifactManager;

 import org.apache.maven.archetype.common.ArchetypeRegistryManager;

 import org.apache.maven.archetype.exception.ArchetypeException;

 import org.apache.maven.archetype.exception.ArchetypeGenerationFailure;

 import org.apache.maven.archetype.exception.ArchetypeNotDefined;

 import org.apache.maven.archetype.exception.UnknownArchetype;

 import org.apache.maven.artifact.repository.ArtifactRepository;

 import org.codehaus.plexus.component.annotations.Component;

 import org.codehaus.plexus.component.annotations.Requirement;

 import org.codehaus.plexus.logging.AbstractLogEnabled;

 import org.codehaus.plexus.util.StringUtils;

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

 import org.dom4j.DocumentException;

 import java.io.File;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.List;

 @Component( role = ArchetypeGenerator.class )

 public class DefaultArchetypeGenerator

     extends AbstractLogEnabled

     implements ArchetypeGenerator

 {

     @Requirement

     private ArchetypeRegistryManager archetypeRegistryManager;

     @Requirement

     private ArchetypeArtifactManager archetypeArtifactManager;

     @Requirement

     private FilesetArchetypeGenerator filesetGenerator;

     @Requirement

     private OldArchetype oldArchetype;

     private File getArchetypeFile( ArchetypeGenerationRequest request, ArtifactRepository localRepository )

         throws IOException, ArchetypeException, XmlPullParserException, DocumentException

     {

         if ( !isArchetypeDefined( request ) )

         {

             throw new ArchetypeNotDefined( "The archetype is not defined" );

         }

         List<ArtifactRepository> repos = new ArrayList<ArtifactRepository>();

         ArtifactRepository remoteRepo = null;

         if ( request != null && request.getArchetypeRepository() != null )

         {

             remoteRepo =

                 archetypeRegistryManager.createRepository( request.getArchetypeRepository(),

                                                            request.getArchetypeArtifactId() + "-repo" );

             repos.add( remoteRepo );

         }

         if ( !archetypeArtifactManager.exists( request.getArchetypeGroupId(), request.getArchetypeArtifactId(),

                                                request.getArchetypeVersion(), remoteRepo, localRepository, repos ) )

         {

             throw new UnknownArchetype( "The desired archetype does not exist (" + request.getArchetypeGroupId() + ":"

                 + request.getArchetypeArtifactId() + ":" + request.getArchetypeVersion() + ")" );

         }

         File archetypeFile =

             archetypeArtifactManager.getArchetypeFile( request.getArchetypeGroupId(), request.getArchetypeArtifactId(),

                                                        request.getArchetypeVersion(), remoteRepo, localRepository,

                                                        repos );

         return archetypeFile;

     }

     private void generateArchetype( ArchetypeGenerationRequest request, File archetypeFile )

         throws IOException, ArchetypeException, XmlPullParserException, DocumentException

     {

         if ( archetypeArtifactManager.isFileSetArchetype( archetypeFile ) )

         {

             processFileSetArchetype( request, archetypeFile );

         }

         else if ( archetypeArtifactManager.isOldArchetype( archetypeFile ) )

         {

             processOldArchetype( request, archetypeFile );

         }

         else

         {

             throw new ArchetypeGenerationFailure( "The defined artifact is not an archetype" );

         }

     }

     /** Common */

     public String getPackageAsDirectory( String packageName )

     {

         return StringUtils.replace( packageName, ".", "/" );

     }

     private boolean isArchetypeDefined( ArchetypeGenerationRequest request )

     {

         return StringUtils.isNotEmpty( request.getArchetypeGroupId() )

             && StringUtils.isNotEmpty( request.getArchetypeArtifactId() )

             && StringUtils.isNotEmpty( request.getArchetypeVersion() );

     }

     /** FileSetArchetype */

     private void processFileSetArchetype( ArchetypeGenerationRequest request, File archetypeFile )

         throws ArchetypeException

     {

         filesetGenerator.generateArchetype( request, archetypeFile );

     }

     private void processOldArchetype( ArchetypeGenerationRequest request, File archetypeFile )

         throws UnknownArchetype, ArchetypeGenerationFailure

     {

         oldArchetype.createArchetype( request, archetypeFile );

     }

     public void generateArchetype( ArchetypeGenerationRequest request, File archetypeFile,

                                    ArchetypeGenerationResult result )

     {

         try

         {

             generateArchetype( request, archetypeFile );

         }

         catch ( IOException ex )

         {

             result.setCause( ex );

         }

         catch ( ArchetypeException ex )

         {

             result.setCause( ex );

         }

         catch ( XmlPullParserException ex )

         {

             result.setCause( ex );

         }

         catch ( DocumentException ex )

         {

             result.setCause( ex );

         }

     }

     public void generateArchetype( ArchetypeGenerationRequest request, ArchetypeGenerationResult result )

     {

         try

         {

             File archetypeFile = getArchetypeFile( request, request.getLocalRepository() );

             generateArchetype( request, archetypeFile, result );

         }

         catch ( IOException ex )

         {

             result.setCause( ex );

         }

         catch ( ArchetypeException ex )

         {

             result.setCause( ex );

         }

         catch ( XmlPullParserException ex )

         {

             result.setCause( ex );

         }

         catch ( DocumentException ex )

         {

             result.setCause( ex );

         }

     }

 }