handle request body

Author: hauner

src/main/groovy/com/github/hauner/openapi/spring/writer/MethodWriter.groovy

@@ -17,6 +17,7 @@
 package com.github.hauner.openapi.spring.writer
 
 import com.github.hauner.openapi.spring.model.Endpoint
+import com.github.hauner.openapi.spring.model.RequestBody
 import com.github.hauner.openapi.spring.model.parameters.Parameter
 import com.github.hauner.openapi.support.Identifier
 
@@ -39,6 +40,11 @@ class MethodWriter {
         mapping += "("
         mapping += 'path = ' + quote(endpoint.path)
 
+        if (!endpoint.requestBodies.empty) {
+            mapping += ", "
+            mapping += 'consumes = {' + quote(endpoint.requestBody.contentType) + '}'
+        }
+
         if (!endpoint.response.empty) {
             mapping += ", "
             mapping += 'produces = {' + quote(endpoint.response.contentType) + '}'
@@ -68,6 +74,17 @@ class MethodWriter {
         param
     }
 
+    private String createRequestBodyAnnotation (RequestBody requestBody) {
+        String param = "${requestBody.annotation}"
+
+        // required is default, so add required only if the parameter is not required
+        if (!requestBody.required) {
+            param += '(required = false)'
+        }
+
+        param
+    }
+
     private String createMethodName (Endpoint endpoint) {
         def tokens = endpoint.path.tokenize ('/')
         tokens = tokens.collect { Identifier.fromJson (it).capitalize () }
@@ -86,6 +103,12 @@ class MethodWriter {
 
         }
 
+        if (!endpoint.requestBodies.empty) {
+            def body = endpoint.requestBody
+            def param = "${createRequestBodyAnnotation(body)} ${body.requestBodyType.name} body"
+            ps.add (param)
+        }
+
         ps.join (', ')
     }
 

src/test/groovy/com/github/hauner/openapi/spring/writer/MethodWriterSpec.groovy

@@ -18,6 +18,7 @@ package com.github.hauner.openapi.spring.writer
 
 import com.github.hauner.openapi.spring.model.Endpoint
 import com.github.hauner.openapi.spring.model.HttpMethod
+import com.github.hauner.openapi.spring.model.RequestBody
 import com.github.hauner.openapi.spring.model.Response
 import com.github.hauner.openapi.spring.model.datatypes.BooleanDataType
 import com.github.hauner.openapi.spring.model.datatypes.CollectionDataType
@@ -322,4 +323,47 @@ class MethodWriterSpec extends Specification {
 """
     }
 
+    void "writes required request body parameter" () {
+        def endpoint = new Endpoint (path: '/foo', method: HttpMethod.POST, responses: [
+            new Response (contentType: 'application/json', responseType: new NoneDataType())
+        ], requestBodies: [
+            new RequestBody(
+                contentType: 'application/json',
+                requestBodyType: new ObjectDataType (type: 'FooRequestBody',
+                    properties: ['foo': new StringDataType ()] as LinkedHashMap),
+                required: true)
+        ])
+
+        when:
+        writer.write (target, endpoint)
+
+        then:
+        target.toString () == """\
+    @PostMapping(path = "${endpoint.path}", consumes = {"application/json"})
+    ResponseEntity<void> postFoo(@RequestBody FooRequestBody body);
+"""
+    }
+
+    void "writes optional request body parameter" () {
+        def endpoint = new Endpoint (path: '/foo', method: HttpMethod.POST, responses: [
+            new Response (contentType: 'application/json', responseType: new NoneDataType())
+        ], requestBodies: [
+            new RequestBody(
+                contentType: 'application/json',
+                requestBodyType: new ObjectDataType (
+                    type: 'FooRequestBody',
+                    properties: ['foo': new StringDataType ()] as LinkedHashMap),
+                required: false)
+        ])
+
+        when:
+        writer.write (target, endpoint)
+
+        then:
+        target.toString () == """\
+    @PostMapping(path = "${endpoint.path}", consumes = {"application/json"})
+    ResponseEntity<void> postFoo(@RequestBody(required = false) FooRequestBody body);
+"""
+    }
+
 }