postgresai 0.11.0-alpha.7 → 0.11.0-alpha.9

package/dist/bin/postgres-ai.js

@@ -0,0 +1,897 @@
+#!/usr/bin/env node
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
+const pkg = __importStar(require("../package.json"));
+const config = __importStar(require("../lib/config"));
+const yaml = __importStar(require("js-yaml"));
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const child_process_1 = require("child_process");
+const util_1 = require("util");
+const readline = __importStar(require("readline"));
+const http = __importStar(require("https"));
+const url_1 = require("url");
+const execPromise = (0, util_1.promisify)(child_process_1.exec);
+const execFilePromise = (0, util_1.promisify)(child_process_1.execFile);
+/**
+ * Get configuration from various sources
+ * @param opts - Command line options
+ * @returns Configuration object
+ */
+function getConfig(opts) {
+    // Priority order:
+    // 1. Command line option (--api-key)
+    // 2. Environment variable (PGAI_API_KEY)
+    // 3. User-level config file (~/.config/postgresai/config.json)
+    // 4. Legacy project-local config (.pgwatch-config)
+    let apiKey = opts.apiKey || process.env.PGAI_API_KEY || "";
+    // Try config file if not provided via CLI or env
+    if (!apiKey) {
+        const fileConfig = config.readConfig();
+        if (!apiKey)
+            apiKey = fileConfig.apiKey || "";
+    }
+    return { apiKey };
+}
+const program = new commander_1.Command();
+program
+    .name("postgres-ai")
+    .description("PostgresAI CLI")
+    .version(pkg.version)
+    .option("--api-key <key>", "API key (overrides PGAI_API_KEY)")
+    .option("--api-base-url <url>", "API base URL for backend RPC (overrides PGAI_API_BASE_URL)")
+    .option("--ui-base-url <url>", "UI base URL for browser routes (overrides PGAI_UI_BASE_URL)");
+/**
+ * Stub function for not implemented commands
+ */
+const stub = (name) => async () => {
+    // Temporary stubs until Node parity is implemented
+    console.error(`${name}: not implemented in Node CLI yet; use bash CLI for now`);
+    process.exitCode = 2;
+};
+/**
+ * Resolve project paths
+ */
+function resolvePaths() {
+    const projectDir = process.cwd();
+    const composeFile = path.resolve(projectDir, "docker-compose.yml");
+    const instancesFile = path.resolve(projectDir, "instances.yml");
+    return { fs, path, projectDir, composeFile, instancesFile };
+}
+/**
+ * Get docker compose command
+ */
+function getComposeCmd() {
+    const tryCmd = (cmd, args) => (0, child_process_1.spawnSync)(cmd, args, { stdio: "ignore" }).status === 0;
+    if (tryCmd("docker-compose", ["version"]))
+        return ["docker-compose"];
+    if (tryCmd("docker", ["compose", "version"]))
+        return ["docker", "compose"];
+    return null;
+}
+/**
+ * Run docker compose command
+ */
+async function runCompose(args) {
+    const { composeFile } = resolvePaths();
+    const cmd = getComposeCmd();
+    if (!cmd) {
+        console.error("docker compose not found (need docker-compose or docker compose)");
+        process.exitCode = 1;
+        return 1;
+    }
+    return new Promise((resolve) => {
+        const child = (0, child_process_1.spawn)(cmd[0], [...cmd.slice(1), "-f", composeFile, ...args], { stdio: "inherit" });
+        child.on("close", (code) => resolve(code || 0));
+    });
+}
+program.command("help", { isDefault: true }).description("show help").action(() => {
+    program.outputHelp();
+});
+// Service lifecycle
+program
+    .command("quickstart")
+    .description("complete setup (generate config, start services)")
+    .option("--demo", "demo mode", false)
+    .action(async () => {
+    const code1 = await runCompose(["run", "--rm", "sources-generator"]);
+    if (code1 !== 0) {
+        process.exitCode = code1;
+        return;
+    }
+    const code2 = await runCompose(["up", "-d"]);
+    if (code2 !== 0)
+        process.exitCode = code2;
+});
+program
+    .command("install")
+    .description("prepare project (no-op in repo checkout)")
+    .action(async () => {
+    console.log("Project files present; nothing to install.");
+});
+program
+    .command("start")
+    .description("start services")
+    .action(async () => {
+    const code = await runCompose(["up", "-d"]);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("stop")
+    .description("stop services")
+    .action(async () => {
+    const code = await runCompose(["down"]);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("restart [service]")
+    .description("restart all services or specific service")
+    .action(async (service) => {
+    const args = ["restart"];
+    if (service)
+        args.push(service);
+    const code = await runCompose(args);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("status")
+    .description("show service status")
+    .action(async () => {
+    const code = await runCompose(["ps"]);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("logs [service]")
+    .option("-f, --follow", "follow logs", false)
+    .option("--tail <lines>", "number of lines to show from the end of logs", "all")
+    .description("show logs for all or specific service")
+    .action(async (service, opts) => {
+    const args = ["logs"];
+    if (opts.follow)
+        args.push("-f");
+    if (opts.tail)
+        args.push("--tail", opts.tail);
+    if (service)
+        args.push(service);
+    const code = await runCompose(args);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("health")
+    .description("health check")
+    .option("--wait <seconds>", "wait time in seconds for services to become healthy", parseInt, 0)
+    .action(async (opts) => {
+    const services = [
+        { name: "Grafana", url: "http://localhost:3000/api/health" },
+        { name: "Prometheus", url: "http://localhost:59090/-/healthy" },
+        { name: "PGWatch (Postgres)", url: "http://localhost:58080/health" },
+        { name: "PGWatch (Prometheus)", url: "http://localhost:58089/health" },
+    ];
+    const waitTime = opts.wait || 0;
+    const maxAttempts = waitTime > 0 ? Math.ceil(waitTime / 5) : 1;
+    console.log("Checking service health...\n");
+    let allHealthy = false;
+    for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+        if (attempt > 1) {
+            console.log(`Retrying (attempt ${attempt}/${maxAttempts})...\n`);
+            await new Promise(resolve => setTimeout(resolve, 5000));
+        }
+        allHealthy = true;
+        for (const service of services) {
+            try {
+                const { stdout } = await execPromise(`curl -sf -o /dev/null -w "%{http_code}" ${service.url}`, { timeout: 5000 });
+                const code = stdout.trim();
+                if (code === "200") {
+                    console.log(`✓ ${service.name}: healthy`);
+                }
+                else {
+                    console.log(`✗ ${service.name}: unhealthy (HTTP ${code})`);
+                    allHealthy = false;
+                }
+            }
+            catch (error) {
+                console.log(`✗ ${service.name}: unreachable`);
+                allHealthy = false;
+            }
+        }
+        if (allHealthy) {
+            break;
+        }
+    }
+    console.log("");
+    if (allHealthy) {
+        console.log("All services are healthy");
+    }
+    else {
+        console.log("Some services are unhealthy");
+        process.exitCode = 1;
+    }
+});
+program
+    .command("config")
+    .description("show configuration")
+    .action(async () => {
+    const { fs, projectDir, composeFile, instancesFile } = resolvePaths();
+    console.log(`Project Directory: ${projectDir}`);
+    console.log(`Docker Compose File: ${composeFile}`);
+    console.log(`Instances File: ${instancesFile}`);
+    if (fs.existsSync(instancesFile)) {
+        console.log("\nInstances configuration:\n");
+        const text = fs.readFileSync(instancesFile, "utf8");
+        process.stdout.write(text);
+        if (!/\n$/.test(text))
+            console.log();
+    }
+});
+program
+    .command("update-config")
+    .description("apply configuration (generate sources)")
+    .action(async () => {
+    const code = await runCompose(["run", "--rm", "sources-generator"]);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("update")
+    .description("update project")
+    .action(async () => {
+    console.log("Updating PostgresAI monitoring stack...\n");
+    try {
+        // Check if we're in a git repo
+        const gitDir = path.resolve(process.cwd(), ".git");
+        if (!fs.existsSync(gitDir)) {
+            console.error("Not a git repository. Cannot update.");
+            process.exitCode = 1;
+            return;
+        }
+        // Fetch latest changes
+        console.log("Fetching latest changes...");
+        await execPromise("git fetch origin");
+        // Check current branch
+        const { stdout: branch } = await execPromise("git rev-parse --abbrev-ref HEAD");
+        const currentBranch = branch.trim();
+        console.log(`Current branch: ${currentBranch}`);
+        // Pull latest changes
+        console.log("Pulling latest changes...");
+        const { stdout: pullOut } = await execPromise("git pull origin " + currentBranch);
+        console.log(pullOut);
+        // Update Docker images
+        console.log("\nUpdating Docker images...");
+        const code = await runCompose(["pull"]);
+        if (code === 0) {
+            console.log("\n✓ Update completed successfully");
+            console.log("\nTo apply updates, restart services:");
+            console.log("  postgres-ai restart");
+        }
+        else {
+            console.error("\n✗ Docker image update failed");
+            process.exitCode = 1;
+        }
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.error(`Update failed: ${message}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("reset [service]")
+    .description("reset all or specific service")
+    .action(async (service) => {
+    const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    const question = (prompt) => new Promise((resolve) => rl.question(prompt, resolve));
+    try {
+        if (service) {
+            // Reset specific service
+            console.log(`\nThis will stop '${service}', remove its volume, and restart it.`);
+            console.log("All data for this service will be lost!\n");
+            const answer = await question("Continue? (y/N): ");
+            if (answer.toLowerCase() !== "y") {
+                console.log("Cancelled");
+                rl.close();
+                return;
+            }
+            console.log(`\nStopping ${service}...`);
+            await runCompose(["stop", service]);
+            console.log(`Removing volume for ${service}...`);
+            await runCompose(["rm", "-f", "-v", service]);
+            console.log(`Restarting ${service}...`);
+            const code = await runCompose(["up", "-d", service]);
+            if (code === 0) {
+                console.log(`\n✓ Service '${service}' has been reset`);
+            }
+            else {
+                console.error(`\n✗ Failed to restart '${service}'`);
+                process.exitCode = 1;
+            }
+        }
+        else {
+            // Reset all services
+            console.log("\nThis will stop all services and remove all data!");
+            console.log("Volumes, networks, and containers will be deleted.\n");
+            const answer = await question("Continue? (y/N): ");
+            if (answer.toLowerCase() !== "y") {
+                console.log("Cancelled");
+                rl.close();
+                return;
+            }
+            console.log("\nStopping services and removing data...");
+            const downCode = await runCompose(["down", "-v"]);
+            if (downCode === 0) {
+                console.log("✓ Environment reset completed - all containers and data removed");
+            }
+            else {
+                console.error("✗ Reset failed");
+                process.exitCode = 1;
+            }
+        }
+        rl.close();
+    }
+    catch (error) {
+        rl.close();
+        const message = error instanceof Error ? error.message : String(error);
+        console.error(`Reset failed: ${message}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("clean")
+    .description("cleanup artifacts")
+    .action(async () => {
+    console.log("Cleaning up Docker resources...\n");
+    try {
+        // Remove stopped containers
+        const { stdout: containers } = await execFilePromise("docker", ["ps", "-aq", "--filter", "status=exited"]);
+        if (containers.trim()) {
+            const containerIds = containers.trim().split('\n');
+            await execFilePromise("docker", ["rm", ...containerIds]);
+            console.log("✓ Removed stopped containers");
+        }
+        else {
+            console.log("✓ No stopped containers to remove");
+        }
+        // Remove unused volumes
+        await execFilePromise("docker", ["volume", "prune", "-f"]);
+        console.log("✓ Removed unused volumes");
+        // Remove unused networks
+        await execFilePromise("docker", ["network", "prune", "-f"]);
+        console.log("✓ Removed unused networks");
+        // Remove dangling images
+        await execFilePromise("docker", ["image", "prune", "-f"]);
+        console.log("✓ Removed dangling images");
+        console.log("\nCleanup completed");
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.error(`Error during cleanup: ${message}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("shell <service>")
+    .description("open service shell")
+    .action(async (service) => {
+    const code = await runCompose(["exec", service, "/bin/sh"]);
+    if (code !== 0)
+        process.exitCode = code;
+});
+program
+    .command("check")
+    .description("system readiness check")
+    .action(async () => {
+    const code = await runCompose(["ps"]);
+    if (code !== 0)
+        process.exitCode = code;
+});
+// Instance management
+program
+    .command("list-instances")
+    .description("list instances")
+    .action(async () => {
+    const instancesPath = path.resolve(process.cwd(), "instances.yml");
+    if (!fs.existsSync(instancesPath)) {
+        console.error(`instances.yml not found in ${process.cwd()}`);
+        process.exitCode = 1;
+        return;
+    }
+    try {
+        const content = fs.readFileSync(instancesPath, "utf8");
+        const instances = yaml.load(content);
+        if (!instances || !Array.isArray(instances) || instances.length === 0) {
+            console.log("No instances configured");
+            console.log("");
+            console.log("To add an instance:");
+            console.log("  postgres-ai add-instance <connection-string> <name>");
+            console.log("");
+            console.log("Example:");
+            console.log("  postgres-ai add-instance 'postgresql://user:pass@host:5432/db' my-db");
+            return;
+        }
+        // Filter out demo placeholder
+        const filtered = instances.filter((inst) => inst.name && inst.name !== "target-database");
+        if (filtered.length === 0) {
+            console.log("No instances configured");
+            console.log("");
+            console.log("To add an instance:");
+            console.log("  postgres-ai add-instance <connection-string> <name>");
+            console.log("");
+            console.log("Example:");
+            console.log("  postgres-ai add-instance 'postgresql://user:pass@host:5432/db' my-db");
+            return;
+        }
+        for (const inst of filtered) {
+            console.log(`Instance: ${inst.name}`);
+        }
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        console.error(`Error parsing instances.yml: ${message}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("add-instance [connStr] [name]")
+    .description("add instance")
+    .action(async (connStr, name) => {
+    const file = path.resolve(process.cwd(), "instances.yml");
+    if (!connStr) {
+        console.error("Connection string required: postgresql://user:pass@host:port/db");
+        process.exitCode = 1;
+        return;
+    }
+    const m = connStr.match(/^postgresql:\/\/([^:]+):([^@]+)@([^:\/]+)(?::(\d+))?\/(.+)$/);
+    if (!m) {
+        console.error("Invalid connection string format");
+        process.exitCode = 1;
+        return;
+    }
+    const host = m[3];
+    const db = m[5];
+    const instanceName = name && name.trim() ? name.trim() : `${host}-${db}`.replace(/[^a-zA-Z0-9-]/g, "-");
+    // Check if instance already exists
+    try {
+        if (fs.existsSync(file)) {
+            const content = fs.readFileSync(file, "utf8");
+            const instances = yaml.load(content) || [];
+            if (Array.isArray(instances)) {
+                const exists = instances.some((inst) => inst.name === instanceName);
+                if (exists) {
+                    console.error(`Instance '${instanceName}' already exists`);
+                    process.exitCode = 1;
+                    return;
+                }
+            }
+        }
+    }
+    catch (err) {
+        // If YAML parsing fails, fall back to simple check
+        const content = fs.existsSync(file) ? fs.readFileSync(file, "utf8") : "";
+        if (new RegExp(`^- name: ${instanceName}$`, "m").test(content)) {
+            console.error(`Instance '${instanceName}' already exists`);
+            process.exitCode = 1;
+            return;
+        }
+    }
+    // Add new instance
+    const body = `- name: ${instanceName}\n  conn_str: ${connStr}\n  preset_metrics: full\n  custom_metrics:\n  is_enabled: true\n  group: default\n  custom_tags:\n    env: production\n    cluster: default\n    node_name: ${instanceName}\n    sink_type: ~sink_type~\n`;
+    const content = fs.existsSync(file) ? fs.readFileSync(file, "utf8") : "";
+    fs.appendFileSync(file, (content && !/\n$/.test(content) ? "\n" : "") + body, "utf8");
+    console.log(`Instance '${instanceName}' added`);
+});
+program
+    .command("remove-instance <name>")
+    .description("remove instance")
+    .action(async (name) => {
+    const file = path.resolve(process.cwd(), "instances.yml");
+    if (!fs.existsSync(file)) {
+        console.error("instances.yml not found");
+        process.exitCode = 1;
+        return;
+    }
+    try {
+        const content = fs.readFileSync(file, "utf8");
+        const instances = yaml.load(content);
+        if (!instances || !Array.isArray(instances)) {
+            console.error("Invalid instances.yml format");
+            process.exitCode = 1;
+            return;
+        }
+        const filtered = instances.filter((inst) => inst.name !== name);
+        if (filtered.length === instances.length) {
+            console.error(`Instance '${name}' not found`);
+            process.exitCode = 1;
+            return;
+        }
+        fs.writeFileSync(file, yaml.dump(filtered), "utf8");
+        console.log(`Instance '${name}' removed`);
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        console.error(`Error processing instances.yml: ${message}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("test-instance <name>")
+    .description("test instance connectivity")
+    .action(async (name) => {
+    const instancesPath = path.resolve(process.cwd(), "instances.yml");
+    if (!fs.existsSync(instancesPath)) {
+        console.error("instances.yml not found");
+        process.exitCode = 1;
+        return;
+    }
+    try {
+        const content = fs.readFileSync(instancesPath, "utf8");
+        const instances = yaml.load(content);
+        if (!instances || !Array.isArray(instances)) {
+            console.error("Invalid instances.yml format");
+            process.exitCode = 1;
+            return;
+        }
+        const instance = instances.find((inst) => inst.name === name);
+        if (!instance) {
+            console.error(`Instance '${name}' not found`);
+            process.exitCode = 1;
+            return;
+        }
+        if (!instance.conn_str) {
+            console.error(`Connection string not found for instance '${name}'`);
+            process.exitCode = 1;
+            return;
+        }
+        console.log(`Testing connection to '${name}'...`);
+        const { stdout, stderr } = await execFilePromise("psql", [instance.conn_str, "-c", "SELECT version();", "--no-psqlrc"], { timeout: 10000, env: { ...process.env, PAGER: 'cat' } });
+        console.log(`✓ Connection successful`);
+        console.log(stdout.trim());
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.error(`✗ Connection failed: ${message}`);
+        process.exitCode = 1;
+    }
+});
+// Authentication and API key management
+program
+    .command("auth")
+    .description("authenticate via browser and obtain API key")
+    .option("--port <port>", "local callback server port (default: random)", parseInt)
+    .option("--debug", "enable debug output")
+    .action(async (opts) => {
+    const pkce = require("../lib/pkce");
+    const authServer = require("../lib/auth-server");
+    console.log("Starting authentication flow...\n");
+    // Generate PKCE parameters
+    const params = pkce.generatePKCEParams();
+    const rootOpts = program.opts();
+    const apiBaseUrl = (rootOpts.apiBaseUrl || process.env.PGAI_API_BASE_URL || "https://postgres.ai/api/general/").replace(/\/$/, "");
+    const uiBaseUrl = (rootOpts.uiBaseUrl || process.env.PGAI_UI_BASE_URL || "https://console.postgres.ai").replace(/\/$/, "");
+    if (opts.debug) {
+        console.log(`Debug: Resolved API base URL: ${apiBaseUrl}`);
+        console.log(`Debug: Resolved UI base URL: ${uiBaseUrl}`);
+    }
+    try {
+        // Step 1: Start local callback server FIRST to get actual port
+        console.log("Starting local callback server...");
+        const requestedPort = opts.port || 0; // 0 = OS assigns available port
+        const callbackServer = authServer.createCallbackServer(requestedPort, params.state, 300000);
+        // Wait a bit for server to start and get port
+        await new Promise(resolve => setTimeout(resolve, 100));
+        const actualPort = callbackServer.getPort();
+        const redirectUri = `http://localhost:${actualPort}/callback`;
+        console.log(`Callback server listening on port ${actualPort}`);
+        // Step 2: Initialize OAuth session on backend
+        console.log("Initializing authentication session...");
+        const initData = JSON.stringify({
+            client_type: "cli",
+            state: params.state,
+            code_challenge: params.codeChallenge,
+            code_challenge_method: params.codeChallengeMethod,
+            redirect_uri: redirectUri,
+        });
+        // Build init URL by appending to the API base path (keep /api/general)
+        const initUrl = new url_1.URL(`${apiBaseUrl}/rpc/oauth_init`);
+        if (opts.debug) {
+            console.log(`Debug: Trying to POST to: ${initUrl.toString()}`);
+            console.log(`Debug: Request data: ${initData}`);
+        }
+        const initReq = http.request(initUrl, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                "Content-Length": Buffer.byteLength(initData),
+            },
+        }, (res) => {
+            let data = "";
+            res.on("data", (chunk) => (data += chunk));
+            res.on("end", async () => {
+                if (res.statusCode !== 200) {
+                    console.error(`Failed to initialize auth session: ${res.statusCode}`);
+                    console.error(data);
+                    callbackServer.server.close();
+                    process.exitCode = 1;
+                    return;
+                }
+                // Step 3: Open browser
+                const authUrl = `${uiBaseUrl}/cli/auth?state=${encodeURIComponent(params.state)}&code_challenge=${encodeURIComponent(params.codeChallenge)}&code_challenge_method=S256&redirect_uri=${encodeURIComponent(redirectUri)}`;
+                if (opts.debug) {
+                    console.log(`Debug: Auth URL: ${authUrl}`);
+                }
+                console.log(`\nOpening browser for authentication...`);
+                console.log(`If browser does not open automatically, visit:\n${authUrl}\n`);
+                // Open browser (cross-platform)
+                const openCommand = process.platform === "darwin" ? "open" :
+                    process.platform === "win32" ? "start" :
+                        "xdg-open";
+                (0, child_process_1.spawn)(openCommand, [authUrl], { detached: true, stdio: "ignore" }).unref();
+                // Step 4: Wait for callback
+                console.log("Waiting for authorization...");
+                try {
+                    const { code } = await callbackServer.promise;
+                    // Step 5: Exchange code for token
+                    console.log("\nExchanging authorization code for API token...");
+                    const exchangeData = JSON.stringify({
+                        authorization_code: code,
+                        code_verifier: params.codeVerifier,
+                        state: params.state,
+                    });
+                    const exchangeUrl = new url_1.URL(`${apiBaseUrl}/rpc/oauth_token_exchange`);
+                    const exchangeReq = http.request(exchangeUrl, {
+                        method: "POST",
+                        headers: {
+                            "Content-Type": "application/json",
+                            "Content-Length": Buffer.byteLength(exchangeData),
+                        },
+                    }, (exchangeRes) => {
+                        let exchangeData = "";
+                        exchangeRes.on("data", (chunk) => (exchangeData += chunk));
+                        exchangeRes.on("end", () => {
+                            if (exchangeRes.statusCode !== 200) {
+                                console.error(`Failed to exchange code for token: ${exchangeRes.statusCode}`);
+                                console.error(exchangeData);
+                                process.exitCode = 1;
+                                return;
+                            }
+                            try {
+                                const result = JSON.parse(exchangeData);
+                                const apiToken = result.api_token;
+                                const orgId = result.org_id;
+                                // Step 6: Save token to config
+                                config.writeConfig({
+                                    apiKey: apiToken,
+                                    baseUrl: apiBaseUrl,
+                                    orgId: orgId,
+                                });
+                                console.log("\nAuthentication successful!");
+                                console.log(`API key saved to: ${config.getConfigPath()}`);
+                                console.log(`Organization ID: ${orgId}`);
+                                console.log(`\nYou can now use the CLI without specifying an API key.`);
+                            }
+                            catch (err) {
+                                const message = err instanceof Error ? err.message : String(err);
+                                console.error(`Failed to parse response: ${message}`);
+                                process.exitCode = 1;
+                            }
+                        });
+                    });
+                    exchangeReq.on("error", (err) => {
+                        console.error(`Exchange request failed: ${err.message}`);
+                        process.exitCode = 1;
+                    });
+                    exchangeReq.write(exchangeData);
+                    exchangeReq.end();
+                }
+                catch (err) {
+                    const message = err instanceof Error ? err.message : String(err);
+                    console.error(`\nAuthentication failed: ${message}`);
+                    process.exitCode = 1;
+                }
+            });
+        });
+        initReq.on("error", (err) => {
+            console.error(`Failed to connect to API: ${err.message}`);
+            callbackServer.server.close();
+            process.exitCode = 1;
+        });
+        initReq.write(initData);
+        initReq.end();
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        console.error(`Authentication error: ${message}`);
+        process.exitCode = 1;
+    }
+});
+program
+    .command("add-key <apiKey>")
+    .description("store API key")
+    .action(async (apiKey) => {
+    config.writeConfig({ apiKey });
+    console.log(`API key saved to ${config.getConfigPath()}`);
+});
+program
+    .command("show-key")
+    .description("show API key (masked)")
+    .action(async () => {
+    const cfg = config.readConfig();
+    if (!cfg.apiKey) {
+        console.log("No API key configured");
+        console.log(`\nTo authenticate, run: pgai auth`);
+        return;
+    }
+    const mask = (k) => {
+        if (k.length <= 8)
+            return "****";
+        if (k.length <= 16)
+            return `${k.slice(0, 4)}${"*".repeat(k.length - 8)}${k.slice(-4)}`;
+        // For longer keys, show more of the beginning to help identify them
+        return `${k.slice(0, Math.min(12, k.length - 8))}${"*".repeat(Math.max(4, k.length - 16))}${k.slice(-4)}`;
+    };
+    console.log(`Current API key: ${mask(cfg.apiKey)}`);
+    if (cfg.orgId) {
+        console.log(`Organization ID: ${cfg.orgId}`);
+    }
+    console.log(`Config location: ${config.getConfigPath()}`);
+});
+program
+    .command("remove-key")
+    .description("remove API key")
+    .action(async () => {
+    // Check both new config and legacy config
+    const newConfigPath = config.getConfigPath();
+    const hasNewConfig = fs.existsSync(newConfigPath);
+    const legacyPath = path.resolve(process.cwd(), ".pgwatch-config");
+    const hasLegacyConfig = fs.existsSync(legacyPath) && fs.statSync(legacyPath).isFile();
+    if (!hasNewConfig && !hasLegacyConfig) {
+        console.log("No API key configured");
+        return;
+    }
+    // Remove from new config
+    if (hasNewConfig) {
+        config.deleteConfigKeys(["apiKey", "orgId"]);
+    }
+    // Remove from legacy config
+    if (hasLegacyConfig) {
+        try {
+            const content = fs.readFileSync(legacyPath, "utf8");
+            const filtered = content
+                .split(/\r?\n/)
+                .filter((l) => !/^api_key=/.test(l))
+                .join("\n")
+                .replace(/\n+$/g, "\n");
+            fs.writeFileSync(legacyPath, filtered, "utf8");
+        }
+        catch (err) {
+            // If we can't read/write the legacy config, just skip it
+            console.warn(`Warning: Could not update legacy config: ${err instanceof Error ? err.message : String(err)}`);
+        }
+    }
+    console.log("API key removed");
+    console.log(`\nTo authenticate again, run: pgai auth`);
+});
+program
+    .command("generate-grafana-password")
+    .description("generate Grafana password")
+    .action(async () => {
+    const cfgPath = path.resolve(process.cwd(), ".pgwatch-config");
+    try {
+        // Generate secure password using openssl
+        const { stdout: password } = await execPromise("openssl rand -base64 12 | tr -d '\n'");
+        const newPassword = password.trim();
+        if (!newPassword) {
+            console.error("Failed to generate password");
+            process.exitCode = 1;
+            return;
+        }
+        // Read existing config
+        let configContent = "";
+        if (fs.existsSync(cfgPath)) {
+            const stats = fs.statSync(cfgPath);
+            if (stats.isDirectory()) {
+                console.error(".pgwatch-config is a directory, expected a file. Skipping read.");
+            }
+            else {
+                configContent = fs.readFileSync(cfgPath, "utf8");
+            }
+        }
+        // Update or add grafana_password
+        const lines = configContent.split(/\r?\n/).filter((l) => !/^grafana_password=/.test(l));
+        lines.push(`grafana_password=${newPassword}`);
+        // Write back
+        fs.writeFileSync(cfgPath, lines.filter(Boolean).join("\n") + "\n", "utf8");
+        console.log("✓ New Grafana password generated and saved");
+        console.log("\nNew credentials:");
+        console.log("  URL:      http://localhost:3000");
+        console.log("  Username: monitor");
+        console.log(`  Password: ${newPassword}`);
+        console.log("\nRestart Grafana to apply:");
+        console.log("  postgres-ai restart grafana");
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.error(`Failed to generate password: ${message}`);
+        console.error("\nNote: This command requires 'openssl' to be installed");
+        process.exitCode = 1;
+    }
+});
+program
+    .command("show-grafana-credentials")
+    .description("show Grafana credentials")
+    .action(async () => {
+    const cfgPath = path.resolve(process.cwd(), ".pgwatch-config");
+    if (!fs.existsSync(cfgPath)) {
+        console.error("Configuration file not found. Run 'quickstart' first.");
+        process.exitCode = 1;
+        return;
+    }
+    const stats = fs.statSync(cfgPath);
+    if (stats.isDirectory()) {
+        console.error(".pgwatch-config is a directory, expected a file. Cannot read credentials.");
+        process.exitCode = 1;
+        return;
+    }
+    const content = fs.readFileSync(cfgPath, "utf8");
+    const lines = content.split(/\r?\n/);
+    let password = "";
+    for (const line of lines) {
+        const m = line.match(/^grafana_password=(.+)$/);
+        if (m) {
+            password = m[1].trim();
+            break;
+        }
+    }
+    if (!password) {
+        console.error("Grafana password not found in configuration");
+        process.exitCode = 1;
+        return;
+    }
+    console.log("\nGrafana credentials:");
+    console.log("  URL:      http://localhost:3000");
+    console.log("  Username: monitor");
+    console.log(`  Password: ${password}`);
+    console.log("");
+});
+program.parseAsync(process.argv);
+//# sourceMappingURL=postgres-ai.js.map