spindb 0.5.4 → 0.5.5

package/engines/postgresql/backup.ts

@@ -0,0 +1,93 @@
+/**
+ * PostgreSQL Backup
+ *
+ * Creates database backups in SQL or custom (.dump) format using pg_dump.
+ */
+
+import { spawn } from 'child_process'
+import { stat } from 'fs/promises'
+import { configManager } from '../../core/config-manager'
+import { defaults } from '../../config/defaults'
+import type { ContainerConfig, BackupOptions, BackupResult } from '../../types'
+
+/**
+ * Get pg_dump path from config, with helpful error message
+ */
+async function getPgDumpPath(): Promise<string> {
+  const pgDumpPath = await configManager.getBinaryPath('pg_dump')
+  if (!pgDumpPath) {
+    throw new Error(
+      'pg_dump not found. Install PostgreSQL client tools:\n' +
+        '  macOS: brew install libpq && brew link --force libpq\n' +
+        '  Ubuntu/Debian: apt install postgresql-client\n\n' +
+        'Or configure manually: spindb config set pg_dump /path/to/pg_dump',
+    )
+  }
+  return pgDumpPath
+}
+
+/**
+ * Create a backup of a PostgreSQL database
+ *
+ * CLI equivalent:
+ * - SQL format: pg_dump -Fp -h 127.0.0.1 -p {port} -U postgres -d {database} -f {outputPath}
+ * - Dump format: pg_dump -Fc -h 127.0.0.1 -p {port} -U postgres -d {database} -f {outputPath}
+ */
+export async function createBackup(
+  container: ContainerConfig,
+  outputPath: string,
+  options: BackupOptions,
+): Promise<BackupResult> {
+  const { port } = container
+  const { database, format } = options
+
+  const pgDumpPath = await getPgDumpPath()
+
+  // -Fp = plain SQL format, -Fc = custom format
+  const formatFlag = format === 'sql' ? '-Fp' : '-Fc'
+
+  return new Promise((resolve, reject) => {
+    const args = [
+      '-h',
+      '127.0.0.1',
+      '-p',
+      String(port),
+      '-U',
+      defaults.superuser,
+      '-d',
+      database,
+      formatFlag,
+      '-f',
+      outputPath,
+    ]
+
+    const proc = spawn(pgDumpPath, args, {
+      stdio: ['pipe', 'pipe', 'pipe'],
+    })
+
+    let stderr = ''
+
+    proc.stderr?.on('data', (data: Buffer) => {
+      stderr += data.toString()
+    })
+
+    proc.on('error', (err: NodeJS.ErrnoException) => {
+      reject(err)
+    })
+
+    proc.on('close', async (code) => {
+      if (code === 0) {
+        // Get file size
+        const stats = await stat(outputPath)
+        resolve({
+          path: outputPath,
+          format: format === 'sql' ? 'sql' : 'custom',
+          size: stats.size,
+        })
+      } else {
+        const errorMessage = stderr || `pg_dump exited with code ${code}`
+        reject(new Error(errorMessage))
+      }
+    })
+  })
+}

package/engines/postgresql/index.ts

@@ -16,10 +16,13 @@ import {
   FALLBACK_VERSION_MAP,
 } from './binary-urls'
 import { detectBackupFormat, restoreBackup } from './restore'
+import { createBackup } from './backup'
 import type {
   ContainerConfig,
   ProgressCallback,
   BackupFormat,
+  BackupOptions,
+  BackupResult,
   RestoreResult,
   DumpResult,
   StatusResult,
@@ -369,6 +372,29 @@ export class PostgreSQLEngine extends BaseEngine {
     }
   }
 
+  /**
+   * Get the size of the container's database in bytes
+   * Uses pg_database_size() to get accurate data size
+   * Returns null if container is not running or query fails
+   */
+  async getDatabaseSize(container: ContainerConfig): Promise<number | null> {
+    const { port, database } = container
+    const db = database || 'postgres'
+
+    try {
+      const psqlPath = await this.getPsqlPath()
+      // Query pg_database_size for the specific database
+      const { stdout } = await execAsync(
+        `"${psqlPath}" -h 127.0.0.1 -p ${port} -U ${defaults.superuser} -d postgres -t -A -c "SELECT pg_database_size('${db}')"`,
+      )
+      const size = parseInt(stdout.trim(), 10)
+      return isNaN(size) ? null : size
+    } catch {
+      // Container not running or query failed
+      return null
+    }
+  }
+
   /**
    * Create a dump from a remote database using a connection string
    * @param connectionString PostgreSQL connection string (e.g., postgresql://user:pass@host:port/dbname)
@@ -420,6 +446,17 @@ export class PostgreSQLEngine extends BaseEngine {
       })
     })
   }
+
+  /**
+   * Create a backup of a PostgreSQL database
+   */
+  async backup(
+    container: ContainerConfig,
+    outputPath: string,
+    options: BackupOptions,
+  ): Promise<BackupResult> {
+    return createBackup(container, outputPath, options)
+  }
 }
 
 export const postgresqlEngine = new PostgreSQLEngine()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "spindb",
-  "version": "0.5.4",
+  "version": "0.5.5",
   "description": "Spin up local database containers without Docker. A DBngin-like CLI for PostgreSQL and MySQL.",
   "type": "module",
   "bin": {

package/types/index.ts

@@ -4,6 +4,7 @@ export type ContainerConfig = {
   version: string
   port: number
   database: string
+  databases?: string[]
   created: string
   status: 'created' | 'running' | 'stopped'
   clonedFrom?: string
@@ -56,6 +57,17 @@ export type RestoreResult = {
   code?: number
 }
 
+export type BackupOptions = {
+  database: string
+  format: 'sql' | 'dump'
+}
+
+export type BackupResult = {
+  path: string
+  format: string
+  size: number
+}
+
 export type DumpResult = {
   filePath: string
   stdout?: string
@@ -85,6 +97,10 @@ export type BinaryTool =
   | 'mysqlpump'
   | 'mysqld'
   | 'mysqladmin'
+  // Enhanced shells (optional)
+  | 'pgcli'
+  | 'mycli'
+  | 'usql'
 
 /**
  * Source of a binary - bundled (downloaded by spindb) or system (found on PATH)
@@ -118,6 +134,10 @@ export type SpinDBConfig = {
     mysqlpump?: BinaryConfig
     mysqld?: BinaryConfig
     mysqladmin?: BinaryConfig
+    // Enhanced shells (optional)
+    pgcli?: BinaryConfig
+    mycli?: BinaryConfig
+    usql?: BinaryConfig
   }
   // Default settings
   defaults?: {