improvement: Send enabled rules in initialize params

Author: tgodzik

protocol/src/main/kotlin/org/jetbrains/bsp/protocol/InitializeBuildData.kt

@@ -6,6 +6,7 @@ data class InitializeBuildData(
   val clientClassesRootDir: String? = null,
   val openTelemetryEndpoint: String? = null,
   val featureFlags: FeatureFlags? = null,
+  val enabledRules: List<String>? = null,
 )
 
 data class FeatureFlags(

server/server/src/main/kotlin/org/jetbrains/bsp/bazel/server/BazelBspServer.kt

@@ -96,6 +96,7 @@ class BazelBspServer(
         compilationManager = compilationManager,
         bspClientLogger = bspClientLogger,
         featureFlags = featureFlags,
+        initializeBuildData = initializeBuildData,
       )
     val bspProjectMapper =
       BspProjectMapper(
@@ -171,6 +172,7 @@ class BazelBspServer(
     compilationManager: BazelBspCompilationManager,
     bspClientLogger: BspClientLogger,
     featureFlags: FeatureFlags,
+    initializeBuildData: InitializeBuildData,
   ): ProjectProvider {
     val aspectsResolver =
       InternalAspectsResolver(
@@ -219,6 +221,7 @@ class BazelBspServer(
         bazelPathsResolver = bazelPathsResolver,
         bspClientLogger = bspClientLogger,
         featureFlags = featureFlags,
+        initializeBuildData = initializeBuildData,
       )
     val firstPhaseProjectResolver =
       FirstPhaseProjectResolver(

server/server/src/main/kotlin/org/jetbrains/bsp/bazel/server/bsp/managers/BazelExternalRulesetsQuery.kt

@@ -21,20 +21,19 @@ interface BazelExternalRulesetsQuery {
   fun fetchExternalRulesetNames(cancelChecker: CancelChecker): List<String>
 }
 
-class BazelEnabledRulesetsQueryImpl(private val enabledRulesSpec: EnabledRulesSpec) : BazelExternalRulesetsQuery {
+class BazelEnabledRulesetsQueryImpl(private val enabledRules: List<String>) : BazelExternalRulesetsQuery {
   override fun fetchExternalRulesetNames(cancelChecker: CancelChecker): List<String> {
-    val specifiedRules = enabledRulesSpec.values
-    val neededTransitiveRules = specifiedRules.mapNotNull { rootRulesToNeededTransitiveRules[it] }.flatten()
+    val neededTransitiveRules = enabledRules.mapNotNull { rootRulesToNeededTransitiveRules[it] }.flatten()
 
-    return (specifiedRules + neededTransitiveRules).distinct()
+    return (enabledRules + neededTransitiveRules).distinct()
   }
 }
 
 class BazelExternalRulesetsQueryImpl(
   private val bazelRunner: BazelRunner,
   private val isBzlModEnabled: Boolean,
   private val isWorkspaceEnabled: Boolean,
-  private val enabledRules: EnabledRulesSpec,
+  private val enabledRules: List<String>,
   private val bspClientLogger: BspClientLogger,
   private val repoMapping: RepoMapping,
 ) : BazelExternalRulesetsQuery {

server/server/src/main/kotlin/org/jetbrains/bsp/bazel/server/sync/ProjectResolver.kt

@@ -26,6 +26,7 @@ import org.jetbrains.bsp.bazel.workspacecontext.TargetsSpec
 import org.jetbrains.bsp.bazel.workspacecontext.WorkspaceContext
 import org.jetbrains.bsp.bazel.workspacecontext.WorkspaceContextProvider
 import org.jetbrains.bsp.protocol.FeatureFlags
+import org.jetbrains.bsp.protocol.InitializeBuildData
 
 /** Responsible for querying bazel and constructing Project instance  */
 class ProjectResolver(
@@ -41,6 +42,7 @@ class ProjectResolver(
   private val bazelPathsResolver: BazelPathsResolver,
   private val bspClientLogger: BspClientLogger,
   private val featureFlags: FeatureFlags,
+  private val initializeBuildData: InitializeBuildData,
 ) {
   private suspend fun <T> measured(description: String, f: suspend () -> T): T = tracer.spanBuilder(description).useWithScope { f() }
 
@@ -60,13 +62,13 @@ class ProjectResolver(
         measured("Calculating external repository mapping") {
           calculateRepoMapping(workspaceContext, bazelRunner, bazelInfo, bspClientLogger)
         }
-
+      val enabledRules = initializeBuildData.enabledRules ?: workspaceContext.enabledRules.values
       val bazelExternalRulesetsQuery =
         BazelExternalRulesetsQueryImpl(
           bazelRunner,
           bazelInfo.isBzlModEnabled,
           bazelInfo.isWorkspaceEnabled,
-          workspaceContext.enabledRules,
+          enabledRules,
           bspClientLogger,
           repoMapping,
         )