jooby-apt: print error on failure

Author: jknack

modules/jooby-apt/src/main/java/io/jooby/apt/JoobyProcessor.java

@@ -15,7 +15,7 @@
 import java.nio.file.Files;
 import java.nio.file.Paths;
 import java.util.*;
-import java.util.function.Consumer;
+import java.util.function.BiConsumer;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -71,10 +71,10 @@ static String string(ProcessingEnvironment environment, String option, String de
   }
 
   protected MvcContext context;
-  private Consumer<String> output;
+  private BiConsumer<Diagnostic.Kind, String> output;
   private final Set<Object> processed = new HashSet<>();
 
-  public JoobyProcessor(Consumer<String> output) {
+  public JoobyProcessor(BiConsumer<Diagnostic.Kind, String> output) {
     this.output = output;
   }
 
@@ -88,38 +88,42 @@ public synchronized void init(ProcessingEnvironment processingEnvironment) {
             ofNullable(output)
                 .orElseGet(
                     () ->
-                        message ->
-                            processingEnvironment
-                                .getMessager()
-                                .printMessage(Diagnostic.Kind.OTHER, message)));
+                        (kind, message) ->
+                            processingEnvironment.getMessager().printMessage(kind, message)));
     super.init(processingEnvironment);
   }
 
   @Override
   public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
-    if (roundEnv.processingOver()) {
-      context.debug("Output:");
-      context.getRouters().forEach(it -> context.debug("  %s.java", it.getGeneratedType()));
-      if (context.generateServices()) {
-        doServices(context.getProcessingEnvironment().getFiler(), context.getRouters());
-      }
-      return false;
-    } else {
-      var routeMap = buildRouteRegistry(annotations, roundEnv);
-      for (var router : routeMap.values()) {
-        try {
-          var sourceCode = router.toSourceCode(null);
-          var sourceLocation = router.getGeneratedFilename();
-          onGeneratedSource(toJavaFileObject(sourceLocation, sourceCode));
-          context.debug("router %s: %s", router.getTargetType(), router.getGeneratedType());
-          router.getRoutes().forEach(it -> context.debug("   %s", it));
-          writeSource(router, sourceLocation, sourceCode);
-          context.add(router);
-        } catch (IOException cause) {
-          throw new RuntimeException("Unable to generate: " + router.getTargetType(), cause);
+    try {
+      if (roundEnv.processingOver()) {
+        context.debug("Output:");
+        context.getRouters().forEach(it -> context.debug("  %s.java", it.getGeneratedType()));
+        if (context.generateServices()) {
+          doServices(context.getProcessingEnvironment().getFiler(), context.getRouters());
+        }
+        return false;
+      } else {
+        var routeMap = buildRouteRegistry(annotations, roundEnv);
+        for (var router : routeMap.values()) {
+          try {
+            var sourceCode = router.toSourceCode(null);
+            var sourceLocation = router.getGeneratedFilename();
+            onGeneratedSource(toJavaFileObject(sourceLocation, sourceCode));
+            context.debug("router %s: %s", router.getTargetType(), router.getGeneratedType());
+            router.getRoutes().forEach(it -> context.debug("   %s", it));
+            writeSource(router, sourceLocation, sourceCode);
+            context.add(router);
+          } catch (IOException cause) {
+            throw new RuntimeException("Unable to generate: " + router.getTargetType(), cause);
+          }
         }
+        return true;
       }
-      return true;
+    } catch (Exception cause) {
+      context.error(
+          Optional.ofNullable(cause.getMessage()).orElse("Unable to generate routes"), cause);
+      throw sneakyThrow0(cause);
     }
   }
 

modules/jooby-apt/src/main/java/io/jooby/internal/apt/MvcContext.java

@@ -5,15 +5,17 @@
  */
 package io.jooby.internal.apt;
 
+import java.io.PrintWriter;
+import java.io.StringWriter;
 import java.util.*;
 import java.util.function.BiConsumer;
-import java.util.function.Consumer;
 import java.util.stream.Collectors;
 
 import javax.annotation.processing.ProcessingEnvironment;
 import javax.lang.model.element.*;
 import javax.lang.model.type.DeclaredType;
 import javax.lang.model.type.TypeMirror;
+import javax.tools.Diagnostic;
 
 import io.jooby.apt.JoobyProcessor.Options;
 
@@ -30,13 +32,14 @@ private record ResultType(String type, String handler, boolean nonBlocking) {}
   private final boolean services;
   private final String routerPrefix;
   private final String routerSuffix;
-  private final Consumer<String> output;
+  private final BiConsumer<Diagnostic.Kind, String> output;
   private final List<MvcRouter> routers = new ArrayList<>();
   private final boolean returnType;
   private final boolean mvcMethod;
   private final Map<TypeElement, ResultType> handler = new HashMap<>();
 
-  public MvcContext(ProcessingEnvironment processingEnvironment, Consumer<String> output) {
+  public MvcContext(
+      ProcessingEnvironment processingEnvironment, BiConsumer<Diagnostic.Kind, String> output) {
     this.processingEnvironment = processingEnvironment;
     this.output = output;
     this.debug = Options.boolOpt(processingEnvironment, Options.DEBUG, false);
@@ -258,13 +261,29 @@ public boolean isIncremental() {
 
   public void debug(String message, Object... args) {
     if (debug) {
-      info(message, args);
+      report(Diagnostic.Kind.OTHER, message, args);
     }
   }
 
-  public void info(String message, Object... args) {
+  public void error(String message, Object... args) {
+    Throwable cause =
+        args.length > 0 && args[args.length - 1] instanceof Throwable
+            ? (Throwable) args[args.length - 1]
+            : null;
+    if (cause != null) {
+      var str = new StringWriter();
+      cause.printStackTrace(new PrintWriter(str, true));
+      var errorMessage = cause.getMessage();
+      args[args.length - 1] =
+          (errorMessage == null || errorMessage.equals(message) ? "" : errorMessage) + ":\n" + str;
+      message += ": %s";
+    }
+    report(Diagnostic.Kind.ERROR, message, args);
+  }
+
+  private void report(Diagnostic.Kind kind, String message, Object... args) {
     var msg = args.length == 0 ? message : message.formatted(args);
-    output.accept(msg);
+    output.accept(kind, msg);
   }
 
   public void generateStaticImports(MvcRouter mvcRouter, BiConsumer<String, String> consumer) {

modules/jooby-apt/src/test/java/io/jooby/apt/ProcessorRunner.java

@@ -60,7 +60,7 @@ private static class HookJoobyProcessor extends JoobyProcessor {
     private JavaFileObject source;
 
     public HookJoobyProcessor(Consumer<String> console) {
-      super(console);
+      super((kind, message) -> console.accept(message));
     }
 
     public GeneratedSourceClassLoader createClassLoader() {