io.inverno.mod.web.compiler.internal.WebServerControllerConfigurerClassGenerator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of inverno-web-compiler Show documentation
Show all versions of inverno-web-compiler Show documentation
Inverno web compiler module providing an Inverno compiler plugin to aggregate module's web routes and web router configurers into a single web router configurer
/*
* Copyright 2021 Jeremy KUHN
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.inverno.mod.web.compiler.internal;
import io.inverno.core.annotation.Bean;
import io.inverno.mod.base.net.URIs;
import io.inverno.mod.http.base.ExchangeContext;
import io.inverno.mod.http.base.ws.WebSocketMessage;
import io.inverno.mod.web.server.ErrorWebInterceptorsConfigurer;
import io.inverno.mod.web.server.ErrorWebRouter;
import io.inverno.mod.web.server.ErrorWebRouterConfigurer;
import io.inverno.mod.web.server.ErrorWebRoutesConfigurer;
import io.inverno.mod.web.server.WebInterceptorsConfigurer;
import io.inverno.mod.web.server.WebRouter;
import io.inverno.mod.web.server.WebRouterConfigurer;
import io.inverno.mod.web.server.WebRoutesConfigurer;
import io.inverno.mod.web.server.WebServerControllerConfigurer;
import io.inverno.mod.web.server.annotation.WebRoutes;
import io.inverno.mod.web.compiler.internal.WebServerControllerConfigurerClassGenerationContext.GenerationMode;
import io.inverno.mod.web.compiler.spi.ErrorWebInterceptorsConfigurerInfo;
import io.inverno.mod.web.compiler.spi.ErrorWebRouterConfigurerInfo;
import io.inverno.mod.web.compiler.spi.ErrorWebRoutesConfigurerInfo;
import io.inverno.mod.web.compiler.spi.WebBasicParameterInfo;
import io.inverno.mod.web.compiler.spi.WebControllerInfo;
import io.inverno.mod.web.compiler.spi.WebCookieParameterInfo;
import io.inverno.mod.web.compiler.spi.WebExchangeContextParameterInfo;
import io.inverno.mod.web.compiler.spi.WebExchangeParameterInfo;
import io.inverno.mod.web.compiler.spi.WebFormParameterInfo;
import io.inverno.mod.web.compiler.spi.WebHeaderParameterInfo;
import io.inverno.mod.web.compiler.spi.WebInterceptorsConfigurerInfo;
import io.inverno.mod.web.compiler.spi.WebParameterInfo;
import io.inverno.mod.web.compiler.spi.WebPathParameterInfo;
import io.inverno.mod.web.compiler.spi.WebQueryParameterInfo;
import io.inverno.mod.web.compiler.spi.WebRequestBodyParameterInfo;
import io.inverno.mod.web.compiler.spi.WebRequestBodyParameterInfo.RequestBodyKind;
import io.inverno.mod.web.compiler.spi.WebRequestBodyParameterInfo.RequestBodyReactiveKind;
import io.inverno.mod.web.compiler.spi.WebResponseBodyInfo;
import io.inverno.mod.web.compiler.spi.WebResponseBodyInfo.ResponseBodyKind;
import io.inverno.mod.web.compiler.spi.WebResponseBodyInfo.ResponseBodyReactiveKind;
import io.inverno.mod.web.compiler.spi.WebRouteInfo;
import io.inverno.mod.web.compiler.spi.WebRouterConfigurerInfo;
import io.inverno.mod.web.compiler.spi.WebRoutesConfigurerInfo;
import io.inverno.mod.web.compiler.spi.WebServerControllerConfigurerInfo;
import io.inverno.mod.web.compiler.spi.WebServerControllerConfigurerInfoVisitor;
import io.inverno.mod.web.compiler.spi.WebSocketBoundPublisherInfo;
import io.inverno.mod.web.compiler.spi.WebSocketExchangeParameterInfo;
import io.inverno.mod.web.compiler.spi.WebSocketInboundParameterInfo;
import io.inverno.mod.web.compiler.spi.WebSocketInboundPublisherParameterInfo;
import io.inverno.mod.web.compiler.spi.WebSocketOutboundParameterInfo;
import io.inverno.mod.web.compiler.spi.WebSocketOutboundPublisherInfo;
import io.inverno.mod.web.compiler.spi.WebSocketParameterInfo;
import io.inverno.mod.web.compiler.spi.WebSocketRouteInfo;
import io.inverno.mod.web.compiler.spi.WebSseEventFactoryParameterInfo;
import java.time.ZonedDateTime;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.ArrayType;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.ExecutableType;
import javax.lang.model.type.PrimitiveType;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.util.ElementFilter;
import org.apache.commons.text.StringEscapeUtils;
/**
*
* A {@link WebServerControllerConfigurerInfoVisitor} implementation used to generates a web router configurer class in an Inverno module.
*
*
* @author Jeremy Kuhn
* @since 1.0
*/
class WebServerControllerConfigurerClassGenerator implements WebServerControllerConfigurerInfoVisitor {
@Override
public StringBuilder visit(WebServerControllerConfigurerInfo controllerConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
String configurerClassName = controllerConfigurerInfo.getQualifiedName().getClassName();
String configurerPackageName = configurerClassName.lastIndexOf(".") != -1 ? configurerClassName.substring(0, configurerClassName.lastIndexOf(".")) : "";
configurerClassName = configurerClassName.substring(configurerPackageName.length() + 1);
if(context.getMode() == GenerationMode.CONFIGURER_CLASS) {
TypeMirror generatedType = context.getElementUtils().getTypeElement(context.getElementUtils().getModuleElement("java.compiler"), "javax.annotation.processing.Generated").asType();
TypeMirror beanAnnotationType = context.getElementUtils().getTypeElement(Bean.class.getCanonicalName()).asType();
TypeMirror webServerControllerConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(WebServerControllerConfigurer.class.getCanonicalName()).asType());
TypeMirror routerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(WebRouter.class.getCanonicalName()).asType());
TypeMirror errorRouterType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(ErrorWebRouter.class.getCanonicalName()).asType());
context.addImport(configurerClassName, configurerPackageName + "." + configurerClassName);
context.addImport("Context", configurerPackageName + "." + configurerClassName + ".Context");
StringBuilder configurerAnnotation = this.visit(controllerConfigurerInfo, context.withIndentDepth(1).withMode(GenerationMode.CONFIGURER_ANNOTATION));
StringBuilder configurer_controller_fields = Arrays.stream(controllerConfigurerInfo.getControllers())
.map(controllerInfo -> this.visit(controllerInfo, context.withIndentDepth(1).withMode(GenerationMode.CONTROLLER_FIELD)))
.collect(context.joining(System.lineSeparator()));
// List>
TypeMirror interceptorsConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(WebInterceptorsConfigurer.class.getCanonicalName()).asType());
// List>
TypeMirror routesConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(WebRoutesConfigurer.class.getCanonicalName()).asType());
// List>
TypeMirror routerConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(WebRouterConfigurer.class.getCanonicalName()).asType());
StringBuilder interceptorsConfigurersDecl = new StringBuilder(context.getListTypeName()).append("<").append(context.getTypeName(interceptorsConfigurerType)).append("> interceptorsConfigurers");
StringBuilder routesConfigurersDecl = new StringBuilder(context.getListTypeName()).append("<").append(context.getTypeName(routesConfigurerType)).append("> routesConfigurers");
StringBuilder routerConfigurersDecl = new StringBuilder(context.getListTypeName()).append("<").append(context.getTypeName(routerConfigurerType)).append("> routerConfigurers");
StringBuilder interceptorsConfigurersField = new StringBuilder(context.indent(1)).append("private ").append(interceptorsConfigurersDecl).append(";");
StringBuilder routesConfigurersField = new StringBuilder(context.indent(1)).append("private ").append(routesConfigurersDecl).append(";");
StringBuilder routerConfigurersField = new StringBuilder(context.indent(1)).append("private ").append(routerConfigurersDecl).append(";");
StringBuilder interceptorsConfigurersSetter = new StringBuilder(context.indent(1)).append("public void setInterceptorsConfigurers(").append(interceptorsConfigurersDecl).append(") {").append(System.lineSeparator());
interceptorsConfigurersSetter.append(context.indent(2)).append("this.interceptorsConfigurers = interceptorsConfigurers;").append(System.lineSeparator());
interceptorsConfigurersSetter.append(context.indent(1)).append("}");
StringBuilder routesConfigurersSetter = new StringBuilder(context.indent(1)).append("public void setRoutesConfigurers(").append(routesConfigurersDecl).append(") {").append(System.lineSeparator());
routesConfigurersSetter.append(context.indent(2)).append("this.routesConfigurers = routesConfigurers;").append(System.lineSeparator());
routesConfigurersSetter.append(context.indent(1)).append("}");
StringBuilder routerConfigurersSetter = new StringBuilder(context.indent(1)).append("public void setRouterConfigurers(").append(routerConfigurersDecl).append(") {").append(System.lineSeparator());
routerConfigurersSetter.append(context.indent(2)).append("this.routerConfigurers = routerConfigurers;").append(System.lineSeparator());
routerConfigurersSetter.append(context.indent(1)).append("}");
// List>
TypeMirror errorInterceptorsConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(ErrorWebInterceptorsConfigurer.class.getCanonicalName()).asType());
// List>
TypeMirror errorRoutesConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(ErrorWebRoutesConfigurer.class.getCanonicalName()).asType());
// List>
TypeMirror errorRouterConfigurerType = context.getTypeUtils().erasure(context.getElementUtils().getTypeElement(ErrorWebRouterConfigurer.class.getCanonicalName()).asType());
StringBuilder errorInterceptorsConfigurersDecl = new StringBuilder(context.getListTypeName()).append("<").append(context.getTypeName(errorInterceptorsConfigurerType)).append("> errorInterceptorsConfigurers");
StringBuilder errorRoutesConfigurersDecl = new StringBuilder(context.getListTypeName()).append("<").append(context.getTypeName(errorRoutesConfigurerType)).append("> errorRoutesConfigurers");
StringBuilder errorRouterConfigurersDecl = new StringBuilder(context.getListTypeName()).append("<").append(context.getTypeName(errorRouterConfigurerType)).append("> errorRouterConfigurers");
StringBuilder errorInterceptorsConfigurersField = new StringBuilder(context.indent(1)).append("private ").append(errorInterceptorsConfigurersDecl).append(";");
StringBuilder errorRoutesConfigurersField = new StringBuilder(context.indent(1)).append("private ").append(errorRoutesConfigurersDecl).append(";");
StringBuilder errorRouterConfigurersField = new StringBuilder(context.indent(1)).append("private ").append(errorRouterConfigurersDecl).append(";");
StringBuilder errorInterceptorsConfigurersSetter = new StringBuilder(context.indent(1)).append("public void setErrorInterceptorsConfigurers(").append(errorInterceptorsConfigurersDecl).append(") {").append(System.lineSeparator());
errorInterceptorsConfigurersSetter.append(context.indent(2)).append("this.errorInterceptorsConfigurers = errorInterceptorsConfigurers;").append(System.lineSeparator());
errorInterceptorsConfigurersSetter.append(context.indent(1)).append("}");
StringBuilder errorRoutesConfigurersSetter = new StringBuilder(context.indent(1)).append("public void setErrorRoutesConfigurers(").append(errorRoutesConfigurersDecl).append(") {").append(System.lineSeparator());
errorRoutesConfigurersSetter.append(context.indent(2)).append("this.errorRoutesConfigurers = errorRoutesConfigurers;").append(System.lineSeparator());
errorRoutesConfigurersSetter.append(context.indent(1)).append("}");
StringBuilder errorRouterConfigurersSetter = new StringBuilder(context.indent(1)).append("public void setErrorRouterConfigurers(").append(errorRouterConfigurersDecl).append(") {").append(System.lineSeparator());
errorRouterConfigurersSetter.append(context.indent(2)).append("this.errorRouterConfigurers = errorRouterConfigurers;").append(System.lineSeparator());
errorRouterConfigurersSetter.append(context.indent(1)).append("}");
StringBuilder configurer_constructor = new StringBuilder(context.indent(1)).append("public ").append(configurerClassName).append("(");
configurer_constructor.append(Arrays.stream(controllerConfigurerInfo.getControllers()).map(controllerInfo -> this.visit(controllerInfo, context.withIndentDepth(0).withMode(GenerationMode.CONTROLLER_PARAMETER))).collect(context.joining(", ")));
configurer_constructor.append(") {").append(System.lineSeparator());
configurer_constructor.append(Arrays.stream(controllerConfigurerInfo.getControllers()).map(controllerInfo -> this.visit(controllerInfo, context.withIndentDepth(2).withMode(GenerationMode.CONTROLLER_ASSIGNMENT))).collect(context.joining(System.lineSeparator())));
configurer_constructor.append(System.lineSeparator()).append(context.indent(1)).append("}");
StringBuilder configurer_configure = new StringBuilder(context.indent(1)).append("@Override").append(System.lineSeparator());
configurer_configure.append(context.indent(1)).append("public void configure(").append(context.getTypeName(routerType)).append("<").append(configurerClassName).append(".Context> router) {");
configurer_configure.append(System.lineSeparator()).append(context.indent(2)).append("router").append(System.lineSeparator());
configurer_configure.append(context.indent(3)).append(".configureInterceptors(this.interceptorsConfigurers)").append(System.lineSeparator());
configurer_configure.append(context.indent(3)).append(".configure(this.routerConfigurers)").append(System.lineSeparator());
configurer_configure.append(context.indent(3)).append(".configureRoutes(this.routesConfigurers)").append(System.lineSeparator());
if(controllerConfigurerInfo.getControllers().length > 0) {
configurer_configure.append(Arrays.stream(controllerConfigurerInfo.getControllers()).map(controllerInfo -> this.visit(controllerInfo, context.withIndentDepth(3).withMode(GenerationMode.ROUTE_DECLARATION))).collect(context.joining(System.lineSeparator())));
}
configurer_configure.append(";").append(System.lineSeparator());
configurer_configure.append(context.indent(1)).append("}");
StringBuilder configurer_configure_error = new StringBuilder(context.indent(1)).append("@Override").append(System.lineSeparator());
configurer_configure_error.append(context.indent(1)).append("public void configure(").append(context.getTypeName(errorRouterType)).append("<").append(configurerClassName).append(".Context> errorRouter) {");
configurer_configure_error.append(System.lineSeparator()).append(context.indent(2)).append("errorRouter").append(System.lineSeparator());
configurer_configure_error.append(context.indent(3)).append(".configureInterceptors(this.errorInterceptorsConfigurers)").append(System.lineSeparator());
configurer_configure_error.append(context.indent(3)).append(".configure(this.errorRouterConfigurers)").append(System.lineSeparator());
configurer_configure_error.append(context.indent(3)).append(".configureRoutes(this.errorRoutesConfigurers);").append(System.lineSeparator());
configurer_configure_error.append(context.indent(1)).append("}");
StringBuilder configurer_context_creator = this.visit(controllerConfigurerInfo, context.withIndentDepth(1).withMode(GenerationMode.CONFIGURER_CONTEXT_CREATOR));
StringBuilder configurer_context = this.visit(controllerConfigurerInfo, context.withIndentDepth(1).withMode(GenerationMode.CONFIGURER_CONTEXT));
StringBuilder configurer_class = new StringBuilder();
configurer_class.append(configurerAnnotation).append(System.lineSeparator());
configurer_class.append("@").append(context.getTypeName(beanAnnotationType)).append("( name = \"").append(controllerConfigurerInfo.getQualifiedName().getBeanName()).append("\" )").append(System.lineSeparator());
configurer_class.append("@").append(context.getTypeName(generatedType)).append("(value=\"").append(WebServerControllerConfigurerCompilerPlugin.class.getCanonicalName()).append("\", date = \"").append(ZonedDateTime.now().toString()).append("\")").append(System.lineSeparator());
configurer_class.append("public final class ").append(configurerClassName).append(" implements ").append(context.getTypeName(webServerControllerConfigurerType)).append("<").append(configurerClassName).append(".Context> {").append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(interceptorsConfigurersField).append(System.lineSeparator());
configurer_class.append(routesConfigurersField).append(System.lineSeparator());
configurer_class.append(routerConfigurersField).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(errorInterceptorsConfigurersField).append(System.lineSeparator());
configurer_class.append(errorRoutesConfigurersField).append(System.lineSeparator());
configurer_class.append(errorRouterConfigurersField).append(System.lineSeparator()).append(System.lineSeparator());
if(controllerConfigurerInfo.getControllers().length > 0) {
configurer_class.append(configurer_controller_fields).append(System.lineSeparator()).append(System.lineSeparator());
}
configurer_class.append(configurer_constructor).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(configurer_configure).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(configurer_configure_error).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(interceptorsConfigurersSetter).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(routesConfigurersSetter).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(routerConfigurersSetter).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(errorInterceptorsConfigurersSetter).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(errorRoutesConfigurersSetter).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(errorRouterConfigurersSetter).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(configurer_context_creator).append(System.lineSeparator()).append(System.lineSeparator());
configurer_class.append(configurer_context).append(System.lineSeparator());
configurer_class.append("}");
context.removeImport(configurerClassName);
context.removeImport("Context");
configurer_class.insert(0, System.lineSeparator() + System.lineSeparator()).insert(0, context.getImports().stream().sorted().filter(i -> i.lastIndexOf(".") > 0 && !i.substring(0, i.lastIndexOf(".")).equals(configurerPackageName)).map(i -> new StringBuilder().append("import ").append(i).append(";")).collect(context.joining(System.lineSeparator())));
if(!configurerPackageName.equals("")) {
configurer_class.insert(0, ";" + System.lineSeparator() + System.lineSeparator()).insert(0, configurerPackageName).insert(0, "package ");
}
return configurer_class;
}
else if(context.getMode() == GenerationMode.CONFIGURER_ANNOTATION) {
TypeMirror webRoutesAnnotationType = context.getElementUtils().getTypeElement(WebRoutes.class.getCanonicalName()).asType();
StringBuilder result = new StringBuilder();
result.append("@").append(context.getTypeName(webRoutesAnnotationType));
WebServerControllerConfigurerClassGenerationContext webSocketRouteAnnotationContext = context.withMode(GenerationMode.WEBSOCKET_ROUTE_ANNOTATION).withIndentDepthAdd(1);
StringBuilder webSocketRoutes = Stream.of(
Arrays.stream(controllerConfigurerInfo.getRoutesConfigurers()).map(routesConfigurerInfo -> this.visit(routesConfigurerInfo, webSocketRouteAnnotationContext)),
Arrays.stream(controllerConfigurerInfo.getRouterConfigurers()).map(routerConfigurerInfo -> this.visit(routerConfigurerInfo, webSocketRouteAnnotationContext)),
Arrays.stream(controllerConfigurerInfo.getControllers())
.flatMap(controllerInfo -> Arrays.stream(controllerInfo.getRoutes()).map(routeInfo -> this.visit(routeInfo, webSocketRouteAnnotationContext.withWebController(controllerInfo))))
)
.flatMap(Function.identity())
.filter(s -> !s.isEmpty())
.collect(context.joining("," + System.lineSeparator()));
WebServerControllerConfigurerClassGenerationContext routeAnnotationContext;
if(webSocketRoutes.isEmpty()) {
routeAnnotationContext= context.withMode(GenerationMode.ROUTE_ANNOTATION);
}
else {
routeAnnotationContext = context.withMode(GenerationMode.ROUTE_ANNOTATION).withIndentDepthAdd(1);
}
StringBuilder webRoutes = Stream.of(
Arrays.stream(controllerConfigurerInfo.getRoutesConfigurers()).map(routesConfigurerInfo -> this.visit(routesConfigurerInfo, routeAnnotationContext)),
Arrays.stream(controllerConfigurerInfo.getRouterConfigurers()).map(routerConfigurerInfo -> this.visit(routerConfigurerInfo, routeAnnotationContext)),
Arrays.stream(controllerConfigurerInfo.getControllers())
.flatMap(controllerInfo -> Arrays.stream(controllerInfo.getRoutes()).map(routeInfo -> this.visit(routeInfo, routeAnnotationContext.withWebController(controllerInfo))))
)
.flatMap(Function.identity())
.filter(s -> !s.isEmpty())
.collect(context.joining("," + System.lineSeparator()));
if(webSocketRoutes.isEmpty()) {
result.append("({").append(System.lineSeparator());
result.append(webRoutes);
result.append(System.lineSeparator()).append("})");
}
else {
result.append("(").append(System.lineSeparator());
result.append(context.indent(0)).append("value = {").append(System.lineSeparator());
result.append(webRoutes).append(System.lineSeparator());
result.append(context.indent(0)).append("},").append(System.lineSeparator());
result.append(context.indent(0)).append("webSockets = {").append(System.lineSeparator());
result.append(webSocketRoutes).append(System.lineSeparator());
result.append(context.indent(0)).append("}").append(System.lineSeparator());
result.append(")");
}
return result;
}
else if(context.getMode() == GenerationMode.CONFIGURER_CONTEXT) {
StringBuilder result = new StringBuilder();
result.append(context.indent(0)).append("public static interface Context extends ");
if(controllerConfigurerInfo.getContextTypes().length > 0) {
result.append(Arrays.stream(controllerConfigurerInfo.getContextTypes()).map(context::getTypeName).collect(Collectors.joining(", ")));
}
else {
result.append(context.getTypeName(context.getElementUtils().getTypeElement(ExchangeContext.class.getCanonicalName()).asType()));
}
result.append(" {}");
return result;
}
else if(context.getMode() == GenerationMode.CONFIGURER_CONTEXT_CREATOR) {
// For each context type I must define the methods in order to implement the context
// This can actually be tricky because some interface may override others
// Let's start by listing all methods
Map context_fields = new HashMap<>();
Map context_methods = new HashMap<>();
Set defaultMethods = new HashSet<>();
for(TypeMirror contextType : controllerConfigurerInfo.getContextTypes()) {
ElementFilter.methodsIn(context.getElementUtils().getAllMembers((TypeElement)context.getTypeUtils().asElement(contextType))).stream()
.filter(exectuableElement -> exectuableElement.getEnclosingElement().getKind() == ElementKind.INTERFACE && !exectuableElement.getModifiers().contains(Modifier.STATIC))
.forEach(exectuableElement -> {
ExecutableType executableType = (ExecutableType)context.getTypeUtils().asMemberOf((DeclaredType)contextType, exectuableElement);
StringBuilder signatureKeyBuilder = new StringBuilder();
signatureKeyBuilder.append(exectuableElement.getSimpleName().toString());
exectuableElement.getParameters().stream().map(variableElement -> context.getTypeUtils().erasure(variableElement.asType()).toString()).forEach(signatureKeyBuilder::append);
String signatureKey = signatureKeyBuilder.toString();
if(exectuableElement.isDefault()) {
defaultMethods.add(signatureKey);
context_methods.remove(signatureKey);
context_fields.remove(signatureKey);
}
else if(!defaultMethods.contains(signatureKey)) {
String fieldName = null;
TypeMirror fieldType = null;
StringBuilder contextMethod = new StringBuilder();
String methodName = exectuableElement.getSimpleName().toString();
if( (methodName.startsWith("get") || methodName.startsWith("is")) && executableType.getParameterTypes().isEmpty() && executableType.getReturnType().getKind() != TypeKind.VOID) {
fieldName = Character.toLowerCase(methodName.charAt(3)) + methodName.substring(4);
fieldType = executableType.getReturnType();
contextMethod.append(context.indent(2)).append("@Override").append(System.lineSeparator());
contextMethod.append(context.indent(2)).append("public ").append(context.getTypeName(executableType.getReturnType())).append(" ").append(methodName).append("() {").append(System.lineSeparator());
contextMethod.append(context.indent(3)).append("return this.").append(fieldName).append(";").append(System.lineSeparator());
contextMethod.append(context.indent(2)).append("}");
}
else if(methodName.startsWith("set") && executableType.getParameterTypes().size() == 1 && executableType.getReturnType().getKind() == TypeKind.VOID) {
fieldName = Character.toLowerCase(methodName.charAt(3)) + methodName.substring(4);
fieldType = executableType.getParameterTypes().get(0);
contextMethod.append(context.indent(2)).append("@Override").append(System.lineSeparator());
contextMethod.append(context.indent(2)).append("public void ").append(methodName).append("(").append(context.getTypeName(executableType.getParameterTypes().get(0))).append(" ").append(fieldName).append(") {").append(System.lineSeparator());
contextMethod.append(context.indent(3)).append("this.").append(fieldName).append(" = ").append(fieldName).append(";").append(System.lineSeparator());
contextMethod.append(context.indent(2)).append("}");
}
else {
contextMethod.append(context.indent(2)).append("public ").append(context.getTypeName(executableType.getReturnType())).append(" ").append(methodName).append("(").append(executableType.getParameterTypes().stream().map(parameterType -> new StringBuilder().append(context.getTypeName(parameterType)).append(" ").append(methodName)).collect(context.joining(", "))).append(") {").append(System.lineSeparator());
if(executableType.getReturnType().getKind() != TypeKind.VOID) {
contextMethod.append(context.indent(3)).append("return ");
switch(executableType.getReturnType().getKind()) {
case BOOLEAN: contextMethod.append("false");
break;
case BYTE:
case CHAR:
case SHORT:
case INT:
case LONG:
case FLOAT:
case DOUBLE: contextMethod.append("0");
break;
default: contextMethod.append("null");
}
contextMethod.append(";").append(System.lineSeparator());
}
contextMethod.append(context.indent(2)).append("}");
}
if(fieldName != null && fieldType != null) {
StringBuilder contextField = new StringBuilder();
contextField.append(context.indent(2)).append("private ").append(context.getTypeName(fieldType)).append(" ").append(fieldName).append(";");
context_fields.put(signatureKey, contextField.toString());
}
context_methods.put(signatureKey, contextMethod.toString());
}
});
}
StringBuilder context_creator = new StringBuilder();
context_creator.append(context.indent(0)).append("@Override").append(System.lineSeparator());
context_creator.append(context.indent(0)).append("public Context createContext() {").append(System.lineSeparator());
context_creator.append(context.indent(1)).append("return new Context() {").append(System.lineSeparator());
context_creator.append(context_fields.values().stream().distinct().collect(Collectors.joining(System.lineSeparator()))).append(System.lineSeparator()).append(System.lineSeparator());
context_creator.append(context_methods.values().stream().collect(Collectors.joining(System.lineSeparator() + System.lineSeparator()))).append(System.lineSeparator());
context_creator.append(context.indent(1)).append("};").append(System.lineSeparator());
context_creator.append(context.indent(0)).append("}");
return context_creator;
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebInterceptorsConfigurerInfo interceptorsConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
return new StringBuilder();
}
@Override
public StringBuilder visit(WebRoutesConfigurerInfo routesConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_ANNOTATION || context.getMode() == GenerationMode.WEBSOCKET_ROUTE_ANNOTATION) {
return Arrays.stream(routesConfigurerInfo.getRoutes())
.map(routeInfo -> this.visit(routeInfo, context))
.filter(s -> !s.isEmpty())
.collect(context.joining("," + System.lineSeparator()));
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebRouterConfigurerInfo routerConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.CONFIGURER_ANNOTATION || context.getMode() == GenerationMode.WEBSOCKET_ROUTE_ANNOTATION) {
return Arrays.stream(routerConfigurerInfo.getRoutes())
.map(routeInfo -> this.visit(routeInfo, context))
.filter(s -> !s.isEmpty())
.collect(context.joining("," + System.lineSeparator()));
}
return new StringBuilder();
}
@Override
public StringBuilder visit(ErrorWebInterceptorsConfigurerInfo errorInterceptorsConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
return new StringBuilder();
}
@Override
public StringBuilder visit(ErrorWebRoutesConfigurerInfo errorRoutesConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
return new StringBuilder();
}
@Override
public StringBuilder visit(ErrorWebRouterConfigurerInfo errorRouterConfigurerInfo, WebServerControllerConfigurerClassGenerationContext context) {
return new StringBuilder();
}
@Override
public StringBuilder visit(WebControllerInfo controllerInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.CONTROLLER_FIELD) {
return new StringBuilder(context.indent(0)).append("private final ").append(context.getTypeName(controllerInfo.getType())).append(" ").append(context.getFieldName(controllerInfo.getQualifiedName())).append(";");
}
else if(context.getMode() == GenerationMode.CONTROLLER_PARAMETER) {
return new StringBuilder(context.indent(0)).append(context.getTypeName(controllerInfo.getType())).append(" ").append(context.getFieldName(controllerInfo.getQualifiedName()));
}
else if(context.getMode() == GenerationMode.CONTROLLER_ASSIGNMENT) {
return new StringBuilder(context.indent(0)).append("this.").append(context.getFieldName(controllerInfo.getQualifiedName())).append(" = ").append(context.getFieldName(controllerInfo.getQualifiedName())).append(";");
}
else if(context.getMode() == GenerationMode.ROUTE_DECLARATION) {
return Arrays.stream(controllerInfo.getRoutes()).map(routeInfo -> this.visit(routeInfo, context.withWebController(controllerInfo))).collect(context.joining(System.lineSeparator()));
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebRouteInfo routeInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(routeInfo instanceof WebSocketRouteInfo) {
return this.visit((WebSocketRouteInfo)routeInfo, context);
}
else {
if(context.getMode() == GenerationMode.ROUTE_ANNOTATION) {
StringBuilder result = new StringBuilder();
result.append(context.indent(0)).append("@").append(context.getWebRouteAnnotationTypeName()).append("(");
result.append("path = { ");
if(routeInfo.getPaths().length > 0) {
result.append(Arrays.stream(routeInfo.getPaths())
.map(path -> "\"" + StringEscapeUtils.escapeJava(routeInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).path(path, false).buildRawPath())
.orElse(path)) + "\""
)
.collect(Collectors.joining(", "))
);
}
else {
routeInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).buildRawPath())
.ifPresent(rootPath -> result.append("\"").append(rootPath).append("\""));
}
result.append(" }");
if(routeInfo.isMatchTrailingSlash()) {
result.append(", matchTrailingSlash = true");
}
if(routeInfo.getMethods() != null && routeInfo.getMethods().length > 0) {
result.append(", method = { ").append(Arrays.stream(routeInfo.getMethods()).map(method -> context.getMethodTypeName() + "." + method.toString()).collect(Collectors.joining(", "))).append(" }");
}
if(routeInfo.getConsumes() != null && routeInfo.getConsumes().length > 0) {
result.append(", consumes = { ").append(Arrays.stream(routeInfo.getConsumes()).map(consumes -> "\"" + consumes + "\"").collect(Collectors.joining(", "))).append(" }");
}
if(routeInfo.getProduces() != null && routeInfo.getProduces().length > 0) {
result.append(", produces = { ").append(Arrays.stream(routeInfo.getProduces()).map(produces -> "\"" + produces + "\"").collect(Collectors.joining(", "))).append(" }");
}
if(routeInfo.getLanguages() != null && routeInfo.getLanguages().length > 0) {
result.append(", language = { ").append(Arrays.stream(routeInfo.getLanguages()).map(language -> "\"" + language + "\"").collect(Collectors.joining(", "))).append(" }");
}
result.append(")");
return result;
}
else if(context.getMode() == GenerationMode.ROUTE_DECLARATION) {
boolean typesMode = context.isTypeMode(routeInfo);
StringBuilder routeHandler = new StringBuilder("exchange -> {").append(System.lineSeparator());
routeHandler.append(this.visit(routeInfo.getResponseBody(), context.withIndentDepth(context.getIndentDepth() + (typesMode ? 2 : 1) ).withWebRoute(routeInfo).withMode(typesMode ? GenerationMode.ROUTE_HANDLER_TYPE : GenerationMode.ROUTE_HANDLER_CLASS)));
routeHandler.append(context.indent(typesMode ? 1 : 0)).append("}");
StringBuilder routeManager = new StringBuilder();
if(routeInfo.getPaths().length > 0) {
routeManager.append(Arrays.stream(routeInfo.getPaths())
.map(path -> ".path(\"" + StringEscapeUtils.escapeJava(routeInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).path(path, false).buildRawPath())
.orElse(path)) + "\", " + routeInfo.isMatchTrailingSlash() + ")"
)
.collect(Collectors.joining())
);
}
else {
routeInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).buildRawPath())
.ifPresent(rootPath -> routeManager.append(".path(\"").append(StringEscapeUtils.escapeJava(rootPath)).append("\", ").append(routeInfo.isMatchTrailingSlash()).append(")"));
}
if(routeInfo.getMethods() != null && routeInfo.getMethods().length > 0) {
routeManager.append(Arrays.stream(routeInfo.getMethods()).map(method -> ".method(" + context.getMethodTypeName() + "." + method.toString() + ")").collect(Collectors.joining()));
}
if(routeInfo.getConsumes() != null && routeInfo.getConsumes().length > 0) {
routeManager.append(Arrays.stream(routeInfo.getConsumes()).map(consumes -> ".consumes(\"" + consumes + "\")").collect(Collectors.joining()));
}
if(routeInfo.getProduces() != null && routeInfo.getProduces().length > 0) {
routeManager.append(Arrays.stream(routeInfo.getProduces()).map(produces -> ".produces(\"" + produces + "\")").collect(Collectors.joining()));
}
if(routeInfo.getLanguages() != null && routeInfo.getLanguages().length > 0) {
routeManager.append(Arrays.stream(routeInfo.getLanguages()).map(language -> ".language(\"" + language + "\")").collect(Collectors.joining()));
}
routeManager.append(".handler(").append(routeHandler).append(")");
if(typesMode) {
StringBuilder routeTypes = new StringBuilder(context.getTypeTypeName()).append("[] routeTypes = new ").append(context.getTypeTypeName()).append("[] {").append(System.lineSeparator());
routeTypes.append(Stream.concat(
Arrays.stream(routeInfo.getParameters())
.map(parameterInfo -> {
if(parameterInfo instanceof WebRequestBodyParameterInfo) {
return parameterInfo.getType();
}
else {
return context.getParameterConverterType(parameterInfo.getType());
}
}),
routeInfo.getResponseBody().getBodyKind() != ResponseBodyKind.EMPTY ? Stream.of(routeInfo.getResponseBody().getType()) : Stream.of()
)
.map(converterType -> new StringBuilder(context.indent(2)).append(context.getTypeGenerator(converterType)))
.collect(context.joining("," + System.lineSeparator()))
);
routeTypes.append(System.lineSeparator()).append(context.indent(1)).append("}");
StringBuilder result = new StringBuilder(context.indent(0)).append(".route(route -> {").append(System.lineSeparator());
result.append(context.indent(1)).append(routeTypes).append(";").append(System.lineSeparator());
result.append(context.indent(1)).append("route").append(routeManager).append(";").append(System.lineSeparator());
result.append(context.indent(0)).append("})");
return result;
}
else {
return new StringBuilder(context.indent(0)).append(".route()").append(routeManager);
}
}
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebResponseBodyInfo responseBodyInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_HANDLER_CLASS || context.getMode() == GenerationMode.ROUTE_HANDLER_TYPE) {
boolean typesMode = context.getMode() == GenerationMode.ROUTE_HANDLER_TYPE;
StringBuilder result = new StringBuilder();
WebRouteInfo routeInfo = context.getWebRoute();
StringBuilder requestParameters = new StringBuilder();
GenerationMode parameterReferenceMode = typesMode ? GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE : GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS;
Integer nonReactiveRequestBodyParameterIndex = null;
boolean hasFormParameters = false;
WebRequestBodyParameterInfo requestBodyInfo = null;
int parameterIndex = 0;
for(Iterator parameterInfoIterator = Arrays.stream(routeInfo.getParameters()).iterator();parameterInfoIterator.hasNext();) {
WebParameterInfo parameterInfo = parameterInfoIterator.next();
if(parameterInfo instanceof WebRequestBodyParameterInfo && ((WebRequestBodyParameterInfo)parameterInfo).getBodyReactiveKind() == RequestBodyReactiveKind.NONE) {
nonReactiveRequestBodyParameterIndex = parameterIndex;
requestParameters.append("body");
}
else {
requestParameters.append(this.visit(parameterInfo, context.withIndentDepth(0).withMode(parameterReferenceMode).withParameterIndex(parameterIndex)));
}
if(parameterInfo instanceof WebFormParameterInfo) {
hasFormParameters = true;
}
if(parameterInfo instanceof WebRequestBodyParameterInfo) {
requestBodyInfo = (WebRequestBodyParameterInfo)parameterInfo;
}
if(parameterInfoIterator.hasNext()) {
requestParameters.append(", ");
}
parameterIndex++;
}
StringBuilder controllerInvoke = new StringBuilder("this.").append(context.getFieldName(context.getWebController().getQualifiedName())).append(".").append(routeInfo.getElement().get().getSimpleName().toString()).append("(").append(requestParameters).append(")");
if(responseBodyInfo.getBodyKind() == ResponseBodyKind.EMPTY && responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.NONE) {
if(hasFormParameters) {
result.append(context.indent(0)).append("exchange.response().body().raw().stream(");
result.append(context.getFluxTypeName()).append(".from(exchange.request().body().get().urlEncoded().stream()).collectMultimap(").append(context.getParameterTypeName()).append("::getName)").append(".flatMap(formParameters -> ");
result.append("{ ").append(controllerInvoke).append("; return ").append(context.getMonoTypeName()).append(".empty(); }");
result.append("));").append(System.lineSeparator());
}
else if(nonReactiveRequestBodyParameterIndex != null) {
result.append(context.indent(0)).append("exchange.response().body().raw().stream(");
result.append(this.visit(routeInfo.getParameters()[nonReactiveRequestBodyParameterIndex], context.withIndentDepth(0).withMode(parameterReferenceMode).withParameterIndex(nonReactiveRequestBodyParameterIndex))).append(".flatMap(body -> ");
result.append("{ ").append(controllerInvoke).append("; return ").append(context.getMonoTypeName()).append(".empty(); }");
result.append("));").append(System.lineSeparator());
}
else {
result.append(context.indent(0)).append(controllerInvoke).append(";").append(System.lineSeparator());
result.append(context.indent(0)).append("exchange.response().body().empty();").append(System.lineSeparator());
}
}
else {
if(nonReactiveRequestBodyParameterIndex != null) {
if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.NONE) {
controllerInvoke.insert(0, this.visit(routeInfo.getParameters()[nonReactiveRequestBodyParameterIndex], context.withIndentDepth(0).withMode(parameterReferenceMode).withParameterIndex(nonReactiveRequestBodyParameterIndex)).append(".map(body -> ")).append(")");
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.ONE) {
controllerInvoke.insert(0, this.visit(routeInfo.getParameters()[nonReactiveRequestBodyParameterIndex], context.withIndentDepth(0).withMode(parameterReferenceMode).withParameterIndex(nonReactiveRequestBodyParameterIndex)).append(".flatMap(body -> ")).append(")");
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.PUBLISHER || responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.MANY) {
controllerInvoke.insert(0, this.visit(routeInfo.getParameters()[nonReactiveRequestBodyParameterIndex], context.withIndentDepth(0).withMode(parameterReferenceMode).withParameterIndex(nonReactiveRequestBodyParameterIndex)).append(".flatMapMany(body -> ")).append(")");
}
else {
throw new IllegalStateException("Unknown response body reactive kind: " + responseBodyInfo.getBodyReactiveKind());
}
}
else if(hasFormParameters) {
String mapFunction;
if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.NONE) {
mapFunction = "map";
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.ONE) {
if(responseBodyInfo.getBodyKind() == ResponseBodyKind.EMPTY) {
mapFunction = "flatMap";
}
else {
mapFunction = "map";
}
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.PUBLISHER || responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.MANY) {
mapFunction = "flatMapMany";
}
else {
throw new IllegalStateException("Unknown response body reactive kind: " + responseBodyInfo.getBodyReactiveKind());
}
controllerInvoke.insert(0, new StringBuilder(context.getFluxTypeName()).append(".from(exchange.request().body().get().urlEncoded().stream()).collectMultimap(").append(context.getParameterTypeName()).append("::getName)").append(".").append(mapFunction).append("(formParameters -> ")).append(")");
}
if(responseBodyInfo.getBodyKind() == ResponseBodyKind.EMPTY) {
// We know we are reactive here
if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.PUBLISHER) {
controllerInvoke.insert(0, new StringBuilder(context.getFluxTypeName()).append(".from(")).append(")");
}
result.append(context.indent(0)).append("exchange.response().body().raw().stream(");
result.append(controllerInvoke).append(".then().cast(").append(context.getByteBufTypeName()).append(".class)");
result.append(");").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.RAW) {
String responseBodyDataMethod;
if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.NONE) {
if(hasFormParameters || requestBodyInfo != null) {
responseBodyDataMethod = "stream";
}
else {
responseBodyDataMethod = "value";
}
}
else {
responseBodyDataMethod = "stream";
}
result.append(context.indent(0)).append("exchange.response().body().raw().").append(responseBodyDataMethod).append("(");
result.append(controllerInvoke);
result.append(");").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.CHARSEQUENCE) {
String responseBodyDataMethod;
if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.NONE) {
if(hasFormParameters || requestBodyInfo != null) {
responseBodyDataMethod = "stream";
}
else {
responseBodyDataMethod = "value";
}
}
else {
responseBodyDataMethod = "stream";
}
result.append(context.indent(0)).append("exchange.response().body().string().").append(responseBodyDataMethod).append("(");
result.append(controllerInvoke);
result.append(");").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.ENCODED) {
String responseBodyDataMethod;
if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.NONE) {
// if a request body is injected, we have to use one
if(hasFormParameters || requestBodyInfo != null) {
responseBodyDataMethod = "one";
}
else {
responseBodyDataMethod = "value";
}
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.RAW) {
responseBodyDataMethod = "stream";
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.ONE) {
responseBodyDataMethod = "one";
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.MANY) {
responseBodyDataMethod = "many";
}
else if(responseBodyInfo.getBodyReactiveKind() == ResponseBodyReactiveKind.PUBLISHER) {
responseBodyDataMethod = "stream";
}
else {
throw new IllegalStateException("Unknown response body reactive kind: " + responseBodyInfo.getBodyReactiveKind());
}
result.append(context.indent(0)).append("exchange.response().body()");
if(typesMode) {
result.append(".<").append(context.getTypeName(responseBodyInfo.getType())).append(">encoder(routeTypes[").append(routeInfo.getParameters().length).append("]).");
}
else {
result.append(".encoder(").append(context.getTypeName(responseBodyInfo.getType())).append(".class).");
}
result.append(responseBodyDataMethod).append("(");
result.append(controllerInvoke);
result.append(");").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.RESOURCE) {
result.append(context.indent(0)).append("exchange.response().body().resource().value(");
result.append(controllerInvoke);
result.append(");").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.SSE_RAW) {
result.append(context.indent(0)).append("exchange.response().body().sse().from((events, data) -> data.stream(");
result.append(controllerInvoke);
result.append("));").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.SSE_CHARSEQUENCE) {
result.append(context.indent(0)).append("exchange.response().body().<").append(context.getTypeName(responseBodyInfo.getType())).append(">sseString().from((events, data) -> data.stream(");
result.append(controllerInvoke);
result.append("));").append(System.lineSeparator());
}
else if(responseBodyInfo.getBodyKind() == ResponseBodyKind.SSE_ENCODED) {
WebSseEventFactoryParameterInfo sseEventFactoryParameter = (WebSseEventFactoryParameterInfo)Arrays.stream(context.getWebRoute().getParameters()).filter(parameter -> parameter instanceof WebSseEventFactoryParameterInfo).findFirst().get();
result.append(context.indent(0)).append("exchange.response().body()");
if(typesMode) {
result.append(".<").append(context.getTypeName(responseBodyInfo.getType())).append(">sseEncoder(\"").append(sseEventFactoryParameter.getDataMediaType().orElse("text/plain")).append("\", ").append("routeTypes[").append(routeInfo.getParameters().length).append("])");
}
else {
result.append(".sseEncoder(\"").append(sseEventFactoryParameter.getDataMediaType().orElse("text/plain")).append("\", ").append(context.getTypeName(responseBodyInfo.getType())).append(".class)");
}
result.append(".from((events, data) -> data.stream(");
result.append(controllerInvoke);
result.append("));").append(System.lineSeparator());
}
}
return result;
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebParameterInfo parameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(parameterInfo instanceof WebCookieParameterInfo) {
return this.visit((WebCookieParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebExchangeParameterInfo) {
return this.visit((WebExchangeParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebExchangeContextParameterInfo) {
return this.visit((WebExchangeContextParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebFormParameterInfo) {
return this.visit((WebFormParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebHeaderParameterInfo) {
return this.visit((WebHeaderParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebPathParameterInfo) {
return this.visit((WebPathParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebQueryParameterInfo) {
return this.visit((WebQueryParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebRequestBodyParameterInfo) {
return this.visit((WebRequestBodyParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebSseEventFactoryParameterInfo) {
return this.visit((WebSseEventFactoryParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebSocketParameterInfo) {
return this.visit((WebSocketParameterInfo)parameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebBasicParameterInfo basicParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
boolean typesMode = context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE;
String parameterType = typesMode ? "routeTypes[" + context.getParameterIndex() + "]" : null;
StringBuilder result = new StringBuilder();
TypeMirror basicParameterType = basicParameterInfo.getType();
if(basicParameterType.getKind() == TypeKind.ARRAY) {
if(!typesMode) {
parameterType = context.getTypeName(((ArrayType)basicParameterType).getComponentType()) + ".class";
}
result.append(context.getOptionalTypeName()).append(".ofNullable(");
result.append(this.visit(basicParameterInfo, context.withMode(GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY))).append(")");
result.append(".<").append(context.getTypeName(basicParameterType)).append(">map(parameters -> parameters.stream().flatMap(parameter -> parameter.asListOf(").append(parameterType).append(").stream())");
result.append(".toArray(").append(context.getTypeName(((ArrayType)basicParameterType).getComponentType())).append("[]::new)).filter(l -> l.length > 0)");
}
else if(context.getTypeUtils().isSameType(context.getCollectionType(), context.getTypeUtils().erasure(basicParameterType))) {
if(!typesMode) {
parameterType = context.getTypeName(((DeclaredType)basicParameterType).getTypeArguments().get(0)) + ".class";
}
result.append(context.getOptionalTypeName()).append(".ofNullable(");
result.append(this.visit(basicParameterInfo, context.withMode(GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY))).append(")");
result.append(".<").append(context.getTypeName(basicParameterType)).append(">map(parameters -> parameters.stream().flatMap(parameter -> parameter.asListOf(").append(parameterType).append(").stream())");
result.append(".collect(").append(context.getCollectorsTypeName()).append(".toList())).filter(l -> !l.isEmpty())");
}
else if(context.getTypeUtils().isSameType(context.getListType(), context.getTypeUtils().erasure(basicParameterType))) {
if(!typesMode) {
parameterType = context.getTypeName(((DeclaredType)basicParameterType).getTypeArguments().get(0)) + ".class";
}
result.append(context.getOptionalTypeName()).append(".ofNullable(");
result.append(this.visit(basicParameterInfo, context.withMode(GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY))).append(")");
result.append(".<").append(context.getTypeName(basicParameterType)).append(">map(parameters -> parameters.stream().flatMap(parameter -> parameter.asListOf(").append(parameterType).append(").stream())");
result.append(".collect(").append(context.getCollectorsTypeName()).append(".toList())).filter(l -> !l.isEmpty())");
}
else if(context.getTypeUtils().isSameType(context.getSetType(), context.getTypeUtils().erasure(basicParameterType))) {
if(!typesMode) {
parameterType = context.getTypeName(((DeclaredType)basicParameterType).getTypeArguments().get(0)) + ".class";
}
result.append(context.getOptionalTypeName()).append(".ofNullable(");
result.append(this.visit(basicParameterInfo, context.withMode(GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY))).append(")");
result.append(".<").append(context.getTypeName(basicParameterType)).append(">map(parameters -> parameters.stream().flatMap(parameter -> parameter.asListOf(").append(parameterType).append(").stream())");
result.append(".collect(").append(context.getCollectorsTypeName()).append(".toSet())).filter(l -> !l.isEmpty())");
}
else {
result.append(this.visit(basicParameterInfo, context.withMode(GenerationMode.ROUTE_PARAMETER_REFERENCE_ONE)));
result.append(".map(parameter -> parameter.");
if(basicParameterType instanceof PrimitiveType) {
// boolean, byte, short, int, long, char, float, and double.
switch(basicParameterType.getKind()) {
case BOOLEAN: result.append("asBoolean())");
break;
case BYTE: result.append("asByte())");
break;
case SHORT: result.append("asShort())");
break;
case INT: result.append("asInteger())");
break;
case LONG: result.append("asLong())");
break;
case CHAR: result.append("asCharacter())");
break;
case FLOAT: result.append("asFloat())");
break;
case DOUBLE: result.append("asDouble())");
break;
default:
throw new IllegalStateException("Unsupported primitive type: " + basicParameterType);
}
}
else {
if(!typesMode) {
parameterType = context.getTypeName(basicParameterType) + ".class";
}
else {
result.append("<").append(context.getTypeName(basicParameterType)).append(">");
}
result.append("as(").append(parameterType).append("))");
}
}
if(basicParameterInfo.isRequired()) {
result.append(".orElseThrow(() -> new ").append(context.getMissingRequiredParameterExceptionTypeName()).append("(\"").append(basicParameterInfo.getQualifiedName().getParameterName()).append("\"))");
}
return result;
}
else {
if(basicParameterInfo instanceof WebCookieParameterInfo) {
return this.visit((WebCookieParameterInfo)basicParameterInfo, context);
}
else if(basicParameterInfo instanceof WebFormParameterInfo) {
return this.visit((WebFormParameterInfo)basicParameterInfo, context);
}
else if(basicParameterInfo instanceof WebHeaderParameterInfo) {
return this.visit((WebHeaderParameterInfo)basicParameterInfo, context);
}
else if(basicParameterInfo instanceof WebPathParameterInfo) {
return this.visit((WebPathParameterInfo)basicParameterInfo, context);
}
else if(basicParameterInfo instanceof WebQueryParameterInfo) {
return this.visit((WebQueryParameterInfo)basicParameterInfo, context);
}
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebCookieParameterInfo cookieParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_ONE) {
return new StringBuilder("exchange.request().cookies().get(\"").append(cookieParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY) {
return new StringBuilder("exchange.request().cookies().getAll(\"").append(cookieParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return this.visit((WebBasicParameterInfo)cookieParameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebFormParameterInfo formParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_ONE) {
return new StringBuilder(context.getOptionalTypeName()).append(".ofNullable(formParameters.get(\"").append(formParameterInfo.getQualifiedName().getParameterName()).append("\")).flatMap(parameter -> parameter.stream().findFirst())");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY) {
return new StringBuilder("formParameters.get(\"").append(formParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return this.visit((WebBasicParameterInfo)formParameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebHeaderParameterInfo headerParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_ONE) {
return new StringBuilder("exchange.request().headers().getParameter(\"").append(headerParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY) {
return new StringBuilder("exchange.request().headers().getAllParameter(\"").append(headerParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return this.visit((WebBasicParameterInfo)headerParameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebPathParameterInfo pathParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_ONE) {
return new StringBuilder("exchange.request().pathParameters().get(\"").append(pathParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY) {
return new StringBuilder("exchange.request().pathParameters().get(\"").append(pathParameterInfo.getQualifiedName().getParameterName()).append("\").map(parameter -> ").append(context.getListTypeName()).append(".of(parameter)).orElse(").append(context.getListTypeName()).append(".of())");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return this.visit((WebBasicParameterInfo)pathParameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebQueryParameterInfo queryParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_ONE) {
return new StringBuilder("exchange.request().queryParameters().get(\"").append(queryParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_MANY) {
return new StringBuilder("exchange.request().queryParameters().getAll(\"").append(queryParameterInfo.getQualifiedName().getParameterName()).append("\")");
}
else if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return this.visit((WebBasicParameterInfo)queryParameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebRequestBodyParameterInfo bodyParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
boolean typesMode = context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE;
String parameterType = typesMode ? "routeTypes[" + context.getParameterIndex() + "]" : null;
StringBuilder result = new StringBuilder();
TypeMirror bodyParameterType = bodyParameterInfo.getType();
if(bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.NONE) {
if(bodyParameterInfo.getBodyKind() == RequestBodyKind.RAW) {
result.append(context.getFluxTypeName()).append(".from(exchange.request().body().get().raw().stream()).reduceWith(() -> ").append(context.getUnpooledTypeName()).append(".unreleasableBuffer(").append(context.getUnpooledTypeName()).append(".buffer()), (acc, chunk) -> { try { return acc.writeBytes(chunk); } finally { chunk.release(); } })");
}
else if(bodyParameterInfo.getBodyKind() == RequestBodyKind.ENCODED) {
result.append("exchange.request().body().get().");
if(!typesMode) {
parameterType = context.getTypeName(bodyParameterType) + ".class";
}
else {
result.append("<").append(context.getTypeName(bodyParameterType)).append(">");
}
result.append("decoder(").append(parameterType).append(").one()");
}
else {
throw new IllegalStateException("Can't generate non-reactive body parameter of kind: " + bodyParameterInfo.getBodyKind());
}
}
else {
// Reactive
result.append("exchange.request().body().get().");
if(bodyParameterInfo.getBodyKind() == RequestBodyKind.ENCODED) {
if(!typesMode) {
parameterType = context.getTypeName(bodyParameterInfo.getType()) + ".class";
}
else {
result.append("<").append(context.getTypeName(bodyParameterType)).append(">");
}
result.append("decoder(").append(parameterType).append(")");
if(bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.ONE) {
result.append(".one()");
}
else if(bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.PUBLISHER || bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.MANY) {
result.append(".many()");
}
else {
throw new IllegalStateException("Unknown request body reactive kind: " + bodyParameterInfo.getBodyReactiveKind());
}
}
else {
if(bodyParameterInfo.getBodyKind() == RequestBodyKind.RAW) {
result.append("raw().stream()");
}
else if(bodyParameterInfo.getBodyKind() == RequestBodyKind.MULTIPART) {
result.append("multipart().stream()");
}
else if(bodyParameterInfo.getBodyKind() == RequestBodyKind.URLENCODED) {
result.append("urlEncoded().stream()");
}
else {
throw new IllegalStateException("Unknown request body kind: " + bodyParameterInfo.getBodyKind());
}
if(bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.ONE) {
result.insert(0, ".from(").insert(0, context.getMonoTypeName()).append(")");
}
else if(bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.MANY || bodyParameterInfo.getBodyReactiveKind() == RequestBodyReactiveKind.PUBLISHER) {
result.insert(0, ".from(").insert(0, context.getFluxTypeName()).append(")");
}
else {
throw new IllegalStateException("Unknown request body reactive kind: " + bodyParameterInfo.getBodyReactiveKind());
}
}
}
return result;
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebExchangeParameterInfo exchangeParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return new StringBuilder("exchange");
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebExchangeContextParameterInfo exchangeContextParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return new StringBuilder("exchange.context()");
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSseEventFactoryParameterInfo sseEventFactoryParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return new StringBuilder("events");
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketRouteInfo webSocketRouteInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.WEBSOCKET_ROUTE_ANNOTATION) {
StringBuilder result = new StringBuilder();
result.append(context.indent(0)).append("@").append(context.getWebSocketRouteAnnotationTypeName()).append("(");
result.append("path = { ");
if(webSocketRouteInfo.getPaths().length > 0) {
result.append(Arrays.stream(webSocketRouteInfo.getPaths())
.map(path -> "\"" + StringEscapeUtils.escapeJava(webSocketRouteInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).path(path, false).buildRawPath())
.orElse(path)) + "\""
)
.collect(Collectors.joining(", "))
);
}
else {
webSocketRouteInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).buildRawPath())
.ifPresent(rootPath -> result.append("\"").append(rootPath).append("\""));
}
result.append(" }");
if(webSocketRouteInfo.isMatchTrailingSlash()) {
result.append(", matchTrailingSlash = true");
}
if(webSocketRouteInfo.getLanguages() != null && webSocketRouteInfo.getLanguages().length > 0) {
result.append(", language = { ").append(Arrays.stream(webSocketRouteInfo.getLanguages()).map(language -> "\"" + language + "\"").collect(Collectors.joining(", "))).append(" }");
}
if(webSocketRouteInfo.getSubprotocols()!= null && webSocketRouteInfo.getSubprotocols().length > 0) {
result.append(", subprotocol = { ").append(Arrays.stream(webSocketRouteInfo.getSubprotocols()).map(subprotocol -> "\"" + subprotocol + "\"").collect(Collectors.joining(", "))).append(" }");
}
if(webSocketRouteInfo.getMessageType() != null) {
result.append(", messageType = ").append(context.getWebSocketMessageKindTypeName()).append(".").append(webSocketRouteInfo.getMessageType().toString());
}
result.append(")");
return result;
}
else if(context.getMode() == GenerationMode.ROUTE_DECLARATION) {
boolean typesMode = context.isTypeMode(webSocketRouteInfo);
StringBuilder routeHandler = new StringBuilder("webSocketExchange -> {").append(System.lineSeparator());
webSocketRouteInfo.getOutboundPublisher().ifPresentOrElse(
outboundPublisherInfo -> routeHandler.append(context.indent(typesMode ? 2 : 1)).append(this.visit(outboundPublisherInfo, context.withIndentDepth(context.getIndentDepth() + (typesMode ? 2 : 1) ).withWebRoute(webSocketRouteInfo).withMode(typesMode ? GenerationMode.WEBSOCKET_ROUTE_HANDLER_TYPE : GenerationMode.WEBSOCKET_ROUTE_HANDLER_CLASS))).append(";").append(System.lineSeparator()),
() -> routeHandler.append(context.indent(typesMode ? 2 : 1)).append(this.visit(webSocketRouteInfo, context.withIndentDepth(context.getIndentDepth() + (typesMode ? 2 : 1) ).withWebRoute(webSocketRouteInfo).withMode(typesMode ? GenerationMode.WEBSOCKET_ROUTE_HANDLER_TYPE : GenerationMode.WEBSOCKET_ROUTE_HANDLER_CLASS))).append(";").append(System.lineSeparator())
);
routeHandler.append(context.indent(typesMode ? 1 : 0)).append("}");
StringBuilder routeManager = new StringBuilder();
if(webSocketRouteInfo.getPaths().length > 0) {
routeManager.append(Arrays.stream(webSocketRouteInfo.getPaths())
.map(path -> ".path(\"" + StringEscapeUtils.escapeJava(webSocketRouteInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).path(path, false).buildRawPath())
.orElse(path)) + "\", " + webSocketRouteInfo.isMatchTrailingSlash() + ")"
)
.collect(Collectors.joining())
);
}
else {
webSocketRouteInfo.getController()
.map(WebControllerInfo::getRootPath)
.map(rootPath -> URIs.uri(rootPath, URIs.RequestTargetForm.ABSOLUTE, URIs.Option.PARAMETERIZED, URIs.Option.NORMALIZED, URIs.Option.PATH_PATTERN).buildRawPath())
.ifPresent(rootPath -> routeManager.append(".path(\"").append(StringEscapeUtils.escapeJava(rootPath)).append("\", ").append(webSocketRouteInfo.isMatchTrailingSlash()).append(")"));
}
if(webSocketRouteInfo.getLanguages() != null && webSocketRouteInfo.getLanguages().length > 0) {
routeManager.append(Arrays.stream(webSocketRouteInfo.getLanguages()).map(language -> ".language(\"" + language + "\")").collect(Collectors.joining()));
}
if(webSocketRouteInfo.getSubprotocols() != null && webSocketRouteInfo.getSubprotocols().length > 0) {
routeManager.append(Arrays.stream(webSocketRouteInfo.getSubprotocols()).map(subprotocol -> ".subprotocol(\"" + subprotocol + "\")").collect(Collectors.joining()));
}
routeManager.append(".handler(").append(routeHandler).append(")");
if(typesMode) {
StringBuilder routeTypes = new StringBuilder(context.getTypeTypeName()).append("[] routeTypes = new ").append(context.getTypeTypeName()).append("[] {").append(System.lineSeparator());
routeTypes.append(Stream.concat(
Arrays.stream(webSocketRouteInfo.getParameters())
.filter(parameterInfo -> parameterInfo instanceof GenericWebSocketInboundPublisherParameterInfo)
.map(parameterInfo -> parameterInfo.getType()),
webSocketRouteInfo.getOutboundPublisher().map(outboundPublisherInfo -> Stream.of(outboundPublisherInfo.getType())).orElse(Stream.of())
)
.map(converterType -> new StringBuilder(context.indent(2)).append(context.getTypeGenerator(converterType)))
.collect(context.joining("," + System.lineSeparator()))
);
routeTypes.append(System.lineSeparator()).append(context.indent(1)).append("}");
StringBuilder result = new StringBuilder(context.indent(0)).append(".webSocketRoute(webSocketRoute -> {").append(System.lineSeparator());
result.append(context.indent(1)).append(routeTypes).append(";").append(System.lineSeparator());
result.append(context.indent(1)).append("webSocketRoute").append(routeManager).append(";").append(System.lineSeparator());
result.append(context.indent(0)).append("})");
return result;
}
else {
return new StringBuilder(context.indent(0)).append(".webSocketRoute()").append(routeManager);
}
}
else if(context.getMode() == GenerationMode.WEBSOCKET_ROUTE_HANDLER_CLASS || context.getMode() == GenerationMode.WEBSOCKET_ROUTE_HANDLER_TYPE) {
boolean typesMode = context.getMode() == GenerationMode.WEBSOCKET_ROUTE_HANDLER_TYPE;
StringBuilder requestParameters = new StringBuilder();
GenerationMode parameterReferenceMode = typesMode ? GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE : GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS;
int parameterIndex = 0;
for(Iterator parameterInfoIterator = Arrays.stream(webSocketRouteInfo.getParameters()).iterator();parameterInfoIterator.hasNext();) {
// WebExchangeContextParameterInfo, WebSocketExchangeParameterInfo, WebSocketInboundParameterInfo, WebSocketInboundPubliserInfo, WebSocketOutboundParameterInfo
requestParameters.append(this.visit(parameterInfoIterator.next(), context.withIndentDepth(0).withMode(parameterReferenceMode).withWebRoute(webSocketRouteInfo).withParameterIndex(parameterIndex)));
if(parameterInfoIterator.hasNext()) {
requestParameters.append(", ");
}
parameterIndex++;
}
return new StringBuilder("this.").append(context.getFieldName(context.getWebController().getQualifiedName())).append(".").append(webSocketRouteInfo.getElement().get().getSimpleName().toString()).append("(").append(requestParameters).append(")");
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketBoundPublisherInfo boundPublisherInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(boundPublisherInfo instanceof WebSocketOutboundPublisherInfo) {
return this.visit((WebSocketOutboundPublisherInfo)boundPublisherInfo, context);
}
else if(boundPublisherInfo instanceof WebSocketInboundPublisherParameterInfo) {
return this.visit((WebSocketInboundPublisherParameterInfo)boundPublisherInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketOutboundPublisherInfo outboundPublisherInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.WEBSOCKET_ROUTE_HANDLER_CLASS || context.getMode() == GenerationMode.WEBSOCKET_ROUTE_HANDLER_TYPE) {
boolean typesMode = context.getMode() == GenerationMode.WEBSOCKET_ROUTE_HANDLER_TYPE;
WebSocketMessage.Kind WebSocketMessageKind = ((WebSocketRouteInfo)context.getWebRoute()).getMessageType();
StringBuilder result = new StringBuilder();
result.append("webSocketExchange.outbound()");
String webSocketInbound = "webSocketExchange.outbound()";
WebSocketRouteInfo webSocketRouteInfo = (WebSocketRouteInfo)context.getWebRoute();
StringBuilder handlerInvoke = this.visit(webSocketRouteInfo, context);
boolean isPublisher = outboundPublisherInfo.getBoundReactiveKind() == WebSocketBoundPublisherInfo.BoundReactiveKind.PUBLISHER;
boolean isCharsequence = false;
switch(outboundPublisherInfo.getBoundKind()) {
case CHARSEQUENCE_REDUCED:
case CHARSEQUENCE_REDUCED_ONE:
case CHARSEQUENCE_PUBLISHER:
case CHARSEQUENCE_MANY: isCharsequence = true;
case RAW_REDUCED:
case RAW_REDUCED_ONE:
case RAW_PUBLISHER:
case RAW_MANY: {
// (Publisher|Flux|Mono)
// webSocketExchange.outbound().messages(factory -> Flux.from(...).map(factory::text_raw))
// webSocketExchange.outbound().messages(factory -> ....map(factory::text_raw))
result.append(".messages(factory -> ");
if(isPublisher) {
result.append(context.getFluxTypeName()).append(".from(").append(handlerInvoke).append(")");
}
else {
result.append(handlerInvoke);
}
result.append(".map(");
switch(WebSocketMessageKind) {
case TEXT: {
if(isCharsequence) {
result.append("factory::text");
}
else {
result.append("factory::text_raw");
}
}
break;
case BINARY: result.append("factory::binary");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
result.append(")");
result.append(")");
return result;
}
case EMPTY: {
// (Publisher|Flux|Mono)
// webSocketExchange.outbound().messages(factory -> Flux.from(...).then(Mono.empty()));
// webSocketExchange.outbound().messages(factory -> ....then(Mono.empty()));
result.append(".messages(factory -> ");
if(isPublisher) {
result.append(context.getFluxTypeName()).append(".from(").append(handlerInvoke).append(")");
}
else {
result.append(handlerInvoke);
}
result.append(".then(").append(context.getMonoTypeName()).append(".empty())");
result.append(")");
return result;
}
case ENCODED: {
// (Publisher|Flux|Mono)
// => GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS
// webSocketExchange.outbound().encodeTextMessages(this.wsx13(), Message.class);
// => GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE
// webSocketExchange.outbound().encodeTextMessages(this.wsx13(), routeTypes[0]);
switch(WebSocketMessageKind) {
case TEXT: result.append(".encodeTextMessages(");
break;
case BINARY: result.append(".encodeBinaryMessages(");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
result.append(handlerInvoke).append(", ");
if(typesMode) {
result.append("routeTypes[").append(webSocketRouteInfo.getParameters().length).append("]");
}
else {
result.append(context.getTypeName(outboundPublisherInfo.getType())).append(".class");
}
result.append(")");
return result;
}
default :
throw new IllegalStateException("Unknown WebSocket bound reactive kind");
}
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketParameterInfo parameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(parameterInfo instanceof WebSocketOutboundParameterInfo) {
return this.visit((WebSocketOutboundParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebSocketInboundPublisherParameterInfo) {
return this.visit((WebSocketInboundPublisherParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebSocketInboundParameterInfo) {
return this.visit((WebSocketInboundParameterInfo)parameterInfo, context);
}
else if(parameterInfo instanceof WebSocketExchangeParameterInfo) {
return this.visit((WebSocketExchangeParameterInfo)parameterInfo, context);
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketOutboundParameterInfo outboundParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS) {
return new StringBuilder("webSocketExchange.outbound()");
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketInboundPublisherParameterInfo inboundPublisherParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS) {
boolean typesMode = context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE;
WebSocketMessage.Kind WebSocketMessageKind = ((WebSocketRouteInfo)context.getWebRoute()).getMessageType();
StringBuilder result = new StringBuilder();
String webSocketInbound = "webSocketExchange.inbound()";
TypeMirror boundReactiveType;
switch(inboundPublisherParameterInfo.getBoundReactiveKind()) {
case PUBLISHER: boundReactiveType = context.getFluxType();
break;
case MANY: boundReactiveType = context.getFluxType();
break;
case ONE: boundReactiveType = context.getMonoType();
break;
default:
throw new IllegalStateException("Unknown WebSocket bound reactive kind");
}
boolean isCharsequence = false;
switch(inboundPublisherParameterInfo.getBoundKind()) {
case CHARSEQUENCE_REDUCED: isCharsequence = true;
case RAW_REDUCED: {
// (Publisher|Flux|Mono)
// Flux.from(webSocketExchange.inbound().textMessages()).flatMap(WebSocketMessage::reducedBinary)
// Flux.from(webSocketExchange.inbound().binaryMessages()).flatMap(WebSocketMessage::reducedBinary)
result.append(context.getTypeName(boundReactiveType)).append(".from(").append(webSocketInbound);
switch(WebSocketMessageKind) {
case TEXT: result.append(".textMessages()");
break;
case BINARY: result.append(".binaryMessages()");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
result.append(").flatMap(").append(context.getWebSocketMessageTypeName()).append("::").append(isCharsequence ? "reducedText" : "reducedBinary").append(")");
return result;
}
case CHARSEQUENCE_REDUCED_ONE: isCharsequence = true;
case RAW_REDUCED_ONE: {
// (Publisher|Flux|Mono)>
// (Flux|Mono).from(webSocketExchange.inbound().textMessages()).map(WebSocketMessage::reducedBinary)
// (Flux|Mono).from(webSocketExchange.inbound().binaryMessages()).map(WebSocketMessage::reducedBinary)
result.append(context.getTypeName(boundReactiveType)).append(".from(").append(webSocketInbound);
switch(WebSocketMessageKind) {
case TEXT: result.append(".textMessages()");
break;
case BINARY: result.append(".binaryMessages()");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
result.append(").map(").append(context.getWebSocketMessageTypeName()).append("::").append(isCharsequence ? "reducedText" : "reduceBinary").append(")");
return result;
}
case CHARSEQUENCE_PUBLISHER: isCharsequence = true;
case RAW_PUBLISHER: {
// (Publisher|Flux|Mono)>
// (Flux|Mono).from(webSocketExchange.inbound().textMessages()).map(WebSocketMessage::binary)
// (Flux|Mono).from(webSocketExchange.inbound().binaryMessages()).map(WebSocketMessage::binary)
result.append(context.getTypeName(boundReactiveType)).append(".from(").append(webSocketInbound);
switch(WebSocketMessageKind) {
case TEXT: result.append(".textMessages()");
break;
case BINARY: result.append(".binaryMessages()");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
result.append(").map(").append(context.getWebSocketMessageTypeName()).append("::").append(isCharsequence ? "binary" : "text").append(")");
return result;
}
case CHARSEQUENCE_MANY: isCharsequence = true;
case RAW_MANY: {
// (Publisher|Flux|Mono)>
// (Flux|Mono).from(webSocketExchange.inbound().textMessages()).map(message -> Flux.from(message.binary()))
// (Flux|Mono).from(webSocketExchange.inbound().binaryMessages()).map(message -> Flux.from(message.binary()))
result.append(context.getTypeName(boundReactiveType)).append(".from(").append(webSocketInbound);
switch(WebSocketMessageKind) {
case TEXT: result.append(".textMessages()");
break;
case BINARY: result.append(".binaryMessages()");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
result.append(").map(message -> ").append(context.getFluxTypeName()).append(".from(message.").append(isCharsequence ? "binary" : "text").append("()))");
return result;
}
case EMPTY: {
// (Publisher|Flux|Mono)
// Flux.from(webSocketExchange.inbound().frames()).doOnNext(WebSocketFrame::release).then()
// Flux.from(webSocketExchange.inbound().frames()).doOnNext(WebSocketFrame::release).thenMany(Flux.empty())
result.append(context.getTypeName(boundReactiveType)).append(".from(").append(webSocketInbound).append(".frames())");
result.append(".doOnNext(").append(context.getWebSocketFrameTypeName()).append("::release)");
if(inboundPublisherParameterInfo.getBoundReactiveKind() == WebSocketBoundPublisherInfo.BoundReactiveKind.MANY) {
result.append(".thenMany(").append(context.getFluxTypeName()).append(".empty())");
}
else {
result.append(".then()");
}
return result;
}
case ENCODED: {
// (Publisher|Flux|Mono)
// => GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS
// webSocketExchange.inbound().decodeTextMessages(Message.class);
// Flux.from(webSocketExchange.inbound().decodeTextMessages(Message.class));
// Mono.from(webSocketExchange.inbound().decodeTextMessages(Message.class));
// => GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE
// webSocketExchange.inbound().decodeTextMessages(routeTypes[0]);
// Flux.from(webSocketExchange.inbound().decodeTextMessages(routeTypes[0]));
// Mono.from(webSocketExchange.inbound().decodeTextMessages(routeTypes[0]));
StringBuilder decodedMessagesPublisher = new StringBuilder();
decodedMessagesPublisher.append(webSocketInbound);
switch(WebSocketMessageKind) {
case TEXT: decodedMessagesPublisher.append(".decodeTextMessages(");
break;
case BINARY: decodedMessagesPublisher.append(".decodeBinaryMessages(");
break;
default :
throw new IllegalStateException("Unknown WebSocket message kind");
}
if(typesMode) {
decodedMessagesPublisher.append("routeTypes[").append(context.getParameterIndex()).append("]");
}
else {
decodedMessagesPublisher.append(context.getTypeName(inboundPublisherParameterInfo.getType())).append(".class");
}
decodedMessagesPublisher.append(")");
if(inboundPublisherParameterInfo.getBoundReactiveKind() == WebSocketBoundPublisherInfo.BoundReactiveKind.PUBLISHER) {
result.append(decodedMessagesPublisher);
}
else {
result.append(context.getTypeName(boundReactiveType)).append(".from(").append(decodedMessagesPublisher).append(")");
}
return result;
}
default :
throw new IllegalStateException("Unknown WebSocket bound reactive kind");
}
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketInboundParameterInfo inboundParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS) {
return new StringBuilder("webSocketExchange.inbound()");
}
return new StringBuilder();
}
@Override
public StringBuilder visit(WebSocketExchangeParameterInfo exchangeParameterInfo, WebServerControllerConfigurerClassGenerationContext context) {
if(context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_CLASS || context.getMode() == GenerationMode.ROUTE_PARAMETER_REFERENCE_TYPE) {
return new StringBuilder("webSocketExchange");
}
return new StringBuilder();
}
}