feat: init for sql ddl

unit-mesh · Jul 12, 2023 · 43970b1 · 43970b1
1 parent 6553535
commit 43970b1
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/actions/chat/ChatBaseAction.kt b/src/main/kotlin/cc/unitmesh/devti/actions/chat/ChatBaseAction.kt
@@ -1,5 +1,6 @@
 package cc.unitmesh.devti.actions.chat
 
+import cc.unitmesh.devti.gui.DevtiFlowToolWindowFactory
 import cc.unitmesh.devti.prompting.JavaActionPrompting
 import cc.unitmesh.devti.gui.chat.ChatBotActionType
 import cc.unitmesh.devti.gui.chat.ChatCodingComponent
@@ -11,8 +12,12 @@ import com.intellij.openapi.wm.ToolWindowManager
 
 abstract class ChatBaseAction : AnAction() {
     override fun actionPerformed(event: AnActionEvent) {
+        executeAction(event)
+    }
+
+    open fun executeAction(event: AnActionEvent) {
         val project = event.project
-        val toolWindowManager = ToolWindowManager.getInstance(project!!).getToolWindow("DevTiFlow")
+        val toolWindowManager = ToolWindowManager.getInstance(project!!).getToolWindow(DevtiFlowToolWindowFactory.id)
         val contentManager = toolWindowManager?.contentManager
 
         val caretModel = event.getData(CommonDataKeys.EDITOR)?.caretModel
@@ -41,7 +46,6 @@ abstract class ChatBaseAction : AnAction() {
             JavaActionPrompting(chatCodingService.actionType, lang, selectedText, file, project),
             getReplaceableAction(event)
         )
-
     }
 
     open fun getReplaceableAction(event: AnActionEvent): ((response: String) -> Unit)? {

diff --git a/src/main/kotlin/cc/unitmesh/devti/actions/chat/CreateDdlAction.kt b/src/main/kotlin/cc/unitmesh/devti/actions/chat/CreateDdlAction.kt
@@ -1,11 +1,20 @@
 package cc.unitmesh.devti.actions.chat
 
 import cc.unitmesh.devti.gui.chat.ChatBotActionType
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.actionSystem.CommonDataKeys.PSI_FILE
 
 class CreateDdlAction : ChatBaseAction() {
     override fun getActionType(): ChatBotActionType {
         return ChatBotActionType.CREATE_DDL
     }
 
+    override fun actionPerformed(event: AnActionEvent) {
+        val fileType = event.getData(PSI_FILE)?.fileType?.name
+        if (fileType != "SQL") {
+            return
+        }
 
+        super.actionPerformed(event)
+    }
 }
diff --git a/src/main/kotlin/cc/unitmesh/devti/gui/DevtiFlowToolWindowFactory.kt b/src/main/kotlin/cc/unitmesh/devti/gui/DevtiFlowToolWindowFactory.kt
@@ -12,6 +12,10 @@ import com.intellij.openapi.wm.ToolWindowFactory
 import com.intellij.ui.content.ContentFactory
 
 class DevtiFlowToolWindowFactory : ToolWindowFactory, DumbAware {
+    companion object {
+        val id = "DevTiFlow"
+    }
+
     private val contentFactory = ApplicationManager.getApplication().getService(
         ContentFactory::class.java
     )