refactor(editor): auto refresh if 'drop/create database' (#547)

* refactor: auto refresh if 'drop/create database'

* style: remove

* refactor: simplified

Author: ZonaHex

src/hooks/query-code.ts

@@ -58,6 +58,30 @@ export default function useQueryCode() {
     })
   }
 
+  // Parse SQL command to extract operation details
+  const parseSqlCommand = (sql: string) => {
+    const regex =
+      /\bSHOW\s+CREATE\s+TABLE\b|(?:CREATE|DROP|ALTER)\s+(DATABASE|TABLE)\s+(?:IF\s+(?:NOT\s+)?EXISTS\s+)?[`"]?([\w.-]+)[`"]?/gi
+
+    // Reset regex state to ensure proper matching
+    regex.lastIndex = 0
+    const match = regex.exec(sql)
+
+    // Return immediately after finding the first match
+    if (match && match[1]) {
+      // Extract action from the full match
+      const fullMatch = match[0]
+      const action = fullMatch.split(/\s+/)[0].toUpperCase() // First word is the action
+      return {
+        action,
+        object: match[1].toUpperCase(),
+        name: match[2],
+      }
+    }
+
+    return null
+  }
+
   const runQuery = async (
     code: string,
     type = queryType.value,
@@ -72,45 +96,36 @@ export default function useQueryCode() {
     }
     if (!res.error && type === 'sql') {
       const sql = sqlFormatter(code)
+      const command = parseSqlCommand(sql)
 
-      const regex =
-        /CREATE TABLE IF NOT EXISTS(\s)+(\S)+|CREATE TABLE(\s)+(\S)+|DROP TABLE(\s)+(\S)+|ALTER TABLE(\s)+(\S)+/g
-      const matchString = sql.match(regex)?.[0] || ''
+      if (command) {
+        const { refreshTables, loadMoreColumns } = useSiderTabs()
+        const { originTablesTree } = storeToRefs(useDataBaseStore())
+        const { fetchDatabases } = useAppStore()
 
-      const { refreshTables, loadMoreColumns, loadMoreDetails } = useSiderTabs()
-      const { originTablesTree } = storeToRefs(useDataBaseStore())
+        const commandType = `${command.action}_${command.object}`
 
-      if (matchString !== '') {
-        const matchArray = matchString.split(' ')
-        let tableName = matchArray[matchArray.length - 1]
-        if (tableName.startsWith(`'`) || tableName.startsWith(`"`)) {
-          tableName = tableName.slice(1, tableName.length - 1)
-        }
-        // TODO: What about `show create table`?
-        if (
-          (matchString.includes('CREATE TABLE') && !matchString.includes('IF NOT EXISTS')) ||
-          matchString.includes('DROP TABLE')
-        ) {
-          // CREATE NEW TABLE OR DROP TABLE
-          refreshTables()
-        } else if (!matchString.includes('ALTER TABLE')) {
-          // CREATE TABLE IF NOT EXIST
-          const isNewTable =
-            originTablesTree.value.findIndex((item: TableTreeParent) => item.title === tableName) === -1
-          if (isNewTable) {
+        switch (commandType) {
+          case 'CREATE_DATABASE':
+          case 'DROP_DATABASE':
+            await fetchDatabases()
+            break
+          case 'CREATE_TABLE':
+          case 'DROP_TABLE':
             refreshTables()
+            break
+          case 'ALTER_TABLE': {
+            const tableNodeData = originTablesTree.value.find((item: TableTreeParent) => item.title === command.name)
+            if (tableNodeData) {
+              await loadMoreColumns(tableNodeData, true)
+              // Update new children
+              originTablesTree.value[tableNodeData.key].children =
+                originTablesTree.value[tableNodeData.key][tableNodeData.childrenType]
+            }
+            break
           }
-        } else {
-          // ALTER TABLE
-          const tableNodeData = originTablesTree.value.find((item: TableTreeParent) => item.title === tableName)
-          if (tableNodeData) {
-            // Silently load more
-            await loadMoreColumns(tableNodeData, true)
-            // await loadMoreDetails(tableNodeData, true)
-            // Update new children
-            originTablesTree.value[tableNodeData.key].children =
-              originTablesTree.value[tableNodeData.key][tableNodeData.childrenType]
-          }
+          default:
+            break
         }
       }
     }
@@ -145,6 +160,7 @@ export default function useQueryCode() {
       throw enhancedError
     }
   }
+
   return {
     getResultsByType,
     runQuery,