poet.CodeBlockBuilder.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-code-generation Show documentation

Wrapping core components for kotlin code generation with kotlin-poet.

The newest version!

package io.toolisticon.kotlin.generation.poet

import com.squareup.kotlinpoet.CodeBlock
import com.squareup.kotlinpoet.CodeBlock.Builder

/**
 * Wraps [CodeBlock.Builder] for typesafe access.
 */
class CodeBlockBuilder(
  override val builder: Builder
) : PoetSpecBuilder {
  companion object {
    val EMPTY_CODE_BLOCK = CodeBlock.builder().build()
    private fun Builder.wrap() = CodeBlockBuilder(this)
    fun of(format: String, vararg args: Any?): CodeBlock = CodeBlock.of(format, *args)
    fun codeBlock(format: String, vararg args: Any?): CodeBlock = of(format, *args)
    fun builder(): CodeBlockBuilder = CodeBlock.builder().wrap()
  }

  fun add(format: String, vararg args: Any?): CodeBlockBuilder = apply { builder.add(format, *args) }
  fun add(codeBlock: CodeBlock): CodeBlockBuilder = apply { builder.add(codeBlock) }
  fun add(builder: CodeBlockBuilder): CodeBlockBuilder = add(builder.build())
  fun addNamed(format: String, arguments: Map): CodeBlockBuilder = apply { builder.addNamed(format, arguments) }
  fun addStatement(format: String, vararg args: Any?): CodeBlockBuilder = apply { builder.addStatement(format, *args) }
  fun beginControlFlow(controlFlow: String, vararg args: Any?): CodeBlockBuilder = apply { builder.beginControlFlow(controlFlow, *args) }
  fun clear(): CodeBlockBuilder = apply { builder.clear() }
  fun endControlFlow(): CodeBlockBuilder = apply { builder.endControlFlow() }
  fun indent(): CodeBlockBuilder = apply { builder.indent() }
  fun isEmpty(): Boolean = builder.isEmpty()
  fun isNotEmpty(): Boolean = builder.isNotEmpty()
  fun nextControlFlow(controlFlow: String, vararg args: Any?): CodeBlockBuilder = apply { builder.nextControlFlow(controlFlow, *args) }
  fun unindent(): CodeBlockBuilder = apply { builder.unindent() }

  override fun build(): CodeBlock = builder.build()
}

interface CodeBlockSupplier : PoetSpecSupplier
typealias CodeBlockBuilderReceiver = CodeBlockBuilder.() -> Unit