maven-archetypeng-plugin/src/main/java/org/codehaus/mojo/archetypeng/mojos/registry/AddLanguagesMojo.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-languages
 */
public class AddLanguagesMojo
extends AbstractMojo
{
    /**
     * @parameter  expression="${user.home}/.m2/archetype.xml"
     */
    private File archetypeRegistryFile;

    /**
     * @component
     */
    ArchetypeRegistryManager archetypeRegistryManager;
    
    /**
     * @parameter expression="${language}"
     */
    String  language;
    
    /**
     * @parameter expression="${languages}"
     */
    String languages;

    public void execute()
    throws MojoExecutionException, MojoFailureException
    {
        if (StringUtils.isEmpty(language)&&StringUtils.isEmpty(languages))
        {
            throw new MojoFailureException("-Dlanguage or -Dlanguages must be set");
        }
        else if (StringUtils.isNotEmpty(language)&& StringUtils.isNotEmpty(languages))
        {
            throw new MojoFailureException("Only one of -Dlanguage or -Dlanguages can be set");
        }
        
        
        try
        {
            List languagesToAdd = new ArrayList();
            if (StringUtils.isNotEmpty(language))
            {
                languagesToAdd.add(language);
            }
            else
            {
                languagesToAdd.addAll(Arrays.asList(StringUtils.split(languages, ",")));
            }
            
            ArchetypeRegistry registry = archetypeRegistryManager.readArchetypeRegistry(
                    archetypeRegistryFile
                );
            
            Iterator languagesToAddIterator = languagesToAdd.iterator();
            while(languagesToAddIterator.hasNext())
            {
                String languageToAdd = (String)languagesToAddIterator.next();
                if (registry.getLanguages().contains(languageToAdd))
                {
                    getLog().debug("Language "+ languageToAdd+" already exists");
                }
                else
                {
                    registry.addLanguage(languageToAdd.trim());
                    getLog().debug("Language "+ languageToAdd+" added");
                }
            }
            archetypeRegistryManager.writeArchetypeRegistry(archetypeRegistryFile, registry);
        }
        catch ( Exception ex )
        {
            throw new MojoExecutionException ( ex.getMessage (), ex );
        }
    }

}