Fix build

Author: DRSchlaubi

build.gradle.kts

@@ -5,7 +5,7 @@ plugins {
 
 allprojects {
     group = "dev.kord.codegen"
-    version = "1.0.1"
+    version = "1.0.2"
 
     repositories {
         mavenCentral()

gradle/libs.versions.toml

@@ -11,9 +11,9 @@ ksp-api = { group = "com.google.devtools.ksp", name = "symbol-processing-api", v
 kotlinpoet = { group = "com.squareup", name = "kotlinpoet-jvm", version.ref = "kotlinpoet" }
 kotlinpoet-ksp = { group = "com.squareup", name = "kotlinpoet-ksp", version.ref = "kotlinpoet" }
 
-codegen-kotlinpoet = { group = "dev.kord.codegen", name = "kotlinpoet", version = "main-SNAPSHOT" }
-codegen-ksp = { group = "dev.kord.codegen", name = "ksp", version = "main-SNAPSHOT" }
-codegen-ksp-processor = { group = "dev.kord.codegen", name = "ksp-processor", version = "main-SNAPSHOT" }
+codegen-kotlinpoet = { group = "dev.kord.codegen", name = "kotlinpoet", version.ref = "codegen-kt" }
+codegen-ksp = { group = "dev.kord.codegen", name = "ksp", version.ref = "codegen-kt" }
+codegen-ksp-processor = { group = "dev.kord.codegen", name = "ksp-processor", version.ref = "codegen-kt" }
 
 mockk = { group = "io.mockk", name = "mockk", version = "1.14.5" }
 

ksp/src/main/kotlin/annotations/AnnotationArguments.kt

@@ -2,6 +2,7 @@ package dev.kord.codegen.ksp.annotations
 
 import com.google.devtools.ksp.isDefault
 import com.google.devtools.ksp.symbol.KSAnnotation
+import com.google.devtools.ksp.symbol.KSClassDeclaration
 import com.google.devtools.ksp.symbol.KSType
 import com.google.devtools.ksp.symbol.KSValueArgument
 import kotlin.reflect.KClass
@@ -74,11 +75,20 @@ public class AnnotationArguments<A : Annotation> private constructor(
         parameter.value as KSType?
 
     @JvmName("getEnum")
-    public inline operator fun <reified T : Enum<T>> get(parameter: KProperty1<A, T>): T? =
-        (parameter.value as KSType?)?.let {
-            enumValueOf<T>(it.declaration.simpleName.asString())
+    public inline operator fun <reified T : Enum<T>> get(parameter: KProperty1<A, T>): T? {
+        val constant = parameter.value ?: return null
+
+        val name = when (constant) {
+            // KSP V1
+            is KSType -> constant.declaration.simpleName.asString()
+            // KSP V2
+            is KSClassDeclaration -> constant.simpleName.asString()
+            else -> error("Unexpected type of $constant")
         }
 
+        return enumValueOf<T>(name)
+    }
+
     /**
      * Returns the value of [parameter] as a [List] of [Numbers][Number] or `null` if it is the default value
      * (see [NonNullAnnotationArguments] for more information).
@@ -202,7 +212,7 @@ public class AnnotationArguments<A : Annotation> private constructor(
 
         @JvmName("getEnumArray")
         public inline operator fun <reified T : Enum<T>> get(parameter: KProperty1<A, Array<out T>>): List<T>? =
-            delegate[parameter]!!
+            delegate[parameter]
 
         public companion object {
             /**