diff options
| -rw-r--r-- | src/main/java/foo/zaaarf/routecompass/Route.java | 17 | ||||
| -rw-r--r-- | src/main/java/foo/zaaarf/routecompass/RouteCompass.java | 39 |
2 files changed, 52 insertions, 4 deletions
diff --git a/src/main/java/foo/zaaarf/routecompass/Route.java b/src/main/java/foo/zaaarf/routecompass/Route.java index ade19a6..f56de0b 100644 --- a/src/main/java/foo/zaaarf/routecompass/Route.java +++ b/src/main/java/foo/zaaarf/routecompass/Route.java @@ -12,8 +12,9 @@ public class Route { public final String produces; public final String consumes; public final boolean deprecated; + public final Param[] params; - public Route(String route, RequestMethod[] methods, MediaType consumes, MediaType produces, boolean deprecated) { + public Route(String route, RequestMethod[] methods, MediaType consumes, MediaType produces, boolean deprecated, Param... params) { this.route = route; StringBuilder methodStringBuilder = new StringBuilder("["); @@ -33,5 +34,19 @@ public class Route { else this.consumes = null; this.deprecated = deprecated; + + this.params = params; + } + + public static class Param { + public final String typeFQN; + public final String name; + public final String defaultValue; + + public Param(String typeFQN, String name, String defaultValue) { + this.typeFQN = typeFQN; + this.name = name; + this.defaultValue = defaultValue; + } } } diff --git a/src/main/java/foo/zaaarf/routecompass/RouteCompass.java b/src/main/java/foo/zaaarf/routecompass/RouteCompass.java index 6b3ab76..94aeb7d 100644 --- a/src/main/java/foo/zaaarf/routecompass/RouteCompass.java +++ b/src/main/java/foo/zaaarf/routecompass/RouteCompass.java @@ -10,6 +10,7 @@ import javax.lang.model.SourceVersion; import javax.lang.model.element.Element; import javax.lang.model.element.ExecutableElement; import javax.lang.model.element.TypeElement; +import javax.lang.model.element.VariableElement; import javax.tools.Diagnostic; import javax.tools.FileObject; import javax.tools.StandardLocation; @@ -50,12 +51,13 @@ public class RouteCompass extends AbstractProcessor { this.getRequestMethods(annotationType, elem), this.getConsumedType(annotationType, elem), this.getProducedType(annotationType, elem), - this.isDeprecated(elem) + this.isDeprecated(elem), + this.getParams(elem.getParameters()) )); }); } - try { //TODO: support param printing + try { FileObject serviceProvider = this.processingEnv.getFiler().createResource( StandardLocation.SOURCE_OUTPUT, "", "routes" ); @@ -66,10 +68,19 @@ public class RouteCompass extends AbstractProcessor { List<Route> routesInClass = this.foundRoutes.get(componentClass); for(Route r : routesInClass) { - out.print("\t- " + r.method + r.route); + out.print("\t- "); + if(r.deprecated) out.print("[DEPRECATED] "); + out.print(r.method + " " + r.route); if(r.consumes != null) out.print("(expects: " + r.consumes + ")"); if(r.produces != null) out.print("(returns: " + r.produces + ")"); out.println(); + + for(Route.Param p : r.params) { + out.print("\t\t- " + p.typeFQN + " " + p.name); + if(p.defaultValue != null) + out.print(" " + "(default: " + p.defaultValue + ")"); + out.println(); + } } } @@ -132,6 +143,28 @@ public class RouteCompass extends AbstractProcessor { || elem.getEnclosingElement().getAnnotation(Deprecated.class) != null; } + private Route.Param[] getParams(List<? extends VariableElement> params) { + return params.stream() + .map(p -> { + RequestParam ann = p.getAnnotation(RequestParam.class); + if(ann == null) return null; + + String name = ann.name(); //first try annotation.name() + name = name.isEmpty() + ? ann.value() //then annotation.value() + : name; + name = name.isEmpty() + ? p.getSimpleName().toString() //fall back on parameter name + : name; + + String defaultValue = ann.defaultValue(); + if(defaultValue.equals(ValueConstants.DEFAULT_NONE)) + defaultValue = null; + + return new Route.Param(name, defaultValue, p.asType().toString()); + }).filter(Objects::nonNull).toArray(Route.Param[]::new); + } + @SuppressWarnings({"OptionalGetWithoutIsPresent", "unchecked"}) private <T> T getAnnotationFieldsValue(TypeElement annotationType, Element element, String ... fieldNames) throws ReflectiveOperationException { |