spec.KotlinEnumClassSpec.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.poet.KDoc
import kotlin.reflect.KClass

/**
 * Represents an enum class.
 */
@ExperimentalKotlinPoetApi
data class KotlinEnumClassSpec(
  override val className: ClassName,
  private val spec: TypeSpec,
) : KotlinGeneratorTypeSpec, KotlinEnumClassSpecSupplier, KotlinDocumentableSpec {

  init {
    require(spec.isEnum) { "Not an enum spec: $spec" }
  }

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

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