spindb 0.5.4 → 0.6.0

package/cli/ui/prompts.ts

@@ -251,7 +251,8 @@ export async function promptDatabaseName(
   engine?: string,
 ): Promise<string> {
   // MySQL uses "schema" terminology (database and schema are synonymous)
-  const label = engine === 'mysql' ? 'Database (schema) name:' : 'Database name:'
+  const label =
+    engine === 'mysql' ? 'Database (schema) name:' : 'Database name:'
 
   const { database } = await inquirer.prompt<{ database: string }>([
     {
@@ -276,6 +277,93 @@ export async function promptDatabaseName(
   return database
 }
 
+/**
+ * Prompt to select a database from a list of databases in a container
+ */
+export async function promptDatabaseSelect(
+  databases: string[],
+  message: string = 'Select database:',
+): Promise<string> {
+  if (databases.length === 0) {
+    throw new Error('No databases available to select')
+  }
+
+  if (databases.length === 1) {
+    return databases[0]
+  }
+
+  const { database } = await inquirer.prompt<{ database: string }>([
+    {
+      type: 'list',
+      name: 'database',
+      message,
+      choices: databases.map((db, index) => ({
+        name: index === 0 ? `${db} ${chalk.gray('(primary)')}` : db,
+        value: db,
+        short: db,
+      })),
+    },
+  ])
+
+  return database
+}
+
+/**
+ * Prompt for backup format selection
+ */
+export async function promptBackupFormat(
+  engine: string,
+): Promise<'sql' | 'dump'> {
+  const sqlDescription =
+    engine === 'mysql'
+      ? 'Plain SQL - human-readable, larger file'
+      : 'Plain SQL - human-readable, larger file'
+  const dumpDescription =
+    engine === 'mysql'
+      ? 'Compressed SQL (.sql.gz) - smaller file'
+      : 'Custom format - smaller file, faster restore'
+
+  const { format } = await inquirer.prompt<{ format: 'sql' | 'dump' }>([
+    {
+      type: 'list',
+      name: 'format',
+      message: 'Select backup format:',
+      choices: [
+        { name: `.sql ${chalk.gray(`- ${sqlDescription}`)}`, value: 'sql' },
+        { name: `.dump ${chalk.gray(`- ${dumpDescription}`)}`, value: 'dump' },
+      ],
+      default: 'sql',
+    },
+  ])
+
+  return format
+}
+
+/**
+ * Prompt for backup filename
+ */
+export async function promptBackupFilename(
+  defaultName: string,
+): Promise<string> {
+  const { filename } = await inquirer.prompt<{ filename: string }>([
+    {
+      type: 'input',
+      name: 'filename',
+      message: 'Backup filename (without extension):',
+      default: defaultName,
+      validate: (input: string) => {
+        if (!input) return 'Filename is required'
+        if (!/^[a-zA-Z0-9_-]+$/.test(input)) {
+          return 'Filename must contain only letters, numbers, underscores, and hyphens'
+        }
+        return true
+      },
+    },
+  ])
+
+  return filename
+}
+
 export type CreateOptions = {
   name: string
   engine: string

package/cli/ui/theme.ts

@@ -155,3 +155,14 @@ export function connectionBox(
 
   return box(lines)
 }
+
+/**
+ * Format bytes into human-readable format (B, KB, MB, GB)
+ */
+export function formatBytes(bytes: number): string {
+  if (bytes === 0) return '0 B'
+  const units = ['B', 'KB', 'MB', 'GB', 'TB']
+  const i = Math.floor(Math.log(bytes) / Math.log(1024))
+  const value = bytes / Math.pow(1024, i)
+  return `${value.toFixed(1)} ${units[i]}`
+}

package/core/config-manager.ts

@@ -18,6 +18,27 @@ const DEFAULT_CONFIG: SpinDBConfig = {
   binaries: {},
 }
 
+// Cache staleness threshold (7 days in milliseconds)
+const CACHE_STALENESS_MS = 7 * 24 * 60 * 60 * 1000
+
+// All tools organized by category
+const POSTGRESQL_TOOLS: BinaryTool[] = [
+  'psql',
+  'pg_dump',
+  'pg_restore',
+  'pg_basebackup',
+]
+
+const MYSQL_TOOLS: BinaryTool[] = ['mysql', 'mysqldump', 'mysqladmin', 'mysqld']
+
+const ENHANCED_SHELLS: BinaryTool[] = ['pgcli', 'mycli', 'usql']
+
+const ALL_TOOLS: BinaryTool[] = [
+  ...POSTGRESQL_TOOLS,
+  ...MYSQL_TOOLS,
+  ...ENHANCED_SHELLS,
+]
+
 export class ConfigManager {
   private config: SpinDBConfig | null = null
 
@@ -170,44 +191,55 @@ export class ConfigManager {
   }
 
   /**
-   * Get common installation paths for PostgreSQL client tools
+   * Get common installation paths for database tools
    */
   private getCommonBinaryPaths(tool: BinaryTool): string[] {
-    const paths: string[] = []
-
-    // Homebrew (macOS)
-    paths.push(`/opt/homebrew/bin/${tool}`)
-    paths.push(`/opt/homebrew/opt/libpq/bin/${tool}`)
-    paths.push(`/usr/local/bin/${tool}`)
-    paths.push(`/usr/local/opt/libpq/bin/${tool}`)
-
-    // Postgres.app (macOS)
-    paths.push(
-      `/Applications/Postgres.app/Contents/Versions/latest/bin/${tool}`,
-    )
-
-    // Linux common paths
-    paths.push(`/usr/bin/${tool}`)
-    paths.push(`/usr/lib/postgresql/16/bin/${tool}`)
-    paths.push(`/usr/lib/postgresql/15/bin/${tool}`)
-    paths.push(`/usr/lib/postgresql/14/bin/${tool}`)
-
-    return paths
+    const commonPaths: string[] = []
+
+    // Homebrew (macOS ARM)
+    commonPaths.push(`/opt/homebrew/bin/${tool}`)
+    // Homebrew (macOS Intel)
+    commonPaths.push(`/usr/local/bin/${tool}`)
+
+    // PostgreSQL-specific paths
+    if (POSTGRESQL_TOOLS.includes(tool) || tool === 'pgcli') {
+      commonPaths.push(`/opt/homebrew/opt/libpq/bin/${tool}`)
+      commonPaths.push(`/usr/local/opt/libpq/bin/${tool}`)
+      // Postgres.app (macOS)
+      commonPaths.push(
+        `/Applications/Postgres.app/Contents/Versions/latest/bin/${tool}`,
+      )
+      // Linux PostgreSQL paths
+      commonPaths.push(`/usr/lib/postgresql/17/bin/${tool}`)
+      commonPaths.push(`/usr/lib/postgresql/16/bin/${tool}`)
+      commonPaths.push(`/usr/lib/postgresql/15/bin/${tool}`)
+      commonPaths.push(`/usr/lib/postgresql/14/bin/${tool}`)
+    }
+
+    // MySQL-specific paths
+    if (MYSQL_TOOLS.includes(tool) || tool === 'mycli') {
+      commonPaths.push(`/opt/homebrew/opt/mysql/bin/${tool}`)
+      commonPaths.push(`/opt/homebrew/opt/mysql-client/bin/${tool}`)
+      commonPaths.push(`/usr/local/opt/mysql/bin/${tool}`)
+      commonPaths.push(`/usr/local/opt/mysql-client/bin/${tool}`)
+      // Linux MySQL/MariaDB paths
+      commonPaths.push(`/usr/bin/${tool}`)
+      commonPaths.push(`/usr/sbin/${tool}`)
+    }
+
+    // General Linux paths
+    commonPaths.push(`/usr/bin/${tool}`)
+
+    return commonPaths
   }
 
   /**
    * Detect all available client tools on the system
    */
   async detectAllTools(): Promise<Map<BinaryTool, string>> {
-    const tools: BinaryTool[] = [
-      'psql',
-      'pg_dump',
-      'pg_restore',
-      'pg_basebackup',
-    ]
     const found = new Map<BinaryTool, string>()
 
-    for (const tool of tools) {
+    for (const tool of ALL_TOOLS) {
       const path = await this.detectSystemBinary(tool)
       if (path) {
         found.set(tool, path)
@@ -219,18 +251,19 @@ export class ConfigManager {
 
   /**
    * Initialize config by detecting all available tools
+   * Groups results by category for better display
    */
-  async initialize(): Promise<{ found: BinaryTool[]; missing: BinaryTool[] }> {
-    const tools: BinaryTool[] = [
-      'psql',
-      'pg_dump',
-      'pg_restore',
-      'pg_basebackup',
-    ]
+  async initialize(): Promise<{
+    found: BinaryTool[]
+    missing: BinaryTool[]
+    postgresql: { found: BinaryTool[]; missing: BinaryTool[] }
+    mysql: { found: BinaryTool[]; missing: BinaryTool[] }
+    enhanced: { found: BinaryTool[]; missing: BinaryTool[] }
+  }> {
     const found: BinaryTool[] = []
     const missing: BinaryTool[] = []
 
-    for (const tool of tools) {
+    for (const tool of ALL_TOOLS) {
       const path = await this.getBinaryPath(tool)
       if (path) {
         found.push(tool)
@@ -239,7 +272,57 @@ export class ConfigManager {
       }
     }
 
-    return { found, missing }
+    return {
+      found,
+      missing,
+      postgresql: {
+        found: found.filter((t) => POSTGRESQL_TOOLS.includes(t)),
+        missing: missing.filter((t) => POSTGRESQL_TOOLS.includes(t)),
+      },
+      mysql: {
+        found: found.filter((t) => MYSQL_TOOLS.includes(t)),
+        missing: missing.filter((t) => MYSQL_TOOLS.includes(t)),
+      },
+      enhanced: {
+        found: found.filter((t) => ENHANCED_SHELLS.includes(t)),
+        missing: missing.filter((t) => ENHANCED_SHELLS.includes(t)),
+      },
+    }
+  }
+
+  /**
+   * Check if the config cache is stale (older than 7 days)
+   */
+  async isStale(): Promise<boolean> {
+    const config = await this.load()
+    if (!config.updatedAt) {
+      return true
+    }
+
+    const updatedAt = new Date(config.updatedAt).getTime()
+    const now = Date.now()
+    return now - updatedAt > CACHE_STALENESS_MS
+  }
+
+  /**
+   * Refresh all tool paths if cache is stale
+   * Returns true if refresh was performed
+   */
+  async refreshIfStale(): Promise<boolean> {
+    if (await this.isStale()) {
+      await this.refreshAllBinaries()
+      return true
+    }
+    return false
+  }
+
+  /**
+   * Force refresh all binary paths
+   * Re-detects all tools and updates versions
+   */
+  async refreshAllBinaries(): Promise<void> {
+    await this.clearAllBinaries()
+    await this.initialize()
   }
 
   /**
@@ -269,3 +352,6 @@ export class ConfigManager {
 }
 
 export const configManager = new ConfigManager()
+
+// Export tool categories for use in commands
+export { POSTGRESQL_TOOLS, MYSQL_TOOLS, ENHANCED_SHELLS, ALL_TOOLS }

package/core/container-manager.ts

@@ -51,6 +51,7 @@ export class ContainerManager {
       version,
       port,
       database,
+      databases: [database],
       created: new Date().toISOString(),
       status: 'created',
     }
@@ -63,6 +64,7 @@ export class ContainerManager {
   /**
    * Get container configuration
    * If engine is not provided, searches all engine directories
+   * Automatically migrates old schemas to include databases array
    */
   async getConfig(
     name: string,
@@ -77,7 +79,8 @@ export class ContainerManager {
         return null
       }
       const content = await readFile(configPath, 'utf8')
-      return JSON.parse(content) as ContainerConfig
+      const config = JSON.parse(content) as ContainerConfig
+      return this.migrateConfig(config)
     }
 
     // Search all engine directories
@@ -86,13 +89,43 @@ export class ContainerManager {
       const configPath = paths.getContainerConfigPath(name, { engine: eng })
       if (existsSync(configPath)) {
         const content = await readFile(configPath, 'utf8')
-        return JSON.parse(content) as ContainerConfig
+        const config = JSON.parse(content) as ContainerConfig
+        return this.migrateConfig(config)
       }
     }
 
     return null
   }
 
+  /**
+   * Migrate old container configs to include databases array
+   * Ensures primary database is always in the databases array
+   */
+  private async migrateConfig(
+    config: ContainerConfig,
+  ): Promise<ContainerConfig> {
+    let needsSave = false
+
+    // If databases array is missing, create it with the primary database
+    if (!config.databases) {
+      config.databases = [config.database]
+      needsSave = true
+    }
+
+    // Ensure primary database is in the array
+    if (!config.databases.includes(config.database)) {
+      config.databases = [config.database, ...config.databases]
+      needsSave = true
+    }
+
+    // Save if we made changes
+    if (needsSave) {
+      await this.saveConfig(config.name, { engine: config.engine }, config)
+    }
+
+    return config
+  }
+
   /**
    * Save container configuration
    */
@@ -333,6 +366,49 @@ export class ContainerManager {
     return /^[a-zA-Z][a-zA-Z0-9_-]*$/.test(name)
   }
 
+  /**
+   * Add a database to the container's databases array
+   */
+  async addDatabase(containerName: string, database: string): Promise<void> {
+    const config = await this.getConfig(containerName)
+    if (!config) {
+      throw new Error(`Container "${containerName}" not found`)
+    }
+
+    // Ensure databases array exists
+    if (!config.databases) {
+      config.databases = [config.database]
+    }
+
+    // Add if not already present
+    if (!config.databases.includes(database)) {
+      config.databases.push(database)
+      await this.saveConfig(containerName, { engine: config.engine }, config)
+    }
+  }
+
+  /**
+   * Remove a database from the container's databases array
+   */
+  async removeDatabase(containerName: string, database: string): Promise<void> {
+    const config = await this.getConfig(containerName)
+    if (!config) {
+      throw new Error(`Container "${containerName}" not found`)
+    }
+
+    // Don't remove the primary database from the array
+    if (database === config.database) {
+      throw new Error(
+        `Cannot remove primary database "${database}" from tracking`,
+      )
+    }
+
+    if (config.databases) {
+      config.databases = config.databases.filter((db) => db !== database)
+      await this.saveConfig(containerName, { engine: config.engine }, config)
+    }
+  }
+
   /**
    * Get connection string for a container
    * Delegates to the appropriate engine

package/core/dependency-manager.ts

@@ -20,6 +20,7 @@ import {
   mycliDependency,
 } from '../config/os-dependencies'
 import { platformService } from './platform-service'
+import { configManager } from './config-manager'
 
 const execAsync = promisify(exec)
 
@@ -268,6 +269,10 @@ export async function installDependency(
       execWithInheritedStdio(cmd)
     }
 
+    // Refresh config cache after package manager interaction
+    // This ensures newly installed tools are detected with correct versions
+    await configManager.refreshAllBinaries()
+
     // Verify installation
     const status = await checkDependency(dependency)
     if (!status.installed) {

package/core/update-manager.ts

@@ -0,0 +1,194 @@
+import { exec } from 'child_process'
+import { promisify } from 'util'
+import { createRequire } from 'module'
+import { configManager } from './config-manager'
+
+const execAsync = promisify(exec)
+const require = createRequire(import.meta.url)
+
+const NPM_REGISTRY_URL = 'https://registry.npmjs.org/spindb'
+const CHECK_THROTTLE_MS = 24 * 60 * 60 * 1000 // 24 hours
+
+export type UpdateCheckResult = {
+  currentVersion: string
+  latestVersion: string
+  updateAvailable: boolean
+  lastChecked: string
+}
+
+export type UpdateResult = {
+  success: boolean
+  previousVersion: string
+  newVersion: string
+  error?: string
+}
+
+export class UpdateManager {
+  /**
+   * Get currently installed version from package.json
+   */
+  getCurrentVersion(): string {
+    const pkg = require('../package.json') as { version: string }
+    return pkg.version
+  }
+
+  /**
+   * Check npm registry for latest version
+   * Throttled to once per 24 hours unless force=true
+   */
+  async checkForUpdate(force = false): Promise<UpdateCheckResult | null> {
+    const config = await configManager.load()
+    const lastCheck = config.update?.lastCheck
+
+    // Return cached result if within throttle period
+    if (!force && lastCheck) {
+      const elapsed = Date.now() - new Date(lastCheck).getTime()
+      if (elapsed < CHECK_THROTTLE_MS && config.update?.latestVersion) {
+        const currentVersion = this.getCurrentVersion()
+        return {
+          currentVersion,
+          latestVersion: config.update.latestVersion,
+          updateAvailable:
+            this.compareVersions(config.update.latestVersion, currentVersion) >
+            0,
+          lastChecked: lastCheck,
+        }
+      }
+    }
+
+    try {
+      const latestVersion = await this.fetchLatestVersion()
+      const currentVersion = this.getCurrentVersion()
+
+      // Update cache
+      config.update = {
+        ...config.update,
+        lastCheck: new Date().toISOString(),
+        latestVersion,
+      }
+      await configManager.save()
+
+      return {
+        currentVersion,
+        latestVersion,
+        updateAvailable:
+          this.compareVersions(latestVersion, currentVersion) > 0,
+        lastChecked: new Date().toISOString(),
+      }
+    } catch {
+      // Offline or registry error - return null
+      return null
+    }
+  }
+
+  /**
+   * Perform self-update via npm
+   */
+  async performUpdate(): Promise<UpdateResult> {
+    const previousVersion = this.getCurrentVersion()
+
+    try {
+      // Execute npm install globally
+      await execAsync('npm install -g spindb@latest', { timeout: 60000 })
+
+      // Verify new version by checking what npm reports
+      const { stdout } = await execAsync('npm list -g spindb --json')
+      const npmData = JSON.parse(stdout) as {
+        dependencies?: { spindb?: { version?: string } }
+      }
+      const newVersion =
+        npmData.dependencies?.spindb?.version || previousVersion
+
+      return {
+        success: true,
+        previousVersion,
+        newVersion,
+      }
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error)
+
+      // Detect permission issues
+      if (message.includes('EACCES') || message.includes('permission')) {
+        return {
+          success: false,
+          previousVersion,
+          newVersion: previousVersion,
+          error: 'Permission denied. Try: sudo npm install -g spindb@latest',
+        }
+      }
+
+      return {
+        success: false,
+        previousVersion,
+        newVersion: previousVersion,
+        error: message,
+      }
+    }
+  }
+
+  /**
+   * Get cached update info (for showing notification without network call)
+   */
+  async getCachedUpdateInfo(): Promise<{
+    latestVersion?: string
+    autoCheckEnabled: boolean
+  }> {
+    const config = await configManager.load()
+    return {
+      latestVersion: config.update?.latestVersion,
+      autoCheckEnabled: config.update?.autoCheckEnabled !== false,
+    }
+  }
+
+  /**
+   * Set whether auto-update checks are enabled
+   */
+  async setAutoCheckEnabled(enabled: boolean): Promise<void> {
+    const config = await configManager.load()
+    config.update = {
+      ...config.update,
+      autoCheckEnabled: enabled,
+    }
+    await configManager.save()
+  }
+
+  /**
+   * Fetch latest version from npm registry
+   */
+  private async fetchLatestVersion(): Promise<string> {
+    const controller = new AbortController()
+    const timeout = setTimeout(() => controller.abort(), 10000)
+
+    try {
+      const response = await fetch(NPM_REGISTRY_URL, {
+        signal: controller.signal,
+      })
+      if (!response.ok) {
+        throw new Error(`Registry returned ${response.status}`)
+      }
+      const data = (await response.json()) as {
+        'dist-tags': { latest: string }
+      }
+      return data['dist-tags'].latest
+    } finally {
+      clearTimeout(timeout)
+    }
+  }
+
+  /**
+   * Compare semver versions
+   * Returns >0 if a > b, <0 if a < b, 0 if equal
+   */
+  compareVersions(a: string, b: string): number {
+    const partsA = a.split('.').map((n) => parseInt(n, 10) || 0)
+    const partsB = b.split('.').map((n) => parseInt(n, 10) || 0)
+
+    for (let i = 0; i < 3; i++) {
+      const diff = (partsA[i] || 0) - (partsB[i] || 0)
+      if (diff !== 0) return diff
+    }
+    return 0
+  }
+}
+
+export const updateManager = new UpdateManager()

package/engines/base-engine.ts

@@ -2,6 +2,8 @@ import type {
   ContainerConfig,
   ProgressCallback,
   BackupFormat,
+  BackupOptions,
+  BackupResult,
   RestoreResult,
   DumpResult,
   StatusResult,
@@ -131,4 +133,22 @@ export abstract class BaseEngine {
     connectionString: string,
     outputPath: string,
   ): Promise<DumpResult>
+
+  /**
+   * Get the size of a database in bytes
+   * Returns null if the container is not running or size cannot be determined
+   */
+  abstract getDatabaseSize(container: ContainerConfig): Promise<number | null>
+
+  /**
+   * Create a backup of a database
+   * @param container - The container configuration
+   * @param outputPath - Path to write the backup file
+   * @param options - Backup options including database name and format
+   */
+  abstract backup(
+    container: ContainerConfig,
+    outputPath: string,
+    options: BackupOptions,
+  ): Promise<BackupResult>
 }