com.puppetlabs.geppetto.forge.maven.plugin.Package.java Source code

Introduction

Here is the source code for com.puppetlabs.geppetto.forge.maven.plugin.Package.java

Source

/**
 * Copyright (c) 2013 Puppet Labs, Inc. and other contributors, as listed below.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *   Puppet Labs
 */
package com.puppetlabs.geppetto.forge.maven.plugin;

import static com.puppetlabs.geppetto.diagnostic.Diagnostic.ERROR;
import static com.puppetlabs.geppetto.forge.Forge.PACKAGE;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.Collection;

import org.apache.maven.artifact.Artifact;
import org.apache.maven.plugins.annotations.Component;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.project.MavenProject;
import org.apache.maven.repository.RepositorySystem;

import com.google.common.base.Charsets;
import com.puppetlabs.geppetto.diagnostic.Diagnostic;
import com.puppetlabs.geppetto.forge.model.Metadata;

/**
 * Goal that builds the module gzipped tarball and optionally generates the <tt>metadata.json</tt> file.
 */
@Mojo(name = "package", requiresProject = false, defaultPhase = LifecyclePhase.PACKAGE)
public class Package extends AbstractForgeMojo {
    @Component
    private RepositorySystem repositorySystem;

    private File buildForge(File moduleSource, File destination, Metadata[] resultingMetadata,
            byte[][] resultingMD5, Diagnostic result) throws IOException {
        return getForgeUtil().build(moduleSource, destination, null, resultingMetadata, resultingMD5, result);
    }

    @Override
    protected String getActionName() {
        return "Package";
    }

    @Override
    protected void invoke(Diagnostic result) throws Exception {
        Collection<File> moduleRoots = findModuleRoots();
        if (moduleRoots.isEmpty()) {
            result.addChild(new Diagnostic(ERROR, PACKAGE, "No modules found in repository"));
            return;
        }

        File buildDir = getBuildDir();
        buildDir.mkdirs();
        if (moduleRoots.size() == 1) {
            MavenProject project = getProject();
            File moduleRoot = moduleRoots.iterator().next();
            Metadata[] resultingMetadata = new Metadata[1];
            byte[][] resultingMD5 = new byte[1][];
            project.getArtifact()
                    .setFile(buildForge(moduleRoot, buildDir, resultingMetadata, resultingMD5, result));

            Artifact pmriArtifact = repositorySystem.createArtifact(project.getGroupId(), project.getArtifactId(),
                    project.getVersion(), "compile", "pmri");

            PuppetModuleReleaseInfo pmri = new PuppetModuleReleaseInfo();
            pmri.setMetadata(resultingMetadata[0]);
            pmri.populate(moduleRoot);

            File pmriFile = new File(buildDir, "release.pmri");
            OutputStream out = new FileOutputStream(pmriFile);
            try {
                Writer writer = new BufferedWriter(new OutputStreamWriter(out, Charsets.UTF_8));
                getGson().toJson(pmri, writer);
                writer.flush();
            } finally {
                out.close();
            }
            pmriArtifact.setFile(pmriFile);
            pmriArtifact.setResolved(true);
            project.addAttachedArtifact(pmriArtifact);
        } else {
            File builtModules = new File(buildDir, "builtModules");
            if (!(builtModules.mkdir() || builtModules.isDirectory())) {
                result.addChild(
                        new Diagnostic(ERROR, PACKAGE, "Unable to create directory" + builtModules.getPath()));
                return;
            }
            for (File moduleRoot : moduleRoots)
                buildForge(moduleRoot, builtModules, null, null, result);
        }
    }
}