KSPGeneratedSources.kt

private fun KotlinCompilation.generatedSourceFor(fileName: String): String {
    return kspSourcesDir.walkTopDown()
        .firstOrNull { it.name == fileName }
        ?.readText()
        ?: throw IllegalArgumentException(
            "Unable to find $fileName in ${
                kspSourcesDir.walkTopDown().filter { it.isFile }.toList()
            }"
        )
}

private val KotlinCompilation.kspWorkingDir: File
    get() = workingDir.resolve("ksp")

private val KotlinCompilation.kspSourcesDir: File
    get() = kspWorkingDir.resolve("sources")