com.igormaznitsa.mvngolang.GolangMvnInstallMojo.java Source code

Introduction

Here is the source code for com.igormaznitsa.mvngolang.GolangMvnInstallMojo.java

Source

/*
 * Copyright 2016 Igor Maznitsa.
 *
 * Licensed 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.
 */
package com.igormaznitsa.mvngolang;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.Writer;
import java.util.Locale;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.model.Model;
import org.apache.maven.model.Resource;
import org.apache.maven.model.io.xpp3.MavenXpp3Writer;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
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.plugins.annotations.Parameter;
import org.apache.maven.plugins.annotations.ResolutionScope;
import org.apache.maven.project.MavenProject;
import org.zeroturnaround.zip.ZipUtil;

/**
 * The Mojo packs all found source and resource project folders and create new artifact in the local repository.
 *
 * @since 2.1.0
 */
@Mojo(name = "mvninstall", defaultPhase = LifecyclePhase.INSTALL, threadSafe = true, requiresDependencyResolution = ResolutionScope.COMPILE)
public class GolangMvnInstallMojo extends AbstractMojo {

    @Component
    private MavenProject project;

    /**
     * Compression level of zip file. Must be 1..9
     *
     * @since 2.1.0
     */
    @Parameter(name = "compression", defaultValue = "9")
    private int compression;

    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        try {
            final File archive = compressProjectFiles();
            Artifact artifact = this.project.getArtifact();
            artifact.setFile(archive);
        } catch (IOException ex) {
            throw new MojoExecutionException("Detected unexpected IOException, check the log!", ex);
        }
    }

    private void safeCopyDirectory(@Nullable final String src, @Nonnull final File dst) throws IOException {
        if (src == null || src.isEmpty()) {
            return;
        }
        final File srcFile = new File(src);
        if (srcFile.isDirectory()) {
            if (getLog().isDebugEnabled()) {
                getLog().debug(String.format("Copying %s => %s", srcFile.getAbsolutePath(), dst.getAbsolutePath()));
            }
            FileUtils.copyDirectoryToDirectory(srcFile, dst);
        }
    }

    private void saveEffectivePom(@Nonnull final File folder) throws IOException {
        final Model model = this.project.getModel();
        Writer writer = null;
        try {
            writer = new FileWriter(new File(folder, "pom.xml"));
            new MavenXpp3Writer().write(writer, model);
            if (getLog().isDebugEnabled()) {
                getLog().debug("Effective pom has been written");
            }
        } finally {
            IOUtils.closeQuietly(writer);
        }
    }

    @Nonnull
    private File compressProjectFiles() throws IOException {
        final Artifact artifact = this.project.getArtifact();
        final File resultZip = new File(this.project.getBuild().getDirectory(),
                artifact.getArtifactId() + '-' + artifact.getVersion() + '.' + artifact.getType());
        if (resultZip.isFile() && !resultZip.delete()) {
            throw new IOException("Can't delete file : " + resultZip);
        }

        final File folderToPack = new File(
                ".tmp_pack_folder_" + Long.toHexString(System.currentTimeMillis()).toUpperCase(Locale.ENGLISH));
        if (folderToPack.isDirectory()) {
            FileUtils.deleteDirectory(folderToPack);
        }
        if (!folderToPack.mkdirs()) {
            throw new IOException("Can't create temp folder : " + folderToPack);
        }

        try {
            saveEffectivePom(folderToPack);

            FileUtils.copyFileToDirectory(this.project.getFile(), folderToPack);
            safeCopyDirectory(this.project.getBuild().getSourceDirectory(), folderToPack);
            safeCopyDirectory(this.project.getBuild().getTestSourceDirectory(), folderToPack);

            for (final Resource res : this.project.getBuild().getResources()) {
                safeCopyDirectory(res.getDirectory(), folderToPack);
            }

            for (final Resource res : this.project.getBuild().getTestResources()) {
                safeCopyDirectory(res.getDirectory(), folderToPack);
            }

            if (getLog().isDebugEnabled()) {
                getLog().debug(String.format("Packing folder %s to %s", folderToPack.getAbsolutePath(),
                        resultZip.getAbsolutePath()));
            }

            ZipUtil.pack(folderToPack, resultZip, Math.min(9, Math.max(1, this.compression)));

        } finally {
            FileUtils.deleteQuietly(folderToPack);
        }

        return resultZip;
    }

}