fix: 修复语言键使用状态不实时更新的问题

Author: kzheart

src/main/kotlin/org/tabooproject/development/inlay/EditorDocumentListener.kt

@@ -38,7 +38,7 @@ class EditorDocumentListener(private val project: Project) : Disposable {
             val document = event.document
             val file = FileDocumentManager.getInstance().getFile(document)
 
-            if (file != null && isLanguageFile(file)) {
+            if (file != null && (isLanguageFile(file) || isKotlinFile(file))) {
                 // 防抖：避免频繁刷新
                 val currentTime = System.currentTimeMillis()
                 if (currentTime - lastRefreshTime < refreshDebounceMs) {
@@ -49,7 +49,11 @@ class EditorDocumentListener(private val project: Project) : Disposable {
                 // 延迟刷新以合并连续的修改
                 ApplicationManager.getApplication().invokeLater({
                     if (!project.isDisposed) {
-                        refreshEditorsForFile(file)
+                        if (isLanguageFile(file)) {
+                            refreshEditorsForFile(file)
+                        } else if (isKotlinFile(file)) {
+                            refreshUsageCache()
+                        }
                     }
                 }, project.disposed)
             }
@@ -77,7 +81,7 @@ class EditorDocumentListener(private val project: Project) : Disposable {
         val openFiles = fileEditorManager.openFiles
 
         for (file in openFiles) {
-            if (isLanguageFile(file)) {
+            if (isLanguageFile(file) || isKotlinFile(file)) {
                 val document = FileDocumentManager.getInstance().getDocument(file)
                 if (document != null) {
                     registerDocument(document)
@@ -188,6 +192,40 @@ class EditorDocumentListener(private val project: Project) : Disposable {
         return LangFiles.isLangFile(file)
     }
 
+    /**
+     * 检查文件是否为Kotlin文件
+     *
+     * @param file 要检查的文件
+     * @return 如果是Kotlin文件返回 true
+     */
+    private fun isKotlinFile(file: VirtualFile): Boolean {
+        return file.name.endsWith(".kt")
+    }
+
+    /**
+     * 刷新使用缓存并更新相关编辑器
+     */
+    private fun refreshUsageCache() {
+        // 清除语言使用分析缓存
+        LangUsageAnalyzer.clearCache(project)
+        
+        // 更新编辑器通知
+        com.intellij.ui.EditorNotifications.updateAll()
+        
+        // 重新分析所有语言文件以更新未使用标记
+        val fileEditorManager = FileEditorManager.getInstance(project)
+        val psiManager = PsiManager.getInstance(project)
+        val daemonCodeAnalyzer = com.intellij.codeInsight.daemon.DaemonCodeAnalyzer.getInstance(project)
+        
+        // 查找所有语言文件并触发重新分析
+        LangIndex.findProjectLangFiles(project).forEach { langFile ->
+            val psiFile = psiManager.findFile(langFile)
+            if (psiFile != null) {
+                daemonCodeAnalyzer.restart(psiFile)
+            }
+        }
+    }
+
     override fun dispose() {
         // 清理所有监听器
         registeredDocuments.forEach { document ->

src/main/kotlin/org/tabooproject/development/inlay/LangUsageAnalyzer.kt

@@ -24,6 +24,8 @@ object LangUsageAnalyzer {
 
     // 缓存使用情况分析结果
     private val usageCache = ConcurrentHashMap<String, LangUsageResult>()
+    // 缓存文件最后修改时间，用于检测缓存是否过期
+    private val fileModificationCache = ConcurrentHashMap<String, Long>()
 
     /**
      * 语言使用结果
@@ -49,8 +51,10 @@ object LangUsageAnalyzer {
     fun analyzeLangUsage(project: Project): LangUsageResult {
         val projectPath = project.basePath ?: return LangUsageResult(emptySet(), emptyMap())
 
-        // 检查缓存
-        usageCache[projectPath]?.let { return it }
+        // 检查缓存是否过期
+        if (isCacheValid(project, projectPath)) {
+            usageCache[projectPath]?.let { return it }
+        }
 
         val usedKeys = mutableSetOf<String>()
         val keyUsages = mutableMapOf<String, MutableList<LangUsageLocation>>()
@@ -86,6 +90,9 @@ object LangUsageAnalyzer {
         val result = LangUsageResult(usedKeys, keyUsages)
         usageCache[projectPath] = result
 
+        // 更新文件修改时间缓存
+        updateFileModificationCache(project, projectPath)
+        
         return result
     }
 
@@ -122,11 +129,43 @@ object LangUsageAnalyzer {
             val projectPath = project.basePath
             if (projectPath != null) {
                 usageCache.remove(projectPath)
+                fileModificationCache.remove(projectPath)
             }
         } else {
             usageCache.clear()
+            fileModificationCache.clear()
         }
     }
+
+    /**
+     * 检查缓存是否有效
+     */
+    private fun isCacheValid(project: Project, projectPath: String): Boolean {
+        if (!usageCache.containsKey(projectPath)) {
+            return false
+        }
+        
+        val cachedTime = fileModificationCache[projectPath] ?: return false
+        val currentMaxModificationTime = getCurrentMaxModificationTime(project)
+        
+        return cachedTime >= currentMaxModificationTime
+    }
+
+    /**
+     * 获取项目中所有Kotlin文件的最大修改时间
+     */
+    private fun getCurrentMaxModificationTime(project: Project): Long {
+        val kotlinFiles = findKotlinFiles(project)
+        return kotlinFiles.maxOfOrNull { it.timeStamp } ?: 0L
+    }
+
+    /**
+     * 更新文件修改时间缓存
+     */
+    private fun updateFileModificationCache(project: Project, projectPath: String) {
+        val maxModificationTime = getCurrentMaxModificationTime(project)
+        fileModificationCache[projectPath] = maxModificationTime
+    }
 
     /**
      * 查找项目中的所有Kotlin文件