postgresai 0.14.0-dev.51 → 0.14.0-dev.53

package/bin/postgres-ai.ts

@@ -7,7 +7,6 @@ import * as yaml from "js-yaml";
 import * as fs from "fs";
 import * as path from "path";
 import * as os from "os";
-import * as crypto from "node:crypto";
 import { Client } from "pg";
 import { startMcpServer } from "../lib/mcp-server";
 import { fetchIssues, fetchIssueComments, createIssueComment, fetchIssue } from "../lib/issues";
@@ -18,8 +17,6 @@ import * as authServer from "../lib/auth-server";
 import { maskSecret } from "../lib/util";
 import { createInterface } from "readline";
 import * as childProcess from "child_process";
-import { REPORT_GENERATORS, CHECK_INFO, generateAllReports } from "../lib/checkup";
-import { createCheckupReport, uploadCheckupReportJson, RpcError, formatRpcErrorForDisplay } from "../lib/checkup-api";
 
 // Singleton readline interface for stdin prompts
 let rl: ReturnType<typeof createInterface> | null = null;
@@ -112,62 +109,6 @@ async function question(prompt: string): Promise<string> {
   });
 }
 
-function expandHomePath(p: string): string {
-  const s = (p || "").trim();
-  if (!s) return s;
-  if (s === "~") return os.homedir();
-  if (s.startsWith("~/") || s.startsWith("~\\")) {
-    return path.join(os.homedir(), s.slice(2));
-  }
-  return s;
-}
-
-function createTtySpinner(
-  enabled: boolean,
-  initialText: string
-): { update: (text: string) => void; stop: (finalText?: string) => void } {
-  if (!enabled) {
-    return {
-      update: () => {},
-      stop: () => {},
-    };
-  }
-
-  const frames = ["|", "/", "-", "\\"];
-  const startTs = Date.now();
-  let text = initialText;
-  let frameIdx = 0;
-  let stopped = false;
-
-  const render = (): void => {
-    if (stopped) return;
-    const elapsedSec = ((Date.now() - startTs) / 1000).toFixed(1);
-    const frame = frames[frameIdx % frames.length]!;
-    frameIdx += 1;
-    process.stdout.write(`\r\x1b[2K${frame} ${text} (${elapsedSec}s)`);
-  };
-
-  const timer = setInterval(render, 120);
-  render(); // immediate feedback
-
-  return {
-    update: (t: string) => {
-      text = t;
-      render();
-    },
-    stop: (finalText?: string) => {
-      if (stopped) return;
-      stopped = true;
-      clearInterval(timer);
-      process.stdout.write("\r\x1b[2K");
-      if (finalText && finalText.trim()) {
-        process.stdout.write(finalText);
-      }
-      process.stdout.write("\n");
-    },
-  };
-}
-
 /**
  * CLI configuration options
  */
@@ -345,20 +286,6 @@ program
     "UI base URL for browser routes (overrides PGAI_UI_BASE_URL)"
   );
 
-program
-  .command("set-default-project <project>")
-  .description("store default project for checkup uploads")
-  .action(async (project: string) => {
-    const value = (project || "").trim();
-    if (!value) {
-      console.error("Error: project is required");
-      process.exitCode = 1;
-      return;
-    }
-    config.writeConfig({ defaultProject: value });
-    console.log(`Default project saved: ${value}`);
-  });
-
 program
   .command("prepare-db [conn]")
   .description("prepare database for monitoring: create monitoring user, required view(s), and grant permissions (idempotent)")
@@ -686,251 +613,6 @@ program
     }
   });
 
-program
-  .command("checkup [conn]")
-  .description("generate health check reports directly from PostgreSQL (express mode)")
-  .option("--check-id <id>", `specific check to run: ${Object.keys(CHECK_INFO).join(", ")}, or ALL`, "ALL")
-  .option("--node-name <name>", "node name for reports", "node-01")
-  .option("--output <path>", "output directory for JSON files")
-  .option("--[no-]upload", "upload JSON results to PostgresAI (default: enabled; requires API key)", undefined)
-  .option(
-    "--project <project>",
-    "project name or ID for remote upload (used with --upload; defaults to config defaultProject; auto-generated on first run)"
-  )
-  .option("--json", "output JSON to stdout (implies --no-upload)")
-  .addHelpText(
-    "after",
-    [
-      "",
-      "Available checks:",
-      ...Object.entries(CHECK_INFO).map(([id, title]) => `  ${id}: ${title}`),
-      "",
-      "Examples:",
-      "  postgresai checkup postgresql://user:pass@host:5432/db",
-      "  postgresai checkup postgresql://user:pass@host:5432/db --check-id A003",
-      "  postgresai checkup postgresql://user:pass@host:5432/db --output ./reports",
-      "  postgresai checkup postgresql://user:pass@host:5432/db --project my_project",
-      "  postgresai set-default-project my_project",
-      "  postgresai checkup postgresql://user:pass@host:5432/db",
-      "  postgresai checkup postgresql://user:pass@host:5432/db --no-upload --json",
-    ].join("\n")
-  )
-  .action(async (conn: string | undefined, opts: {
-    checkId: string;
-    nodeName: string;
-    output?: string;
-    upload?: boolean;
-    project?: string;
-    json?: boolean;
-  }, cmd: Command) => {
-    if (!conn) {
-      // No args — show help like other commands do (instead of a bare error).
-      cmd.outputHelp();
-      process.exitCode = 1;
-      return;
-    }
-
-    const shouldPrintJson = !!opts.json;
-    // `--json` implies "local output" mode — do not upload by default.
-    const shouldUpload = opts.upload !== false && !shouldPrintJson;
-    const generateDefaultProjectName = (): string => {
-      // Must start with a letter; use only letters/numbers/underscores.
-      return `project_${crypto.randomBytes(6).toString("hex")}`;
-    };
-
-    // Preflight: validate/create output directory BEFORE connecting / running checks.
-    // This avoids waiting on network/DB work only to fail at the very end.
-    let outputPath: string | undefined;
-    if (opts.output) {
-      const outputDir = expandHomePath(opts.output);
-      outputPath = path.isAbsolute(outputDir) ? outputDir : path.resolve(process.cwd(), outputDir);
-      if (!fs.existsSync(outputPath)) {
-        try {
-          fs.mkdirSync(outputPath, { recursive: true });
-        } catch (e) {
-          const errAny = e as any;
-          const code = typeof errAny?.code === "string" ? errAny.code : "";
-          const msg = errAny instanceof Error ? errAny.message : String(errAny);
-          if (code === "EACCES" || code === "EPERM" || code === "ENOENT") {
-            console.error(`Error: Failed to create output directory: ${outputPath}`);
-            console.error(`Reason: ${msg}`);
-            console.error("Tip: choose a writable path, e.g. --output ./reports or --output ~/reports");
-            process.exitCode = 1;
-            return;
-          }
-          throw e;
-        }
-      }
-    }
-
-    // Preflight: validate upload flags/credentials BEFORE connecting / running checks.
-    // This allows "fast-fail" for missing API key / project name.
-    let uploadCfg:
-      | { apiKey: string; apiBaseUrl: string; project: string; epoch: number }
-      | undefined;
-    let projectWasGenerated = false;
-    if (shouldUpload) {
-      const rootOpts = program.opts() as CliOptions;
-      const { apiKey } = getConfig(rootOpts);
-      if (!apiKey) {
-        console.error("Error: API key is required for upload");
-        console.error("Tip: run 'postgresai auth' or pass --api-key / set PGAI_API_KEY");
-        process.exitCode = 1;
-        return;
-      }
-
-      const cfg = config.readConfig();
-      const { apiBaseUrl } = resolveBaseUrls(rootOpts, cfg);
-      let project = ((opts.project || cfg.defaultProject) || "").trim();
-      if (!project) {
-        project = generateDefaultProjectName();
-        projectWasGenerated = true;
-        try {
-          config.writeConfig({ defaultProject: project });
-        } catch (e) {
-          const message = e instanceof Error ? e.message : String(e);
-          console.error(`Warning: Failed to save generated default project: ${message}`);
-        }
-      }
-      uploadCfg = {
-        apiKey,
-        apiBaseUrl,
-        project,
-      };
-    }
-
-    // Use the same SSL behavior as prepare-db:
-    // - Default: sslmode=prefer (try SSL first, fallback to non-SSL)
-    // - Respect PGSSLMODE env and ?sslmode=... in connection URI
-    const adminConn = resolveAdminConnection({
-      conn,
-      envPassword: process.env.PGPASSWORD,
-    });
-    let client: Client | undefined;
-    const spinnerEnabled = !!process.stdout.isTTY && !!shouldUpload;
-    const spinner = createTtySpinner(spinnerEnabled, "Connecting to Postgres");
-
-    try {
-      spinner.update("Connecting to Postgres");
-      const connResult = await connectWithSslFallback(Client, adminConn);
-      client = connResult.client as Client;
-
-      let reports: Record<string, any>;
-      let uploadSummary:
-        | { project: string; reportId: number; uploaded: Array<{ checkId: string; filename: string; chunkId: number }> }
-        | undefined;
-
-      if (opts.checkId === "ALL") {
-        reports = await generateAllReports(client, opts.nodeName, (p) => {
-          spinner.update(`Running ${p.checkId}: ${p.checkTitle} (${p.index}/${p.total})`);
-        });
-      } else {
-        const checkId = opts.checkId.toUpperCase();
-        const generator = REPORT_GENERATORS[checkId];
-        if (!generator) {
-          spinner.stop();
-          console.error(`Unknown check ID: ${opts.checkId}`);
-          console.error(`Available: ${Object.keys(CHECK_INFO).join(", ")}, ALL`);
-          process.exitCode = 1;
-          return;
-        }
-        spinner.update(`Running ${checkId}: ${CHECK_INFO[checkId] || checkId}`);
-        reports = { [checkId]: await generator(client, opts.nodeName) };
-      }
-
-      // Optional: upload to PostgresAI API.
-      if (uploadCfg) {
-        spinner.update("Creating remote checkup report");
-        const created = await createCheckupReport({
-          apiKey: uploadCfg.apiKey,
-          apiBaseUrl: uploadCfg.apiBaseUrl,
-          project: uploadCfg.project,
-        });
-
-        const reportId = created.reportId;
-        // Keep upload progress out of stdout when JSON is printed to stdout.
-        const logUpload = (msg: string): void => {
-          if (shouldPrintJson) {
-            console.error(msg);
-          } else {
-            console.log(msg);
-          }
-        };
-        logUpload(`Created remote checkup report: ${reportId}`);
-
-        const uploaded: Array<{ checkId: string; filename: string; chunkId: number }> = [];
-        for (const [checkId, report] of Object.entries(reports)) {
-          spinner.update(`Uploading ${checkId}.json`);
-          const jsonText = JSON.stringify(report, null, 2);
-          const r = await uploadCheckupReportJson({
-            apiKey: uploadCfg.apiKey,
-            apiBaseUrl: uploadCfg.apiBaseUrl,
-            reportId,
-            filename: `${checkId}.json`,
-            checkId,
-            jsonText,
-          });
-          uploaded.push({ checkId, filename: `${checkId}.json`, chunkId: r.reportChunkId });
-        }
-        logUpload("Upload completed");
-        uploadSummary = { project: uploadCfg.project, reportId, uploaded };
-      }
-
-      spinner.stop();
-      // Output results
-      if (opts.output) {
-        // Write to files
-        // outputPath is preflight-validated above
-        const outDir = outputPath || path.resolve(process.cwd(), expandHomePath(opts.output));
-        for (const [checkId, report] of Object.entries(reports)) {
-          const filePath = path.join(outDir, `${checkId}.json`);
-          fs.writeFileSync(filePath, JSON.stringify(report, null, 2), "utf8");
-          console.log(`✓ ${checkId}: ${filePath}`);
-        }
-      }
-
-      if (uploadSummary) {
-        const out = shouldPrintJson ? console.error : console.log;
-        out("\nCheckup report uploaded");
-        out("======================\n");
-        if (projectWasGenerated) {
-          out(`Project: ${uploadSummary.project} (generated and saved as default)`);
-        } else {
-          out(`Project: ${uploadSummary.project}`);
-        }
-        out(`Report ID: ${uploadSummary.reportId}`);
-        out("View in Console: console.postgres.ai → Support → checkup reports");
-        out("");
-        out("Files:");
-        for (const item of uploadSummary.uploaded) {
-          out(`- ${item.checkId}: ${item.filename}`);
-        }
-      }
-
-      if (shouldPrintJson) {
-        console.log(JSON.stringify(reports, null, 2));
-      } else if (!shouldUpload && !opts.output) {
-        // Offline mode keeps historical behavior: print JSON when not uploading and no --output.
-        console.log(JSON.stringify(reports, null, 2));
-      }
-    } catch (error) {
-      spinner.stop();
-      if (error instanceof RpcError) {
-        for (const line of formatRpcErrorForDisplay(error)) {
-          console.error(line);
-        }
-      } else {
-        const message = error instanceof Error ? error.message : String(error);
-        console.error(`Error: ${message}`);
-      }
-      process.exitCode = 1;
-    } finally {
-      if (client) {
-        await client.end();
-      }
-    }
-  });
-
 /**
  * Stub function for not implemented commands
  */
@@ -1915,9 +1597,6 @@ auth
       }
       
       config.writeConfig({ apiKey: trimmedKey });
-      // When API key is set directly, invalidate org/project selection
-      // as it likely belongs to a different account/session.
-      config.deleteConfigKeys(["orgId", "defaultProject"]);
       console.log(`API key saved to ${config.getConfigPath()}`);
       return;
     }
@@ -2096,9 +1775,6 @@ auth
             baseUrl: apiBaseUrl,
             orgId: orgId,
           });
-          // When re-authing via OAuth, orgId will be refreshed,
-          // but defaultProject may no longer be valid in the new org.
-          config.deleteConfigKeys(["defaultProject"]);
 
           console.log("\nAuthentication successful!");
           console.log(`API key saved to: ${config.getConfigPath()}`);

package/bun.lock

@@ -14,8 +14,6 @@
         "@types/bun": "^1.1.14",
         "@types/js-yaml": "^4.0.9",
         "@types/pg": "^8.15.6",
-        "ajv": "^8.17.1",
-        "ajv-formats": "^3.0.1",
         "typescript": "^5.3.3",
       },
     },
@@ -131,7 +129,7 @@
 
     "jose": ["jose@6.1.3", "", {}, "sha512-0TpaTfihd4QMNwrz/ob2Bp7X04yuxJkjRGi4aKmOqwhov54i6u79oCv7T+C7lo70MKH6BesI3vscD1yb/yzKXQ=="],
 
-    "js-yaml": ["js-yaml@4.1.1", "", { "dependencies": { "argparse": "^2.0.1" }, "bin": { "js-yaml": "bin/js-yaml.js" } }, "sha512-qQKT4zQxXl8lLwBtHMWwaTcGfFOZviOJet3Oy/xmGk2gZH677CJM9EvtfdSkgWcATZhj/55JZ0rmy3myCT5lsA=="],
+    "js-yaml": ["js-yaml@4.1.1", "", { "dependencies": { "argparse": "^2.0.1" }, "bin": "bin/js-yaml.js" }, "sha512-qQKT4zQxXl8lLwBtHMWwaTcGfFOZviOJet3Oy/xmGk2gZH677CJM9EvtfdSkgWcATZhj/55JZ0rmy3myCT5lsA=="],
 
     "json-schema-traverse": ["json-schema-traverse@1.0.0", "", {}, "sha512-NM8/P9n3XjXhIZn1lLhkFaACTOURQXjWhV4BA/RnOv8xvgqtqpAX9IO4mRQxSx1Rlo4tqzeqb0sOlruaOy3dug=="],