com.antwerkz.critter.CritterMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of critter-maven Show documentation
There is a newer version: 3.0.5
package com.antwerkz.critter;

import java.io.File;
import java.io.FileNotFoundException;
import java.util.function.Consumer;

import static java.util.Arrays.asList;
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.codehaus.plexus.util.DirectoryWalkListener;
import org.codehaus.plexus.util.DirectoryWalker;
import org.jboss.forge.roaster.Roaster;
import org.jboss.forge.roaster.model.JavaType;
import org.jboss.forge.roaster.model.source.JavaClassSource;

@Mojo(name = "generate", defaultPhase = LifecyclePhase.GENERATE_SOURCES)
public class CritterMojo extends AbstractMojo {
  @Parameter(property = "directory", defaultValue = "src/main/java")
  private File directory;

  @Parameter(property = "includes", defaultValue = "**/*.java")
  private String includes;

  @Parameter(property = "package", defaultValue = "com.antwerkz.critter.criteria")
  private String criteriaPkg;

  @Override
  public void execute() throws MojoExecutionException, MojoFailureException {
    CritterContext context = new CritterContext(criteriaPkg);
//    final FileSet fileSet = new FileSet();
//    fileSet.setDirectory(directory);
//    fileSet.setIncludes(asList(includes.split(",")));
//    fileSet.includes()
    final DirectoryWalker walker = new DirectoryWalker();
    walker.setBaseDir(directory);
    walker.setIncludes(asList(includes.split(",")));

    walker.addDirectoryWalkListener(new DirectoryWalkListener() {
      @Override
      public void directoryWalkStarting(final File basedir) {
      }

      @Override
      public void directoryWalkStep(final int percentage, final File file) {
        if (!file.getName().endsWith("Criteria.java")) {
          final JavaType type;
          try {
            type = Roaster.parse(file);
          } catch (FileNotFoundException e) {
            throw new RuntimeException(e.getMessage(), e);
          }
          if (type instanceof JavaClassSource) {
            context.add(new CritterClass(context, type));
          }
        }
      }

      @Override
      public void directoryWalkFinished() {
      }

      @Override
      public void debug(final String message) {
      }
    });
    walker.scan();

    context.getClasses().stream()
        .forEach(critterClass -> critterClass.build(directory));
  }
}