support.ThrowsAnnotation.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.support

import com.squareup.kotlinpoet.CodeBlock
import com.squareup.kotlinpoet.ExperimentalKotlinPoetApi
import com.squareup.kotlinpoet.TypeName
import com.squareup.kotlinpoet.asTypeName
import io.toolisticon.kotlin.generation.KotlinCodeGeneration.buildAnnotation
import io.toolisticon.kotlin.generation.KotlinCodeGeneration.format.FORMAT_KCLASS
import io.toolisticon.kotlin.generation.spec.KotlinAnnotationSpec
import io.toolisticon.kotlin.generation.spec.KotlinAnnotationSpecSupplier
import io.toolisticon.kotlin.generation.support.CodeBlockArray.Companion.Format
import kotlin.collections.fold
import kotlin.reflect.KClass

/**
 * Creates a `@Throws` annotation.
 */
@ExperimentalKotlinPoetApi
data class ThrowsAnnotation(private val members: CodeBlockArray) : KotlinAnnotationSpecSupplier {
  companion object {
    val EXCEPTION = Throws::class
    val EMPTY = CodeBlockArray.codeBlockArray(Format(format = "", prefix = "", suffix = ""))
    private val TypeName.codeBlock: CodeBlock get() = CodeBlock.of(FORMAT_KCLASS, this)
  }

  /**
   * Create new instance.
   */
  constructor(exceptions: List) : this(members = exceptions.fold(EMPTY) { acc, cur -> acc + cur.codeBlock })

  /**
   * Create new instance.
   */
  constructor(exception: TypeName, vararg exceptions: TypeName) : this(listOf(exception, *exceptions))

  /**
   * Create new instance.
   */
  constructor(exception: KClass, vararg exceptions: KClass) : this(listOf(exception, *exceptions).map { it.asTypeName() })

  override fun spec(): KotlinAnnotationSpec = buildAnnotation(EXCEPTION) {
    addMember(members.build())
  }
}