Fixed request body enum

Fixed micronaut-projects#1896
altro3 · Dec 18, 2024 · c2debf2 · c2debf2
1 parent 0dcac7a
commit c2debf2
Show file tree

Hide file tree

Showing 7 changed files with 196 additions and 6 deletions.
diff --git a/...-generator/src/main/java/io/micronaut/openapi/generator/AbstractMicronautJavaCodegen.java b/...-generator/src/main/java/io/micronaut/openapi/generator/AbstractMicronautJavaCodegen.java
@@ -26,6 +26,7 @@
 import io.swagger.v3.oas.models.media.MediaType;
 import io.swagger.v3.oas.models.media.Schema;
 import io.swagger.v3.oas.models.parameters.Parameter;
+import io.swagger.v3.oas.models.parameters.RequestBody;
 import io.swagger.v3.oas.models.responses.ApiResponse;
 import io.swagger.v3.oas.models.servers.Server;
 import io.swagger.v3.parser.util.SchemaTypeUtil;
@@ -732,6 +733,62 @@ public String toModelTestFilename(String name) {
         return toModelName(name) + "Test";
     }
 
+    @Override
+    public CodegenParameter fromRequestBody(RequestBody body, Set<String> imports, String bodyParameterName) {
+        var rqBody = super.fromRequestBody(body, imports, bodyParameterName);
+
+        var rqBodySchema = body.getContent() != null && !body.getContent().isEmpty() ? body.getContent().entrySet().iterator().next().getValue().getSchema() : null;
+        CodegenProperty codegenProperty = fromProperty(bodyParameterName, rqBodySchema, false);
+
+        if (rqBodySchema != null) {
+            rqBodySchema = unaliasSchema(rqBodySchema);
+
+            if (getUseInlineModelResolver()) {
+                codegenProperty = fromProperty(bodyParameterName, getReferencedSchemaWhenNotEnum(rqBodySchema), false);
+            } else {
+                codegenProperty = fromProperty(bodyParameterName, rqBodySchema, false);
+            }
+            rqBody.setSchema(codegenProperty);
+        }
+
+        if (Boolean.TRUE.equals(codegenProperty.isModel)) {
+            rqBody.isModel = true;
+        }
+
+        rqBody.dataFormat = codegenProperty.dataFormat;
+        if (body.getRequired() != null) {
+            rqBody.required = body.getRequired();
+        }
+
+        // set containerType
+        rqBody.containerType = codegenProperty.containerType;
+        rqBody.containerTypeMapped = codegenProperty.containerTypeMapped;
+
+        // enum
+        updateCodegenPropertyEnum(codegenProperty);
+        rqBody.isEnum = codegenProperty.isEnum;
+        rqBody.isEnumRef = codegenProperty.isEnumRef;
+        rqBody._enum = codegenProperty._enum;
+        rqBody.allowableValues = codegenProperty.allowableValues;
+
+        if (codegenProperty.isEnum || codegenProperty.isEnumRef) {
+            rqBody.datatypeWithEnum = codegenProperty.datatypeWithEnum;
+            rqBody.enumName = codegenProperty.enumName;
+            if (codegenProperty.defaultValue != null) {
+                rqBody.enumDefaultValue = codegenProperty.defaultValue.replace(codegenProperty.enumName + ".", "");
+            }
+        }
+        return rqBody;
+    }
+
+    private Schema getReferencedSchemaWhenNotEnum(Schema parameterSchema) {
+        Schema referencedSchema = ModelUtils.getReferencedSchema(openAPI, parameterSchema);
+        if (referencedSchema.getEnum() != null && !referencedSchema.getEnum().isEmpty()) {
+            referencedSchema = parameterSchema;
+        }
+        return referencedSchema;
+    }
+
     @Override
     public CodegenParameter fromParameter(Parameter p, Set<String> imports) {
         var parameter = super.fromParameter(p, imports);
@@ -816,7 +873,11 @@ public CodegenProperty fromProperty(String name, Schema schema, boolean required
         property.vendorExtensions.put("realName", realName);
 
         if (schema != null && schema.get$ref() != null) {
-            schema = ModelUtils.getSchemaFromRefToSchemaWithProperties(openAPI, schema.get$ref());
+            var refSchema = ModelUtils.getSchemaFromRefToSchemaWithProperties(openAPI, schema.get$ref());
+            if (refSchema == null) {
+                refSchema = ModelUtils.getReferencedSchema(openAPI, schema);
+            }
+            schema = refSchema;
         }
 
         String defaultValueInit;

diff --git a/...enerator/src/main/java/io/micronaut/openapi/generator/AbstractMicronautKotlinCodegen.java b/...enerator/src/main/java/io/micronaut/openapi/generator/AbstractMicronautKotlinCodegen.java
@@ -1425,13 +1425,64 @@ public String toModelName(final String name) {
     @Override
     public CodegenParameter fromRequestBody(RequestBody body, Set<String> imports, String bodyParameterName) {
         var rqBody = super.fromRequestBody(body, imports, bodyParameterName);
+
+        var rqBodySchema = body.getContent() != null && !body.getContent().isEmpty() ? body.getContent().entrySet().iterator().next().getValue().getSchema() : null;
+        CodegenProperty codegenProperty = fromProperty(bodyParameterName, rqBodySchema, false);
+
+        if (rqBodySchema != null) {
+            rqBodySchema = unaliasSchema(rqBodySchema);
+
+            if (getUseInlineModelResolver()) {
+                codegenProperty = fromProperty(bodyParameterName, getReferencedSchemaWhenNotEnum(rqBodySchema), false);
+            } else {
+                codegenProperty = fromProperty(bodyParameterName, rqBodySchema, false);
+            }
+            rqBody.setSchema(codegenProperty);
+        }
+
+        if (Boolean.TRUE.equals(codegenProperty.isModel)) {
+            rqBody.isModel = true;
+        }
+
+        rqBody.dataFormat = codegenProperty.dataFormat;
+        if (body.getRequired() != null) {
+            rqBody.required = body.getRequired();
+        }
+
+        // set containerType
+        rqBody.containerType = codegenProperty.containerType;
+        rqBody.containerTypeMapped = codegenProperty.containerTypeMapped;
+
+        // enum
+        updateCodegenPropertyEnum(codegenProperty);
+        rqBody.isEnum = codegenProperty.isEnum;
+        rqBody.isEnumRef = codegenProperty.isEnumRef;
+        rqBody._enum = codegenProperty._enum;
+        rqBody.allowableValues = codegenProperty.allowableValues;
+
+        if (codegenProperty.isEnum || codegenProperty.isEnumRef) {
+            rqBody.datatypeWithEnum = codegenProperty.datatypeWithEnum;
+            rqBody.enumName = codegenProperty.enumName;
+            if (codegenProperty.defaultValue != null) {
+                rqBody.enumDefaultValue = codegenProperty.defaultValue.replace(codegenProperty.enumName + ".", "");
+            }
+        }
+
         if (!rqBody.required) {
             rqBody.vendorExtensions.put("defaultValueInit", "null");
         }
 
         return rqBody;
     }
 
+    private Schema getReferencedSchemaWhenNotEnum(Schema parameterSchema) {
+        Schema referencedSchema = ModelUtils.getReferencedSchema(openAPI, parameterSchema);
+        if (referencedSchema.getEnum() != null && !referencedSchema.getEnum().isEmpty()) {
+            referencedSchema = parameterSchema;
+        }
+        return referencedSchema;
+    }
+
     @Override
     public CodegenParameter fromParameter(Parameter p, Set<String> imports) {
         var parameter = super.fromParameter(p, imports);
@@ -1512,7 +1563,11 @@ public CodegenProperty fromProperty(String name, Schema schema, boolean required
         property.vendorExtensions.put("realName", realName);
 
         if (schema != null && schema.get$ref() != null) {
-            schema = ModelUtils.getSchemaFromRefToSchemaWithProperties(openAPI, schema.get$ref());
+            var refSchema = ModelUtils.getSchemaFromRefToSchemaWithProperties(openAPI, schema.get$ref());
+            if (refSchema == null) {
+                refSchema = ModelUtils.getReferencedSchema(openAPI, schema);
+            }
+            schema = refSchema;
         }
 
         String defaultValueInit;

diff --git a/...enerator/src/test/java/io/micronaut/openapi/generator/JavaMicronautClientCodegenTest.java b/...enerator/src/test/java/io/micronaut/openapi/generator/JavaMicronautClientCodegenTest.java
@@ -17,7 +17,7 @@
 class JavaMicronautClientCodegenTest extends AbstractMicronautCodegenTest {
 
     @Test
-    void clientOptsUnicity() {
+    void clientOptsUniqueness() {
         var codegen = new JavaMicronautClientCodegen();
         codegen.cliOptions()
             .stream()
@@ -1460,4 +1460,20 @@ public int hashCode() {
                 }
             """);
     }
+
+    @Test
+    void testBodyEnum() {
+
+        var codegen = new JavaMicronautClientCodegen();
+        String outputPath = generateFiles(codegen, "src/test/resources/3_0/body-enum.yml", CodegenConstants.APIS, CodegenConstants.MODELS);
+        String path = outputPath + "src/main/java/org/openapitools/";
+
+        assertFileContains(path + "api/MyCustomApi.java", """
+                @Post("/api/v1/colors/{name}")
+                Mono<@NotNull String> selectColor(
+                    @Body @NotNull Color body
+                );
+            """);
+        assertFileContains(path + "model/Color.java", "public enum Color {");
+    }
 }
diff --git a/...enerator/src/test/java/io/micronaut/openapi/generator/JavaMicronautServerCodegenTest.java b/...enerator/src/test/java/io/micronaut/openapi/generator/JavaMicronautServerCodegenTest.java
@@ -18,7 +18,7 @@ class JavaMicronautServerCodegenTest extends AbstractMicronautCodegenTest {
     static String MULTI_TAGS_TEST_PATH = "src/test/resources/3_0/micronaut/multi-tags-test.yaml";
 
     @Test
-    void clientOptsUnicity() {
+    void clientOptsUniqueness() {
         var codegen = new JavaMicronautServerCodegen();
         codegen.cliOptions()
             .stream()

diff --git a/...erator/src/test/java/io/micronaut/openapi/generator/KotlinMicronautClientCodegenTest.java b/...erator/src/test/java/io/micronaut/openapi/generator/KotlinMicronautClientCodegenTest.java
@@ -17,7 +17,7 @@
 class KotlinMicronautClientCodegenTest extends AbstractMicronautCodegenTest {
 
     @Test
-    void clientOptsUnicity() {
+    void clientOptsUniqueness() {
         var codegen = new KotlinMicronautClientCodegen();
         codegen.cliOptions()
             .stream()
@@ -1468,4 +1468,20 @@ fun sendPrimitives(
                         ): Mono<String>
                      """);
     }
+
+    @Test
+    void testBodyEnum() {
+
+        var codegen = new KotlinMicronautClientCodegen();
+        String outputPath = generateFiles(codegen, "src/test/resources/3_0/body-enum.yml", CodegenConstants.APIS, CodegenConstants.MODELS);
+        String path = outputPath + "src/main/kotlin/org/openapitools/";
+
+        assertFileContains(path + "api/MyCustomApi.kt", """
+                @Post("/api/v1/colors/{name}")
+                fun selectColor(
+                    @Body @NotNull body: Color
+                ): Mono<String>
+            """);
+        assertFileContains(path + "model/Color.kt", "enum class Color(");
+    }
 }
diff --git a/...erator/src/test/java/io/micronaut/openapi/generator/KotlinMicronautServerCodegenTest.java b/...erator/src/test/java/io/micronaut/openapi/generator/KotlinMicronautServerCodegenTest.java
@@ -17,7 +17,7 @@ class KotlinMicronautServerCodegenTest extends AbstractMicronautCodegenTest {
     static String MULTI_TAGS_TEST_PATH = "src/test/resources/3_0/micronaut/multi-tags-test.yaml";
 
     @Test
-    void clientOptsUnicity() {
+    void clientOptsUniqueness() {
         var codegen = new KotlinMicronautServerCodegen();
         codegen.cliOptions()
             .stream()

diff --git a/openapi-generator/src/test/resources/3_0/body-enum.yml b/openapi-generator/src/test/resources/3_0/body-enum.yml
@@ -0,0 +1,42 @@
+openapi: 3.1.0
+info:
+  version: '1.0.0'
+  title: 'OpenAPI BUG REST API'
+servers:
+  - url: 'localhost:3000'
+
+paths:
+  /api/v1/colors/{name}:
+    post:
+      tags: [ my-custom ]
+      operationId: "selectColor"
+      requestBody:
+        required: true
+        content:
+          application/json:
+            schema:
+              $ref: "#/components/schemas/Color"
+      responses:
+        "200":
+          description: "OK"
+          content:
+            application/json:
+              schema:
+                type: string
+
+components:
+  schemas:
+
+    Color:
+      type: string
+      enum:
+        - GREEN
+        - RED
+        - WHITE
+        - BLACK
+        - YELLOW
+        - BLUE
+
+tags:
+  - name: my-custom
+    description: 'All API operations'