builder.KotlinPropertySpecBuilder.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.builder

import com.squareup.kotlinpoet.*
import io.toolisticon.kotlin.generation.BuilderSupplier
import io.toolisticon.kotlin.generation.PropertyName
import io.toolisticon.kotlin.generation.poet.*
import io.toolisticon.kotlin.generation.poet.PropertySpecBuilder.Companion.wrap
import io.toolisticon.kotlin.generation.spec.KotlinAnnotationSpecSupplier
import io.toolisticon.kotlin.generation.spec.KotlinPropertySpec
import io.toolisticon.kotlin.generation.spec.KotlinPropertySpecSupplier
import java.lang.reflect.Type
import javax.lang.model.element.Element
import kotlin.reflect.KClass

/**
 * Builder for [KotlinPropertySpec].
 */
@ExperimentalKotlinPoetApi
class KotlinPropertySpecBuilder internal constructor(
  private val delegate: PropertySpecBuilder
) : BuilderSupplier,
  DelegatingBuilder,
  KotlinAnnotatableDocumentableModifiableBuilder,
  KotlinContextReceivableBuilder,
  KotlinPropertySpecSupplier {

  companion object {

    /**
     * Creates new builder.
     */
    fun builder(name: PropertyName, type: TypeName, vararg modifiers: KModifier): KotlinPropertySpecBuilder = KotlinPropertySpecBuilder(
      delegate = PropertySpecBuilder.builder(name, type, *modifiers)
    )

    /**
     * Creates new builder.
     */
    fun builder(name: PropertyName, type: Type, vararg modifiers: KModifier): KotlinPropertySpecBuilder = builder(
      name = name,
      type = type.asTypeName(),
      modifiers = modifiers
    )

    /**
     * Creates new builder.
     */
    fun builder(name: PropertyName, type: KClass<*>, vararg modifiers: KModifier): KotlinPropertySpecBuilder = builder(name, type.asTypeName(), *modifiers)

    /**
     * Creates new builder.
     */
    fun builder(name: PropertyName, type: TypeName, modifiers: Iterable): KotlinPropertySpecBuilder = KotlinPropertySpecBuilder(
      delegate = PropertySpecBuilder.builder(name, type, modifiers)
    )

    /**
     * Creates new builder.
     */
    fun builder(name: PropertyName, type: KClass<*>, modifiers: Iterable): KotlinPropertySpecBuilder = KotlinPropertySpecBuilder(
      delegate = PropertySpecBuilder.builder(name, type.asTypeName(), modifiers)
    )

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

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


  internal fun addOriginatingElement(originatingElement: Element) = builder { this.addOriginatingElement(originatingElement) }

  fun mutable(mutable: Boolean = true) = builder { this.mutable(mutable) }

  fun addTypeVariables(typeVariables: Iterable) = builder { this.addTypeVariables(typeVariables) }
  fun addTypeVariable(typeVariable: TypeVariableName) = builder { this.addTypeVariable(typeVariable) }
  fun initializer(format: String, vararg args: Any?) = builder { this.initializer(format, *args) }
  fun initializer(codeBlock: CodeBlock?) = builder { this.initializer(codeBlock) }
  fun delegate(format: String, vararg args: Any?) = builder { this.delegate(format, *args) }
  fun delegate(codeBlock: CodeBlock) = builder { this.delegate(codeBlock) }
  fun getter(getter: FunSpecSupplier?) = builder { this.getter(getter?.get()) }
  fun setter(setter: FunSpecSupplier?) = builder { this.setter(setter?.get()) }
  fun receiver(receiverType: TypeName?) = builder { this.receiver(receiverType) }
  fun receiver(receiverType: KClass<*>) = builder { this.receiver(receiverType) }

  override fun build(): KotlinPropertySpec = KotlinPropertySpec(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): KotlinPropertySpecBuilder = 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: PropertySpecBuilderReceiver) = apply { delegate.builder.block() }
  override fun get(): PropertySpec = build().get()
  override fun spec(): KotlinPropertySpec = build()
  // endregion [overrides]
}

@ExperimentalKotlinPoetApi
typealias KotlinPropertySpecBuilderReceiver = KotlinPropertySpecBuilder.() -> Unit