feat(export): support database setup and batch SQL scripts

apps/desktop/src/components/export/DatabaseExportDialog.vue

@@ -59,6 +59,7 @@ const tableError = ref<string | null>(null);
 const includeStructure = ref(true);
 const includeData = ref(true);
 const includeObjects = ref(true);
+const includeCreateDatabase = ref(false);
 const dropTableIfExists = ref(false);
 const omitAutoIncrement = ref(false);
 // `AUTO_INCREMENT` stripping is a MySQL-only DDL transform (backend gates on
@@ -77,6 +78,7 @@ const pendingPrefillTables = ref<string[]>([]);
 const exportAllDatabases = ref(false);
 const batchDatabaseIndex = ref(0);
 const batchDatabaseTotal = ref(0);
+const batchRowsExported = ref(0);
 const activeDatabaseExportId = ref("");
 
 const sqlConnections = computed(() => store.connections.filter((c) => !["redis", "mongodb", "elasticsearch", "qdrant", "milvus", "weaviate", "chromadb", "etcd", "zookeeper", "mq", "nacos"].includes(c.db_type)));
@@ -110,11 +112,6 @@ function joinExportPath(directory: string, fileName: string): string {
   return `${directory.replace(/[\\/]+$/, "")}${separator}${fileName}`;
 }
 
-function currentRowsExported(): number {
-  const progress: ExportProgress | null = exportProgress.value;
-  return progress?.rowsExported ?? 0;
-}
-
 async function loadDatabases(connId: string) {
   if (!connId) return;
   loadingMeta.value = true;
@@ -276,6 +273,7 @@ async function startExport() {
     includeStructure: includeStructure.value,
     includeData: includeData.value,
     includeObjects: includeObjects.value,
+    includeCreateDatabase: includeCreateDatabase.value,
     dropTableIfExists: dropTableIfExists.value,
     omitAutoIncrement: omitAutoIncrement.value,
     batchSize: 1000,
@@ -343,6 +341,7 @@ async function startAllDatabasesExport() {
   exportCancelled.value = false;
   exportProgress.value = null;
   batchDatabaseIndex.value = 0;
+  batchRowsExported.value = 0;
 
   const dbs = [...selectedDatabases.value];
   const batchId = generateDatabaseExportId();
@@ -370,19 +369,30 @@ async function startAllDatabasesExport() {
         includeStructure: includeStructure.value,
         includeData: includeData.value,
         includeObjects: includeObjects.value,
+        includeCreateDatabase: includeCreateDatabase.value,
         dropTableIfExists: dropTableIfExists.value,
         omitAutoIncrement: omitAutoIncrement.value,
         batchSize: 1000,
       };
+      let currentDatabaseRowsExported = 0;
 
       await runDatabaseExportUntilTerminal(request, (progress) => {
-        exportProgress.value = { ...progress, exportId: batchId, currentObject: `${item.displayName}: ${progress.currentObject || item.displayName}` };
+        const nextRowsExported = Math.max(0, progress.rowsExported);
+        batchRowsExported.value += Math.max(0, nextRowsExported - currentDatabaseRowsExported);
+        currentDatabaseRowsExported = nextRowsExported;
+        exportProgress.value = {
+          ...progress,
+          exportId: batchId,
+          currentObject: `${item.displayName}: ${progress.currentObject || item.displayName}`,
+          rowsExported: batchRowsExported.value,
+        };
         updateDatabaseExportTask(batchId, {
           ...progress,
           exportId: batchId,
           currentObject: item.displayName,
           objectIndex: index,
           totalObjects: exportPlan.length,
+          rowsExported: batchRowsExported.value,
         });
         if (progress.status === "Error") {
           exportError.value = progress.error;
@@ -400,16 +410,18 @@ async function startAllDatabasesExport() {
     if (!exportError.value && !exportCancelled.value) {
       exportDone.value = true;
       isExporting.value = false;
-      updateDatabaseExportTask(batchId, {
+      const finalProgress: api.ExportProgress = {
         exportId: batchId,
         currentObject: t("databaseExport.allDatabasesTask", { count: dbs.length }),
         objectIndex: exportPlan.length,
         totalObjects: exportPlan.length,
-        rowsExported: currentRowsExported(),
+        rowsExported: batchRowsExported.value,
         totalRows: null,
         status: "Done",
         error: null,
-      });
+      };
+      exportProgress.value = finalProgress;
+      updateDatabaseExportTask(batchId, finalProgress);
       toast(t("databaseExport.exportAllSuccess", { count: dbs.length }), 3000);
     }
   } catch (e: any) {
@@ -419,7 +431,7 @@ async function startAllDatabasesExport() {
       currentObject: t("databaseExport.allDatabasesTask", { count: dbs.length }),
       objectIndex: Math.max(0, batchDatabaseIndex.value - 1),
       totalObjects: batchDatabaseTotal.value || dbs.length,
-      rowsExported: currentRowsExported(),
+      rowsExported: batchRowsExported.value,
       totalRows: null,
       status: "Error",
       error: exportError.value,
@@ -459,6 +471,7 @@ function resetState() {
   includeStructure.value = true;
   includeData.value = true;
   includeObjects.value = true;
+  includeCreateDatabase.value = false;
   dropTableIfExists.value = false;
   omitAutoIncrement.value = false;
   isExporting.value = false;
@@ -469,10 +482,17 @@ function resetState() {
   exportId.value = "";
   batchDatabaseIndex.value = 0;
   batchDatabaseTotal.value = 0;
+  batchRowsExported.value = 0;
   activeDatabaseExportId.value = "";
 }
 
 const progressPercent = computed(() => {
+  if (exportAllDatabases.value && batchDatabaseTotal.value > 0) {
+    if (exportDone.value) return 100;
+    const current = exportProgress.value;
+    const currentDatabaseProgress = current && current.totalObjects > 0 ? current.objectIndex / current.totalObjects : 0;
+    return Math.round(Math.min(1, (Math.max(0, batchDatabaseIndex.value - 1) + currentDatabaseProgress) / batchDatabaseTotal.value) * 100);
+  }
   const p = exportProgress.value;
   if (!p || p.totalObjects === 0) return 0;
   return Math.round((p.objectIndex / p.totalObjects) * 100);
@@ -679,6 +699,11 @@ watch(
               <Square v-else class="w-3.5 h-3.5 text-muted-foreground/40 shrink-0" />
               {{ t("databaseExport.includeStructure") }}
             </div>
+            <div v-if="isMysqlFamily" class="flex items-center gap-2 cursor-pointer text-xs" @click="includeCreateDatabase = !includeCreateDatabase">
+              <CheckSquare v-if="includeCreateDatabase" class="w-3.5 h-3.5 text-primary shrink-0" />
+              <Square v-else class="w-3.5 h-3.5 text-muted-foreground/40 shrink-0" />
+              {{ t("databaseExport.includeCreateDatabase") }}
+            </div>
             <div class="flex items-center gap-2 text-xs" :class="includeStructure ? 'cursor-pointer' : 'cursor-not-allowed text-muted-foreground/50'" @click="includeStructure && (dropTableIfExists = !dropTableIfExists)">
               <CheckSquare v-if="dropTableIfExists && includeStructure" class="w-3.5 h-3.5 text-primary shrink-0" />
               <Square v-else class="w-3.5 h-3.5 text-muted-foreground/40 shrink-0" />
@@ -708,7 +733,7 @@ watch(
             {{ t("databaseExport.currentDatabase", { current: batchDatabaseIndex, total: batchDatabaseTotal }) }}
           </div>
           <div v-if="exportProgress" class="space-y-2">
-            <div class="text-xs text-muted-foreground">
+            <div v-if="!exportAllDatabases || !exportDone" class="text-xs text-muted-foreground">
               {{
                 t("databaseExport.currentTable", {
                   table: exportProgress.currentObject,
@@ -723,7 +748,7 @@ watch(
             </div>
 
             <div class="text-xs text-muted-foreground">
-              {{ t("databaseExport.rowsExported", { count: exportProgress.rowsExported.toLocaleString() }) }}
+              {{ exportAllDatabases ? t("databaseExport.allRowsExported", { count: exportProgress.rowsExported.toLocaleString() }) : t("databaseExport.rowsExported", { current: exportProgress.objectIndex, total: exportProgress.totalObjects, count: exportProgress.rowsExported.toLocaleString() }) }}
             </div>
           </div>
 

apps/desktop/src/components/sql-file/SqlFileExecutionDialog.vue

@@ -17,7 +17,7 @@ import { useProductionSafetyStore } from "@/stores/productionSafetyStore";
 import { productionContextForDatabase } from "@/lib/database/productionSafety";
 import { fetchSqlFileTargetOptions } from "@/composables/useDatabaseOptions";
 import { requiresSqlFileTargetDatabaseSelection } from "@/lib/connection/connectionLevelDatabaseBootstrap";
-import { cancelSqlFileExecution, executeSqlFile, listenSqlFileProgress, previewSqlFile, type SqlFilePreview, type SqlFileProgress, type SqlFileStatus } from "@/lib/backend/api";
+import { cancelSqlFileExecution, executeSqlFiles, listenSqlFileProgress, previewSqlFile, type SqlFilePreview, type SqlFileProgress, type SqlFileStatus } from "@/lib/backend/api";
 import { useExportTracker } from "@/composables/useExportTracker";
 import { Check, CheckSquare, FileCode, FolderOpen, Loader2, Play, Square, X } from "@lucide/vue";
 
@@ -38,7 +38,7 @@ const productionSafetyStore = useProductionSafetyStore();
 
 const fileInput = ref<HTMLInputElement | null>(null);
 const filePath = ref("");
-const preview = ref<SqlFilePreview | null>(null);
+const previews = ref<SqlFilePreview[]>([]);
 const selectingFile = ref(false);
 const loadingPreview = ref(false);
 const connectionId = ref("");
@@ -56,6 +56,7 @@ const progress = ref<SqlFileProgress | null>(null);
 const terminalStatus = ref<SqlFileStatus | "idle">("idle");
 const terminalError = ref("");
 const refreshedTarget = ref(false);
+const MAX_WEB_SQL_FILE_BYTES = 200 * 1024 * 1024;
 
 const sqlConnections = computed(() => store.connections.filter((c) => !["redis", "mongodb", "elasticsearch", "qdrant", "milvus", "weaviate", "chromadb", "etcd", "zookeeper", "mq", "nacos"].includes(c.db_type)));
 
@@ -63,8 +64,14 @@ const selectedConnection = computed(() => sqlConnections.value.find((c) => c.id
 
 const canStart = computed(() => {
   const connection = selectedConnection.value;
-  if (!preview.value || !connection || running.value || loadingPreview.value || loadingDatabases.value) return false;
-  return !!database.value.trim() || !requiresSqlFileTargetDatabaseSelection(connection, preview.value.canExecuteWithoutSelectedDatabase);
+  if (previews.value.length === 0 || !connection || running.value || loadingPreview.value || loadingDatabases.value) return false;
+  let hasDatabaseContext = false;
+  const canExecuteWithoutSelectedDatabase = previews.value.every((item) => {
+    if (!hasDatabaseContext && !item.canExecuteWithoutSelectedDatabase) return false;
+    hasDatabaseContext ||= item.establishesDatabaseContext === true;
+    return true;
+  });
+  return !!database.value.trim() || !requiresSqlFileTargetDatabaseSelection(connection, canExecuteWithoutSelectedDatabase);
 });
 
 const statusTone = computed(() => {
@@ -90,6 +97,7 @@ const progressPercent = computed(() => {
   if (current <= 0) return running.value ? 8 : 0;
   return Math.min(95, Math.max(8, Math.round((attempted / current) * 100)));
 });
+const preview = computed(() => previews.value[0] ?? null);
 const previewLineCount = computed(() => preview.value?.preview.split(/\r\n|\r|\n/).length ?? 0);
 const previewLineNumbers = computed(() => Array.from({ length: previewLineCount.value }, (_, index) => index + 1));
 const previewIsTruncated = computed(() => {
@@ -127,10 +135,6 @@ function statusLabel(status: SqlFileStatus | "idle") {
   return t(`sqlFile.status.${status}`);
 }
 
-function isTerminalStatus(status: SqlFileStatus | "idle") {
-  return status === "done" || status === "error" || status === "cancelled";
-}
-
 function resolveInitialConnectionId() {
   if (props.prefillConnectionId && sqlConnections.value.some((c) => c.id === props.prefillConnectionId)) {
     return props.prefillConnectionId;
@@ -162,7 +166,7 @@ function resetExecution() {
 
 function resetState() {
   filePath.value = "";
-  preview.value = null;
+  previews.value = [];
   selectingFile.value = false;
   loadingPreview.value = false;
   connectionId.value = resolveInitialConnectionId();
@@ -209,17 +213,26 @@ async function previewSelectedSqlFile(fileOrPath: string | File) {
   if (isTauriRuntime()) {
     return previewSqlFile(fileOrPath as string);
   }
+  const file = fileOrPath as File;
+  if (file.size > MAX_WEB_SQL_FILE_BYTES) {
+    throw new Error(`File too large: ${file.size} bytes (max ${MAX_WEB_SQL_FILE_BYTES} bytes)`);
+  }
   const { previewSqlFile: previewWebSqlFile } = await import("@/lib/backend/http");
-  return previewWebSqlFile(fileOrPath as File);
+  return previewWebSqlFile(file);
 }
 
-async function loadPreview(fileOrPath: string | File) {
+async function loadPreviews(filesOrPaths: Array<string | File>) {
   loadingPreview.value = true;
-  preview.value = null;
+  previews.value = [];
+  filePath.value = "";
+  resetExecution();
   try {
-    preview.value = await previewSelectedSqlFile(fileOrPath);
-    filePath.value = preview.value.filePath;
-    resetExecution();
+    const nextPreviews: SqlFilePreview[] = [];
+    for (const fileOrPath of filesOrPaths) {
+      nextPreviews.push(await previewSelectedSqlFile(fileOrPath));
+    }
+    previews.value = nextPreviews;
+    filePath.value = nextPreviews.map((item) => item.filePath).join("; ");
   } catch (e: any) {
     toast(e?.message || String(e), 5000);
   } finally {
@@ -237,11 +250,12 @@ async function selectFile() {
   try {
     const { open } = await import("@tauri-apps/plugin-dialog");
     const selected = await open({
-      multiple: false,
+      multiple: true,
       filters: [{ name: "SQL", extensions: ["sql"] }],
     });
-    if (typeof selected === "string") {
-      await loadPreview(selected);
+    const paths = Array.isArray(selected) ? selected : selected ? [selected] : [];
+    if (paths.length > 0) {
+      await loadPreviews(paths);
     }
   } catch (e: any) {
     toast(e?.message || String(e), 5000);
@@ -252,23 +266,27 @@ async function selectFile() {
 
 async function handleFileInputChange(event: Event) {
   const input = event.target as HTMLInputElement;
-  const file = input.files?.[0];
+  const files = Array.from(input.files ?? []);
   input.value = "";
-  if (!file || running.value) return;
+  if (files.length === 0 || running.value) return;
   selectingFile.value = true;
   try {
-    await loadPreview(file);
+    await loadPreviews(files);
   } finally {
     selectingFile.value = false;
   }
 }
 
-async function listenProgress(id: string, handler: (next: SqlFileProgress) => void): Promise<() => void> {
+async function listenProgress(id: string, handler: (next: SqlFileProgress) => void, onError?: (error: Error) => void): Promise<() => void> {
   if (isTauriRuntime()) {
     return listenSqlFileProgress(handler);
   }
   const { listenSqlFileProgressById } = await import("@/lib/sql/httpSqlFileProgress");
-  return listenSqlFileProgressById(id, handler);
+  return listenSqlFileProgressById(id, handler, onError);
+}
+
+function isTerminalProgress(status: SqlFileStatus): boolean {
+  return status === "done" || status === "error" || status === "cancelled";
 }
 
 async function refreshTargetAfterImport() {
@@ -282,12 +300,15 @@ async function refreshTargetAfterImport() {
 }
 
 async function startExecution() {
-  if (!canStart.value || !preview.value) return;
+  if (!canStart.value || previews.value.length === 0) return;
   const productionContext = productionContextForDatabase(selectedConnection.value, database.value);
   if (productionContext.active) {
     // File previews are truncated, so production file execution is always reviewed instead of inferring safety from a partial preview.
     const confirmed = await productionSafetyStore.requestConfirmation({
-      sql: preview.value.preview,
+      sql: previews.value
+        .map((item) => item.preview)
+        .join("\n\n")
+        .slice(0, 20_000),
       connectionName: selectedConnection.value?.name,
       database: database.value,
       productionDatabases: productionContext.databases,
@@ -296,8 +317,8 @@ async function startExecution() {
     if (!confirmed) return;
   }
 
-  const id = uuid();
-  executionId.value = id;
+  const batchId = uuid();
+  executionId.value = batchId;
   running.value = true;
   cancelling.value = false;
   cancelRequested.value = false;
@@ -305,9 +326,9 @@ async function startExecution() {
   terminalStatus.value = "running";
   terminalError.value = "";
   progress.value = null;
-  addSqlFileTask(id, preview.value.fileName, preview.value.filePath);
+  const taskLabel = previews.value.length === 1 ? previews.value[0]!.fileName : `${previews.value[0]!.fileName} (+${previews.value.length - 1})`;
+  addSqlFileTask(batchId, taskLabel, filePath.value);
 
-  let unlisten: (() => void) | undefined;
   try {
     await store.ensureConnected(connectionId.value);
     if (cancelRequested.value) {
@@ -315,58 +336,60 @@ async function startExecution() {
       return;
     }
 
-    unlisten = await listenProgress(id, (next) => {
-      if (next.executionId !== id) return;
-      progress.value = next;
-      terminalStatus.value = next.status;
-      terminalError.value = next.error ?? terminalError.value;
-      updateSqlFileTask(id, next);
-      if (isTerminalStatus(next.status)) {
-        running.value = false;
-        cancelling.value = false;
-      }
-      if (next.status === "done") {
-        void refreshTargetAfterImport();
-      }
+    let resolveTerminalProgress: (progress: SqlFileProgress) => void = () => {};
+    let rejectTerminalProgress: (error: Error) => void = () => {};
+    const terminalProgress = new Promise<SqlFileProgress>((resolve, reject) => {
+      resolveTerminalProgress = resolve;
+      rejectTerminalProgress = reject;
     });
+    let completedSuccessfully = false;
+    const unlisten = await listenProgress(
+      batchId,
+      (next) => {
+        if (next.executionId !== batchId) return;
+        progress.value = next;
+        terminalStatus.value = next.status;
+        terminalError.value = next.error ?? terminalError.value;
+        updateSqlFileTask(batchId, next);
+        if (isTerminalProgress(next.status)) {
+          resolveTerminalProgress(next);
+        }
+      },
+      rejectTerminalProgress,
+    );
 
-    if (cancelRequested.value) {
-      terminalStatus.value = "cancelled";
-      return;
+    try {
+      executionStarted.value = true;
+      await executeSqlFiles(
+        {
+          executionId: batchId,
+          connectionId: connectionId.value,
+          database: database.value.trim(),
+          filePath: previews.value[0]!.filePath,
+          continueOnError: continueOnError.value,
+        },
+        previews.value.map((item) => item.filePath),
+      );
+      const terminal = await terminalProgress;
+      if (terminal.status === "error") {
+        throw new Error(terminal.error || "SQL file execution failed");
+      }
+      if (terminal.status === "cancelled") {
+        cancelRequested.value = true;
+      }
+      completedSuccessfully = terminal.status === "done";
+    } finally {
+      executionStarted.value = false;
+      unlisten();
     }
 
-    executionStarted.value = true;
-    await executeSqlFile({
-      executionId: id,
-      connectionId: connectionId.value,
-      database: database.value.trim(),
-      filePath: preview.value.filePath,
-      continueOnError: continueOnError.value,
-    });
-    if (!isTerminalStatus(terminalStatus.value)) {
-      terminalStatus.value = cancelRequested.value ? "cancelled" : "done";
-      const lastProgress = progress.value as SqlFileProgress | null;
-      updateSqlFileTask(id, {
-        executionId: id,
-        status: terminalStatus.value,
-        statementIndex: lastProgress?.statementIndex ?? 0,
-        successCount: lastProgress?.successCount ?? 0,
-        failureCount: lastProgress?.failureCount ?? 0,
-        affectedRows: lastProgress?.affectedRows ?? 0,
-        elapsedMs: lastProgress?.elapsedMs ?? 0,
-        statementSummary: lastProgress?.statementSummary ?? "",
-        error: lastProgress?.error ?? null,
-      });
-      if (terminalStatus.value === "done") {
-        await refreshTargetAfterImport();
-      }
-    }
+    if (completedSuccessfully) await refreshTargetAfterImport();
   } catch (e: any) {
     terminalStatus.value = cancelRequested.value ? "cancelled" : "error";
     terminalError.value = e?.message || String(e);
     const lastProgress = progress.value as SqlFileProgress | null;
-    updateSqlFileTask(id, {
-      executionId: id,
+    updateSqlFileTask(batchId, {
+      executionId: batchId,
       status: terminalStatus.value,
       statementIndex: lastProgress?.statementIndex ?? 0,
       successCount: lastProgress?.successCount ?? 0,
@@ -380,7 +403,6 @@ async function startExecution() {
       toast(terminalError.value, 5000);
     }
   } finally {
-    unlisten?.();
     running.value = false;
     cancelling.value = false;
     executionStarted.value = false;
@@ -429,7 +451,7 @@ watch(
     // When opened from the SQL Files panel with a pre-selected file, load its
     // preview automatically so the user can review statements before running.
     if (props.prefillFilePath) {
-      void loadPreview(props.prefillFilePath);
+      void loadPreviews([props.prefillFilePath]);
     }
   },
   { immediate: true },
@@ -454,7 +476,7 @@ watch(
           </div>
 
           <div class="flex items-center gap-2">
-            <input ref="fileInput" type="file" accept=".sql,text/sql" class="hidden" @change="handleFileInputChange" />
+            <input ref="fileInput" type="file" accept=".sql,text/sql" multiple class="hidden" @change="handleFileInputChange" />
             <Input :model-value="filePath" readonly class="h-8 text-xs font-mono" :placeholder="t('sqlFile.selectSqlFile')" />
             <Button variant="outline" size="sm" class="h-8 shrink-0" :disabled="running || selectingFile" @click="selectFile">
               <Loader2 v-if="selectingFile || loadingPreview" class="w-3.5 h-3.5 mr-1.5 animate-spin" />
@@ -468,6 +490,7 @@ watch(
               <div class="min-w-0 flex items-center gap-2">
                 <FileCode class="w-3.5 h-3.5 text-muted-foreground shrink-0" />
                 <span class="font-medium truncate">{{ preview.fileName }}</span>
+                <span v-if="previews.length > 1" class="shrink-0 text-muted-foreground">(+{{ previews.length - 1 }})</span>
               </div>
               <div class="flex shrink-0 items-center gap-2 text-muted-foreground">
                 <span>{{ previewLineSummary }}</span>