feat(database): add GenerateUnittestAction

This commit adds the `GenerateUnittestAction` class to the database module. This action generates Java unit test code for Oracle PL/SQL code. It is available only for files written in the Oracle dialect. The action takes the selected PL/SQL code, renders it using a template, and prompts the user to provide additional information. The generated code is then sent to the chat window.

Author: phodal

exts/database/src/main/kotlin/cc/unitmesh/database/actions/GenerateEntityAction.kt

@@ -1,5 +1,6 @@
 package cc.unitmesh.database.actions
 
+import cc.unitmesh.database.actions.base.SqlMigrationContext
 import cc.unitmesh.devti.gui.sendToChatWindow
 import cc.unitmesh.devti.intentions.action.base.AbstractChatIntention
 import cc.unitmesh.devti.provider.ContextPrompter
@@ -15,7 +16,7 @@ class GenerateEntityAction : AbstractChatIntention() {
 
     override fun priority() = 901
 
-    override fun getFamilyName(): String = "Generate Java Entity"
+    override fun getFamilyName(): String = "PL/SQL Migration"
 
     override fun getText(): String = "Generate Java Entity"
 
@@ -31,7 +32,7 @@ class GenerateEntityAction : AbstractChatIntention() {
 
         val templateRender = TemplateRender("genius/migration")
         val template = templateRender.getTemplate("gen-entity.vm")
-        templateRender.context = GenEntityContext(
+        templateRender.context = SqlMigrationContext(
             lang = file.language.displayName ?: "",
             sql = selectedText,
         )
@@ -48,7 +49,3 @@ class GenerateEntityAction : AbstractChatIntention() {
     }
 }
 
-data class GenEntityContext(
-    val lang: String = "",
-    var sql: String = "",
-)

exts/database/src/main/kotlin/cc/unitmesh/database/actions/GenerateFunctionAction.kt

@@ -1,5 +1,6 @@
 package cc.unitmesh.database.actions
 
+import cc.unitmesh.database.actions.base.SqlMigrationContext
 import cc.unitmesh.devti.gui.sendToChatWindow
 import cc.unitmesh.devti.intentions.action.base.AbstractChatIntention
 import cc.unitmesh.devti.provider.ContextPrompter
@@ -13,7 +14,7 @@ import com.intellij.sql.dialects.oracle.OraDialect
 class GenerateFunctionAction : AbstractChatIntention() {
     override fun priority() = 901
 
-    override fun getFamilyName(): String = "Generate Java Function"
+    override fun getFamilyName(): String = "PL/SQL Migration"
 
     override fun getText(): String = "Generate Java Function"
 
@@ -38,7 +39,7 @@ class GenerateFunctionAction : AbstractChatIntention() {
 
         val templateRender = TemplateRender("genius/migration")
         val template = templateRender.getTemplate("gen-function.vm")
-        templateRender.context = GenFunctionContext(
+        templateRender.context = SqlMigrationContext(
             lang = file.language.displayName ?: "",
             sql = selectedText,
         )
@@ -57,5 +58,4 @@ class GenerateFunctionAction : AbstractChatIntention() {
 
 }
 
-data class GenFunctionContext(val lang: String, val sql: String)
 

exts/database/src/main/kotlin/cc/unitmesh/database/actions/GenerateUnittestAction.kt

@@ -0,0 +1,49 @@
+package cc.unitmesh.database.actions
+
+import cc.unitmesh.database.actions.base.SqlMigrationContext
+import cc.unitmesh.devti.gui.sendToChatWindow
+import cc.unitmesh.devti.intentions.action.base.AbstractChatIntention
+import cc.unitmesh.devti.provider.ContextPrompter
+import cc.unitmesh.devti.template.TemplateRender
+import com.intellij.openapi.diagnostic.logger
+import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiFile
+import com.intellij.sql.dialects.oracle.OraDialect
+
+class GenerateUnittestAction : AbstractChatIntention() {
+    override fun priority(): Int = 899
+
+    override fun getFamilyName(): String = "PL/SQL Migration"
+
+    override fun getText(): String = "Generate Java Unit Test"
+
+    override fun isAvailable(project: Project, editor: Editor?, file: PsiFile?): Boolean {
+        if (editor == null || file == null) return false
+        return file.language is OraDialect
+    }
+
+    val logger = logger<GenerateUnittestAction>()
+
+    override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {
+        if (editor == null || file == null) return
+        val selectedText = editor.selectionModel.selectedText ?: return
+
+        val templateRender = TemplateRender("genius/migration")
+        val template = templateRender.getTemplate("gen-unittest.vm")
+        templateRender.context = SqlMigrationContext(
+            lang = file.language.displayName ?: "",
+            sql = selectedText,
+        )
+        val prompter = templateRender.renderTemplate(template)
+
+        logger.info("Prompt: $prompter")
+
+        sendToChatWindow(project, getActionType()) { panel, service ->
+            service.handlePromptAndResponse(panel, object : ContextPrompter() {
+                override fun displayPrompt(): String = prompter
+                override fun requestPrompt(): String = prompter
+            }, null, false)
+        }
+    }
+}

exts/database/src/main/kotlin/cc/unitmesh/database/actions/base/SqlMigrationContext.kt

@@ -0,0 +1,6 @@
+package cc.unitmesh.database.actions.base
+
+data class SqlMigrationContext(
+    val lang: String = "",
+    var sql: String = "",
+)

exts/database/src/main/resources/cc.unitmesh.database.xml

@@ -15,6 +15,11 @@
             <bundleName>messages.AutoDevBundle</bundleName>
             <categoryKey>intention.category.llm</categoryKey>
         </autoDevIntention>
+        <autoDevIntention>
+            <className>cc.unitmesh.database.actions.GenerateUnittestAction</className>
+            <bundleName>messages.AutoDevBundle</bundleName>
+            <categoryKey>intention.category.llm</categoryKey>
+        </autoDevIntention>
 
         <livingDocumentation
                 language="SQL"

exts/database/src/main/resources/genius/migration/gen-unittest.vm

@@ -0,0 +1,12 @@
+You are a professional application migration programmer.
+Write java unit test code to test the following Oracle PL/SQL code.
+
+— When some function is missing in Java, just skip it.
+- If you find some function is not correct, please fix it.
+- Follow the Java coding style.
+
+```${context.lang}
+${context.sql}
+```
+
+