ru.yandex.qatools.allure.inject.AbstractInjectPlugin Maven / Gradle / Ivy

Go to download
package ru.yandex.qatools.allure.inject;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.CanWriteFileFilter;
import org.apache.commons.io.filefilter.RegexFileFilter;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Parameter;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Collections;
import java.util.Iterator;

import static com.google.common.collect.Iterators.concat;

/**
 * @author Dmitry Baev [email protected]
 *         Date: 23.11.13
 */
@SuppressWarnings("unused")
public abstract class AbstractInjectPlugin extends AbstractAspectJPluginExecutor {

    protected static final String CLASS_FILTER = ".*\\.class";

    @Parameter(defaultValue = "${project.build.testOutputDirectory}")
    protected File testOutputDirectory;

    @Parameter(defaultValue = "${project.build.outputDirectory}")
    protected File outputDirectory;

    protected void inject() throws MojoExecutionException, MojoFailureException {
        Iterator testClasses = getClassFiles(testOutputDirectory);
        Iterator classes = getClassFiles(outputDirectory);
        Iterator files = concat(classes, testClasses);

        while (files.hasNext()) {
            File testClassFile = files.next();
            processFile(testClassFile);
        }
    }

    protected void processFile(File testClassFile) {
        getLog().info("Processing file: " + testClassFile);
        try {
            byte[] bytes = getInjector().inject(new FileInputStream(testClassFile));
            FileUtils.writeByteArrayToFile(testClassFile, bytes);
        } catch (IOException e) {
            getLog().error("IO trouble while reading file " + testClassFile, e);
        }
    }

    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        super.execute();
        inject();
    }

    protected static Iterator getClassFiles(File directory) {
        if (directory.exists() && directory.isDirectory() && directory.canRead()) {
            return FileUtils.iterateFiles(directory,
                    new RegexFileFilter(CLASS_FILTER), CanWriteFileFilter.CAN_WRITE);
        } else {
            return Collections.emptyList().iterator();
        }
    }

    protected abstract Injector getInjector();
}