gsonpath.util.TypeSpecExt.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gsonpath-compiler-base Show documentation

An annotation processor which generates Type Adapters for the Google Gson library

There is a newer version: 2.0.0

Show newest version

package gsonpath.util

import com.squareup.javapoet.*
import java.lang.reflect.Type
import javax.lang.model.element.Modifier

object TypeSpecExt {
    fun finalClassBuilder(name: String): TypeSpec.Builder {
        return TypeSpec.classBuilder(name)
                .addModifiers(Modifier.PUBLIC, Modifier.FINAL)
    }

    fun finalClassBuilder(className: ClassName): TypeSpec.Builder {
        return TypeSpec.classBuilder(className)
                .addModifiers(Modifier.PUBLIC, Modifier.FINAL)
    }
}

fun TypeSpec.Builder.applyAndBuild(func: TypeSpec.Builder.() -> Unit): TypeSpec {
    return apply(func).build()
}

fun TypeSpec.Builder.field(name: String, type: TypeName, func: FieldSpec.Builder.() -> Unit) {
    addField(FieldSpec.builder(type, name).apply(func).build())
}

fun TypeSpec.Builder.field(name: String, type: Type, func: FieldSpec.Builder.() -> Unit) {
    field(name, TypeName.get(type), func)
}

fun TypeSpec.Builder.method(name: String, func: MethodSpec.Builder.() -> Unit) {
    addMethod(MethodSpec.methodBuilder(name).applyAndBuild(func))
}

fun TypeSpec.Builder.overrideMethod(name: String, func: MethodSpec.Builder.() -> Unit) {
    addMethod(MethodSpecExt.overrideMethodBuilder(name).applyAndBuild(func))
}

fun TypeSpec.Builder.constructor(func: MethodSpec.Builder.() -> Unit) {
    addMethod(MethodSpec.constructorBuilder().applyAndBuild(func))
}