ksp: support ksp options

Author: xinzhengzhang

kotlin/internal/defs.bzl

@@ -63,5 +63,6 @@ KtCompilerPluginInfo = provider(
 KspPluginInfo = provider(
     fields = {
         "plugins": "List of JavaPLuginInfo providers for the plugins to run with KSP",
+        "options": "List of plugin options, represented as structs with key=value, to be passed to the compiler",
     },
 )

kotlin/internal/jvm/compile.bzl

@@ -317,6 +317,7 @@ def _run_kapt_builder_actions(
         compile_deps = compile_deps,
         deps_artifacts = deps_artifacts,
         annotation_processors = annotation_processors,
+        ksp_options = [],
         transitive_runtime_jars = transitive_runtime_jars,
         plugins = plugins,
         outputs = {
@@ -343,6 +344,7 @@ def _run_ksp_builder_actions(
         compile_deps,
         deps_artifacts,
         annotation_processors,
+        options,
         transitive_runtime_jars,
         plugins):
     """Runs KSP using the KotlinBuilder tool
@@ -362,6 +364,7 @@ def _run_ksp_builder_actions(
         compile_deps = compile_deps,
         deps_artifacts = deps_artifacts,
         annotation_processors = annotation_processors,
+        ksp_options = options,
         transitive_runtime_jars = transitive_runtime_jars,
         plugins = plugins,
         outputs = {
@@ -383,6 +386,7 @@ def _run_kt_builder_action(
         compile_deps,
         deps_artifacts,
         annotation_processors,
+        ksp_options,
         transitive_runtime_jars,
         plugins,
         outputs,
@@ -408,6 +412,7 @@ def _run_kt_builder_action(
     args.add_all("--deps_artifacts", deps_artifacts, omit_if_empty = True)
     args.add_all("--kotlin_friend_paths", associates.jars, map_each = _associate_utils.flatten_jars)
     args.add("--instrument_coverage", ctx.coverage_instrumented())
+    args.add_all("--ksp_options", ksp_options, omit_if_empty = True)
 
     # Collect and prepare plugin descriptor for the worker.
     args.add_all(
@@ -535,6 +540,8 @@ def kt_jvm_produce_jar_actions(ctx, rule_kind):
     )
     annotation_processors = _plugin_mappers.targets_to_annotation_processors(ctx.attr.plugins + ctx.attr.deps)
     ksp_annotation_processors = _plugin_mappers.targets_to_ksp_annotation_processors(ctx.attr.plugins + ctx.attr.deps)
+    ksp_options = _plugin_mappers.targets_to_ksp_options(ctx.attr.plugins + ctx.attr.deps).to_list()
+
     transitive_runtime_jars = _plugin_mappers.targets_to_transitive_runtime_jars(ctx.attr.plugins + ctx.attr.deps)
     plugins = ctx.attr.plugins + _exported_plugins(deps = ctx.attr.deps)
     deps_artifacts = _deps_artifacts(toolchains, ctx.attr.deps + associates.targets)
@@ -558,6 +565,7 @@ def kt_jvm_produce_jar_actions(ctx, rule_kind):
         deps_artifacts = deps_artifacts,
         annotation_processors = annotation_processors,
         ksp_annotation_processors = ksp_annotation_processors,
+        ksp_options = ksp_options,
         transitive_runtime_jars = transitive_runtime_jars,
         plugins = plugins,
         compile_jar = compile_jar,
@@ -650,6 +658,7 @@ def _run_kt_java_builder_actions(
         deps_artifacts,
         annotation_processors,
         ksp_annotation_processors,
+        ksp_options,
         transitive_runtime_jars,
         plugins,
         compile_jar,
@@ -699,6 +708,7 @@ def _run_kt_java_builder_actions(
             compile_deps = compile_deps,
             deps_artifacts = deps_artifacts,
             annotation_processors = ksp_annotation_processors,
+            options = ksp_options,
             transitive_runtime_jars = transitive_runtime_jars,
             plugins = plugins,
         )
@@ -736,6 +746,7 @@ def _run_kt_java_builder_actions(
             compile_deps = compile_deps,
             deps_artifacts = deps_artifacts,
             annotation_processors = [],
+            ksp_options = [],
             transitive_runtime_jars = transitive_runtime_jars,
             plugins = plugins,
             outputs = outputs,

kotlin/internal/jvm/impl.bzl

@@ -437,5 +437,5 @@ def kt_ksp_plugin_impl(ctx):
                 # processors out of the public ABI.
                 generates_api = True,
             ),
-        ]),
+        ], options = ctx.attr.options),
     ]

kotlin/internal/jvm/jvm.bzl

@@ -597,6 +597,9 @@ kt_jvm_library(
             doc = " The fully qualified class name that the Java compiler uses as an entry point to the annotation processor.",
             mandatory = True,
         ),
+        "options": attr.string_list(
+            doc = "Options to be passed to the annotation processor. The syntax is key=value",
+        ),
     },
     implementation = _kt_ksp_plugin_impl,
     provides = [_KspPluginInfo],

kotlin/internal/jvm/plugins.bzl

@@ -49,6 +49,13 @@ def _targets_to_ksp_annotation_processors(targets):
                 plugins.append(plugin.plugins)
     return depset(plugins)
 
+def _targets_to_ksp_options(targets):
+    options = []
+    for t in targets:
+        if _KspPluginInfo in t:
+            options.extend(t[_KspPluginInfo].options)
+    return depset(options)
+
 def _targets_to_annotation_processors_java_plugin_info(targets):
     return [t[JavaPluginInfo] for t in targets if JavaPluginInfo in t]
 
@@ -66,6 +73,7 @@ def _targets_to_transitive_runtime_jars(targets):
 mappers = struct(
     targets_to_annotation_processors = _targets_to_annotation_processors,
     targets_to_ksp_annotation_processors = _targets_to_ksp_annotation_processors,
+    targets_to_ksp_options = _targets_to_ksp_options,
     targets_to_annotation_processors_java_plugin_info = _targets_to_annotation_processors_java_plugin_info,
     targets_to_transitive_runtime_jars = _targets_to_transitive_runtime_jars,
     kt_plugin_to_processor = _kt_plugin_to_processor,

src/main/kotlin/io/bazel/kotlin/builder/tasks/KotlinBuilder.kt

@@ -87,6 +87,7 @@ class KotlinBuilder @Inject internal constructor(
       REDUCED_CLASSPATH_MODE("--reduced_classpath_mode"),
       INSTRUMENT_COVERAGE("--instrument_coverage"),
       KSP_GENERATED_JAVA_SRCJAR("--ksp_generated_java_srcjar"),
+      KSP_OPTIONS("--ksp_options"),
     }
   }
 
@@ -295,6 +296,9 @@ class KotlinBuilder @Inject internal constructor(
         addAllCompilerPluginClasspath(
           argMap.optional(KotlinBuilderFlags.COMPILER_PLUGIN_CLASS_PATH) ?: emptyList(),
         )
+        addAllKspOptions(
+          argMap.optional(KotlinBuilderFlags.KSP_OPTIONS) ?: emptyList(),
+        )
 
         argMap.optional(KotlinBuilderFlags.SOURCES)
           ?.iterator()

src/main/kotlin/io/bazel/kotlin/builder/tasks/jvm/CompilationTask.kt

@@ -208,6 +208,10 @@ internal fun JvmCompilationTask.kspArgs(
           flag(pair.first, value)
         }
       }
+
+      inputs.kspOptionsList.forEach { option ->
+        flag("apoption", option)
+      }
     }
   }
 }
@@ -281,6 +285,9 @@ private fun JvmCompilationTask.runKspPlugin(
       .flag("-d", directories.generatedClasses)
       .values(inputs.kotlinSourcesList)
       .values(inputs.javaSourcesList).list().let { args ->
+        context.whenTracing {
+          printLines("Ksp args:", args)
+        }
         context.executeCompilerTask(
           args,
           compiler::compile,

src/main/protobuf/kotlin_model.proto

@@ -158,6 +158,8 @@ message JvmCompilationTask {
       repeated string javac_flags = 15;
       // JDeps dependency artifacts
       repeated string deps_artifacts = 16;
+      // Kotlin symbol processor  options
+      repeated string ksp_options = 17;
   }
 
   CompilationTaskInfo info = 1;