maven-archetypeng-plugin/src/main/java/org/codehaus/mojo/archetypeng/mojos/registry/AddExtensionsMojo.java

package org.codehaus.mojo.archetypeng.mojos.registry;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;
import org.apache.maven.artifact.repository.ArtifactRepository;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.codehaus.mojo.archetypeng.ArchetypeArtifactManager;
import org.codehaus.mojo.archetypeng.ArchetypeConfiguration;
import org.codehaus.mojo.archetypeng.ArchetypeDefinition;
import org.codehaus.mojo.archetypeng.ArchetypeFactory;
import org.codehaus.mojo.archetypeng.ArchetypePropertiesManager;
import org.codehaus.mojo.archetypeng.ArchetypeRegistryManager;
import org.codehaus.mojo.archetypeng.Constants;
import org.codehaus.mojo.archetypeng.exception.ArchetypeGenerationConfigurationFailure;
import org.codehaus.mojo.archetypeng.exception.ArchetypeNotDefined;
import org.codehaus.mojo.archetypeng.exception.UnknownArchetype;
import org.codehaus.mojo.archetypeng.registry.ArchetypeRegistry;
import org.codehaus.plexus.util.StringUtils;

/**
 *
 * @author rafale

 * @requiresProject  false
 * @goal             add-extensions
 */
public class AddExtensionsMojo
extends AbstractMojo
{
    /**
     * @parameter  expression="${user.home}/.m2/archetype.xml"
     */
    private File archetypeRegistryFile;

    /**
     * @component
     */
    ArchetypeRegistryManager archetypeRegistryManager;
    
    /**
     * @parameter expression="${extension}"
     */
    String extension;
    
    /**
     * @parameter expression="${extensions}"
     */
    String extensions;

    public void execute()
    throws MojoExecutionException, MojoFailureException
    {
        if (StringUtils.isEmpty(extension)&&StringUtils.isEmpty(extensions))
        {
            throw new MojoFailureException("-Dextension or -Dextensions must be set");
        }
        else if (StringUtils.isNotEmpty(extension)&& StringUtils.isNotEmpty(extensions))
        {
            throw new MojoFailureException("Only one of -Dextension or -Dextensions can be set");
        }
        
        
        try
        {
            List extensionsToAdd = new ArrayList();
            if (StringUtils.isNotEmpty(extension))
            {
                extensionsToAdd.add(extension);
            }
            else
            {
                extensionsToAdd.addAll(Arrays.asList(StringUtils.split(extensions, ",")));
            }
            
            ArchetypeRegistry registry = archetypeRegistryManager.readArchetypeRegistry(
                    archetypeRegistryFile
                );
            
            Iterator extensionsToAddIterator = extensionsToAdd.iterator();
            while(extensionsToAddIterator.hasNext())
            {
                String extensionToAdd = (String)extensionsToAddIterator.next();
                if (registry.getFilteredExtensions().contains(extensionToAdd))
                {
                    getLog().debug("Extension "+ extensionToAdd+" already exists");
                }
                else
                {
                    registry.addFilteredExtension(extensionToAdd.trim());
                    getLog().debug("Extension "+ extensionToAdd+" added");
                }
            }
            archetypeRegistryManager.writeArchetypeRegistry(archetypeRegistryFile, registry);
        }
        catch ( Exception ex )
        {
            throw new MojoExecutionException ( ex.getMessage (), ex );
        }
    }

}