@indiekitai/pg-dash 0.3.2 → 0.3.3

package/dist/cli.js

@@ -661,6 +661,28 @@ SELECT pg_reload_conf();`,
     const ignoredSet = new Set(ignoredIds);
     const activeIssues = issues.filter((i) => !ignoredSet.has(i.id));
     const ignoredCount = issues.length - activeIssues.length;
+    const batchFixes = [];
+    const groups = /* @__PURE__ */ new Map();
+    for (const issue of activeIssues) {
+      const prefix = issue.id.replace(/-[^-]+$/, "");
+      if (!groups.has(prefix)) groups.set(prefix, []);
+      groups.get(prefix).push(issue);
+    }
+    const BATCH_TITLES = {
+      "schema-fk-no-idx": "Create all missing FK indexes",
+      "schema-unused-idx": "Drop all unused indexes",
+      "schema-no-pk": "Fix all tables missing primary keys",
+      "maint-vacuum": "VACUUM all overdue tables",
+      "maint-analyze": "ANALYZE all tables missing statistics",
+      "perf-bloated-idx": "REINDEX all bloated indexes",
+      "perf-bloat": "VACUUM FULL all bloated tables"
+    };
+    for (const [prefix, group] of groups) {
+      if (group.length <= 1) continue;
+      const title = BATCH_TITLES[prefix] || `Fix all ${group.length} ${prefix} issues`;
+      const sql = group.map((i) => i.fix.split("\n").filter((l) => !l.trim().startsWith("--")).join("\n").trim()).filter(Boolean).join(";\n") + ";";
+      batchFixes.push({ type: prefix, title: `${title} (${group.length})`, count: group.length, sql });
+    }
     const score = computeAdvisorScore(activeIssues);
     return {
       score,
@@ -668,7 +690,8 @@ SELECT pg_reload_conf();`,
       issues: activeIssues,
       breakdown: computeBreakdown(activeIssues),
       skipped,
-      ignoredCount
+      ignoredCount,
+      batchFixes
     };
   } finally {
     client.release();
@@ -943,6 +966,7 @@ var Collector = class extends EventEmitter {
   pruneTimer = null;
   prev = null;
   lastSnapshot = {};
+  collectCount = 0;
   start() {
     this.collect().catch((err) => console.error("[collector] Initial collection failed:", err));
     this.timer = setInterval(() => {
@@ -1045,6 +1069,28 @@ var Collector = class extends EventEmitter {
       console.error("[collector] Error collecting metrics:", err.message);
       return snapshot;
     }
+    this.collectCount++;
+    if (this.collectCount % 10 === 0) {
+      try {
+        const client = await this.pool.connect();
+        try {
+          const tableRes = await client.query(`
+            SELECT schemaname, relname,
+                   pg_total_relation_size(quote_ident(schemaname) || '.' || quote_ident(relname)) as total_size
+            FROM pg_stat_user_tables
+            ORDER BY pg_total_relation_size(quote_ident(schemaname) || '.' || quote_ident(relname)) DESC
+            LIMIT 20
+          `);
+          for (const row of tableRes.rows) {
+            this.store.insert(`table_size:${row.schemaname}.${row.relname}`, parseInt(row.total_size), now);
+          }
+        } finally {
+          client.release();
+        }
+      } catch (err) {
+        console.error("[collector] Error collecting table sizes:", err.message);
+      }
+    }
     const points = Object.entries(snapshot).map(([metric, value]) => ({
       timestamp: now,
       metric,
@@ -1896,7 +1942,12 @@ function registerActivityRoutes(app, pool) {
 
 // src/server/routes/advisor.ts
 init_advisor();
-function registerAdvisorRoutes(app, pool, longQueryThreshold) {
+var RANGE_MAP2 = {
+  "24h": 24 * 60 * 60 * 1e3,
+  "7d": 7 * 24 * 60 * 60 * 1e3,
+  "30d": 30 * 24 * 60 * 60 * 1e3
+};
+function registerAdvisorRoutes(app, pool, longQueryThreshold, store) {
   app.get("/api/advisor", async (c) => {
     try {
       return c.json(await getAdvisorReport(pool, longQueryThreshold));
@@ -1931,6 +1982,18 @@ function registerAdvisorRoutes(app, pool, longQueryThreshold) {
       return c.json({ error: err.message }, 500);
     }
   });
+  app.get("/api/advisor/history", (c) => {
+    if (!store) return c.json([]);
+    try {
+      const range = c.req.query("range") || "24h";
+      const rangeMs = RANGE_MAP2[range] || RANGE_MAP2["24h"];
+      const now = Date.now();
+      const data = store.query("health_score", now - rangeMs, now);
+      return c.json(data);
+    } catch (err) {
+      return c.json({ error: err.message }, 500);
+    }
+  });
   app.post("/api/fix", async (c) => {
     try {
       const body = await c.req.json();
@@ -2214,7 +2277,7 @@ var DiskPredictor = class {
 };
 
 // src/server/routes/disk.ts
-var RANGE_MAP2 = {
+var RANGE_MAP3 = {
   "24h": 24 * 60 * 60 * 1e3,
   "7d": 7 * 24 * 60 * 60 * 1e3,
   "30d": 30 * 24 * 60 * 60 * 1e3
@@ -2274,10 +2337,22 @@ function registerDiskRoutes(app, pool, store) {
       return c.json({ error: err.message }, 500);
     }
   });
+  app.get("/api/disk/table-history/:table", (c) => {
+    try {
+      const table = c.req.param("table");
+      const range = c.req.query("range") || "24h";
+      const rangeMs = RANGE_MAP3[range] || RANGE_MAP3["24h"];
+      const now = Date.now();
+      const data = store.query(`table_size:${table}`, now - rangeMs, now);
+      return c.json(data);
+    } catch (err) {
+      return c.json({ error: err.message }, 500);
+    }
+  });
   app.get("/api/disk/history", (c) => {
     try {
       const range = c.req.query("range") || "24h";
-      const rangeMs = RANGE_MAP2[range] || RANGE_MAP2["24h"];
+      const rangeMs = RANGE_MAP3[range] || RANGE_MAP3["24h"];
       const now = Date.now();
       const data = store.query("db_size_bytes", now - rangeMs, now);
       return c.json(data);
@@ -2476,7 +2551,7 @@ var QueryStatsStore = class {
 };
 
 // src/server/routes/query-stats.ts
-var RANGE_MAP3 = {
+var RANGE_MAP4 = {
   "1h": 60 * 60 * 1e3,
   "6h": 6 * 60 * 60 * 1e3,
   "24h": 24 * 60 * 60 * 1e3,
@@ -2488,7 +2563,7 @@ function registerQueryStatsRoutes(app, store) {
       const range = c.req.query("range") || "1h";
       const orderBy = c.req.query("orderBy") || "total_time";
       const limit = parseInt(c.req.query("limit") || "20", 10);
-      const rangeMs = RANGE_MAP3[range] || RANGE_MAP3["1h"];
+      const rangeMs = RANGE_MAP4[range] || RANGE_MAP4["1h"];
       const now = Date.now();
       const data = store.getTopQueries(now - rangeMs, now, orderBy, limit);
       return c.json(data);
@@ -2500,7 +2575,7 @@ function registerQueryStatsRoutes(app, store) {
     try {
       const queryid = c.req.param("queryid");
       const range = c.req.query("range") || "1h";
-      const rangeMs = RANGE_MAP3[range] || RANGE_MAP3["1h"];
+      const rangeMs = RANGE_MAP4[range] || RANGE_MAP4["1h"];
       const now = Date.now();
       const data = store.getTrend(queryid, now - rangeMs, now);
       return c.json(data);
@@ -2510,6 +2585,74 @@ function registerQueryStatsRoutes(app, store) {
   });
 }
 
+// src/server/routes/export.ts
+init_advisor();
+function registerExportRoutes(app, pool, longQueryThreshold) {
+  app.get("/api/export", async (c) => {
+    const format = c.req.query("format") || "json";
+    try {
+      const [overview, advisor] = await Promise.all([
+        getOverview(pool),
+        getAdvisorReport(pool, longQueryThreshold)
+      ]);
+      if (format === "md") {
+        const lines = [];
+        lines.push(`# pg-dash Health Report`);
+        lines.push(`
+Generated: ${(/* @__PURE__ */ new Date()).toISOString()}
+`);
+        lines.push(`## Overview
+`);
+        lines.push(`- **PostgreSQL**: ${overview.version}`);
+        lines.push(`- **Database Size**: ${overview.dbSize}`);
+        lines.push(`- **Connections**: ${overview.connections.active} active / ${overview.connections.idle} idle / ${overview.connections.max} max`);
+        lines.push(`
+## Health Score: ${advisor.score}/100 (Grade: ${advisor.grade})
+`);
+        lines.push(`### Category Breakdown
+`);
+        lines.push(`| Category | Grade | Score | Issues |`);
+        lines.push(`|----------|-------|-------|--------|`);
+        for (const [cat, b] of Object.entries(advisor.breakdown)) {
+          lines.push(`| ${cat} | ${b.grade} | ${b.score}/100 | ${b.count} |`);
+        }
+        if (advisor.issues.length > 0) {
+          lines.push(`
+### Issues (${advisor.issues.length})
+`);
+          for (const issue of advisor.issues) {
+            const icon = issue.severity === "critical" ? "\u{1F534}" : issue.severity === "warning" ? "\u{1F7E1}" : "\u{1F535}";
+            lines.push(`#### ${icon} [${issue.severity}] ${issue.title}
+`);
+            lines.push(`${issue.description}
+`);
+            lines.push(`**Impact**: ${issue.impact}
+`);
+            lines.push(`**Fix**:
+\`\`\`sql
+${issue.fix}
+\`\`\`
+`);
+          }
+        } else {
+          lines.push(`
+\u2705 No issues found!
+`);
+        }
+        const md = lines.join("\n");
+        c.header("Content-Type", "text/markdown; charset=utf-8");
+        c.header("Content-Disposition", `attachment; filename="pg-dash-report-${(/* @__PURE__ */ new Date()).toISOString().slice(0, 10)}.md"`);
+        return c.body(md);
+      }
+      const data = { overview, advisor, exportedAt: (/* @__PURE__ */ new Date()).toISOString() };
+      c.header("Content-Disposition", `attachment; filename="pg-dash-report-${(/* @__PURE__ */ new Date()).toISOString().slice(0, 10)}.json"`);
+      return c.json(data);
+    } catch (err) {
+      return c.json({ error: err.message }, 500);
+    }
+  });
+}
+
 // src/server/index.ts
 import Database3 from "better-sqlite3";
 import { WebSocketServer, WebSocket } from "ws";
@@ -2610,12 +2753,13 @@ async function startServer(opts) {
   registerOverviewRoutes(app, pool);
   registerMetricsRoutes(app, store, collector);
   registerActivityRoutes(app, pool);
-  registerAdvisorRoutes(app, pool, longQueryThreshold);
+  registerAdvisorRoutes(app, pool, longQueryThreshold, store);
   registerSchemaRoutes(app, pool, schemaTracker);
   registerAlertsRoutes(app, alertManager);
   registerExplainRoutes(app, pool);
   registerDiskRoutes(app, pool, store);
   registerQueryStatsRoutes(app, queryStatsStore);
+  registerExportRoutes(app, pool, longQueryThreshold);
   const uiPath = path3.resolve(__dirname, "ui");
   const MIME_TYPES = {
     ".html": "text/html",
@@ -2763,6 +2907,7 @@ async function startServer(opts) {
           try {
             const report = await getAdvisorReport(pool, longQueryThreshold);
             alertMetrics.health_score = report.score;
+            store.insert("health_score", report.score);
           } catch (err) {
             console.error("[alerts] Error checking health score:", err.message);
           }
@@ -2918,7 +3063,7 @@ Options:
   --query-stats-interval <min> Query stats snapshot interval in minutes (default: 5)
   --long-query-threshold <min> Long query threshold in minutes (default: 5)
   --threshold <score>    Health score threshold for check command (default: 70)
-  -f, --format <fmt>     Output format: text|json (default: text)
+  -f, --format <fmt>     Output format: text|json|md (default: text)
   -v, --version          Show version
   -h, --help             Show this help
 
@@ -2957,6 +3102,38 @@ if (subcommand === "check") {
     const report = await getAdvisorReport2(pool, lqt);
     if (format === "json") {
       console.log(JSON.stringify(report, null, 2));
+    } else if (format === "md") {
+      console.log(`# pg-dash Health Report
+`);
+      console.log(`Generated: ${(/* @__PURE__ */ new Date()).toISOString()}
+`);
+      console.log(`## Health Score: ${report.score}/100 (Grade: ${report.grade})
+`);
+      console.log(`| Category | Grade | Score | Issues |`);
+      console.log(`|----------|-------|-------|--------|`);
+      for (const [cat, b] of Object.entries(report.breakdown)) {
+        console.log(`| ${cat} | ${b.grade} | ${b.score}/100 | ${b.count} |`);
+      }
+      if (report.issues.length > 0) {
+        console.log(`
+### Issues (${report.issues.length})
+`);
+        for (const issue of report.issues) {
+          const icon = issue.severity === "critical" ? "\u{1F534}" : issue.severity === "warning" ? "\u{1F7E1}" : "\u{1F535}";
+          console.log(`#### ${icon} [${issue.severity}] ${issue.title}
+`);
+          console.log(`${issue.description}
+`);
+          console.log(`**Fix**:
+\`\`\`sql
+${issue.fix}
+\`\`\`
+`);
+        }
+      } else {
+        console.log(`
+\u2705 No issues found!`);
+      }
     } else {
       console.log(`
   Health Score: ${report.score}/100 (Grade: ${report.grade})