// これから
@GenerateUiState // 生成してほしいので
data class DetailScreen(
   val name: String,
   val description: String
)

// これを作りたい
sealed interface GenerateDetailScreen {

   data object Loading: GenerateDetailScreen

   data class Success(
      val name: String,
      val description: String
   ) : GenerateDetailScreen

   data class Error(val throwable: Throwable): GenerateDetailScreen
}

@Target(AnnotationTarget.CLASS)
annotation class GenerateUiState

dependencies {
   // 省略...

   implementation(project(":annotation")) // アノテーションを取り込む
   implementation("com.google.devtools.ksp:symbol-processing-api:2.3.4") // ksp も取り込む
}

class GenerateUiStateProvider : SymbolProcessorProvider {
    override fun create(environment: SymbolProcessorEnvironment): SymbolProcessor {
        return GenerateUiStateProcessor(
            options = environment.options,
            logger = environment.logger,
            codeGenerator = environment.codeGenerator
        )
    }
}

class GenerateUiStateProcessor(
    options: Map<String, String>,
    logger: KSPLogger,
    private val codeGenerator: CodeGenerator
) : SymbolProcessor {
    override fun process(resolver: Resolver): List<KSAnnotated> {
        // TODO このあとすぐ
    }
}

class GenerateUiStateProcessor(
    options: Map<String, String>,
    logger: KSPLogger,
    private val codeGenerator: CodeGenerator
) : SymbolProcessor {

    /** ファイルが二回作られるのを防ぐ */
    private var invoked = false

    override fun process(resolver: Resolver): List<KSAnnotated> {
        // アノテーションがついているクラスを探す
        val symbols = resolver
            .getSymbolsWithAnnotation(GenerateUiState::class.qualifiedName!!) // implementation(project(":annotation")) したのでアクセス可。名前をべた書きしてもよかったかも・・・
            .filterIsInstance<KSClassDeclaration>() // クラスに限定する、Function とかもあります

        if (!invoked) {
            invoked = true

            // ファイルを作る
            symbols.forEach { symbol ->

                // 自動生成するクラス名
                val generatedClassName = "Generated${symbol.qualifiedName!!.getShortName()}"

                // クラスの引数を val hoge: String にする
                val parameterKotlinCode = symbol
                    .getAllProperties()
                    .joinToString(separator = ", ") {
                        "val ${it.simpleName.getShortName()}: ${it.type.resolve().declaration.qualifiedName!!.asString()}"
                    }

                // ファイルを作る
                // ビルドすると IDEA / AndroidStudio から参照可能になる
                val file = codeGenerator.createNewFile(
                    dependencies = Dependencies(false, *resolver.getAllFiles().toList().toTypedArray()),
                    packageName = "io.github.takusan23.kspuistategenerator",
                    fileName = generatedClassName
                )

                // 自動生成する Kotlin コード
                val generatedKotlinCode = """
                package io.github.takusan23.kspuistategenerator

                sealed interface $generatedClassName {

                    data object Loading : $generatedClassName

                    data class Success(
                        $parameterKotlinCode
                    ) : $generatedClassName

                    data class Error(val throwable: Throwable) : $generatedClassName
                }
                """.trimIndent()

                // 書き込む
                file.write(generatedKotlinCode.toByteArray())
            }

        }

        val unableToProcess = symbols.filterNot { it.validate() }.toList()
        return unableToProcess
    }
}

// アノテーションがついているクラスを探す
val symbols = resolver
   .getSymbolsWithAnnotation(GenerateUiState::class.qualifiedName!!) // implementation(project(":annotation")) したのでアクセス可。名前をべた書きしてもよかったかも・・・
   .filterIsInstance<KSClassDeclaration>() // クラスに限定する、Function とかもあります

@GenerateUiState data class HomeState // この場合は
println(symbols.map { it.qualifiedName!!.asString() }.toList()) // [io.github.takusan23.io.github.takusan23.kspuistategenerator.example.HomeState] こうなる

val parameterKotlinCode = symbol
   .getAllProperties()
   .joinToString(separator = ", ") {
      "val ${it.simpleName.getShortName()}: ${it.type.resolve().declaration.qualifiedName!!.asString()}"
   }

// これを渡すと
@GenerateUiState
data class HomeState(
   val title: String, 
   val description: String
)

println(parameterKotlinCode) // こうなる→ val title: kotlin.String, val description: kotlin.String

// ファイルを作る
// ビルドすると IDEA / AndroidStudio から参照可能になる
val file = codeGenerator.createNewFile(
   dependencies = Dependencies(false, *resolver.getAllFiles().toList().toTypedArray()),
   packageName = "io.github.takusan23.kspuistategenerator",
   fileName = generatedClassName
)

// 自動生成する Kotlin コード
val generatedKotlinCode = """
package io.github.takusan23.kspuistategenerator

sealed interface $generatedClassName {

   data object Loading : $generatedClassName

   data class Success(
      $parameterKotlinCode
   ) : $generatedClassName

   data class Error(val throwable: Throwable) : $generatedClassName
}
""".trimIndent()

// 書き込む
file.write(generatedKotlinCode.toByteArray())

plugins {
    kotlin("jvm")
    id("com.google.devtools.ksp") version "2.3.3" // KSP 入れる
}

group = "io.github.takusan23.kspuistategenerator.example"
version = "1.0-SNAPSHOT"

repositories {
    mavenCentral()
}

dependencies {
    testImplementation(kotlin("test"))

    // アノテーションと ksp を入れる
    implementation(project(":annotation"))
    ksp(project(":processor"))
}

tasks.test {
    useJUnitPlatform()
}

@GenerateUiState
data class DetailScreen(
    val name: String,
    val description: String
)

fun main() {
    
}

Execution failed for task ':example:kspKotlin'.
> A failure occurred while executing com.google.devtools.ksp.gradle.KspAAWorkerAction
   > com/google/devtools/ksp/impl/KotlinSymbolProcessing$ExitCode

annotation class ClassArg(val clazz: KClass<*>) // KSP ではクラスが存在しないエラーになってしまう

class ExampleB

@ClassArg(ExampleB::class) // 引数に渡しても KSP からは見つけることができない
class ExampleA