org.swaggertools.maven.ProcessorMojo Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of swagger-tools-maven-plugin Show documentation

There is a newer version: 0.8.3

Show newest version

package org.swaggertools.maven;

import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
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.project.MavenProject;
import org.swaggertools.core.config.HelpPrinter;
import org.swaggertools.core.run.ProcessorFactory;

import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.regex.Pattern;

@Mojo(name = "generate", defaultPhase = LifecyclePhase.GENERATE_SOURCES)
public class ProcessorMojo extends AbstractMojo {
    private static final Pattern TARGET = Pattern.compile("target\\.(.+?)\\.location");

    @Parameter
    private Boolean skip;

    @Parameter
    private Boolean help;

    @Parameter
    private Map options;

    @Parameter(readonly = true, required = true, defaultValue = "${project}")
    private MavenProject project;

    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        if (skip != null && skip) {
            getLog().info("Swagger processing is skipped.");
        } else if ((help != null && help) || options == null) {
            printHelp();
        } else {
            getLog().info("Process swagger definition");
            try {
                new ProcessorFactory().create(options).process();
                addCompileSourceRoot();
            } catch (Exception e) {
                getLog().error(e);
                throw new MojoFailureException(e.getMessage());
            }
        }
    }

    private void printHelp() {
        HelpPrinter printer = new HelpPrinter("");
        printer.printProperties();
        getLog().info(printer.getHelp());
    }

    private void addCompileSourceRoot() {
        Set targets = findTargets();
        for (String target : targets) {
            project.addCompileSourceRoot(target);
        }
    }

    private Set findTargets() {
        Set targets = new HashSet<>();
        options.forEach((k, v) -> {
            if (TARGET.matcher(k).matches()) {
                targets.add(v);
            }
        });
        return targets;
    }
}