feat(go): implement module retrieval functionality in GoBuildSystemProvider #308

Author: phodal

goland/src/main/kotlin/cc/unitmesh/go/provider/GoBuildSystemProvider.kt

@@ -6,7 +6,12 @@ import cc.unitmesh.devti.template.context.DockerfileContext
 import com.goide.vgo.mod.psi.VgoModuleSpec
 import com.goide.vgo.mod.psi.VgoRequireDirective
 import com.goide.vgo.project.VgoDependency
+import com.goide.vgo.project.VgoModulesRegistry
+import com.intellij.openapi.actionSystem.DataContext
+import com.intellij.openapi.module.ModuleManager
+import com.intellij.openapi.progress.ProgressManager
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiFile
 import com.intellij.psi.util.PsiTreeUtil
 
@@ -24,6 +29,37 @@ class GoBuildSystemProvider : BuildSystemProvider() {
             .toList()
     }
 
+
+    fun getAllModules(project: Project): Iterable<DevModuleModel> {
+        val result = mutableListOf<DevModuleModel>()
+
+        ModuleManager.getInstance(project).modules.forEach { module ->
+            ProgressManager.checkCanceled()
+            VgoModulesRegistry.getInstance(project).getModules(module).forEach { vgoModule ->
+                val dependencies = vgoModule.dependencies.mapNotNull { vgoDependency ->
+                    toPackage(vgoDependency)
+                }.toSet()
+
+                val importPath = vgoModule.importPath
+                val buildFile = vgoModule.root.findChild("go.mod")
+                val moduleModel = DevModuleModel(importPath, module, buildFile, dependencies, project)
+                result.add(moduleModel)
+            }
+        }
+
+        return result
+    }
+
+    data class DevModuleModel(
+        val id: String,
+        val platformModule: com.intellij.openapi.module.Module?,
+        val buildFile: VirtualFile?,
+        var dependencies: Set<DevPackage>,
+        val project: Project,
+        val dataContext: DataContext? = null,
+        val parentModuleName: String? = null
+    )
+
     fun getDependencies(modFile: PsiFile): List<VgoModuleSpec> {
         return PsiTreeUtil.getChildrenOfTypeAsList(modFile, VgoRequireDirective::class.java)
             .flatMap { requireDirective ->