generate allOf class

Author: hauner

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/model/Api.kt

@@ -17,6 +17,7 @@
 package io.openapiprocessor.core.model
 
 import io.openapiprocessor.core.model.datatypes.DataType
+import io.openapiprocessor.core.model.datatypes.ModelDataType
 import io.openapiprocessor.core.model.datatypes.ObjectDataType
 import io.openapiprocessor.core.model.datatypes.StringEnumDataType
 import java.util.function.Consumer
@@ -60,6 +61,10 @@ class Api(
         models.getObjectDataTypes().forEach(action)
     }
 
+    fun forEachModelDataType(action: Consumer<ModelDataType>) {
+        models.getModelDataTypes().forEach(action)
+    }
+
     fun forEachEnumDataType(action: Consumer<StringEnumDataType>) {
         models.getEnumDataTypes().forEach(action)
     }

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/model/DataTypes.kt

@@ -16,10 +16,7 @@
 
 package io.openapiprocessor.core.model
 
-import io.openapiprocessor.core.model.datatypes.DataType
-import io.openapiprocessor.core.model.datatypes.MappedDataType
-import io.openapiprocessor.core.model.datatypes.ObjectDataType
-import io.openapiprocessor.core.model.datatypes.StringEnumDataType
+import io.openapiprocessor.core.model.datatypes.*
 
 /**
  * Container of data types from OpenAPI '#/component/schemas'.
@@ -52,6 +49,21 @@ class DataTypes {
             .toList()
     }
 
+    /**
+     * provides the *object* data types (model classes) used by the api endpoints.
+     * For this objects the processor will create POJOs classes.
+     *
+     * experimental: will probably replace getObjectDataTypes().
+     *
+     * @return list of object data types
+     */
+    fun getModelDataTypes(): Collection<ModelDataType> {
+        return types.values
+            .filterIsInstance<ModelDataType>()
+            .filter { it.isModel() }
+            .toList()
+    }
+
     /**
      * provides the enum data types (model classes) used by the api endpoints.
      * For this objects the processor will create enum classes.

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/model/datatypes/ComposedObjectDataType.kt

@@ -30,7 +30,7 @@ class ComposedObjectDataType(
     constraints: DataTypeConstraints? = null,
     deprecated: Boolean = false
 
-): DataTypeBase(constraints, deprecated) {
+): DataTypeBase(constraints, deprecated), ModelDataType {
 
     override fun getName(): String {
         return type
@@ -51,17 +51,22 @@ class ComposedObjectDataType(
             .toSet()
     }
 
+    // todo find better name
     override fun isComposed(): Boolean {
         return of != "allOf"
     }
 
-    fun getProperties(): Map<String, DataType> {
+    override fun isModel(): Boolean {
+        return of == "allOf"
+    }
+
+    override fun getProperties(): Map<String, DataType> {
         val properties = linkedMapOf<String, DataType>()
 
         if (of == "allOf") {
             items.forEach {
                 if (it is ObjectDataType) {
-                    properties.putAll(it.getObjectProperties())
+                    properties.putAll(it.getProperties())
                 }
             }
         }

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/model/datatypes/ModelDataType.kt

@@ -0,0 +1,14 @@
+/*
+ * Copyright © 2020 https://github.com/openapi-processor/openapi-processor-core
+ * PDX-License-Identifier: Apache-2.0
+ */
+
+package io.openapiprocessor.core.model.datatypes
+
+interface ModelDataType: DataType {
+
+    fun isModel(): Boolean
+
+    fun getProperties(): Map<String, DataType>
+
+}

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/model/datatypes/ObjectDataType.kt

@@ -30,7 +30,7 @@ class ObjectDataType(
     constraints: DataTypeConstraints? = null,
     deprecated: Boolean = false
 
-): DataTypeBase(constraints, deprecated) {
+): DataTypeBase(constraints, deprecated), ModelDataType {
 
     override fun getName(): String {
         return type
@@ -55,7 +55,16 @@ class ObjectDataType(
         return properties[name]!!
     }
 
+    @Deprecated("do not override groovys getProperties()", ReplaceWith("getProperties()"))
     fun getObjectProperties(): Map<String, DataType> {
+        return getProperties()
+    }
+
+    override fun isModel(): Boolean {
+        return true
+    }
+
+    override fun getProperties(): Map<String, DataType> {
         return properties
     }
 

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/writer/java/ApiWriter.kt

@@ -21,9 +21,9 @@ import com.google.googlejavaformat.java.JavaFormatterOptions
 import io.openapiprocessor.core.converter.ApiOptions
 import io.openapiprocessor.core.model.Api
 import io.openapiprocessor.core.model.Interface
-import io.openapiprocessor.core.model.datatypes.ObjectDataType
 import io.openapiprocessor.core.model.datatypes.StringEnumDataType
 import io.openapiprocessor.core.misc.toURL
+import io.openapiprocessor.core.model.datatypes.ModelDataType
 import java.io.BufferedWriter
 import java.io.StringWriter
 import java.io.Writer
@@ -73,7 +73,7 @@ class ApiWriter(
     }
 
     private fun writeObjectDataTypes(api: Api) {
-        api.forEachObjectDataType {
+        api.forEachModelDataType {
             val target = modelFolder.resolve ("${it.getName()}.java")
             val writer = BufferedWriter(PathWriter(target))
             writeDataType(writer, it)
@@ -96,7 +96,7 @@ class ApiWriter(
         writer.write(format(raw.toString()))
     }
 
-    private fun writeDataType(writer: Writer, dataType: ObjectDataType) {
+    private fun writeDataType(writer: Writer, dataType: ModelDataType) {
         val raw = StringWriter()
         dataTypeWriter.write(raw, dataType)
         writer.write(format(raw.toString ()))

openapi-processor-core/src/main/kotlin/io/openapiprocessor/core/writer/java/DataTypeWriter.kt

@@ -18,7 +18,7 @@ package io.openapiprocessor.core.writer.java
 
 import io.openapiprocessor.core.converter.ApiOptions
 import io.openapiprocessor.core.model.datatypes.DataType
-import io.openapiprocessor.core.model.datatypes.ObjectDataType
+import io.openapiprocessor.core.model.datatypes.ModelDataType
 import java.io.Writer
 
 /**
@@ -33,7 +33,7 @@ class DataTypeWriter(
     private val validationAnnotations: BeanValidationFactory = BeanValidationFactory()
 ) {
 
-    fun write(target: Writer, dataType: ObjectDataType) {
+    fun write(target: Writer, dataType: ModelDataType) {
         headerWriter.write(target)
         target.write("package ${dataType.getPackageName()};\n\n")
 
@@ -52,16 +52,14 @@ class DataTypeWriter(
 
         target.write("public class ${dataType.getName()} {\n\n")
 
-        val propertyNames = dataType.getObjectProperties().keys
-        propertyNames.forEach {
-            val javaPropertyName = toCamelCase(it)
-            val propDataType = dataType.getObjectProperty(it)
-            target.write(getProp(it, javaPropertyName, propDataType))
+        val properties = dataType.getProperties()
+        properties.forEach { (propName, propDataType) ->
+            val javaPropertyName = toCamelCase(propName)
+            target.write(getProp(propName, javaPropertyName, propDataType))
         }
 
-        propertyNames.forEach {
-            val javaPropertyName = toCamelCase(it)
-            val propDataType = dataType.getObjectProperty(it)
+        properties.forEach { (propName, propDataType) ->
+            val javaPropertyName = toCamelCase(propName)
             target.write(getGetter(javaPropertyName, propDataType))
             target.write(getSetter(javaPropertyName, propDataType))
         }
@@ -122,14 +120,14 @@ class DataTypeWriter(
         return result
     }
 
-    private fun collectImports(packageName: String, dataType: ObjectDataType): List<String> {
+    private fun collectImports(packageName: String, dataType: ModelDataType): List<String> {
         val imports = mutableSetOf<String>()
         imports.add("com.fasterxml.jackson.annotation.JsonProperty")
 
         imports.addAll(dataType.getReferencedImports())
 
         if (apiOptions.beanValidation) {
-            dataType.getObjectProperties().values.forEach {
+            dataType.getProperties().values.forEach {
                 imports.addAll(validationAnnotations.collectImports(it))
             }
         }

openapi-processor-core/src/test/kotlin/io/openapiprocessor/core/writer/java/ApiWriterSpec.kt

@@ -218,7 +218,7 @@ class ApiWriterSpec: StringSpec({
     "re-formats model sources" {
         val dtWriter = io.mockk.mockk<DataTypeWriter>()
         every { dtWriter.write(any(), any()) } answers {
-            arg<Writer>(0).write("  class \n  ${arg<ObjectDataType>(1).getName()} {   }\n")
+            arg<Writer>(0).write("  class \n  ${arg<ModelDataType>(1).getName()} {   }\n")
         }
 
         val dt = DataTypes()

openapi-processor-core/src/testInt/groovy/com/github/hauner/openapi/processor/core/ProcessorPendingTest.groovy

@@ -25,15 +25,15 @@ import org.junit.runner.RunWith
 import org.junit.runners.Parameterized
 import spock.lang.Ignore
 
-@Ignore
+//@Ignore
 @RunWith(Parameterized)
 class ProcessorPendingTest extends ProcessorTestBase {
 
     @Parameterized.Parameters(name = "{0}")
     static Collection<TestSet> sources () {
         return [
-            new TestSet(name: 'ref-loop-array', processor: new TestProcessor(), parser: ParserType.SWAGGER),
-            new TestSet(name: 'ref-loop-array', processor: new TestProcessor(), parser: ParserType.OPENAPI4J)
+            new TestSet(name: 'schema-composed-allof', processor: new TestProcessor(), parser: ParserType.SWAGGER),
+            new TestSet(name: 'schema-composed-allof', processor: new TestProcessor(), parser: ParserType.OPENAPI4J)
         ]
     }