feat(database): add support for parsing and verifying SQL scripts before inserting them into the editor.

This commit introduces a new feature to the database extension that enhances the user experience by adding support for parsing and verifying SQL scripts before they are inserted into the editor. This ensures that the scripts are syntactically correct and adhere to the specified dialect, improving the overall quality and reliability of the SQL code.

Author: phodal

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

@@ -16,6 +16,7 @@ import com.intellij.openapi.progress.ProgressManager
 import com.intellij.openapi.progress.impl.BackgroundableProcessIndicator
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiFile
+import com.intellij.sql.dialects.sqlite.SqliteDialect
 
 
 class AutoSqlAction : ChatBaseIntention() {
@@ -43,7 +44,9 @@ class AutoSqlAction : ChatBaseIntention() {
         val schemaName = rawDataSource.name.substringBeforeLast('@')
         val dasTables = rawDataSource.let {
             val tables = DasUtil.getTables(it)
-            tables.filter { table -> table.kind == ObjectKind.TABLE && table.dasParent?.name == schemaName }
+            tables.filter { table -> table.kind == ObjectKind.TABLE &&
+                    (table.dasParent?.name == schemaName || (file.language == SqliteDialect.INSTANCE && table.dasParent?.name == "main"))
+            }
         }.toList()
 
         val genSqlContext = AutoSqlContext(

exts/database/src/main/kotlin/cc/unitmesh/database/flow/AutoSqlBackgroundTask.kt

@@ -3,6 +3,7 @@ package cc.unitmesh.database.flow
 import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.util.parser.parseCodeFromString
 import com.intellij.lang.Language
+import com.intellij.openapi.application.runReadAction
 import com.intellij.openapi.command.WriteCommandAction
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.editor.Editor
@@ -45,9 +46,11 @@ class AutoSqlBackgroundTask(
         val sqlScript = flow.design(tableNames)[0]
 
         try {
-            val sqlFile =
-                PsiFileFactory.getInstance(project).createFileFromText("temp.sql", language, sqlScript)
+            val sqlCode = parseCodeFromString(sqlScript).first()
+            val sqlFile = runReadAction {
+                PsiFileFactory.getInstance(project).createFileFromText("temp.sql", language, sqlCode)
                         as SqlFile
+            }
 
             val errors = sqlFile.verifySqlElement()
             if (errors.isNotEmpty()) {
@@ -89,6 +92,8 @@ fun SqlFile.verifySqlElement(): MutableList<String> {
 
 abstract class SqlSyntaxCheckingVisitor : com.intellij.psi.PsiElementVisitor() {
     override fun visitElement(element: PsiElement) {
-        element.children.forEach { it.accept(this) }
+        runReadAction {
+            element.children.forEach { it.accept(this) }
+        }
     }
 }