#78, write data type with implements

Author: hauner

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

@@ -72,6 +72,6 @@ open class ObjectDataType(
 
     private val implementsImports: Set<String>
         get() {
-            return implementsDataType?.referencedImports ?: emptySet()
+            return implementsDataType?.getImports() ?: emptySet()
         }
 }

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

@@ -53,7 +53,12 @@ class DataTypeWriter(
             }
         }
 
-        target.write("public class ${dataType.getTypeName()} {\n\n")
+        val implements: DataType? = dataType.implementsDataType
+        if (implements != null) {
+            writeClassImplementsHeader(target, dataType, implements)
+        } else {
+            writeClassHeader(target, dataType)
+        }
 
         dataType.forEach { propName, propDataType ->
             val javaPropertyName = toCamelCase(propName)
@@ -70,6 +75,21 @@ class DataTypeWriter(
         target.write ("}\n")
     }
 
+    private fun writeClassImplementsHeader(
+        target: Writer,
+        dataType: ModelDataType,
+        implements: DataType
+    ) {
+        target.write("public class ${dataType.getTypeName()} implements ${implements.getTypeName()} {\n\n")
+    }
+
+    private fun writeClassHeader(
+        target: Writer,
+        dataType: ModelDataType
+    ) {
+        target.write("public class ${dataType.getTypeName()} {\n\n")
+    }
+
     private fun getProp(
         propertyName: String,
         javaPropertyName: String,

src/test/kotlin/io/openapiprocessor/core/writer/java/DataTypeWriterSpec.kt

@@ -11,11 +11,8 @@ import io.kotest.matchers.string.shouldContain
 import io.mockk.mockk
 import io.openapiprocessor.core.converter.ApiOptions
 import io.openapiprocessor.core.extractImports
-import io.openapiprocessor.core.model.datatypes.DataTypeConstraints
-import io.openapiprocessor.core.model.datatypes.ModelDataType
-import io.openapiprocessor.core.model.datatypes.PropertyDataType
+import io.openapiprocessor.core.model.datatypes.*
 import io.openapiprocessor.core.support.datatypes.ObjectDataType
-import io.openapiprocessor.core.model.datatypes.StringDataType
 import io.openapiprocessor.core.support.datatypes.ListDataType
 import io.openapiprocessor.core.support.datatypes.propertyDataType
 import io.openapiprocessor.core.support.datatypes.propertyDataTypeString
@@ -143,4 +140,22 @@ class DataTypeWriterSpec: StringSpec({
             """.trimMargin()
     }
 
+    "writes class with implements" {
+        options.oneOfInterface = true
+
+        val dataType = ObjectDataType ("Foo", "pkg", linkedMapOf(
+            "foo" to PropertyDataType (false, false, StringDataType ())
+        ))
+
+        val ifDataType = InterfaceDataType(
+            DataTypeName("MarkerInterface"), "pkg", listOf(dataType))
+
+        dataType.implementsDataType = ifDataType
+
+        // when:
+        writer.write (target, dataType)
+
+        target.toString() shouldContain ("public class Foo implements MarkerInterface {")
+    }
+
 })