openapi-processor/openapi-processor-spring#136, property data type

Author: hauner

src/main/kotlin/io/openapiprocessor/core/converter/DataTypeConverter.kt

@@ -162,6 +162,13 @@ class DataTypeConverter(
             if (propSchemaInfo.getNullable()) {
                 propDataType = nullWrapper.wrap(propDataType, schemaInfo)
             }
+
+            propDataType = PropertyDataType(
+                propSchemaInfo.readOnly,
+                propSchemaInfo.writeOnly,
+                propDataType
+            )
+
             properties[propName] = propDataType
         }
 

src/main/kotlin/io/openapiprocessor/core/model/DataTypeCollector.kt

@@ -53,6 +53,9 @@ class DataTypeCollector(
             is StringEnumDataType -> {
                 dataTypes.addRef(dataType.getName())
             }
+            is PropertyDataType -> {
+                collect(dataType.dataType)
+            }
         }
     }
 

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

@@ -0,0 +1,31 @@
+/*
+ * Copyright 2021 https://github.com/openapi-processor/openapi-processor-core
+ * PDX-License-Identifier: Apache-2.0
+ */
+
+package io.openapiprocessor.core.model.datatypes
+
+/**
+ * schema "properties" data type wrapper. readOnly/writeOnly may be different on each use of the
+ * same schema as a property in another schema.
+ */
+open class PropertyDataType(
+    val readOnly: Boolean,
+    val writeOnly: Boolean,
+    val dataType: DataType
+): DataType {
+
+    override fun getName(): String {
+        return dataType.getName()
+    }
+
+    override fun getPackageName(): String {
+        return dataType.getPackageName()
+    }
+
+    override fun getImports(): Set<String> {
+        return dataType.getImports()
+    }
+
+}
+

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

@@ -9,6 +9,7 @@ import io.openapiprocessor.core.converter.ApiOptions
 import io.openapiprocessor.core.model.datatypes.DataType
 import io.openapiprocessor.core.model.datatypes.ModelDataType
 import io.openapiprocessor.core.model.datatypes.NullDataType
+import io.openapiprocessor.core.model.datatypes.PropertyDataType
 import io.openapiprocessor.core.support.capitalizeFirstChar
 import java.io.Writer
 
@@ -56,7 +57,7 @@ class DataTypeWriter(
 
         dataType.forEach { propName, propDataType ->
             val javaPropertyName = toCamelCase(propName)
-            target.write(getProp(propName, javaPropertyName, propDataType,
+            target.write(getProp(propName, javaPropertyName, propDataType as PropertyDataType,
                 dataType.isRequired(propName)))
         }
 
@@ -70,8 +71,10 @@ class DataTypeWriter(
     }
 
     private fun getProp(
-        propertyName: String, javaPropertyName: String,
-        propDataType: DataType, required: Boolean): String {
+        propertyName: String,
+        javaPropertyName: String,
+        propDataType: PropertyDataType,
+        required: Boolean): String {
 
         var result = ""
 
@@ -91,18 +94,44 @@ class DataTypeWriter(
             propTypeName = prop.dataTypeValue
         }
 
-        result += "    @JsonProperty(\"$propertyName\")\n"
+        result += "    ${getPropertyAnnotation(propertyName, propDataType)}\n"
         result += "    private $propTypeName $javaPropertyName"
 
-        // null may have an init value
-        if (propDataType is NullDataType && propDataType.init != null) {
-            result += " = ${propDataType.init}"
+        // null (JsonNullable) may have an init value
+        val dataType = propDataType.dataType
+        if (dataType is NullDataType && dataType.init != null) {
+            result += " = ${dataType.init}"
         }
 
         result += ";\n\n"
         return result
     }
 
+    private fun getPropertyAnnotation(propertyName: String, propDataType: PropertyDataType): String {
+        val access = getAccess(propDataType)
+
+        var result = "@JsonProperty("
+        if (access != null) {
+            result += "value = \"$propertyName\", access = JsonProperty.Access.${access.value}"
+        } else {
+            result += "\"$propertyName\""
+        }
+
+        result += ")"
+        return result
+    }
+
+    private fun getAccess(propDataType: PropertyDataType): PropertyAccess? {
+        if (!propDataType.readOnly && !propDataType.writeOnly)
+            return null
+
+        return when {
+            propDataType.readOnly -> PropertyAccess("READ_ONLY")
+            propDataType.writeOnly -> PropertyAccess("WRITE_ONLY")
+            else -> throw IllegalStateException()
+        }
+    }
+
     private fun getGetter(propertyName: String, propDataType: DataType): String {
         var result = ""
         result += ifDeprecated(propDataType)
@@ -170,3 +199,5 @@ class DataTypeWriter(
     }
 
 }
+
+class PropertyAccess(val value: String)