spec.KotlinDataClassSpec.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.spec

import com.squareup.kotlinpoet.ClassName
import com.squareup.kotlinpoet.ExperimentalKotlinPoetApi
import com.squareup.kotlinpoet.TypeSpec
import io.toolisticon.kotlin.generation.KotlinCodeGeneration.typeSpec.isDataClass
import io.toolisticon.kotlin.generation.poet.KDoc
import kotlin.reflect.KClass

/**
 * Represents a data class.
 */
@ExperimentalKotlinPoetApi
data class KotlinDataClassSpec(
  override val className: ClassName,
  private val spec: TypeSpec
) : KotlinGeneratorTypeSpec, KotlinDataClassSpecSupplier, KotlinDocumentableSpec {

  init {
    require(spec.isDataClass) { "Not a dataClass spec: $spec." }
  }

  override fun  tag(type: KClass): T? = get().tag(type)
  override val kdoc: KDoc get() = KDoc(spec.kdoc)

  override fun spec(): KotlinDataClassSpec = this
  override fun get(): TypeSpec = spec
}

/**
 * Marks the builder and the spec so they are interchangeable.
 */
@ExperimentalKotlinPoetApi
interface KotlinDataClassSpecSupplier : KotlinGeneratorSpecSupplier, ToFileTypeSpecSupplier {
  override fun get(): TypeSpec = spec().get()
}