livedash-node/scripts/d1-manager.js

#!/usr/bin/env node
/**
 * Comprehensive D1 Database Management Script
 *
 * Usage Examples:
 *   node scripts/d1-manager.js tables
 *   node scripts/d1-manager.js schema Company
 *   node scripts/d1-manager.js count User
 *   node scripts/d1-manager.js query "SELECT * FROM User LIMIT 5"
 *   node scripts/d1-manager.js backup
 *   node scripts/d1-manager.js --remote query "SELECT COUNT(*) FROM Session"
 */

import { execSync } from "child_process";
import { writeFileSync, mkdirSync } from "fs";
import { join } from "path";

const DB_NAME = "d1-notso-livedash";
const args = process.argv.slice(2);

// Parse flags
const isRemote = args.includes("--remote");
const filteredArgs = args.filter((arg) => !arg.startsWith("--"));

if (filteredArgs.length === 0) {
  showHelp();
  process.exit(1);
}

const command = filteredArgs[0];
const params = filteredArgs.slice(1);

function showHelp() {
  console.log(`
🗄️  D1 Database Manager for ${DB_NAME}

Usage: node scripts/d1-manager.js [--remote] <command> [params...]

Commands:
  info                     Show database information
  tables                   List all tables
  schema <table>           Show table schema
  count <table>            Count rows in table
  query "<sql>"            Execute custom SQL query
  backup [filename]        Export database to SQL file
  backup-schema            Export just the schema
  recent-logs              Show recent query activity

Flags:
  --remote                 Execute against remote D1 (production)

Examples:
  node scripts/d1-manager.js tables
  node scripts/d1-manager.js schema User
  node scripts/d1-manager.js count Company
  node scripts/d1-manager.js query "SELECT * FROM User WHERE role = 'admin'"
  node scripts/d1-manager.js backup
  node scripts/d1-manager.js --remote info
`);
}

function execute(sql, silent = false) {
  const remoteFlag = isRemote ? "--remote" : "";
  const cmd = `npx wrangler d1 execute ${DB_NAME} ${remoteFlag} --command "${sql}"`;

  if (!silent) {
    console.log(
      `🔍 Executing${isRemote ? " (remote)" : " (local)"}: ${sql}\\n`
    );
  }

  try {
    return execSync(cmd, { encoding: "utf8" });
  } catch (error) {
    console.error("❌ Query failed:", error.message);
    process.exit(1);
  }
}

function wranglerCommand(subcommand, silent = false) {
  const remoteFlag = isRemote ? "--remote" : "";
  const cmd = `npx wrangler d1 ${subcommand} ${DB_NAME} ${remoteFlag}`;

  if (!silent) {
    console.log(`📊 Running: ${cmd}\\n`);
  }

  try {
    return execSync(cmd, { stdio: "inherit" });
  } catch (error) {
    console.error("❌ Command failed:", error.message);
    process.exit(1);
  }
}

switch (command) {
  case "info":
    wranglerCommand("info");
    break;

  case "tables":
    console.log("📋 Listing all tables:\\n");
    execute(
      "SELECT name, type FROM sqlite_master WHERE type IN ('table', 'view') AND name NOT LIKE 'sqlite_%' ORDER BY name;"
    );
    break;

  case "schema":
    if (!params[0]) {
      console.error("❌ Please specify a table name");
      console.log("Usage: node scripts/d1-manager.js schema <table_name>");
      process.exit(1);
    }
    console.log(`🏗️  Schema for table '${params[0]}':\\n`);
    execute(`PRAGMA table_info(${params[0]});`);
    break;

  case "count":
    if (!params[0]) {
      console.error("❌ Please specify a table name");
      console.log("Usage: node scripts/d1-manager.js count <table_name>");
      process.exit(1);
    }
    console.log(`🔢 Row count for table '${params[0]}':\\n`);
    execute(`SELECT COUNT(*) as row_count FROM ${params[0]};`);
    break;

  case "query":
    if (!params[0]) {
      console.error("❌ Please specify a SQL query");
      console.log(
        'Usage: node scripts/d1-manager.js query "SELECT * FROM table"'
      );
      process.exit(1);
    }
    execute(params[0]);
    break;

  case "backup":
    const timestamp = new Date()
      .toISOString()
      .replace(/[:.]/g, "-")
      .slice(0, 19);
    const filename = params[0] || `backup_${timestamp}.sql`;

    try {
      mkdirSync("backups", { recursive: true });
    } catch (e) {
      // Directory might already exist
    }

    const backupPath = join("backups", filename);
    console.log(`💾 Creating backup: ${backupPath}\\n`);
    wranglerCommand(`export --output ${backupPath}`);
    console.log(`\\n✅ Backup created successfully: ${backupPath}`);
    break;

  case "backup-schema":
    try {
      mkdirSync("backups", { recursive: true });
    } catch (e) {
      // Directory might already exist
    }

    console.log("📜 Exporting schema only...\\n");
    wranglerCommand("export --no-data --output backups/schema.sql");
    console.log("\\n✅ Schema exported to backups/schema.sql");
    break;

  case "recent-logs":
    console.log("📊 Recent database activity:\\n");
    try {
      wranglerCommand("insights");
    } catch (error) {
      console.log("ℹ️  Insights not available for this database");
    }
    break;

  case "all-tables-info":
    console.log("📊 Information about all tables:\\n");
    const tables = ["Company", "User", "Session"];
    for (const table of tables) {
      console.log(`\\n🏷️  Table: ${table}`);
      console.log("─".repeat(50));
      execute(`SELECT COUNT(*) as row_count FROM ${table};`);
    }
    break;

  default:
    console.error(`❌ Unknown command: ${command}`);
    showHelp();
    process.exit(1);
}