refactor(indexer): improve domain dictionary generation and UI handling #358

Extract presentation update logic into separate method for reuse. Exclude test/utility classes from dictionary. Clean up imports and variable naming. This enhances maintainability and clarifies dictionary scope.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/indexer/DomainDictGenerateAction.kt

@@ -1,6 +1,5 @@
 package cc.unitmesh.devti.indexer
 
-import cc.unitmesh.devti.AutoDevIcons
 import cc.unitmesh.devti.indexer.provider.LangDictProvider
 import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.settings.coder.coderSetting
@@ -15,6 +14,8 @@ import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.project.guessProjectDir
 import kotlinx.coroutines.launch
+import cc.unitmesh.devti.AutoDevIcons
+import com.intellij.openapi.actionSystem.Presentation
 
 class DomainDictGenerateAction : AnAction() {
     init {
@@ -23,6 +24,8 @@ class DomainDictGenerateAction : AnAction() {
 
     override fun actionPerformed(event: AnActionEvent) {
         val project = event.project ?: return
+        val presentation = event.presentation
+        
         AutoDevCoroutineScope.scope(project).launch {
             val names = LangDictProvider.all(project)
 
@@ -33,10 +36,10 @@ class DomainDictGenerateAction : AnAction() {
             val prompt = templateRender.renderTemplate(template, context)
 
             try {
-                this@DomainDictGenerateAction.templatePresentation.icon = AutoDevIcons.InProgress
-                this@DomainDictGenerateAction.templatePresentation.isEnabled = false
-                val result = StringBuilder()
+                updatePresentation(presentation, AutoDevIcons.InProgress, false)
                 AutoDevStatusService.notifyApplication(AutoDevStatus.InProgress)
+                
+                val result = StringBuilder()
                 LlmFactory.create(project).stream(prompt, "").collect {
                     result.append(it)
                 }
@@ -53,13 +56,18 @@ class DomainDictGenerateAction : AnAction() {
                 AutoDevStatusService.notifyApplication(AutoDevStatus.Error)
                 e.printStackTrace()
             } finally {
-                this@DomainDictGenerateAction.templatePresentation.icon = AutoDevIcons.AI_COPILOT
-                this@DomainDictGenerateAction.templatePresentation.isEnabled = true
+                // Restore icon and enable the action
+                updatePresentation(presentation, AutoDevIcons.AI_COPILOT, true)
             }
         }
     }
+    
+    private fun updatePresentation(presentation: Presentation, icon: javax.swing.Icon, enabled: Boolean) {
+        presentation.icon = icon
+        presentation.isEnabled = enabled
+    }
 }
 
 data class DomainDictGenerateContext(
     val code: String
-) : TemplateContext
+) : TemplateContext

core/src/main/resources/genius/zh/code/indexer.vm

@@ -3,6 +3,7 @@
 - 它不是公共的库 API（如 Spring 等标准库或者平台的 API）、三方库 API（如 OkHttp、Retrofit 等），也不是常用的类名（如 List、Map 等）
 - 它是关键业务概念、无法理解的单词或者拼音缩写
 - 代码翻译中，不包含任何技术词汇，比如：Controller、Exception、Set、Code、Service、Repository、Mapper、DTO、VO、PO 等
+- 通用测试类和工具类（如 MathHelper）不应该被包含在内
 
 你的返回格式：
 

java/src/main/kotlin/cc/unitmesh/idea/indexer/provider/JavaLangDictProvider.kt

@@ -14,8 +14,10 @@ class JavaLangDictProvider : LangDictProvider {
             FileTypeIndex.getFiles(JavaFileType.INSTANCE, searchScope)
         }
 
-        return javaFiles.mapNotNull {
+        val filenames = javaFiles.mapNotNull {
             it.nameWithoutExtension
         }
+
+        return filenames
     }
 }