org.web3j.openapi.codegen.utils.KPoetUtils.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of web3j-openapi-codegen Show documentation

web3j-openapi project code generators

There is a newer version: 4.8.4

Show newest version

/*
 * Copyright 2020 Web3 Labs Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package org.web3j.openapi.codegen.utils

import com.fasterxml.jackson.annotation.JsonProperty
import com.squareup.kotlinpoet.FileSpec
import com.squareup.kotlinpoet.FunSpec
import com.squareup.kotlinpoet.KModifier
import com.squareup.kotlinpoet.PropertySpec
import com.squareup.kotlinpoet.TypeSpec
import com.squareup.kotlinpoet.AnnotationSpec
import org.web3j.protocol.core.methods.response.AbiDefinition.NamedType

internal fun List.toDataClass(
    modelPackageName: String,
    name: String,
    type: String,
    basePackageName: String = "",
    contractName: String = ""
): FileSpec {
    val outputFile = FileSpec.builder(
        modelPackageName,
        "${name.capitalize()}$type"
    )

    val constructor = TypeSpec
        .classBuilder("${name.capitalize()}$type")

    // FIXME: Events with no parameters require no field class
    if (isNotEmpty()) constructor.addModifiers(KModifier.DATA)

    val constructorBuilder = FunSpec.constructorBuilder()

    forEachIndexed { index, input ->
        val inputName = GeneratorUtils.argumentName(input.name, index)
        val inputType =
            if (input.type == "tuple") input.type.mapType(true, input.internalType.structName, basePackageName, contractName.toLowerCase())
            else input.type.mapType()

        constructorBuilder.addParameter(
            inputName,
            inputType
        )
        constructor.addProperty(
            PropertySpec.builder(
                inputName,
                inputType
            ).initializer(inputName)
                .addAnnotation(
                    AnnotationSpec.builder(JsonProperty::class.java)
                        .addMember("value = %S", inputName)
                        .build()
                )
                .build()
        )
    }
    constructor.primaryConstructor(constructorBuilder.build())

    return outputFile
        .addType(constructor.build())
        .build()
}