fix(java): add Java language check in AutoCrudAction

Author: phodal

exts/database/src/main/kotlin/cc/unitmesh/database/provider/SqlLivingDocumentationProvider.kt

@@ -8,6 +8,7 @@ import com.intellij.openapi.editor.SelectionModel
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiNameIdentifierOwner
 import com.intellij.psi.codeStyle.CodeStyleManager
+import com.intellij.sql.psi.SqlDefinition
 
 class SqlLivingDocumentationProvider : LivingDocumentation {
     override val forbiddenRules: List<String>
@@ -34,6 +35,13 @@ class SqlLivingDocumentationProvider : LivingDocumentation {
     }
 
     override fun findNearestDocumentationTarget(psiElement: PsiElement): PsiNameIdentifierOwner? {
+        // todo: find nearest sql definition
+        when (psiElement) {
+            is SqlDefinition -> {
+                return psiElement
+            }
+        }
+
         return null
     }
 

exts/database/src/test/resources/create_blog.sql

@@ -0,0 +1,10 @@
+-- sample for create blog
+CREATE TABLE `blog`
+(
+    `id`          int(11) NOT NULL AUTO_INCREMENT,
+    `title`       varchar(255) DEFAULT NULL,
+    `content`     varchar(255) DEFAULT NULL,
+    `create_time` datetime     DEFAULT NULL,
+    `update_time` datetime     DEFAULT NULL,
+    PRIMARY KEY (`id`)
+) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4;

java/src/main/kotlin/cc/unitmesh/idea/actions/AutoCrudAction.kt

@@ -5,6 +5,7 @@ import cc.unitmesh.devti.intentions.action.base.AbstractChatIntention
 import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.provider.DevFlowProvider
 import cc.unitmesh.devti.gui.sendToChatPanel
+import com.intellij.lang.java.JavaLanguage
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.progress.ProgressIndicator
@@ -22,8 +23,11 @@ class AutoCrudAction : AbstractChatIntention() {
 
 
     override fun isAvailable(project: Project, editor: Editor?, file: PsiFile?): Boolean {
+        if (editor == null || file == null) return false
+        if (file.language !is JavaLanguage) return false
+
         val isEnvironmentAvailable = super.isAvailable(project, editor, file)
-        return isEnvironmentAvailable && editor?.selectionModel?.hasSelection() == true
+        return isEnvironmentAvailable && editor.selectionModel?.hasSelection() == true
     }
 
     override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {