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

This commit introduces a new feature to the devins-lang project that enhances the user experience when working with SQL scripts. The commit enables the parsing and verification of SQL scripts before they are inserted into the editor, ensuring that the scripts are syntactically correct and free from errors. This feature significantly improves the quality of the generated SQL code and reduces the likelihood of runtime issues.

The commit also includes a new method `fix` in the `TaskFlow` class, which is used to address any errors found in the SQL scripts. This method is designed to be flexible and can be extended to handle a variety of error-fixing scenarios.

By implementing these changes, the devins-lang project now provides a more robust and user-friendly environment for developers to work with SQL scripts.

Author: phodal

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

@@ -44,28 +44,31 @@ class AutoSqlBackgroundTask(
         indicator.text = AutoDevBundle.message("autosql.generate.generate")
         val sqlScript = flow.design(tableNames)[0]
 
-        logger.info("SQL Script: $sqlScript")
-        // verify sql script with parser
         try {
             val sqlFile =
                 PsiFileFactory.getInstance(project).createFileFromText("temp.sql", language, sqlScript)
                         as SqlFile
 
             val errors = sqlFile.verifySqlElement()
             if (errors.isNotEmpty()) {
-                logger.error("SQL Script parse error: ${errors[0]}")
+                val response = flow.fix(errors.joinToString("\n"))
+                val code = parseCodeFromString(response).last()
+                writeToFile(code, indicator)
             }
         } catch (e: Exception) {
             logger.error("SQL Script parse error: $e")
         }
 
+        writeToFile(sqlScript, indicator)
+        indicator.fraction = 1.0
+    }
+
+    private fun writeToFile(sqlScript: String, indicator: ProgressIndicator) {
         WriteCommandAction.runWriteCommandAction(project, "Gen SQL", "cc.unitmesh.livingDoc", {
             editor.document.insertString(editor.caretModel.offset, "\n")
             val code = parseCodeFromString(sqlScript).first()
             editor.document.insertString(editor.caretModel.offset + "\n".length, code)
         })
-
-        indicator.fraction = 1.0
     }
 }
 

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

@@ -74,6 +74,16 @@ class AutoSqlFlow(
         return prompter
     }
 
+    override fun fix(errors: String): String {
+        panel.addMessage(errors, true, errors)
+        panel.addMessage(AutoDevBundle.message("autodev.loading"))
+
+        return runBlocking {
+            val prompt = llm.stream(errors, "")
+            return@runBlocking panel.updateMessage(prompt)
+        }
+    }
+
     fun getAllTables(): List<String> {
         return actions.dasTables.map { it.name }
     }

src/main/kotlin/cc/unitmesh/devti/flow/TaskFlow.kt

@@ -38,4 +38,14 @@ interface TaskFlow<Tasking> {
     fun execute(context: Any): String {
         return ""
     }
+
+    /**
+     * This method is used to fix the errors in the task flow.
+     *
+     * @param errors The errors that need to be fixed.
+     * @return A string representing the fixed errors.
+     */
+    fun fix(errors: String): String {
+        return ""
+    }
 }