CreateBundleMojo.java

/*
 * Sonatype Application Build Lifecycle
 * Copyright (C) 2009 Sonatype, Inc.                                                                                                                          
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see http://www.gnu.org/licenses/.
 *
 */
package com.redhat.tools.nexus.maven.plugin.bundle;

import org.apache.maven.execution.MavenSession;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.assembly.InvalidAssemblerConfigurationException;
import org.apache.maven.plugin.assembly.archive.ArchiveCreationException;
import org.apache.maven.plugin.assembly.archive.AssemblyArchiver;
import org.apache.maven.plugin.assembly.format.AssemblyFormattingException;
import org.apache.maven.plugin.assembly.model.Assembly;
import org.apache.maven.plugin.assembly.model.FileItem;
import org.apache.maven.project.MavenProject;
import org.apache.maven.project.MavenProjectHelper;

import com.redhat.tools.nexus.maven.plugin.ClasspathUtils;

import java.io.File;
import java.io.IOException;
import java.util.Iterator;
import java.util.Properties;

/**
 * Create a plugin bundle artifact attach it to the plugin's project.
 * 
 * @goal create-bundle
 * @phase package
 */
public class CreateBundleMojo
    extends AbstractMojo
{
    /**
     * The current plugin project being built.
     * 
     * @parameter default-value="${project}"
     * @required
     * @readonly
     */
    private MavenProject project;

    /**
     * The current build session, for reference from the Assembly API.
     * 
     * @parameter default-value="${session}"
     * @required
     * @readonly
     */
    private MavenSession session;

    /**
     * Supplemental plugin assembly configuration.
     * 
     * @parameter
     */
    private BundleConfiguration bundle;

    /**
     * Assembly manager component that is responsible for creating the plugin bundle assembly and attaching it to the
     * current project.
     * 
     * @component
     */
    private AssemblyArchiver archiver;

    /**
     * Component used by the {@link AssemblyArchiver} to attach the bundle artifact to the current project.
     * 
     * @component
     */
    private MavenProjectHelper projectHelper;

    @SuppressWarnings( "unchecked" )
    public void execute()
        throws MojoExecutionException
    {
        final Assembly assembly = new Assembly();
        assembly.addFormat( "zip" );
        assembly.setId( "bundle" );
        assembly.setIncludeBaseDirectory( false );

        try
        {
            final Properties cpArtifacts = ClasspathUtils.read( project );
            final String outputDirectory = project.getArtifactId() + "-" + project.getVersion() + "/dependencies";

            for ( final Iterator it = cpArtifacts.keySet().iterator(); it.hasNext(); )
            {
                final String destName = (String) it.next();
                final String sourcePath = cpArtifacts.getProperty( destName );

                final FileItem fi = new FileItem();
                fi.setSource( sourcePath );
                fi.setOutputDirectory( outputDirectory );
                fi.setDestName( destName );

                assembly.addFile( fi );
            }
        }
        catch ( final IOException e )
        {
            throw new MojoExecutionException( "Failed to create plugin bundle: " + e.getMessage(), e );
        }

        final FileItem fi = new FileItem();
        fi.setSource( project.getArtifact().getFile().getPath() );
        fi.setOutputDirectory( project.getArtifactId() + "-" + project.getVersion() );

        assembly.addFile( fi );

        if ( bundle == null )
        {
            bundle = new BundleConfiguration( project, session );
        }
        else
        {
            bundle.initDefaults( project, session );
        }

        try
        {
            final File assemblyFile =
                archiver.createArchive( assembly, bundle.getAssemblyFileName( assembly ), "zip", bundle );
            projectHelper.attachArtifact( project, "zip", assembly.getId(), assemblyFile );
        }
        catch ( final ArchiveCreationException e )
        {
            throw new MojoExecutionException( "Failed to create plugin bundle: " + e.getMessage(), e );
        }
        catch ( final AssemblyFormattingException e )
        {
            throw new MojoExecutionException( "Failed to create plugin bundle: " + e.getMessage(), e );
        }
        catch ( final InvalidAssemblerConfigurationException e )
        {
            throw new MojoExecutionException( "Failed to create plugin bundle: " + e.getMessage(), e );
        }
    }
}