sqlite-hub 1.2.0 → 1.5.0

package/server/services/databaseCommandService.js

@@ -1,10 +1,12 @@
 const path = require("node:path");
 const { NotFoundError, ReadOnlyError, ValidationError } = require("../utils/errors");
 const { ConnectionManager } = require("./sqlite/connectionManager");
+const { BackupService } = require("./sqlite/backupService");
 const { DataBrowserService } = require("./sqlite/dataBrowserService");
 const { ExportService } = require("./sqlite/exportService");
 const { getTableDetail } = require("./sqlite/introspection");
 const { SqlExecutor } = require("./sqlite/sqlExecutor");
+const { TypeGenerationService } = require("./typeGenerationService");
 
 function normalizeLookupValue(value, label) {
   const normalized = String(value ?? "").trim();
@@ -166,6 +168,7 @@ class DatabaseCommandService {
   constructor({ appStateStore, runtimeFactory } = {}) {
     this.appStateStore = appStateStore;
     this.runtimeFactory = runtimeFactory ?? ((connection, options) => this.createRuntime(connection, options));
+    this.typeGenerationService = new TypeGenerationService();
   }
 
   listDatabases() {
@@ -239,6 +242,17 @@ class DatabaseCommandService {
     }
   }
 
+  async withDatabaseAsync(databaseReference, callback, options = {}) {
+    const connection = this.getDatabase(databaseReference);
+    const runtime = this.runtimeFactory(connection, options);
+
+    try {
+      return await callback({ connection, runtime });
+    } finally {
+      runtime.close?.();
+    }
+  }
+
   listTables(databaseReference) {
     return this.withDatabase(databaseReference, ({ runtime }) => runtime.dataBrowserService.listTables());
   }
@@ -250,6 +264,18 @@ class DatabaseCommandService {
     );
   }
 
+  generateTableTypes(databaseReference, tableName, target, options = {}) {
+    const normalizedTableName = normalizeLookupValue(tableName, "Table name");
+    return this.withDatabase(databaseReference, ({ runtime }) =>
+      this.typeGenerationService.generateTypesFromDatabase(
+        runtime.db,
+        normalizedTableName,
+        target,
+        options
+      )
+    );
+  }
+
   getTableRow(databaseReference, tableName, exportTarget) {
     const normalizedTableName = normalizeLookupValue(tableName, "Table name");
     normalizeLookupValue(exportTarget, "Row key");
@@ -367,6 +393,39 @@ class DatabaseCommandService {
     };
   }
 
+  async createBackup(databaseReference, options = {}) {
+    const backupOptions = {
+      name: normalizeOptionalValue(options.name),
+      notes: normalizeOptionalValue(options.notes),
+      type: normalizeOptionalValue(options.type) ?? "manual",
+      context: normalizeOptionalValue(options.context) ?? "automation",
+    };
+
+    return this.withDatabaseAsync(
+      databaseReference,
+      async ({ runtime }) => {
+        const backupService = new BackupService({
+          appStateStore: this.appStateStore,
+          connectionManager: runtime.connectionManager,
+        });
+
+        return backupService.createActiveBackup(backupOptions);
+      },
+      { readOnly: true }
+    );
+  }
+
+  listBackups(databaseReference) {
+    return this.withDatabase(databaseReference, ({ connection, runtime }) => {
+      const backupService = new BackupService({
+        appStateStore: this.appStateStore,
+        connectionManager: runtime.connectionManager,
+      });
+
+      return backupService.listBackups({ connectionId: connection.id });
+    });
+  }
+
   exportSavedQuery(databaseReference, queryName, format = "csv") {
     const query = this.requireQuery(databaseReference, queryName);
     const result = this.withDatabase(databaseReference, ({ runtime }) =>