allow reference to a model from OpenApiContentProperty

javalin · Sep 30, 2024 · 3e0056d · 3e0056d
1 parent cff4deb
commit 3e0056d
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 13 deletions.
diff --git a/examples/javalin-gradle-kotlin/src/main/java/io/javalin/openapi/plugin/test/JavalinTest.java b/examples/javalin-gradle-kotlin/src/main/java/io/javalin/openapi/plugin/test/JavalinTest.java
@@ -203,6 +203,7 @@ public static void main(String[] args) {
                 @OpenApiContent(mimeType = "image/png", type = "string", format = "base64"), // single file upload,
                 @OpenApiContent(mimeType = "multipart/form-data", properties = {
                     @OpenApiContentProperty(name = "form-element", type = "integer"), // random element in form-data
+                    @OpenApiContentProperty(name = "reference", from = KotlinEntity.class), // reference to another object
                     @OpenApiContentProperty(name = "file-name", isArray = true, type = "string", format = "base64") // multi-file upload
                 })
             }

diff --git a/examples/javalin-maven-java/src/main/java/io/javalin/openapi/plugin/test/JavalinTest.java b/examples/javalin-maven-java/src/main/java/io/javalin/openapi/plugin/test/JavalinTest.java
@@ -175,6 +175,7 @@ public static void main(String[] args) {
                 @OpenApiContent(mimeType = "image/png", type = "string", format = "base64"), // single file upload,
                 @OpenApiContent(mimeType = "multipart/form-data", properties = {
                         @OpenApiContentProperty(name = "form-element", type = "integer"), // random element in form-data
+                        @OpenApiContentProperty(name = "reference", from = KotlinEntity.class) // reference to another object
                         @OpenApiContentProperty(name = "file-name", isArray = true, type = "string", format = "base64") // multi-file upload
                 })
             }

diff --git a/...ion-processor/src/main/kotlin/io/javalin/openapi/processor/generators/OpenApiGenerator.kt b/...ion-processor/src/main/kotlin/io/javalin/openapi/processor/generators/OpenApiGenerator.kt
@@ -444,22 +444,29 @@ internal class OpenApiGenerator {
                     schema.addProperty("type", "object")
 
                     for (contentProperty in properties) {
-                        val propertyScheme = JsonObject()
                         val propertyFormat = contentProperty.format.takeIf { it != NULL_STRING }
 
-                        if (contentProperty.isArray) {
-                            propertyScheme.addProperty("type", "array")
-
-                            val items = JsonObject()
-                            items.addProperty("type", contentProperty.type)
-                            propertyFormat?.let { items.addProperty("format", it) }
-                            propertyScheme.add("items", items)
+                        val contentPropertyFrom = contentAnnotation.getTypeMirror { contentProperty.from }
+                        val propertyScheme = if (contentPropertyFrom.getFullName() != NULL_CLASS::class.java.name) {
+                            createTypeDescriptionWithReferences(contentPropertyFrom)
                         } else {
-                            propertyScheme.addProperty("type", contentProperty.type)
-                            propertyFormat?.let { propertyScheme.addProperty("format", it) }
+                            JsonObject().apply {
+                                addProperty("type", contentProperty.type)
+                                propertyFormat?.let { addProperty("format", it) }
+                            }
                         }
 
-                        propertiesSchema.add(contentProperty.name, propertyScheme)
+                        propertiesSchema.add(contentProperty.name,
+                            if (contentProperty.isArray) {
+                                // wrap into OpenAPI array object
+                                JsonObject().apply {
+                                    addProperty("type", "array")
+                                    add("items", propertyScheme)
+                                }
+                            } else {
+                                propertyScheme
+                            }
+                        )
                     }
 
                     schema.add("properties", propertiesSchema)

diff --git a/openapi-specification/src/main/kotlin/io/javalin/openapi/OpenApiAnnotations.kt b/openapi-specification/src/main/kotlin/io/javalin/openapi/OpenApiAnnotations.kt
@@ -147,9 +147,10 @@ annotation class OpenApiContent(
 @Target()
 @Retention(RUNTIME)
 annotation class OpenApiContentProperty(
+    val from: KClass<*> = NULL_CLASS::class,
     val name: String,
     val isArray: Boolean = false,
-    val type: String,
+    val type: String = NULL_STRING,
     val format: String = NULL_STRING
 )
 

diff --git a/wiki b/wiki