io.dinject.javalin.generator.Processor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of javalin-generator Show documentation

There is a newer version: 1.22

package io.dinject.javalin.generator;

import io.dinject.controller.Controller;

import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.ProcessingEnvironment;
import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import java.util.LinkedHashSet;
import java.util.Set;

public class Processor extends AbstractProcessor {

  private ProcessingContext processingContext;

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

  @Override
  public Set getSupportedAnnotationTypes() {

    Set annotations = new LinkedHashSet<>();
    annotations.add(Controller.class.getCanonicalName());
    return annotations;
  }

  @Override
  public synchronized void init(ProcessingEnvironment processingEnv) {
    super.init(processingEnv);
    this.processingContext = new ProcessingContext(processingEnv);
  }

  @Override
  public boolean process(Set annotations, RoundEnvironment roundEnv) {

    Set controllers = roundEnv.getElementsAnnotatedWith(Controller.class);
    for (Element controller : controllers) {
      writeControllerAdapter(controller);
    }

    return false;
  }

  private void writeControllerAdapter(Element controller) {
    if (controller instanceof TypeElement) {
      BeanReader reader = new BeanReader((TypeElement) controller);
      reader.read();
      try {
        ControllerRouteWriter writer = new ControllerRouteWriter(reader, processingContext);
        writer.write();
      } catch (Exception e) {
        e.printStackTrace();
        processingContext.logError(reader.getBeanType(), "Failed to write $route class");
      }
    }
  }

}