builder.KotlinFunSpecBuilder.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.
There is a newer version: 0.1.0
Show newest version
package io.toolisticon.kotlin.generation.builder

import com.squareup.kotlinpoet.*
import io.toolisticon.kotlin.generation.BuilderSupplier
import io.toolisticon.kotlin.generation.poet.FunSpecBuilder
import io.toolisticon.kotlin.generation.poet.FunSpecBuilder.Companion.wrap
import io.toolisticon.kotlin.generation.poet.FunSpecBuilderReceiver
import io.toolisticon.kotlin.generation.poet.KDoc
import io.toolisticon.kotlin.generation.spec.KotlinAnnotationSpecSupplier
import io.toolisticon.kotlin.generation.spec.KotlinFunSpec
import io.toolisticon.kotlin.generation.spec.KotlinFunSpecSupplier
import io.toolisticon.kotlin.generation.spec.KotlinParameterSpecSupplier
import javax.lang.model.element.Element
import javax.lang.model.element.Modifier
import kotlin.reflect.KClass

/**
 * Builder for [KotlinFunSpec].
 */
@ExperimentalKotlinPoetApi
class KotlinFunSpecBuilder internal constructor(
  private val delegate: FunSpecBuilder
) : BuilderSupplier,
  DelegatingBuilder,
  KotlinAnnotatableDocumentableModifiableBuilder,
  KotlinContextReceivableBuilder,
  KotlinFunSpecSupplier {

  companion object {
    /**
     * Creates new builder.
     */
    fun builder(name: String): KotlinFunSpecBuilder = KotlinFunSpecBuilder(
      delegate = FunSpecBuilder.builder(name)
    )

    /**
     * Creates new builder.
     */
    fun builder(memberName: MemberName): KotlinFunSpecBuilder = KotlinFunSpecBuilder(
      delegate = FunSpecBuilder.builder(memberName)
    )

    /**
     * Creates new builder.
     */
    fun constructorBuilder(): KotlinFunSpecBuilder = KotlinFunSpecBuilder(
      delegate = FunSpecBuilder.constructorBuilder()
    )

    /**
     * Creates new builder.
     */
    fun getterBuilder(): KotlinFunSpecBuilder = KotlinFunSpecBuilder(
      delegate = FunSpecBuilder.getterBuilder()
    )

    /**
     * Creates new builder.
     */
    fun setterBuilder(): KotlinFunSpecBuilder = KotlinFunSpecBuilder(
      delegate = FunSpecBuilder.setterBuilder()
    )

    /**
     * Creates new builder.
     */
    fun builder(spec: KotlinFunSpec) = builder(spec.get())

    /**
     * Creates new builder.
     */
    fun builder(spec: FunSpec) = KotlinFunSpecBuilder(delegate = spec.toBuilder().wrap())
  }

  fun addParameter(parameter: KotlinParameterSpecSupplier) = builder { this.addParameter(parameter.get()) }

  internal fun addOriginatingElement(originatingElement: Element) = builder { this.addOriginatingElement(originatingElement) }
  fun jvmModifiers(modifiers: Iterable) = builder { this.jvmModifiers(modifiers) }
  fun addTypeVariables(typeVariables: Iterable) = builder { this.addTypeVariables(typeVariables) }
  fun addTypeVariable(typeVariable: TypeVariableName) = builder { this.addTypeVariable(typeVariable) }

  fun receiver(receiverType: TypeName) = builder { this.receiver(receiverType) }
  fun receiver(receiverType: TypeName, kdoc: CodeBlock) = builder { this.receiver(receiverType, kdoc) }

  fun receiver(receiverType: KClass<*>) = builder { this.receiver(receiverType) }
  fun receiver(receiverType: KClass<*>, kdoc: CodeBlock) = builder { this.receiver(receiverType, kdoc) }

  fun receiver(receiverType: KClass<*>, kdoc: String, vararg args: Any) = builder { this.receiver(receiverType, kdoc, *args) }

  fun returns(returnType: TypeName) = builder { this.returns(returnType) }
  fun returns(returnType: TypeName, kdoc: CodeBlock) = builder { this.returns(returnType, kdoc) }

  fun returns(returnType: KClass<*>) = builder { this.returns(returnType) }
  fun returns(returnType: KClass<*>, kdoc: CodeBlock) = builder { this.returns(returnType, kdoc) }

  fun returns(returnType: KClass<*>, kdoc: String, vararg args: Any) = builder { this.returns(returnType, kdoc, *args) }

  fun addParameters(parameterSpecs: Iterable) = builder { this.addParameters(parameterSpecs) }
  fun addParameter(parameterSpec: ParameterSpec) = builder { this.addParameter(parameterSpec) }
  fun callThisConstructor(args: List) = builder { this.callThisConstructor(args) }

  fun callThisConstructor(args: Iterable) = builder { this.callThisConstructor(args) }
  fun callThisConstructor(vararg args: String) = builder { this.callThisConstructor(*args) }
  fun callThisConstructor(vararg args: CodeBlock = emptyArray()) = builder { this.callThisConstructor(*args) }
  fun callSuperConstructor(args: Iterable) = builder { this.callSuperConstructor(args) }
  fun callSuperConstructor(args: List) = builder { this.callSuperConstructor(args) }
  fun callSuperConstructor(vararg args: String) = builder { this.callSuperConstructor(*args) }
  fun callSuperConstructor(vararg args: CodeBlock = emptyArray()) = builder { this.callSuperConstructor(*args) }
  fun addParameter(name: String, type: TypeName, vararg modifiers: KModifier) = builder { this.addParameter(name, type, *modifiers) }
  fun addParameter(name: String, type: KClass<*>, vararg modifiers: KModifier) = builder { this.addParameter(name, type, *modifiers) }
  fun addParameter(name: String, type: TypeName, modifiers: Iterable) = builder { this.addParameter(name, type, modifiers) }
  fun addParameter(name: String, type: KClass<*>, modifiers: Iterable) = builder { this.addParameter(name, type, modifiers) }
  fun addCode(format: String, vararg args: Any?) = builder { this.addCode(format, *args) }
  fun addNamedCode(format: String, args: Map) = builder { this.addNamedCode(format, args) }
  fun addCode(codeBlock: CodeBlock) = builder { this.addCode(codeBlock) }
  fun addComment(format: String, vararg args: Any) = builder { this.addComment(format, *args) }

  fun beginControlFlow(controlFlow: String, vararg args: Any) = builder { this.beginControlFlow(controlFlow, *args) }
  fun nextControlFlow(controlFlow: String, vararg args: Any) = builder { this.nextControlFlow(controlFlow, *args) }
  fun endControlFlow() = builder { this.endControlFlow() }
  fun addStatement(format: String, vararg args: Any) = builder { this.addStatement(format, *args) }

  override fun build(): KotlinFunSpec = KotlinFunSpec(spec = delegate.build())

  // region [overrides]
  override fun addAnnotation(spec: KotlinAnnotationSpecSupplier) = apply { delegate.addAnnotation(spec.get()) }
  override fun contextReceivers(vararg receiverTypes: TypeName) = builder { this.contextReceivers(*receiverTypes) }
  override fun addKdoc(kdoc: KDoc) = apply { delegate.addKdoc(kdoc.get()) }
  override fun addModifiers(vararg modifiers: KModifier) = builder { this.addModifiers(*modifiers) }
  override fun addTag(type: KClass<*>, tag: Any?) = builder { this.tag(type, tag) }
  override fun builder(block: FunSpecBuilderReceiver): KotlinFunSpecBuilder = apply { delegate.builder.block() }
  override fun get(): FunSpec = build().get()
  override fun spec(): KotlinFunSpec = build()
  // endregion [overrides]
}

@ExperimentalKotlinPoetApi
typealias KotlinFunSpecBuilderReceiver = KotlinFunSpecBuilder.() -> Unit