#3 - Only add @generated if available

Author: rbygrave

src/main/java/io/dinject/javalin/generator/BeanReader.java

@@ -28,11 +28,12 @@ class BeanReader {
 
   private final Set<String> importTypes = new TreeSet<>();
 
-  BeanReader(TypeElement beanType) {
+  BeanReader(TypeElement beanType, ProcessingContext ctx) {
     this.beanType = beanType;
     this.roles = Util.findRoles(beanType);
-
-    importTypes.add(Constants.GENERATED);
+    if (ctx.isGeneratedAvailable()) {
+      importTypes.add(Constants.GENERATED);
+    }
     importTypes.add(Constants.SINGLETON);
     importTypes.add(Constants.API_BUILDER);
     importTypes.add(Constants.WEB_ROUTES);

src/main/java/io/dinject/javalin/generator/ControllerRouteWriter.java

@@ -71,7 +71,9 @@ private void writeClassEnd() {
 
   private void writeClassStart() {
 
-    writer.append(Constants.AT_GENERATED).eol();
+    if (ctx.isGeneratedAvailable()) {
+      writer.append(Constants.AT_GENERATED).eol();
+    }
     writer.append("@Singleton").eol();
     writer.append("public class ").append(shortName).append("$route implements WebRoutes {").eol().eol();
 

src/main/java/io/dinject/javalin/generator/ProcessingContext.java

@@ -4,18 +4,33 @@
 import javax.annotation.processing.Messager;
 import javax.annotation.processing.ProcessingEnvironment;
 import javax.lang.model.element.Element;
+import javax.lang.model.util.Elements;
 import javax.tools.Diagnostic;
 import javax.tools.JavaFileObject;
 import java.io.IOException;
 
+import static io.dinject.javalin.generator.Constants.GENERATED;
+
 class ProcessingContext {
 
   private final Messager messager;
   private final Filer filer;
+  private final Elements elementUtils;
+  private final boolean generatedAvailable;
 
   ProcessingContext(ProcessingEnvironment env) {
     this.messager = env.getMessager();
     this.filer = env.getFiler();
+    this.elementUtils = env.getElementUtils();
+    this.generatedAvailable = isTypeAvailable(GENERATED);
+  }
+
+  private boolean isTypeAvailable(String canonicalName) {
+    return null != elementUtils.getTypeElement(canonicalName);
+  }
+
+  boolean isGeneratedAvailable() {
+    return generatedAvailable;
   }
 
   void logError(Element e, String msg, Object... args) {

src/main/java/io/dinject/javalin/generator/Processor.java

@@ -47,7 +47,7 @@ public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment
 
   private void writeControllerAdapter(Element controller) {
     if (controller instanceof TypeElement) {
-      BeanReader reader = new BeanReader((TypeElement) controller);
+      BeanReader reader = new BeanReader((TypeElement) controller, processingContext);
       reader.read();
       try {
         ControllerRouteWriter writer = new ControllerRouteWriter(reader, processingContext);