@launchql/cli/esm/commands/kill.js

LaunchQL CLI

import { Logger } from '@launchql/logger';
import { getPgPool } from 'pg-cache';
const log = new Logger('db-kill');
const killUsageText = `
LaunchQL Kill Command:

  lql kill [OPTIONS]

  Terminate database connections and optionally drop databases.

Options:
  --help, -h              Show this help message
  --drop                  Drop databases after killing connections (default: true)
  --no-drop               Only kill connections, don't drop databases
  --cwd <directory>       Working directory (default: current directory)

Examples:
  lql kill                Kill connections and drop selected databases
  lql kill --no-drop      Only kill connections, preserve databases
`;
export default async (argv, prompter, _options) => {
    // Show usage if explicitly requested
    if (argv.help || argv.h) {
        console.log(killUsageText);
        process.exit(0);
    }
    const db = await getPgPool({
        database: 'postgres'
    });
    const databasesResult = await db.query(`
    SELECT datname FROM pg_catalog.pg_database
    WHERE datistemplate = FALSE AND datname NOT IN ('postgres')
      AND datname !~ '^pg_';
  `);
    if (!databasesResult.rows.length) {
        log.info('ℹ️  No databases found to process. Exiting.');
        return;
    }
    let databases;
    ({ databases } = await prompter.prompt(argv, [
        {
            type: 'checkbox',
            name: 'databases',
            message: 'Select database(s) to terminate connections and optionally drop',
            options: databasesResult.rows.map(row => row.datname),
            required: true
        }
    ]));
    const selectedDbNames = databases.filter(d => d.selected).map(d => d.value);
    const { yes } = await prompter.prompt(argv, [
        {
            type: 'confirm',
            name: 'yes',
            message: `Are you sure you want to kill connections${argv.drop === false ? '' : ' and DROP'}: ${selectedDbNames.join(', ')}?`,
            default: false
        }
    ]);
    if (!yes) {
        log.info('❌ Aborted. No actions were taken.');
        return;
    }
    for (const dbname of selectedDbNames) {
        const killResult = await db.query(`
      SELECT pg_terminate_backend(pid)
      FROM pg_stat_activity
      WHERE datname = $1 AND pid <> pg_backend_pid();
      `, [dbname]);
        log.warn(`💀 Terminated ${killResult.rowCount} connection(s) to "${dbname}".`);
        if (argv.drop === false) {
            log.info(`⚠️  Skipping DROP for "${dbname}" due to --no-drop flag.`);
            continue;
        }
        try {
            await db.query(`DROP DATABASE "${dbname}";`);
            log.success(`🗑️  Dropped database "${dbname}" successfully.`);
        }
        catch (err) {
            log.error(`❌ Failed to drop "${dbname}": ${err.message}`);
        }
    }
    log.success('✅ Done processing databases.');
};