feat(structure): enhance file structure retrieval and presentation #309

Refactor `StructureInCommand` to improve the retrieval and presentation of file structures. Added a new method `getFileStructure` to handle file structure extraction and a helper method `traverseStructure` for formatting the structure tree. Removed redundant imports and cleaned up code in `VueRelatedClassProvider`.

Author: phodal

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/exec/StructureShireCommand.kt

@@ -1,18 +1,22 @@
 package cc.unitmesh.devti.language.compiler.exec
 
+import cc.unitmesh.devti.devin.InsCommand
+import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
+import cc.unitmesh.devti.language.utils.lookupFile
+import com.intellij.ide.structureView.StructureView
+import com.intellij.ide.structureView.StructureViewTreeElement
+import com.intellij.lang.LanguageStructureViewBuilder
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.runReadAction
 import com.intellij.openapi.diagnostic.logger
+import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiFile
 import com.intellij.psi.PsiManager
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.withContext
-import cc.unitmesh.devti.language.utils.lookupFile
-import cc.unitmesh.devti.context.FileContextProvider
-import cc.unitmesh.devti.devin.InsCommand
-import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
+
 
 class StructureInCommand(val myProject: Project, val prop: String) : InsCommand {
     override val commandName: BuiltinCommand = BuiltinCommand.STRUCTURE
@@ -33,9 +37,38 @@ class StructureInCommand(val myProject: Project, val prop: String) : InsCommand
             }.get()
         } ?: return null
 
-        return FileContextProvider().from(psiFile)?.let {
-            return it?.format()
+        return getFileStructure(myProject, virtualFile, psiFile)
+    }
+
+    fun getFileStructure(project: Project, file: VirtualFile, psiFile: PsiFile): String {
+        val openFile = FileEditorManager.getInstance(myProject).openFile(file, true)
+        val fileEditor = openFile.firstOrNull() ?: return "No FileEditor found."
+
+        val factory = LanguageStructureViewBuilder.INSTANCE.forLanguage(psiFile.language)
+
+        if (factory != null) {
+            val view: StructureView = factory.getStructureViewBuilder(psiFile)!!
+                .createStructureView(fileEditor, project)
+
+            val root: StructureViewTreeElement = view.treeModel.root
+            return traverseStructure(root, 0, StringBuilder()).toString()
         }
+
+        return "No StructureViewModel found."
+    }
+
+    private fun traverseStructure(element: StructureViewTreeElement, depth: Int, sb: StringBuilder): StringBuilder {
+        val indent = "  ".repeat(depth)
+        /// todo: add element line
+        sb.append(indent).append(element.presentation.presentableText).append("\n")
+
+        for (child in element.children) {
+            if (child is StructureViewTreeElement) {
+                traverseStructure(child, depth + 1, sb)
+            }
+        }
+
+        return sb
     }
 
     fun file(project: Project, path: String): VirtualFile? {

exts/ext-vue/src/main/kotlin/cc/unitmesh/vue/provider/VueRelatedClassProvider.kt

@@ -16,7 +16,6 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
 import com.intellij.psi.impl.source.html.HtmlFileImpl
 import com.intellij.util.asSafely
-import com.intellij.webSymbols.SymbolKind
 import org.jetbrains.vuejs.index.findModule
 import org.jetbrains.vuejs.index.findScriptTag
 import org.jetbrains.vuejs.lang.html.VueFile
@@ -62,7 +61,6 @@ class VueRelatedClassProvider : RelatedClassesProvider {
 
         logger<VueRelatedClassProvider>().info("imports: $localImports")
         println("imports: $localImports")
-        /// resolve file in local
         return emptyList()
     }