feat(coder): add custom for generate test #54

unit-mesh · Jan 5, 2024 · b10a0f5 · b10a0f5
1 parent 60deb26
commit b10a0f5
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 1 deletion.
diff --git a/src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatActionType.kt b/src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatActionType.kt
@@ -37,7 +37,12 @@ enum class ChatActionType {
                 }
             }
             CODE_COMPLETE -> "Complete $lang code, return rest code, no explaining"
-            GENERATE_TEST -> "Write unit test for given $lang code"
+            GENERATE_TEST -> {
+                devCoderSettings?.generateTest.let {
+                    val defaultPrompt = "Generate test for $lang code"
+                    compilePrompt(it, defaultPrompt, lang)
+                }
+            }
             FIX_ISSUE -> {
                 devCoderSettings?.fixIssueCode.let {
                     val defaultPrompt = "Help me fix this issue"

diff --git a/src/main/kotlin/cc/unitmesh/devti/settings/coder/AutoDevCoderConfigurable.kt b/src/main/kotlin/cc/unitmesh/devti/settings/coder/AutoDevCoderConfigurable.kt
@@ -17,6 +17,7 @@ class AutoDevCoderConfigurable(project: Project) : BoundConfigurable(AutoDevBund
     private val explainCodeField = JTextField()
     private val refactorCodeField = JTextField()
     private val fixIssueCodeField = JTextField()
+    private val generateTestField = JTextField()
 
     val settings = project.service<AutoDevCoderSettingService>()
     val state = settings.state.copy()
@@ -71,6 +72,14 @@ class AutoDevCoderConfigurable(project: Project) : BoundConfigurable(AutoDevBund
                     prop = state::fixIssueCode.toMutableProperty()
                 )
         }
+        row(AutoDevBundle.message("settings.autodev.coder.generateTest")) {
+            fullWidthCell(generateTestField)
+                .bind(
+                    componentGet = { it.text },
+                    componentSet = { component, value -> component.text = value },
+                    prop = state::generateTest.toMutableProperty()
+                )
+        }
 
         onApply {
             settings.modify {
@@ -80,6 +89,7 @@ class AutoDevCoderConfigurable(project: Project) : BoundConfigurable(AutoDevBund
                 it.explainCode = state.explainCode
                 it.refactorCode = state.refactorCode
                 it.fixIssueCode = state.fixIssueCode
+                it.generateTest = state.generateTest
             }
         }
     }

diff --git a/src/main/kotlin/cc/unitmesh/devti/settings/coder/AutoDevCoderSettingService.kt b/src/main/kotlin/cc/unitmesh/devti/settings/coder/AutoDevCoderSettingService.kt
@@ -25,6 +25,7 @@ class AutoDevCoderSettingService(
         var explainCode: String by property("Explain \$lang code") { it.isEmpty() }
         var refactorCode: String by property("Refactor the given \$lang code") { it.isEmpty() }
         var fixIssueCode: String by property("Help me fix this issue") { it.isEmpty() }
+        var generateTest: String by property("Generate test for \$lang code") { it.isEmpty() }
 
         override fun copy(): AutoDevCoderSettings {
             val state = AutoDevCoderSettings()

diff --git a/src/main/resources/messages/AutoDevBundle.properties b/src/main/resources/messages/AutoDevBundle.properties
@@ -94,4 +94,5 @@ settings.autodev.coder.inEditorCompletion=Completion in Editor
 settings.autodev.coder.explainCode=Explain code
 settings.autodev.coder.refactorCode=Refactor code
 settings.autodev.coder.fixIssueCode=Fix issue
+settings.autodev.coder.generateTest=Generate test