apt: add route injection

Author: jknack

jooby/src/main/java/io/jooby/Route.java

@@ -16,7 +16,6 @@
 import java.util.Set;
 import java.util.HashSet;
 import java.util.Map;
-import java.util.HashMap;
 import java.util.List;
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -654,7 +653,7 @@ public Route(@Nonnull String method, @Nonnull String pattern, @Nonnull Handler h
   public @Nonnull Route setAttributes(@Nonnull Map<String, Object> attributes) {
     if (attributes.size() > 0) {
       if (this.attributes == EMPTY_MAP) {
-        this.attributes = new HashMap<>();
+        this.attributes = new TreeMap<>(String.CASE_INSENSITIVE_ORDER);
       }
       this.attributes.putAll(attributes);
     }

jooby/src/main/java/io/jooby/Usage.java

@@ -1,3 +1,8 @@
+/**
+ * Jooby https://jooby.io
+ * Apache License Version 2.0 https://jooby.io/LICENSE.txt
+ * Copyright 2014 Edgar Espina
+ */
 package io.jooby;
 
 import javax.annotation.Nonnull;

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

@@ -12,6 +12,7 @@
 import io.jooby.Multipart;
 import io.jooby.QueryString;
 import io.jooby.Reified;
+import io.jooby.Route;
 import io.jooby.Session;
 import io.jooby.Value;
 import io.jooby.apt.Annotations;
@@ -260,6 +261,9 @@ private boolean isTypeInjection() {
     if (is(Session.class) || is(Optional.class, Session.class)) {
       return true;
     }
+    if (is(Route.class)) {
+      return true;
+    }
     return false;
   }
 

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

@@ -155,6 +155,15 @@ public enum ParamKind {
       return Context.class.getDeclaredMethod("session");
     }
   },
+  ROUTE_PARAM {
+    @Override public Set<String> annotations() {
+      return Collections.emptySet();
+    }
+
+    @Override public Method valueObject(ParamDefinition param) throws NoSuchMethodException {
+      return Context.class.getDeclaredMethod("getRoute");
+    }
+  },
 
   BODY_PARAM {
     @Override public Method valueObject(ParamDefinition param) throws NoSuchMethodException {

modules/jooby-apt/src/test/java/source/RouteInjection.java

@@ -0,0 +1,14 @@
+package source;
+
+import io.jooby.Route;
+import io.jooby.annotations.GET;
+import io.jooby.annotations.Path;
+
+@Path("/route")
+public class RouteInjection {
+
+  @GET
+  public Route route(Route route) {
+    return route;
+  }
+}

modules/jooby-apt/src/test/java/tests/HandlerCompilerTest.java

@@ -8,6 +8,7 @@
 import io.jooby.Multipart;
 import io.jooby.ProvisioningException;
 import io.jooby.Reified;
+import io.jooby.Route;
 import io.jooby.StatusCode;
 import io.jooby.apt.MvcHandlerCompilerRunner;
 import org.junit.jupiter.api.Test;
@@ -17,6 +18,7 @@
 import source.NoPathRoute;
 import source.NullRoutes;
 import source.Provisioning;
+import source.RouteInjection;
 
 import java.io.InputStream;
 import java.math.BigDecimal;
@@ -419,6 +421,16 @@ public void nullRoutes() throws Exception {
     ;
   }
 
+  @Test
+  public void routeParam() throws Exception {
+    new MvcHandlerCompilerRunner(new RouteInjection())
+        .compile("/route", handler -> {
+          Route route = mock(Route.class);
+          assertEquals(route, handler.apply(new MockContext().setRoute(route)));
+        })
+    ;
+  }
+
   private Context mockContext(String method, String path) {
     Context ctx = mock(Context.class);
     when(ctx.getMethod()).thenReturn(method);

tests/src/test/java/examples/InstanceRouter.java

@@ -1,16 +1,21 @@
 package examples;
 
 import io.jooby.Context;
+import io.jooby.Route;
 import io.jooby.annotations.GET;
 import io.jooby.annotations.POST;
 import io.jooby.annotations.Path;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
 @Path("/")
 public class InstanceRouter {
 
   @GET
   @POST
-  public String getIt() {
+  @Role("some")
+  public String getIt(Route route) {
+    assertEquals("some", route.attribute("role"));
     return "Got it!";
   }