[feat] GO support

buildSrc/src/main/kotlin/versions.kt

@@ -7,23 +7,24 @@ object Plugin {
 
   // See https://plugins.jetbrains.com/docs/intellij/build-number-ranges.html
 // for insight into build numbers and IntelliJ Platform versions.
-  const val sinceBuild = "241.14494.17"
+  const val sinceBuild = "241.14494.158"
   const val untilBuild = "241.*"
 }
 
 // Plugin Verifier integration -> https://github.com/JetBrains/gradle-intellij-plugin//plugin-verifier-dsl
 // See https://jb.gg/intellij-platform-builds-list for available build versions.
-const val pluginVerifierIdeVersions = "241.14494.17"
+const val pluginVerifierIdeVersions = "241.14494.158"
 
 object Platform {
-  const val type = "IC"
-  const val version = "241.14494.17-EAP-SNAPSHOT"
+  const val type = "IU"
+  const val version = "241.14494.158-EAP-SNAPSHOT"
   const val downloadSources = true
 
   // Plugin Dependencies -> https://plugins.jetbrains.com/docs/intellij/plugin-dependencies.html
   // Example: platformPlugins =" com.intellij.java, com.jetbrains.php:203.4449.22"
   val plugins =
-    listOf("com.intellij.java", "org.jetbrains.kotlin", "PythonCore:241.14494.17", "org.jetbrains.android:241.14494.17")
+    listOf("com.intellij.java", "org.jetbrains.kotlin", "PythonCore:241.14494.158", "org.jetbrains.android:241.14494.158",
+      "org.jetbrains.plugins.go:241.14494.158")
 }
 
 const val javaVersion = "17"

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

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

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

@@ -1,14 +1,17 @@
 package org.jetbrains.bsp.protocol
 
 import ch.epfl.scala.bsp4j.BuildTargetIdentifier
+import java.net.URI
 
 public data class LibraryItem(
   val id: BuildTargetIdentifier,
   val dependencies: List<BuildTargetIdentifier>,
   val ijars: List<String>,
   val jars: List<String>,
   val sourceJars: List<String>,
-)
+  val goImportPath: String? = "",
+  val goRoot: URI? = URI(""),
+  )
 
 public data class WorkspaceLibrariesResult(
   val libraries: List<LibraryItem>,

protocol/src/main/kotlin/org/jetbrains/bsp/protocol/utils/Extractors.kt

@@ -7,6 +7,7 @@ import ch.epfl.scala.bsp4j.PythonBuildTarget
 import ch.epfl.scala.bsp4j.ScalaBuildTarget
 import com.google.gson.Gson
 import com.google.gson.JsonObject
+import org.jetbrains.bsp.GoBuildTarget
 import org.jetbrains.bsp.protocol.AndroidBuildTarget
 import org.jetbrains.bsp.protocol.KotlinBuildTarget
 
@@ -32,6 +33,9 @@ public fun extractKotlinBuildTarget(target: BuildTarget): KotlinBuildTarget? =
   extractData(target, "kotlin")
     ?: extractAndroidBuildTarget(target)?.kotlinBuildTarget
 
+public fun extractGoBuildTarget(target: BuildTarget): GoBuildTarget? =
+  extractData(target, "go")
+
 public fun extractJvmBuildTarget(target: BuildTarget): JvmBuildTarget? =
   extractData(target, BuildTargetDataKind.JVM)
     ?: extractAndroidBuildTarget(target)?.jvmBuildTarget

src/main/kotlin/org/jetbrains/plugins/bsp/config/BspFeatureFlags.kt

@@ -5,6 +5,7 @@ import com.intellij.openapi.util.registry.Registry
 private const val PYTHON_SUPPORT = "bsp.python.support"
 private const val SCALA_SUPPORT = "bsp.scala.support"
 private const val ANDROID_SUPPORT = "bsp.android.support"
+private const val GO_SUPPORT = "bsp.go.support"
 private const val BUILD_PROJECT_ON_SYNC = "bsp.build.project.on.sync"
 
 public object BspFeatureFlags {
@@ -17,6 +18,9 @@ public object BspFeatureFlags {
   public val isAndroidSupportEnabled: Boolean
     get() = Registry.`is`(ANDROID_SUPPORT)
 
+  public val isGoSupportEnabled: Boolean
+    get() = Registry.`is`(GO_SUPPORT)
+
   public val isBuildProjectOnSyncEnabled: Boolean
     get() = Registry.`is`(BUILD_PROJECT_ON_SYNC)
 }

src/main/kotlin/org/jetbrains/plugins/bsp/extension/points/GoSdkGetterExtension.kt

@@ -0,0 +1,85 @@
+package org.jetbrains.plugins.bsp.extension.points
+
+import ch.epfl.scala.bsp4j.BuildTarget
+import com.goide.project.GoModuleSettings
+import com.goide.sdk.GoSdk
+import com.goide.sdk.GoSdkService
+import com.goide.vgo.project.workspaceModel.VgoWorkspaceModelUpdater
+import com.intellij.openapi.application.writeAction
+import com.intellij.openapi.extensions.ExtensionPointName
+import com.intellij.openapi.module.Module
+import com.intellij.openapi.project.Project
+import org.jetbrains.bsp.protocol.utils.extractGoBuildTarget
+import org.jetbrains.plugins.bsp.magicmetamodel.ProjectDetails
+
+public interface GoSdkGetterExtension {
+  public suspend fun addGoSdks(
+    project: Project,
+  )
+
+  public fun calculateAllGoSdkInfos(
+    projectDetails: ProjectDetails,
+  )
+
+  public fun enableGoSupportForModule(
+    module: Module,
+  )
+
+  public fun restoreGoModulesRegistry(
+    project: Project,
+  )
+}
+
+private val ep = ExtensionPointName.create<GoSdkGetterExtension>(
+  "org.jetbrains.bsp.goSdkGetterExtension",
+)
+
+public fun goSdkExtension(): GoSdkGetterExtension? = ep.extensionList.firstOrNull()
+
+public fun goSdkExtensionExists(): Boolean = ep.extensionList.isNotEmpty()
+
+public class GoSdkGetter: GoSdkGetterExtension {
+  private var goSdks: Set<GoSdk> = emptySet()
+
+  override suspend fun addGoSdks(
+    project: Project,
+  ) {
+    val goSdkService = GoSdkService.getInstance(project)
+    goSdks.forEach {
+      writeAction {
+        goSdkService.setSdk(it)
+      }
+    }
+  }
+
+  override fun calculateAllGoSdkInfos(
+    projectDetails: ProjectDetails,
+  ) {
+    goSdks = projectDetails.targets
+      .mapNotNull {
+        createGoSdk(it)
+      }
+      .toSet()
+  }
+
+  override fun enableGoSupportForModule(
+    module: Module,
+  ) {
+    GoModuleSettings.getInstance(module).isGoSupportEnabled = true
+  }
+
+  override fun restoreGoModulesRegistry(
+    project: Project,
+  ) {
+    VgoWorkspaceModelUpdater(project).restoreModulesRegistry()
+  }
+
+  private fun createGoSdk(target: BuildTarget): GoSdk? =
+    extractGoBuildTarget(target)?.let {
+      if (it.sdkHomePath == null) {
+        GoSdk.NULL
+      } else {
+        GoSdk.fromHomePath(it.sdkHomePath?.path)
+      }
+    }
+}

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/DefaultMagicMetaModelState.kt

@@ -8,6 +8,9 @@ import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.BuildTargetI
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.ContentRoot
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.GenericModuleInfo
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.GenericSourceRoot
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.GoAddendum
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.GoModule
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.GoModuleDependency
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.IntermediateLibraryDependency
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.IntermediateModuleDependency
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.JavaAddendum
@@ -200,8 +203,9 @@ public data class ModuleState(
   var scalaAddendum: ScalaAddendumState? = null,
   var javaAddendum: JavaAddendumState? = null,
   var androidAddendum: AndroidAddendumState? = null,
+  var goAddendum: GoAddendumState? = null,
 ) : ConvertableFromState<Module> {
-  public fun toJavaModule(): JavaModule = JavaModule(
+  private fun toJavaModule(): JavaModule = JavaModule(
     genericModuleInfo = module.fromState(),
     baseDirContentRoot = baseDirContentRoot?.fromState(),
     sourceRoots = sourceRoots.map { it.toJavaSourceRoot() },
@@ -214,17 +218,29 @@ public data class ModuleState(
     androidAddendum = androidAddendum?.fromState(),
   )
 
-  public fun toPythonModule(): PythonModule = PythonModule(
+  private fun toPythonModule(): PythonModule = PythonModule(
     module = module.fromState(),
     sourceRoots = sourceRoots.map { it.toGenericSourceRoot() },
     libraries = libraries?.map { it.toPythonLibrary() }.orEmpty(),
     resourceRoots = resourceRoots.map { it.toResourceRoot() },
     sdkInfo = sdkInfo?.fromState(),
   )
 
+  private fun toGoModule(): GoModule =
+    GoModule(
+      module = module.fromState(),
+      importPath = goAddendum?.importPath ?: "",
+      root = Path(goAddendum?.root ?: ""),
+      sourceRoots = sourceRoots.map { it.toGenericSourceRoot() },
+      resourceRoots = resourceRoots.map { it.toResourceRoot() },
+      goDependencies = goAddendum?.goDependencies?.map { it.fromState() }.orEmpty()
+    )
+
   override fun fromState(): Module =
     if (module.languageIds.includesPython())
       toPythonModule()
+    else if (module.languageIds.contains("go"))
+      toGoModule()
     else
       toJavaModule()
 }
@@ -287,6 +303,28 @@ public data class AndroidAddendumState(
   )
 }
 
+public data class GoAddendumState(
+  var importPath: String = "",
+  var root: String = "",
+  var goDependencies: List<GoModuleDependencyState> = emptyList(),
+) : ConvertableFromState<GoAddendum> {
+  override fun fromState(): GoAddendum = GoAddendum(
+    importPath = importPath,
+    root = Path(root),
+    goDependencies = goDependencies.map { it.fromState() },
+  )
+}
+
+public data class GoModuleDependencyState(
+  var importPath: String = "",
+  var root: String = "",
+) : ConvertableFromState<GoModuleDependency> {
+  override fun fromState(): GoModuleDependency = GoModuleDependency(
+    importPath = importPath,
+    root = Path(root),
+  )
+}
+
 public data class ModuleCapabilitiesState(
   var canRun: Boolean = false,
   var canTest: Boolean = false,
@@ -318,6 +356,17 @@ public fun AndroidAddendum.toState(): AndroidAddendumState = AndroidAddendumStat
   resourceFolders = resourceFolders.map { it.toString() },
 )
 
+public fun GoAddendum.toState(): GoAddendumState = GoAddendumState(
+  importPath = importPath,
+  root = root.toString(),
+  goDependencies = goDependencies.map { it.toState() },
+)
+
+public fun GoModuleDependency.toState(): GoModuleDependencyState = GoModuleDependencyState(
+  importPath = importPath,
+  root = root.toString(),
+)
+
 public fun ModuleCapabilities.toState(): ModuleCapabilitiesState = ModuleCapabilitiesState(
   canRun = canRun,
   canTest = canTest,

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/MagicMetaModelImpl.kt

@@ -126,6 +126,8 @@ public class MagicMetaModelImpl : MagicMetaModel, ConvertableToState<DefaultMagi
       iJars = it.ijars,
       classJars = it.jars,
       sourceJars = it.sourceJars,
+      goImportPath = it.goImportPath,
+      goRoot = it.goRoot,
     )
   }
 

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/TargetIdToModuleEntitiesMap.kt

@@ -5,9 +5,11 @@ import org.jetbrains.plugins.bsp.magicmetamodel.ProjectDetails
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.BuildTargetId
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.BuildTargetInfo
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.Module
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.impl.updaters.transformers.ModuleDetailsToGoModuleTransformer
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.impl.updaters.transformers.ModuleDetailsToJavaModuleTransformer
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.impl.updaters.transformers.ModuleDetailsToPythonModuleTransformer
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.impl.updaters.transformers.ProjectDetailsToModuleDetailsTransformer
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.includesGo
 import org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel.includesPython
 import java.nio.file.Path
 
@@ -32,13 +34,21 @@ public object TargetIdToModuleEntitiesMap {
       projectBasePath,
       hasDefaultPythonInterpreter,
     )
+    val moduleDetailsToGoModuleTransformer = ModuleDetailsToGoModuleTransformer(
+      targetsMap,
+      projectDetails,
+      moduleNameProvider,
+      projectBasePath,
+    )
 
     val transformer = ProjectDetailsToModuleDetailsTransformer(projectDetails)
 
     return projectDetails.targetsId.associate {
       val moduleDetails = transformer.moduleDetailsForTargetId(it)
       val module = if (moduleDetails.target.languageIds.includesPython()) {
         moduleDetailsToPythonModuleTransformer.transform(moduleDetails)
+      } else if (moduleDetails.target.languageIds.includesGo()) {
+        moduleDetailsToGoModuleTransformer.transform(moduleDetails)
       } else {
         moduleDetailsToJavaModuleTransformer.transform(moduleDetails)
       }

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/workspacemodel/GoWorkspaceModelEntities.kt

@@ -0,0 +1,36 @@
+package org.jetbrains.plugins.bsp.magicmetamodel.impl.workspacemodel
+
+import org.jetbrains.bsp.protocol.LibraryItem
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.ModuleState
+import org.jetbrains.plugins.bsp.magicmetamodel.impl.toState
+import java.nio.file.Path
+
+public data class GoModuleDependency(
+  val importPath: String,
+  val root: Path,
+)
+
+public data class GoModule(
+  val module: GenericModuleInfo,
+  val sourceRoots: List<GenericSourceRoot>,
+  val resourceRoots: List<ResourceRoot>,
+  val importPath: String,
+  val root: Path,
+  val goDependencies: List<GoModuleDependency>,
+  val goLibraries: List<LibraryItem> = emptyList(),
+) : WorkspaceModelEntity(), Module {
+  override fun toState(): ModuleState = ModuleState(
+    module = module.toState(),
+    sourceRoots = sourceRoots.map { it.toState() },
+    resourceRoots = resourceRoots.map { it.toState() },
+    goAddendum = GoAddendum(importPath, root, goDependencies).toState(),
+  )
+
+  override fun getModuleName(): String = module.name
+}
+
+public data class GoAddendum(
+  var importPath: String,
+  var root: Path,
+  val goDependencies: List<GoModuleDependency>,
+)

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/workspacemodel/WorkspaceModelEntities.kt

@@ -53,7 +53,9 @@ public data class Library(
   val iJars: List<String> = listOf(),
   val sourceJars: List<String> = listOf(),
   val classJars: List<String> = listOf(),
-) : WorkspaceModelEntity(), ResourceRootEntity {
+  val goImportPath: String? = "",
+  val goRoot: URI? = URI(""),
+  ) : WorkspaceModelEntity(), ResourceRootEntity {
   public companion object {
     public fun formatJarString(jar: String): String =
       if (jar.endsWith(".jar")) {

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/workspacemodel/WorkspaceModelEntitiesExtensions.kt

@@ -11,6 +11,7 @@ public fun LanguageIds.includesKotlin(): Boolean = contains("kotlin")
 public fun LanguageIds.includesJava(): Boolean = contains("java")
 public fun LanguageIds.includesScala(): Boolean = contains("scala")
 public fun LanguageIds.includesAndroid(): Boolean = contains("android")
+public fun LanguageIds.includesGo(): Boolean = contains("go")
 
 public fun LanguageIds.includesJavaOrScala(): Boolean = includesJava() || includesScala()
 public fun LanguageIds.isJvmTarget(): Boolean = (includesJava() || includesKotlin()) && !includesAndroid()

src/main/kotlin/org/jetbrains/plugins/bsp/magicmetamodel/impl/workspacemodel/WorkspaceModelToModulesMapTransformer.kt

@@ -30,6 +30,7 @@ import org.jetbrains.workspacemodel.entities.AndroidTargetType.TEST
 import org.jetbrains.workspacemodel.entities.androidAddendumEntity
 import org.jetbrains.workspacemodel.entities.jvmBinaryJarsEntity
 import java.nio.file.Path
+import kotlin.io.path.Path
 
 public object WorkspaceModelToModulesMapTransformer {
   public operator fun invoke(
@@ -95,6 +96,7 @@ public object WorkspaceModelToModulesMapTransformer {
         languageIds = entity.customImlData?.customModuleOptions.orEmpty(),
         associates = entity.getKotlinSettings()?.getAssociates().orEmpty(),
       )
+      // TODO: add support for GoModule
       val module = if (genericModuleInfo.languageIds.includesPython()) {
         PythonModule(
           module = genericModuleInfo,