跳转

Author: cyz-home

src/main/kotlin/com/company/plugin/completion/ZyCompletionContributor.kt

@@ -62,6 +62,9 @@ class ZyCompletionContributor : CompletionContributor() {
                 val text = document.text
                 lspService.ensureDidOpen(uri, "zy", text)
 
+                // 通知 LSP 文档变更（确保同步最新内容）
+                lspService.notifyDocumentChange(uri, text)
+                
                 LOG.debug("Requesting completion at ${file.name}:${lineNumber}:${character}")
 
                 val completionFuture = lspService.getCompletion(uri, lineNumber, character)

src/main/kotlin/com/company/plugin/lsp/ZyDocumentListener.kt

@@ -0,0 +1,62 @@
+package com.company.plugin.lsp
+
+import com.intellij.openapi.diagnostic.Logger
+import com.intellij.openapi.editor.Document
+import com.intellij.openapi.editor.event.DocumentEvent
+import com.intellij.openapi.editor.event.DocumentListener
+import com.intellij.openapi.fileEditor.FileDocumentManager
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.startup.ProjectActivity
+
+/**
+ * ZY 文档同步组件
+ * 监听文档变更并通知 LSP 服务器
+ */
+class ZyDocumentSyncComponent : ProjectActivity {
+    
+    companion object {
+        private val LOG = Logger.getInstance(ZyDocumentSyncComponent::class.java)
+    }
+    
+    override suspend fun execute(project: Project) {
+        // 添加全局文档监听器
+        val documentListener = object : DocumentListener {
+            override fun documentChanged(event: DocumentEvent) {
+                handleDocumentChange(project, event)
+            }
+        }
+        
+        // 使用 EditorFactory 添加全局监听器
+        com.intellij.openapi.editor.EditorFactory.getInstance()
+            .eventMulticaster.addDocumentListener(documentListener, project)
+        
+        LOG.info("ZY document sync component initialized for project: ${project.name}")
+    }
+    
+    private fun handleDocumentChange(project: Project, event: DocumentEvent) {
+        try {
+            val document = event.document
+            val file = FileDocumentManager.getInstance().getFile(document)
+            
+            // 只处理 .zy 文件
+            if (file == null || !file.name.endsWith(".zy")) {
+                return
+            }
+            
+            val lspService = project.getService(ZyLspService::class.java)
+            if (lspService == null || !lspService.isStarted()) {
+                return
+            }
+            
+            // 通知 LSP 服务器文档变更
+            val uri = file.url
+            val newText = document.text
+            lspService.notifyDocumentChange(uri, newText)
+            
+            LOG.debug("Document change notified to LSP: ${file.name}")
+            
+        } catch (e: Exception) {
+            LOG.debug("Error handling document change", e)
+        }
+    }
+}

src/main/kotlin/com/company/plugin/lsp/ZyDocumentSyncComponent.kt

@@ -29,6 +29,8 @@ class ZyLspClient(private val project: Project) : LanguageClient {
     private var serverCapabilities: ServerCapabilities? = null
     // 已同步到 LSP 的文档 URI 集合，用于避免重复 didOpen
     private val openedDocuments = mutableSetOf<String>()
+    // 文档版本号，用于跟踪文档变更
+    private val documentVersions = mutableMapOf<String, Int>()
 
     /**
      * 检查进程状态，确保 LSP 服务器仍在运行
@@ -47,6 +49,7 @@ class ZyLspClient(private val project: Project) : LanguageClient {
     private fun cleanupState() {
         isInitialized = false
         openedDocuments.clear()
+        documentVersions.clear()
         serverCapabilities = null
     }
 
@@ -211,15 +214,51 @@ class ZyLspClient(private val project: Project) : LanguageClient {
         }
         if (openedDocuments.contains(uri)) return
         try {
-            val item = TextDocumentItem(uri, languageId, /*version*/1, text)
+            val version = documentVersions.getOrPut(uri) { 1 }
+            val item = TextDocumentItem(uri, languageId, version, text)
             languageServer!!.textDocumentService.didOpen(DidOpenTextDocumentParams(item))
             openedDocuments.add(uri)
-            LOG.debug("didOpen sent for uri=$uri, bytes=${text.toByteArray().size}")
+            LOG.debug("didOpen sent for uri=$uri, version=$version, bytes=${text.toByteArray().size}")
         } catch (e: Exception) {
             LOG.debug("Failed to send didOpen for uri=$uri", e)
         }
     }
 
+    /**
+     * 通知 LSP 服务器文档内容变更
+     */
+    fun didChange(uri: String, newText: String) {
+        if (!isInitialized || languageServer == null || !isProcessAlive()) {
+            if (!isProcessAlive()) {
+                LOG.warn("LSP process is dead, cannot send didChange")
+                cleanupState()
+            }
+            return
+        }
+        if (!openedDocuments.contains(uri)) {
+            LOG.debug("Document not opened, skipping didChange for uri=$uri")
+            return
+        }
+        
+        try {
+            val version = documentVersions.getOrPut(uri) { 1 }
+            documentVersions[uri] = version + 1
+            
+            // 使用全量变更（简单实现）
+            val changes = listOf(TextDocumentContentChangeEvent(newText))
+            
+            val params = DidChangeTextDocumentParams(
+                VersionedTextDocumentIdentifier(uri, documentVersions[uri]!!),
+                changes
+            )
+            
+            languageServer!!.textDocumentService.didChange(params)
+            LOG.debug("didChange sent for uri=$uri, version=${documentVersions[uri]}")
+        } catch (e: Exception) {
+            LOG.debug("Failed to send didChange for uri=$uri", e)
+        }
+    }
+    
     /**
      * 解析 LSP 可执行文件路径
      */

src/main/kotlin/com/company/plugin/lsp/ZyLspClient.kt

@@ -91,6 +91,13 @@ class ZyLspService(private val project: Project) {
         lspClient?.ensureDidOpen(uri, languageId, text)
     }
 
+    /**
+     * 通知文档内容变更
+     */
+    fun notifyDocumentChange(uri: String, newText: String) {
+        lspClient?.didChange(uri, newText)
+    }
+    
     /**
      * 检查 LSP 服务是否已启动
      */

src/main/kotlin/com/company/plugin/lsp/ZyLspService.kt

@@ -61,6 +61,9 @@ class ZyGotoDeclarationHandler : GotoDeclarationHandler {
             // 确保文档已同步到LSP
             val text = document.text
             lspService.ensureDidOpen(uri, "zy", text)
+            
+            // 通知 LSP 文档变更（确保同步最新内容）
+            lspService.notifyDocumentChange(uri, text)
 
             LOG.debug("Requesting definition for uri=$uri, line=$lineNumber, char=$character")
             val future = lspService.getDefinition(uri, lineNumber, character)

src/main/kotlin/com/company/plugin/navigation/ZyGotoDeclarationHandler.kt

@@ -47,6 +47,9 @@
 
         <!-- Project startup activity -->
         <postStartupActivity implementation="com.company.plugin.lsp.ZyProjectStartupActivity" />
+        
+        <!-- Document sync activity -->
+        <postStartupActivity implementation="com.company.plugin.lsp.ZyDocumentSyncComponent" />
 
         <!-- Enter handler: 保持作用域内缩进 -->
         <enterHandlerDelegate implementation="com.company.plugin.language.ZyEnterHandler" order="first"/>