imachine.timachine-maven-plugin.1.1.2.source-code.StatusMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of timachine-maven-plugin Show documentation
The maven plugin to manage migrations
The newest version!
import com.worksap.timachine.Executor;
import com.worksap.timachine.Migrations;
import com.worksap.timachine.VersionChecker;
import com.worksap.timachine.model.Migration;
import com.worksap.timachine.model.VersionDifference;
import com.worksap.timachine.spi.VersionProvider;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.*;
import org.reflections.Reflections;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.PrintStream;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;

/**
 * Show migration status.
 */
@Mojo(name = "status", requiresDependencyResolution = ResolutionScope.COMPILE_PLUS_RUNTIME)
@Execute(phase = LifecyclePhase.COMPILE)
public class StatusMojo extends AbstractMojo {
    @Parameter(required = true)
    private String packageName;
    @Parameter(property = "project.runtimeClasspathElements")
    private List elements;

    @Parameter(property = "project.build.outputDirectory")
    private String outDir;

    @Parameter(required = true)
    private String executor;

    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        PrintStream ps = new PrintStream(os);
        Thread.currentThread().setContextClassLoader(getClassLoader());
        Reflections reflections = new Reflections(packageName);
        List> list = new ArrayList<>(reflections.getTypesAnnotatedWith(Migration.class));
        Migrations migrations = new Migrations(list);
        ps.println();
        ps.print(String.format("Total migrations count: %d", migrations.getVersions().size()));
        if (migrations.getVersions().isEmpty()) {
            ps.println();
        } else {
            ps.println(String.format(", from %s to %s", migrations.getVersions().get(0), migrations.getVersions().get(migrations.getVersions().size() - 1)));
        }
        try {
            Executor executorImpl = (Executor) Thread.currentThread().getContextClassLoader().loadClass(executor).newInstance();
            VersionProvider versionProvider = executorImpl.getVersionProvider();
            VersionChecker versionChecker = new VersionChecker(versionProvider, migrations);
            VersionDifference versionDifference = versionChecker.versionDifference(null);
            ps.print(String.format("Current version: %s", versionDifference.getLatestExecutedVersion()));
            if (!migrations.getVersions().isEmpty()) {
                String status;
                String latestVersion = migrations.getVersions().get(migrations.getVersions().size() - 1);
                if (versionDifference.getSteps().isEmpty()) {
                    status = "up to date";
                } else {
                    status = String.format("behind %d steps", versionDifference.getSteps().size());
                }
                ps.println(", " + status);
            }

            ps.println();
            System.out.println(os.toString());
        } catch (Exception e) {
            throw new MojoExecutionException("Failed to initialize dependencies.", e);
        }
    }

    private ClassLoader getClassLoader() throws MojoExecutionException {
        elements.add(outDir);
        try {
            URL urls[] = new URL[elements.size()];

            for (int i = 0; i < elements.size(); ++i) {
                urls[i] = new File((String) elements.get(i)).toURI().toURL();
            }
            return new URLClassLoader(urls, getClass().getClassLoader());
        } catch (Exception e)//gotta catch em all
        {
            throw new MojoExecutionException("Couldn't create a classloader.", e);
        }
    }
}