se.bjurr.jmib.processor.AnnotationProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-method-invocation-builder Show documentation

Generates builders for invoking methods on instantiated objects

There is a newer version: 1.4

Show newest version

package se.bjurr.jmib.processor;

import static com.google.common.collect.Sets.newHashSet;
import static java.util.logging.Level.SEVERE;

import java.util.Set;
import java.util.logging.Logger;
import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.tools.Diagnostic;
import se.bjurr.jmib.anotations.GenerateMethodInvocationBuilder;

public class AnnotationProcessor extends AbstractProcessor {

  public AnnotationProcessor() {}

  @Override
  public Set getSupportedAnnotationTypes() {
    return newHashSet( //
        GenerateMethodInvocationBuilder.class.getName());
  }

  @Override
  public SourceVersion getSupportedSourceVersion() {
    return SourceVersion.latestSupported();
  }

  @Override
  public boolean process(Set annotations, RoundEnvironment roundEnv) {
    ElementHandler elementHandler =
        new ElementHandler(
            this.processingEnv.getElementUtils(),
            this.processingEnv.getFiler(),
            this.processingEnv.getMessager());

    Set elements =
        roundEnv.getElementsAnnotatedWith(GenerateMethodInvocationBuilder.class);
    for (Element element : elements) {
      try {
        elementHandler.handle((TypeElement) element);

      } catch (Exception e) {
        Logger logger = Logger.getLogger(this.getClass().getSimpleName());
        logger.log(SEVERE, e.getMessage(), e);
        this.processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, e.getMessage());
      }
    }

    return false;
  }
}