com.ancientlightstudios.quarkus.kotlin.openapi.emitter.SafeObjectModelEmitter.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-kotlin-openapi-maven-plugin Show documentation

A Maven plugin to use the OpenAPI generator.

There is a newer version: 0.4.14

Show newest version

package com.ancientlightstudios.quarkus.kotlin.openapi.emitter

import com.ancientlightstudios.quarkus.kotlin.openapi.emitter.statements.writeSerializationStatement
import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.*
import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.expression.PathExpression.Companion.pathExpression
import com.ancientlightstudios.quarkus.kotlin.openapi.models.kotlin.expression.StringExpression.Companion.stringExpression
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformed.RequestSuite
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformed.name.MethodName.Companion.methodName
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformed.name.TypeName.SimpleTypeName.Companion.rawTypeName
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformed.name.VariableName.Companion.variableName
import com.ancientlightstudios.quarkus.kotlin.openapi.models.transformed.typedefinition.ObjectTypeDefinition
import com.ancientlightstudios.quarkus.kotlin.openapi.transformer.TypeDefinitionRegistry

class SafeObjectModelEmitter : CodeEmitter {

    override fun EmitterContext.emit(suite: RequestSuite, typeDefinitionRegistry: TypeDefinitionRegistry) {
        typeDefinitionRegistry.getAllTypeDefinitions()
            .filterIsInstance()
            .forEach {
                emitModel(it)
            }
    }

    private fun EmitterContext.emitModel(definition: ObjectTypeDefinition) {
        kotlinFile(modelPackage(), definition.name) {
            registerImport(apiPackage(), wildcardImport = true)
            registerImport("com.fasterxml.jackson.databind.JsonNode")
            registerImports(additionalImports)

            kotlinClass(fileName, asDataClass = true) {

                definition.properties.forEach { (name, _, propertyTypeUsage) ->
                    kotlinMember(
                        name.variableName(),
                        propertyTypeUsage.safeType,
                        accessModifier = null,
                        default = propertyTypeUsage.defaultValue
                    )
                }

                // toJsonNode
                kotlinMethod("toJsonNode".methodName(), returnType = "JsonNode".rawTypeName(), bodyAsAssignment = true) {
                    kotlinStatement {
                        writeln("objectNode()")
                        indent {
                            definition.properties.forEach { (name, _, propertyTypeUsage) ->
                                if (omitNullsInSerialization) {
                                    write(".setNonNull(")
                                }
                                else {
                                    write(".setAny(")
                                }
                                write(name.stringExpression().evaluate())
                                write(", ")
                                writeSerializationStatement(name.variableName().pathExpression(), propertyTypeUsage)
                                writeln(")")
                            }
                        }
                    }
                }
            }
        }.also { generateFile(it) }
    }
}