JavaPoet DSL

Lightweight Kotlin extension of JavaPoet, providing Kotlin DSL functionality and other convenient solutions.

• Full of convenient methods to achieve minimum code writing possible.
• Options to invoke DSL. For example, methods.add("main") { ... } is as good as methods { "main" { ... } }. Scroll down for more information.
• Smooth transition, existing JavaPoet native specs can still be configured with DSL.

buildJavaFile("com.example.helloworld") {
    types.addClass("HelloWorld") {
        addModifiers(PUBLIC, FINAL)
        methods {
            "main" {
                addModifiers(PUBLIC, STATIC)
                returns = VOID
                parameters.add(STRING.array, "args")
                appendLine("%T.out.println(%S)", System::class, "Hello, JavaPoet!")
            }
        }
    }
}.writeTo(System.out)

Download

repositories {
    mavenCentral()
}
dependencies {
    implementation "com.hanggrian:javapoet-dsl:$version"
}

Usage

Use % in string formatter

JavaPoet uses char prefix $ when formatting literals ($L), strings ($S), types ($T), an names ($N) within strings. However in Kotlin, $ in strings is reserved for variable referral. Avoid using \$ and instead use % as the prefix, this is also the approach taken by KotlinPoet.

buildMethodSpec("getName") {
    setReturns<String>()
    appendLine("%S", name)
}

buildCodeBlock {
    appendLine("int result = 0")
    beginFlow("for (int i = %L; i < %L; i++)", 0, 10)
    appendLine("result = result %L i", "+=")
    endFlow()
    appendLine("return result")
}

Use T::class as parameters

KClass<*> can now be used as format arguments. There is also inline reified type function whenever possible.

buildMethodSpec("sortList") {
    returns = INT
    parameters.add(classNamed("java.util", "List").parameterizedBy(hoverboard), "list")
    appendLine("%T.sort(list)", Collections::class)
    appendLine("return list")
}

buildFieldSpec("count", INT) {
    setInitializer("%L", 0)
}

Optional DSL

Some elements (field, method, parameter, etc.) are wrapped in container class. These containers have ability to add components with/without invoking DSL.

For example, 2 examples below will produce the same result.

types {
    addClass("Car") {
        methods {
            "getWheels" {
                returns = INT
                appendLine("return wheels")
            }
            "setWheels" {
                parameters {
                    add(INT, "wheels")
                }
                appendLine("this.wheels = wheels")
            }
        }
    }
}

types.addClass("Car") {
    methods.add("getWheels") {
        returns = INT
        appendLine("return wheels")
    }
    methods.add("setWheels") {
        parameters.add(INT, "wheels")
        appendLine("this.wheels = wheels")
    }
}

Property delegation

In spirit of Gradle Kotlin DSL, creating a spec can be done by delegating to a property.

val setWheels by methods.adding {
    val wheels by parameters.adding(INT)
    appendLine("this.wheels = wheels")
}

Fluent TypeName API

Write TypeName and all its subtypes fluently.

val customClass: ClassName =
    classNamed("com.example", "MyClass")

val arrayOfString: ArrayTypeName =
    String::class.name.array

val pairOfInteger: ParameterizedTypeName =
    classNamed("android.util", "Pair")
        .parameterizedBy(Integer::class, Integer::class)

val aGenerics: TypeVariableName =
    "T".generics

val subtypeOfCharSequence: WildcardTypeName =
    CharSequence::class.name.subtype