[feature] Go support

aspects/rules/go/go_info.bzl

@@ -0,0 +1,19 @@
+load("//aspects:utils/utils.bzl", "create_proto", "create_struct", "file_location")
+load("@io_bazel_rules_go//go:def.bzl", "go_context")
+
+def extract_sdk(ctx):
+    go = go_context(ctx)
+    if go == None:
+        return None
+    return file_location(go.sdk.go)
+
+def extract_go_info(target, ctx, **kwargs):
+    importpath = getattr(ctx.rule.attr, "importpath", [])
+    sdk_home_path = extract_sdk(ctx)
+
+    go_target_info = create_struct(
+        importpath = importpath,
+        sdk_home_path = sdk_home_path,
+    )
+
+    return create_proto(target, ctx, go_target_info, "go_target_info"), None

docs/dev/ASPECTS.md

@@ -4,6 +4,6 @@
 
 ### The `extensions.bzl` file
 The server generates the `extensions.bzl` file on each sync according to a list of external rules relevant to the project.
-See `server/src/main/kotlin/org/jetbrains/bsp/bazel/server/bsp/managers/BazelBspEnvironmentManager.kt` for the implementation.
+See `server/src/main/kotlin/org/jetbrains/bsp/bazel/server/bsp/managers/BazelBspLanguageExtensionsGenerator.kt` for the implementation.
 
 The file itself contains a declaration of functions invoked in `core.bzl`. Each function is supposed to provide language-specific data.

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

@@ -0,0 +1,9 @@
+package org.jetbrains.bsp
+
+import java.net.URI
+
+data class GoBuildTarget(
+  val sdkHomePath: URI?,
+  val importPath: String?,
+)
+

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

@@ -35,6 +35,7 @@ import org.jetbrains.bsp.bazel.server.sync.TargetKindResolver
 import org.jetbrains.bsp.bazel.server.sync.languages.LanguagePluginsService
 import org.jetbrains.bsp.bazel.server.sync.languages.android.AndroidLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.cpp.CppLanguagePlugin
+import org.jetbrains.bsp.bazel.server.sync.languages.go.GoLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JavaLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JdkResolver
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JdkVersionResolver
@@ -120,6 +121,7 @@ class BazelBspServer(
     val pythonLanguagePlugin = PythonLanguagePlugin(bazelPathsResolver)
     val rustLanguagePlugin = RustLanguagePlugin(bazelPathsResolver)
     val androidLanguagePlugin = AndroidLanguagePlugin(javaLanguagePlugin, bazelPathsResolver)
+    val goLanguagePlugin = GoLanguagePlugin(bazelPathsResolver)
 
     return LanguagePluginsService(
       scalaLanguagePlugin,
@@ -129,7 +131,8 @@ class BazelBspServer(
       thriftLanguagePlugin,
       pythonLanguagePlugin,
       rustLanguagePlugin,
-      androidLanguagePlugin
+      androidLanguagePlugin,
+      goLanguagePlugin
     )
   }
 

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

@@ -1,12 +1,12 @@
 package org.jetbrains.bsp.bazel.server.bsp.managers
 
+import java.nio.file.Paths
+import java.util.Properties
+import kotlin.io.path.writeText
 import org.apache.velocity.app.VelocityEngine
 import org.jetbrains.bsp.bazel.bazelrunner.BazelRelease
 import org.jetbrains.bsp.bazel.commons.Constants
 import org.jetbrains.bsp.bazel.server.bsp.utils.InternalAspectsResolver
-import java.nio.file.Paths
-import java.util.Properties
-import kotlin.io.path.writeText
 
 enum class Language(private val fileName: String, val ruleNames: List<String>, val functions: List<String>, val isTemplate: Boolean) {
   Java("//aspects:rules/java/java_info.bzl", listOf(), listOf("extract_java_toolchain", "extract_java_runtime"), false),
@@ -16,7 +16,8 @@ enum class Language(private val fileName: String, val ruleNames: List<String>, v
   Cpp("//aspects:rules/cpp/cpp_info.bzl", listOf("rules_cc"), listOf("extract_cpp_info"), false),
   Kotlin("//aspects:rules/kt/kt_info.bzl", listOf("io_bazel_rules_kotlin", "rules_kotlin"), listOf("extract_kotlin_info"), true),
   Rust("//aspects:rules/rust/rust_info.bzl", listOf("rules_rust"), listOf("extract_rust_crate_info"), false),
-  Android("//aspects:rules/android/android_info.bzl", listOf(), listOf("extract_android_info"), false);
+  Android("//aspects:rules/android/android_info.bzl", listOf(), listOf("extract_android_info"), false),
+  Go("//aspects:rules/go/go_info.bzl", listOf("io_bazel_rules_go"), listOf("extract_go_info"), false);
 
   fun toLoadStatement(): String =
     this.functions.joinToString(
@@ -83,6 +84,7 @@ class BazelBspLanguageExtensionsGenerator(internalAspectsResolver: InternalAspec
         Language.Kotlin -> """"@${it.ruleName}//kotlin/internal:kt_toolchain_type""""
         Language.Rust -> """"@${it.ruleName}//rust:toolchain_type""""
         Language.Android -> getAndroidToolchain()
+        Language.Go -> """"@io_bazel_rules_go//go:toolchain""""
         else -> null
       }
     }

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

@@ -345,7 +345,8 @@ class BazelProjectMapper(
         it.relativePath.endsWith(".scala") ||
         it.relativePath.endsWith(".py") ||
         it.relativePath.endsWith(".sh") ||
-        it.relativePath.endsWith(".rs")
+        it.relativePath.endsWith(".rs") ||
+        it.relativePath.endsWith(".go")
     }
 
   private fun isWorkspaceTarget(target: TargetInfo): Boolean =

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

@@ -7,6 +7,7 @@ import kotlinx.coroutines.flow.asFlow
 import kotlinx.coroutines.runBlocking
 import org.jetbrains.bsp.bazel.info.BspTargetInfo.AndroidTargetInfo
 import org.jetbrains.bsp.bazel.info.BspTargetInfo.CppTargetInfo
+import org.jetbrains.bsp.bazel.info.BspTargetInfo.GoTargetInfo
 import org.jetbrains.bsp.bazel.info.BspTargetInfo.JavaRuntimeInfo
 import org.jetbrains.bsp.bazel.info.BspTargetInfo.JavaToolchainInfo
 import org.jetbrains.bsp.bazel.info.BspTargetInfo.JvmTargetInfo
@@ -125,6 +126,12 @@ class TargetInfoReader {
             targetInfoBuilder.setAndroidTargetInfo(info)
         }
 
+        "go_target_info" -> {
+            val builder: GoTargetInfo.Builder = readFromFile(path, GoTargetInfo.newBuilder())
+            val info = builder.build()
+            targetInfoBuilder.setGoTargetInfo(info)
+        }
+
         "general" -> {
             val builder: TargetInfo.Builder = readFromFile(path, TargetInfo.newBuilder())
             val info = builder.buildPartial()

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

@@ -4,6 +4,7 @@ import org.jetbrains.bsp.bazel.info.BspTargetInfo
 import org.jetbrains.bsp.bazel.server.sync.languages.android.AndroidLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.cpp.CppLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.cpp.CppModule
+import org.jetbrains.bsp.bazel.server.sync.languages.go.GoLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JavaLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.kotlin.KotlinLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.python.PythonLanguagePlugin
@@ -23,6 +24,7 @@ class LanguagePluginsService(
     val pythonLanguagePlugin: PythonLanguagePlugin,
     val rustLanguagePlugin: RustLanguagePlugin,
     private val androidLanguagePlugin: AndroidLanguagePlugin,
+    val goLanguagePlugin: GoLanguagePlugin,
 ) {
     private val emptyLanguagePlugin: EmptyLanguagePlugin = EmptyLanguagePlugin()
 
@@ -33,6 +35,7 @@ class LanguagePluginsService(
         thriftLanguagePlugin.prepareSync(targetInfos)
         pythonLanguagePlugin.prepareSync(targetInfos)
         rustLanguagePlugin.prepareSync(targetInfos)
+        goLanguagePlugin.prepareSync(targetInfos)
     }
 
     fun getPlugin(languages: Set<Language>): LanguagePlugin<*> =
@@ -45,6 +48,7 @@ class LanguagePluginsService(
             languages.contains(Language.THRIFT) -> thriftLanguagePlugin
             languages.contains(Language.PYTHON) -> pythonLanguagePlugin
             languages.contains(Language.RUST) -> rustLanguagePlugin
+            languages.contains(Language.GO) -> goLanguagePlugin
             else -> emptyLanguagePlugin
         }
 

server/src/main/kotlin/org/jetbrains/bsp/bazel/server/sync/languages/go/GoLanguagePlugin.kt

@@ -0,0 +1,43 @@
+package org.jetbrains.bsp.bazel.server.sync.languages.go
+
+import ch.epfl.scala.bsp4j.BuildTarget
+import org.jetbrains.bsp.GoBuildTarget
+import org.jetbrains.bsp.bazel.info.BspTargetInfo
+import org.jetbrains.bsp.bazel.server.paths.BazelPathsResolver
+import org.jetbrains.bsp.bazel.server.sync.languages.LanguagePlugin
+import java.net.URI
+import kotlin.io.path.toPath
+
+class GoLanguagePlugin(
+  private val bazelPathsResolver: BazelPathsResolver
+) : LanguagePlugin<GoModule>() {
+  override fun applyModuleData(moduleData: GoModule, buildTarget: BuildTarget) {
+    val goBuildTarget = with(moduleData) {
+      GoBuildTarget(
+        sdkHomePath = sdkHomePath,
+        importPath = importPath,
+      )
+    }
+
+    buildTarget.dataKind = "go"
+    buildTarget.data = goBuildTarget
+  }
+
+  override fun resolveModule(targetInfo: BspTargetInfo.TargetInfo): GoModule? {
+    if (!targetInfo.hasGoTargetInfo()) return null
+
+    val goTargetInfo = targetInfo.goTargetInfo
+    return GoModule(
+      sdkHomePath = calculateSdkURI(goTargetInfo.sdkHomePath),
+      importPath = goTargetInfo.importpath,
+    )
+  }
+
+  private fun calculateSdkURI(sdk: BspTargetInfo.FileLocation?): URI? =
+    sdk?.takeUnless { it.relativePath.isNullOrEmpty() }
+      ?.let {
+        val goBinaryPath = bazelPathsResolver.resolveUri(it).toPath()
+        val goSdkDir = goBinaryPath.parent.parent
+        goSdkDir.toUri()
+      }
+}

server/src/main/kotlin/org/jetbrains/bsp/bazel/server/sync/languages/go/GoModule.kt

@@ -0,0 +1,9 @@
+package org.jetbrains.bsp.bazel.server.sync.languages.go
+
+import org.jetbrains.bsp.bazel.server.sync.languages.LanguageData
+import java.net.URI
+
+data class GoModule (
+  val sdkHomePath: URI?,
+  val importPath: String?,
+): LanguageData

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

@@ -21,7 +21,8 @@ enum class Language(
       hashSetOf(JAVA.id),
       // This should be removed once https://github.com/bazelbuild/rules_kotlin/issues/273 is fixed
       "@@rules_kotlin.*//third_party:android_sdk".toRegex(),
-    );
+    ),
+    GO("go", hashSetOf(".go"));
 
     val allNames: Set<String> = dependentNames + id
 

server/src/main/kotlin/org/jetbrains/bsp/bazel/server/sync/proto/bsp_target_info.proto

@@ -103,6 +103,11 @@ message AndroidTargetInfo {
   string kotlin_target_id = 5;
 }
 
+message GoTargetInfo {
+  string importpath = 1;
+  FileLocation sdk_home_path = 2;
+}
+
 message TargetInfo {
   string id = 10;
   string kind = 20;
@@ -122,4 +127,5 @@ message TargetInfo {
   PythonTargetInfo python_target_info = 8000;
   RustCrateInfo rust_crate_info = 9000;
   AndroidTargetInfo android_target_info = 10000;
+  GoTargetInfo go_target_info = 11000;
 }

server/src/test/kotlin/org/jetbrains/bsp/bazel/server/bsp/managers/BazelBspLanguageExtensionsGeneratorTest.kt

@@ -48,15 +48,24 @@ class BazelBspLanguageExtensionsGeneratorTest {
             EXTENSIONS=[extract_java_toolchain,extract_java_runtime,extract_jvm_info,extract_python_info,extract_cpp_info]
             TOOLCHAINS=["@bazel_tools//tools/jdk:runtime_toolchain_type"]
         """.replace(" ", "").replace("\n", "")
+    private val goFileContent =
+        """ load("//aspects:rules/java/java_info.bzl","extract_java_toolchain","extract_java_runtime")
+            load("//aspects:rules/jvm/jvm_info.bzl","extract_jvm_info")
+            load("//aspects:rules/python/python_info.bzl","extract_python_info")
+            load("//aspects:rules/go/go_info.bzl","extract_go_info")
+            EXTENSIONS=[extract_java_toolchain,extract_java_runtime,extract_jvm_info,extract_python_info,extract_go_info]
+            TOOLCHAINS=["@bazel_tools//tools/jdk:runtime_toolchain_type","@io_bazel_rules_go//go:toolchain"]
+        """.replace(" ", "").replace("\n", "")
     private val allExtensionsFileContent =
         """ load("//aspects:rules/java/java_info.bzl","extract_java_toolchain","extract_java_runtime")
             load("//aspects:rules/jvm/jvm_info.bzl","extract_jvm_info")
             load("//aspects:rules/python/python_info.bzl","extract_python_info")
             load("//aspects:rules/cpp/cpp_info.bzl","extract_cpp_info")
             load("//aspects:rules/kt/kt_info.bzl","extract_kotlin_info")
             load("//aspects:rules/scala/scala_info.bzl","extract_scala_info","extract_scala_toolchain_info")
-            EXTENSIONS=[extract_java_toolchain,extract_java_runtime,extract_jvm_info,extract_python_info,extract_cpp_info,extract_kotlin_info,extract_scala_info,extract_scala_toolchain_info]
-            TOOLCHAINS=["@bazel_tools//tools/jdk:runtime_toolchain_type","@io_bazel_rules_kotlin//kotlin/internal:kt_toolchain_type","@io_bazel_rules_scala//scala:toolchain_type"]
+            load("//aspects:rules/go/go_info.bzl","extract_go_info")
+            EXTENSIONS=[extract_java_toolchain,extract_java_runtime,extract_jvm_info,extract_python_info,extract_cpp_info,extract_kotlin_info,extract_scala_info,extract_scala_toolchain_info,extract_go_info]
+            TOOLCHAINS=["@bazel_tools//tools/jdk:runtime_toolchain_type","@io_bazel_rules_kotlin//kotlin/internal:kt_toolchain_type","@io_bazel_rules_scala//scala:toolchain_type","@io_bazel_rules_go//go:toolchain"]
         """.replace(" ", "").replace("\n", "")
     private val defaultRuleLanguages =
       listOf(
@@ -117,13 +126,32 @@ class BazelBspLanguageExtensionsGeneratorTest {
         fileContent shouldBeEqual cppFileContent
     }
 
+    @Test
+    fun `should create the extensions dot bzl file with one import and one toolchain (go)`() {
+        // given
+        val ruleLanguages = defaultRuleLanguages + listOf(
+          RuleLanguage("io_bazel_rules_go", Language.Go)
+        )
+        BazelExternalRulesQueryMock(listOf("io_bazel_rules_go"))
+        val bazelBspLanguageExtensionsGenerator =
+            BazelBspLanguageExtensionsGenerator(internalAspectsResolverMock, bazelRelease)
+
+        // when
+        bazelBspLanguageExtensionsGenerator.generateLanguageExtensions(ruleLanguages)
+
+        // then
+        val fileContent = getExtensionsFileContent()
+        fileContent shouldBeEqual goFileContent
+    }
+
     @Test
     fun `should create the extensions dot bzl file with all possible imports and toolchains`() {
         // given
         val ruleLanguages = defaultRuleLanguages + listOf(
           RuleLanguage("rules_cc", Language.Cpp),
           RuleLanguage("io_bazel_rules_kotlin", Language.Kotlin),
           RuleLanguage("io_bazel_rules_scala", Language.Scala),
+          RuleLanguage("io_bazel_rules_go", Language.Go)
         )
         val bazelBspLanguageExtensionsGenerator =
             BazelBspLanguageExtensionsGenerator(internalAspectsResolverMock, bazelRelease)
@@ -143,6 +171,7 @@ class BazelBspLanguageExtensionsGeneratorTest {
           RuleLanguage("rules_cc", Language.Cpp),
           RuleLanguage("io_bazel_rules_kotlin", Language.Kotlin),
           RuleLanguage("io_bazel_rules_scala", Language.Scala),
+          RuleLanguage("io_bazel_rules_go", Language.Go)
         )
         val emptyBazelBspLanguageExtensionsGenerator =
             BazelBspLanguageExtensionsGenerator(internalAspectsResolverMock, bazelRelease)

server/src/test/kotlin/org/jetbrains/bsp/bazel/server/sync/languages/LanguagePluginServiceTest.kt

@@ -8,6 +8,7 @@ import org.jetbrains.bsp.bazel.bazelrunner.orLatestSupported
 import org.jetbrains.bsp.bazel.server.paths.BazelPathsResolver
 import org.jetbrains.bsp.bazel.server.sync.languages.android.AndroidLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.cpp.CppLanguagePlugin
+import org.jetbrains.bsp.bazel.server.sync.languages.go.GoLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JavaLanguagePlugin
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JdkResolver
 import org.jetbrains.bsp.bazel.server.sync.languages.java.JdkVersionResolver
@@ -51,6 +52,7 @@ class LanguagePluginServiceTest {
     val pythonLanguagePlugin = PythonLanguagePlugin(bazelPathsResolver)
     val rustLanguagePlugin = RustLanguagePlugin(bazelPathsResolver)
     val androidLanguagePlugin = AndroidLanguagePlugin(javaLanguagePlugin, bazelPathsResolver)
+    val goLanguagePlugin = GoLanguagePlugin(bazelPathsResolver)
     languagePluginsService = LanguagePluginsService(
         scalaLanguagePlugin,
         javaLanguagePlugin,
@@ -59,7 +61,8 @@ class LanguagePluginServiceTest {
         thriftLanguagePlugin,
         pythonLanguagePlugin,
         rustLanguagePlugin,
-        androidLanguagePlugin
+        androidLanguagePlugin,
+        goLanguagePlugin
     )
   }
 
@@ -138,6 +141,18 @@ class LanguagePluginServiceTest {
       // then
       plugin shouldNotBe null
     }
+
+    @Test
+    fun `should return GoLanguagePlugin for Go Language`() {
+      // given
+      val languages: Set<Language> = hashSetOf(Language.GO)
+
+      // when
+      val plugin = languagePluginsService.getPlugin(languages) as? GoLanguagePlugin
+
+      // then
+      plugin shouldNotBe null
+    }
   }
 
   @Test