gsonpath.util.CodeBlockExt.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.CodeBlock
import com.squareup.javapoet.TypeName

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

fun codeBlock(func: CodeBlock.Builder.() -> Unit): CodeBlock {
    return CodeBlock.builder().applyAndBuild(func)
}

fun CodeBlock.Builder.addWithNewLine(format: String, vararg args: Any): CodeBlock.Builder {
    this.add(format, *args)
    this.newLine()
    return this
}

fun CodeBlock.Builder.newLine(): CodeBlock.Builder {
    this.add("\n")
    return this
}

fun CodeBlock.Builder.comment(comment: String): CodeBlock.Builder {
    this.add("// $comment\n")
    return this
}

fun CodeBlock.Builder.`return`(format: String? = null, vararg args: Any): CodeBlock.Builder {
    if (format != null) {
        this.addStatement("return $format", *args)
    } else {
        this.addStatement("return")
    }
    return this
}

fun CodeBlock.Builder.addEscaped(format: String): CodeBlock.Builder {
    this.add(format.replace("$", "$$"))
    return this
}

fun CodeBlock.Builder.multiLinedNewObject(typeName: TypeName, variables: List) {
    addWithNewLine("return new \$T(", typeName)
    indent()
    add(variables.joinToString(",\n"))
    unindent()
    addStatement(")")
}

fun CodeBlock.Builder.addEscapedStatement(format: String): CodeBlock.Builder {
    this.addStatement(format.replace("$", "$$"))
    return this
}

fun CodeBlock.Builder.createVariable(type: String, name: String, assignment: String, vararg args: Any): CodeBlock.Builder {
    addStatement("$type $name = $assignment", *args)
    return this
}

fun CodeBlock.Builder.createVariable(typeName: TypeName, name: String, assignment: String, vararg args: Any): CodeBlock.Builder {
    val fullArgs = arrayOf(typeName, *args)
    return createVariable("\$T", name, assignment, *fullArgs)
}

fun CodeBlock.Builder.createVariable(typeName: Class<*>, name: String, assignment: String, vararg args: Any): CodeBlock.Builder {
    return createVariable(TypeName.get(typeName), name, assignment, *args)
}

fun CodeBlock.Builder.createVariableNew(type: String, name: String, assignment: String, vararg args: Any): CodeBlock.Builder {
    addStatement("$type $name = new $assignment", *args)
    return this
}

fun CodeBlock.Builder.assign(name: String, assignment: String, vararg args: Any): CodeBlock.Builder {
    addStatement("$name = $assignment", *args)
    return this
}

fun CodeBlock.Builder.assignNew(name: String, assignment: String, vararg args: Any): CodeBlock.Builder {
    addStatement("$name = new $assignment", *args)
    return this
}

fun  CodeBlock.Builder.autoControlFlow(controlFlow: String, vararg args: Any, func: CodeBlock.Builder.() -> T): T {
    beginControlFlow(controlFlow, *args)
    val result = func(this)
    endControlFlow()
    return result
}

fun  CodeBlock.Builder.`if`(
        condition: String,
        vararg args: Any,
        func: CodeBlock.Builder.() -> T): T = autoControlFlow("if ($condition)", *args, func = func)

fun  CodeBlock.Builder.ifWithoutClose(
        condition: String,
        vararg args: Any,
        func: CodeBlock.Builder.() -> T): T {

    beginControlFlow("if ($condition)", *args)
    return func(this)
}

fun  CodeBlock.Builder.`else`(func: CodeBlock.Builder.() -> T): T {
    nextControlFlow("else")
    val result = func(this)
    endControlFlow()
    return result
}

fun  CodeBlock.Builder.elseIf(
        condition: String,
        vararg args: Any,
        func: CodeBlock.Builder.() -> T): T {

    nextControlFlow("else if ($condition)", *args)
    return func(this)
}

fun  CodeBlock.Builder.`while`(
        condition: String,
        vararg args: Any,
        func: CodeBlock.Builder.() -> T): T = autoControlFlow("while ($condition)", *args, func = func)

fun  CodeBlock.Builder.switch(
        condition: String,
        vararg args: Any,
        func: CodeBlock.Builder.() -> T): T = autoControlFlow("switch ($condition)", *args, func = func)

fun  CodeBlock.Builder.`for`(
        condition: String,
        vararg args: Any,
        func: CodeBlock.Builder.() -> T): T = autoControlFlow("for ($condition)", *args, func = func)

fun  CodeBlock.Builder.case(
        label: String,
        func: CodeBlock.Builder.() -> T): T {

    addEscaped("case $label:")
    newLine()
    indent()
    val result = func(this)
    addStatement("break")
    unindent()
    newLine()
    return result
}

fun  CodeBlock.Builder.default(func: CodeBlock.Builder.() -> T): T {
    add("default:")
    newLine()
    indent()
    val result = func(this)
    addStatement("break")
    unindent()
    newLine()
    return result
}