extract KspOptions into common-util module

1 files changed, 264 insertions, 0 deletions

diff --git a/common-util/src/main/kotlin/com/google/devtools/ksp/KspOptions.kt b/common-util/src/main/kotlin/com/google/devtools/ksp/KspOptions.kt
new file mode 100644
index 00000000..907e650f
--- /dev/null
+++ b/common-util/src/main/kotlin/com/google/devtools/ksp/KspOptions.kt
@@ -0,0 +1,264 @@
+package com.google.devtools.ksp
+
+import org.jetbrains.kotlin.compiler.plugin.AbstractCliOption
+import org.jetbrains.kotlin.config.ApiVersion
+import org.jetbrains.kotlin.config.LanguageVersion
+import org.jetbrains.kotlin.config.toKotlinVersion
+import java.io.File
+
+class KspOptions(
+    val projectBaseDir: File,
+    val compileClasspath: List<File>,
+    val javaSourceRoots: List<File>,
+
+    val classOutputDir: File,
+    val javaOutputDir: File,
+    val kotlinOutputDir: File,
+    val resourceOutputDir: File,
+
+    val processingClasspath: List<File>,
+    val processors: List<String>,
+
+    val processingOptions: Map<String, String>,
+
+    val knownModified: List<File>,
+    val knownRemoved: List<File>,
+
+    val cachesDir: File,
+    val kspOutputDir: File,
+    val incremental: Boolean,
+    val incrementalLog: Boolean,
+    val allWarningsAsErrors: Boolean,
+    val withCompilation: Boolean,
+    val returnOkOnError: Boolean,
+    val changedClasses: List<String>,
+
+    val languageVersion: KotlinVersion,
+    val apiVersion: KotlinVersion,
+    val compilerVersion: KotlinVersion,
+) {
+    class Builder {
+        var projectBaseDir: File? = null
+        val compileClasspath: MutableList<File> = mutableListOf()
+        val javaSourceRoots: MutableList<File> = mutableListOf()
+
+        var classOutputDir: File? = null
+        var javaOutputDir: File? = null
+        var kotlinOutputDir: File? = null
+        var resourceOutputDir: File? = null
+
+        val processingClasspath: MutableList<File> = mutableListOf()
+        val processors: MutableList<String> = mutableListOf()
+
+        val processingOptions: MutableMap<String, String> = mutableMapOf()
+
+        val knownModified: MutableList<File> = mutableListOf()
+        val knownRemoved: MutableList<File> = mutableListOf()
+
+        var cachesDir: File? = null
+        var kspOutputDir: File? = null
+        var incremental: Boolean = false
+        var incrementalLog: Boolean = false
+        var allWarningsAsErrors: Boolean = false
+        var withCompilation: Boolean = false
+        // Default is false. It can be turned on to workaround KT-30172.
+        var returnOkOnError: Boolean = false
+        var changedClasses: MutableList<String> = mutableListOf()
+
+        var languageVersion: KotlinVersion = LanguageVersion.LATEST_STABLE.toKotlinVersion()
+        var apiVersion: KotlinVersion = ApiVersion.LATEST_STABLE.toKotlinVersion()
+        var compilerVersion: KotlinVersion = KotlinVersion.CURRENT
+
+        fun build(): KspOptions {
+            return KspOptions(
+                projectBaseDir!!, compileClasspath, javaSourceRoots,
+                classOutputDir!!,
+                javaOutputDir!!,
+                kotlinOutputDir!!,
+                resourceOutputDir!!,
+                processingClasspath, processors, processingOptions,
+                knownModified, knownRemoved, cachesDir!!, kspOutputDir!!, incremental, incrementalLog,
+                allWarningsAsErrors, withCompilation, returnOkOnError, changedClasses,
+                languageVersion, apiVersion, compilerVersion
+            )
+        }
+    }
+}
+
+fun String?.toKotlinVersion(): KotlinVersion {
+    if (this == null)
+        return KotlinVersion.CURRENT
+
+    return split('-').first().split('.').map { it.toInt() }.let {
+        when (it.size) {
+            1 -> KotlinVersion(it[0], 0, 0)
+            2 -> KotlinVersion(it[0], it[1], 0)
+            3 -> KotlinVersion(it[0], it[1], it[2])
+            else -> KotlinVersion.CURRENT
+        }
+    }
+}
+
+fun ApiVersion.toKotlinVersion(): KotlinVersion = version.canonical.toKotlinVersion()
+
+enum class KspCliOption(
+    override val optionName: String,
+    override val valueDescription: String,
+    override val description: String,
+    override val required: Boolean = false,
+    override val allowMultipleOccurrences: Boolean = false
+) : AbstractCliOption {
+    CLASS_OUTPUT_DIR_OPTION(
+        "classOutputDir",
+        "<classOutputDir>",
+        "Dir of generated classes",
+        false
+    ),
+
+    JAVA_OUTPUT_DIR_OPTION(
+        "javaOutputDir",
+        "<javaOutputDir>",
+        "Dir of generated Java sources",
+        false
+    ),
+
+    KOTLIN_OUTPUT_DIR_OPTION(
+        "kotlinOutputDir",
+        "<kotlinOutputDir>",
+        "Dir of generated Kotlin sources",
+        false
+    ),
+
+    RESOURCE_OUTPUT_DIR_OPTION(
+        "resourceOutputDir",
+        "<resourceOutputDir>",
+        "Dir of generated resources",
+        false
+    ),
+
+    CACHES_DIR_OPTION(
+        "cachesDir",
+        "<cachesDir>",
+        "Dir of caches",
+        false
+    ),
+
+    PROJECT_BASE_DIR_OPTION(
+        "projectBaseDir",
+        "<projectBaseDir>",
+        "path to gradle project",
+        false
+    ),
+
+    KSP_OUTPUT_DIR_OPTION(
+        "kspOutputDir",
+        "<kspOutputDir>",
+        "root of ksp output dirs",
+        false
+    ),
+
+    PROCESSING_OPTIONS_OPTION(
+        "apoption",
+        "<apOption>",
+        "processor defined option",
+        false,
+        true
+    ),
+
+    PROCESSOR_CLASSPATH_OPTION(
+        "apclasspath",
+        "<classpath>",
+        "processor classpath",
+        false
+    ),
+
+    KNOWN_MODIFIED_OPTION(
+        "knownModified",
+        "<knownModified>",
+        "known modified files",
+        false,
+        false
+    ),
+
+    KNOWN_REMOVED_OPTION(
+        "knownRemoved",
+        "<knownRemoved>",
+        "known removed fiels",
+        false,
+        false
+    ),
+
+    INCREMENTAL_OPTION(
+        "incremental",
+        "<incremental>",
+        "processing incrementally",
+        false,
+        false
+    ),
+
+    INCREMENTAL_LOG_OPTION(
+        "incrementalLog",
+        "<incrementalLog>",
+        "log dirty files",
+        false,
+        false
+    ),
+
+    ALL_WARNINGS_AS_ERRORS_OPTION(
+        "allWarningsAsErrors",
+        "<allWarningsAsErrors>",
+        "treat all warnings as errors",
+        false,
+        false
+    ),
+
+    WITH_COMPILATION_OPTION(
+        "withCompilation",
+        "<withCompilation>",
+        "Run processors and compilation in a single compiler invocation",
+        false,
+        false
+    ),
+
+    CHANGED_CLASSES_OPTION(
+        "changedClasses",
+        "<changedClasses>",
+        "canonical / dot-separated names of dirty classes in classpath",
+        false,
+        false
+    ),
+
+    RETURN_OK_ON_ERROR_OPTION(
+        "returnOkOnError",
+        "<returnOkOnError>",
+        "Return OK even if there are errors",
+        false,
+        false
+    ),
+}
+
+@Suppress("IMPLICIT_CAST_TO_ANY")
+fun KspOptions.Builder.processOption(option: KspCliOption, value: String) = when (option) {
+    KspCliOption.PROCESSOR_CLASSPATH_OPTION -> processingClasspath += value.split(File.pathSeparator).map {
+        File(it)
+    }
+    KspCliOption.CLASS_OUTPUT_DIR_OPTION -> classOutputDir = File(value)
+    KspCliOption.JAVA_OUTPUT_DIR_OPTION -> javaOutputDir = File(value)
+    KspCliOption.KOTLIN_OUTPUT_DIR_OPTION -> kotlinOutputDir = File(value)
+    KspCliOption.RESOURCE_OUTPUT_DIR_OPTION -> resourceOutputDir = File(value)
+    KspCliOption.CACHES_DIR_OPTION -> cachesDir = File(value)
+    KspCliOption.KSP_OUTPUT_DIR_OPTION -> kspOutputDir = File(value)
+    KspCliOption.PROJECT_BASE_DIR_OPTION -> projectBaseDir = File(value)
+    KspCliOption.PROCESSING_OPTIONS_OPTION -> {
+        val (k, v) = value.split('=', ignoreCase = false, limit = 2)
+        processingOptions.put(k, v)
+    }
+    KspCliOption.KNOWN_MODIFIED_OPTION -> knownModified.addAll(value.split(File.pathSeparator).map { File(it) })
+    KspCliOption.KNOWN_REMOVED_OPTION -> knownRemoved.addAll(value.split(File.pathSeparator).map { File(it) })
+    KspCliOption.INCREMENTAL_OPTION -> incremental = value.toBoolean()
+    KspCliOption.INCREMENTAL_LOG_OPTION -> incrementalLog = value.toBoolean()
+    KspCliOption.ALL_WARNINGS_AS_ERRORS_OPTION -> allWarningsAsErrors = value.toBoolean()
+    KspCliOption.WITH_COMPILATION_OPTION -> withCompilation = value.toBoolean()
+    KspCliOption.CHANGED_CLASSES_OPTION -> changedClasses.addAll(value.split(':'))
+    KspCliOption.RETURN_OK_ON_ERROR_OPTION -> returnOkOnError = value.toBoolean()
+}