@deaquinodev/querky 0.4.3 → 0.4.5

package/dist/index.js

@@ -3,7 +3,7 @@
 // src/index.tsx
 import { useState as useState5, useEffect as useEffect5 } from "react";
 import { parseArgs } from "util";
-import { render, Box as Box8, Text as Text8 } from "ink";
+import { render, Box as Box9, Text as Text9 } from "ink";
 
 // src/db/client.ts
 import { Client } from "pg";
@@ -39,10 +39,11 @@ var PgDbClient = class {
   }
   pg;
   async query(sql) {
-    const result = await this.pg.query(sql);
+    const raw = await this.pg.query(sql);
+    const result = Array.isArray(raw) ? raw[raw.length - 1] : raw;
     return {
-      fields: result.fields.map((f) => f.name),
-      rows: result.rows,
+      fields: (result.fields ?? []).map((f) => f.name),
+      rows: result.rows ?? [],
       rowCount: result.rowCount ?? 0
     };
   }
@@ -196,7 +197,7 @@ import { useState as useState3, useEffect as useEffect4, useRef as useRef2 } fro
 import { writeFileSync as writeFileSync5 } from "fs";
 import { homedir as homedir4 } from "os";
 import { join as join6 } from "path";
-import { Box as Box6, Text as Text6, Static, useApp, useInput as useInput2, useStdin as useStdin2 } from "ink";
+import { Box as Box7, Text as Text7, Static, useApp, useInput as useInput2, useStdin as useStdin2 } from "ink";
 
 // src/db/query.ts
 async function runQuery(client, sql) {
@@ -625,8 +626,12 @@ var COMMANDS = {
     description: "Save last query as a named alias",
     category: "Aliases",
     usage: "/save <name>",
-    detail: "Saves the last executed SQL query as a named alias scoped to the current database. Run it later with /<name>. Supports positional ($1, $2) and named (:param) substitution.",
-    example: "/save active-orders",
+    detail: "Saves the last executed SQL query as a named alias scoped to the current database. Run it later with /<name>. If the query contains $1, $2 placeholders or :param placeholders, values are substituted at run time.",
+    examples: [
+      "/save active-orders                 (no params \u2014 run with /active-orders)",
+      "/save user-by-id                    (query had WHERE id = $1 \u2014 run with /user-by-id 42)",
+      "/save by-status                     (query had WHERE status = :status \u2014 run with /by-status :status=active)"
+    ],
     run: (ctx) => {
       const name = ctx.args.trim();
       if (!name) return { ok: false, message: "Usage: /save <name>" };
@@ -640,8 +645,14 @@ var COMMANDS = {
     description: "Define an alias inline",
     category: "Aliases",
     usage: "/alias <name> <SQL>",
-    detail: "Defines a named alias without running a query first. Equivalent to /save but lets you specify the SQL directly.",
-    example: "/alias recent SELECT * FROM events ORDER BY created_at DESC LIMIT 20",
+    detail: "Defines a named alias without running a query first. Use $1, $2 for positional args or :param for named args \u2014 values are substituted when the alias is invoked.",
+    examples: [
+      "/alias recent SELECT * FROM events ORDER BY created_at DESC LIMIT 20",
+      "/alias user SELECT * FROM users WHERE id = $1",
+      "/alias by-email SELECT * FROM users WHERE email = :email",
+      "/user 42                             (positional)",
+      "/by-email :email=alice@example.com   (named)"
+    ],
     run: (ctx) => {
       const [name, ...rest] = ctx.args.trim().split(/\s+/);
       if (!name || rest.length === 0) return { ok: false, message: "Usage: /alias <name> <SQL>" };
@@ -678,6 +689,16 @@ var COMMANDS = {
       return removed ? { ok: true, message: `Removed /${name}` } : { ok: false, message: `No alias named /${name}` };
     }
   },
+  "erd": {
+    description: "Visualize tables and relationships as a schema diagram",
+    category: "Schema",
+    usage: "/erd",
+    detail: "Fetches all tables, columns, and foreign key relationships from the connected database and renders a color-coded schema diagram. Each table gets a unique color; FK references use the color of the table they point to.",
+    run: (ctx) => {
+      ctx.onErd();
+      return { ok: true, message: "" };
+    }
+  },
   "help": {
     description: "Show help for all commands or a specific command",
     category: "Session",
@@ -696,7 +717,8 @@ var COMMANDS = {
           description: cmd.description,
           psqlAlias: PSQL_REVERSE[name],
           detail: cmd.detail,
-          example: cmd.example
+          example: cmd.example,
+          examples: cmd.examples
         };
         return { ok: true, message: "", helpData: { mode: "detail", entry } };
       }
@@ -745,6 +767,83 @@ function runCommand(input, ctx) {
   return { ok: false, message: `Unknown command: ${prefix}${rawName}` };
 }
 
+// src/db/erd.ts
+function erdSql(driver) {
+  if (driver === "postgresql") {
+    return `
+      SELECT c.table_name, c.column_name, c.data_type,
+        (pk.column_name IS NOT NULL) AS is_pk,
+        fk.foreign_table AS fk_table
+      FROM information_schema.columns c
+      LEFT JOIN (
+        SELECT kcu.table_name, kcu.column_name
+        FROM information_schema.table_constraints tc
+        JOIN information_schema.key_column_usage kcu
+          ON tc.constraint_name = kcu.constraint_name AND tc.table_schema = kcu.table_schema
+        WHERE tc.constraint_type = 'PRIMARY KEY' AND tc.table_schema = 'public'
+      ) pk ON pk.table_name = c.table_name AND pk.column_name = c.column_name
+      LEFT JOIN (
+        SELECT kcu.table_name, kcu.column_name, ccu.table_name AS foreign_table
+        FROM information_schema.table_constraints tc
+        JOIN information_schema.key_column_usage kcu
+          ON tc.constraint_name = kcu.constraint_name AND tc.table_schema = kcu.table_schema
+        JOIN information_schema.constraint_column_usage ccu
+          ON tc.constraint_name = ccu.constraint_name AND tc.table_schema = ccu.table_schema
+        WHERE tc.constraint_type = 'FOREIGN KEY' AND tc.table_schema = 'public'
+      ) fk ON fk.table_name = c.table_name AND fk.column_name = c.column_name
+      WHERE c.table_schema = 'public'
+      ORDER BY c.table_name, c.ordinal_position
+    `;
+  }
+  if (driver === "mysql") {
+    return `
+      SELECT c.TABLE_NAME AS table_name, c.COLUMN_NAME AS column_name,
+        c.DATA_TYPE AS data_type,
+        (c.COLUMN_KEY = 'PRI') AS is_pk,
+        kcu.REFERENCED_TABLE_NAME AS fk_table
+      FROM information_schema.COLUMNS c
+      LEFT JOIN information_schema.KEY_COLUMN_USAGE kcu
+        ON kcu.TABLE_SCHEMA = c.TABLE_SCHEMA
+        AND kcu.TABLE_NAME = c.TABLE_NAME
+        AND kcu.COLUMN_NAME = c.COLUMN_NAME
+        AND kcu.REFERENCED_TABLE_NAME IS NOT NULL
+      WHERE c.TABLE_SCHEMA = DATABASE()
+      ORDER BY c.TABLE_NAME, c.ORDINAL_POSITION
+    `;
+  }
+  return `
+    SELECT m.name AS table_name, p.name AS column_name,
+      p.type AS data_type,
+      (p.pk > 0) AS is_pk,
+      f."table" AS fk_table
+    FROM sqlite_master m
+    JOIN pragma_table_info(m.name) p
+    LEFT JOIN pragma_foreign_key_list(m.name) f ON f."from" = p.name
+    WHERE m.type = 'table' AND m.name NOT LIKE 'sqlite_%'
+    ORDER BY m.name, p.cid
+  `;
+}
+async function fetchErd(client, driver) {
+  const result = await client.query(erdSql(driver));
+  const tableMap = /* @__PURE__ */ new Map();
+  for (const row of result.rows) {
+    const tableName = String(row["table_name"] ?? "");
+    if (!tableMap.has(tableName)) tableMap.set(tableName, []);
+    const cols = tableMap.get(tableName);
+    const colName = String(row["column_name"] ?? "");
+    if (cols.some((c) => c.name === colName)) continue;
+    cols.push({
+      name: colName,
+      type: String(row["data_type"] ?? ""),
+      isPk: Boolean(row["is_pk"]),
+      fkTable: row["fk_table"] ? String(row["fk_table"]) : void 0
+    });
+  }
+  return {
+    tables: Array.from(tableMap.entries()).map(([name, columns]) => ({ name, columns }))
+  };
+}
+
 // src/commands/shell.ts
 import { exec } from "child_process";
 import { readFileSync as readFileSync2, writeFileSync as writeFileSync2, unlinkSync } from "fs";
@@ -1626,7 +1725,7 @@ import { Box as Box2, Text as Text2 } from "ink";
 import { jsx as jsx2, jsxs as jsxs2 } from "react/jsx-runtime";
 var COL_PAD = 1;
 var INDIGO = "#818cf8";
-var BORDER = "white";
+var BORDER2 = "white";
 var NULL_COLOR = "#6366f1";
 var NULL_MARKER = "\u2205";
 function isNull(val) {
@@ -1657,10 +1756,10 @@ function hline(widths, left, mid, right) {
 function ExpandedTable({ columns, rows }) {
   const keyWidth = columns.reduce((max, col) => Math.max(max, col.length), 0);
   return /* @__PURE__ */ jsx2(Box2, { flexDirection: "column", children: rows.map((row, i) => /* @__PURE__ */ jsxs2(Box2, { flexDirection: "column", marginBottom: i < rows.length - 1 ? 1 : 0, children: [
-    /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: `\u2500[ Record ${i + 1} ]${"\u2500".repeat(Math.max(0, keyWidth + 14 - String(i + 1).length))}` }),
+    /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: `\u2500[ Record ${i + 1} ]${"\u2500".repeat(Math.max(0, keyWidth + 14 - String(i + 1).length))}` }),
     columns.map((col) => /* @__PURE__ */ jsxs2(Box2, { children: [
       /* @__PURE__ */ jsx2(Text2, { color: INDIGO, bold: true, children: col.padEnd(keyWidth) }),
-      /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: " \u2502 " }),
+      /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: " \u2502 " }),
       isNull(row[col]) ? /* @__PURE__ */ jsx2(Text2, { color: NULL_COLOR, dimColor: true, children: NULL_MARKER }) : /* @__PURE__ */ jsx2(Text2, { children: cellValue(row[col]) })
     ] }, col))
   ] }, i)) });
@@ -1673,31 +1772,31 @@ function Table({ columns, rows, expanded = false }) {
   const botLine = hline(widths, "\u2570", "\u2534", "\u256F");
   function renderHeaderRow(cols) {
     return /* @__PURE__ */ jsxs2(Box2, { children: [
-      /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: "\u2502" }),
+      /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: "\u2502" }),
       cols.map((v, i) => /* @__PURE__ */ jsxs2(Box2, { children: [
         /* @__PURE__ */ jsx2(Text2, { color: INDIGO, bold: true, children: " ".repeat(COL_PAD) + pad(v, widths[i]) + " ".repeat(COL_PAD) }),
-        /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: "\u2502" })
+        /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: "\u2502" })
       ] }, i))
     ] });
   }
   function renderDataRow(row) {
     return /* @__PURE__ */ jsxs2(Box2, { children: [
-      /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: "\u2502" }),
+      /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: "\u2502" }),
       columns.map((col, i) => /* @__PURE__ */ jsxs2(Box2, { children: [
         isNull(row[col]) ? /* @__PURE__ */ jsx2(Text2, { color: NULL_COLOR, dimColor: true, children: " ".repeat(COL_PAD) + pad(NULL_MARKER, widths[i]) + " ".repeat(COL_PAD) }) : /* @__PURE__ */ jsx2(Text2, { children: " ".repeat(COL_PAD) + pad(cellValue(row[col]), widths[i]) + " ".repeat(COL_PAD) }),
-        /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: "\u2502" })
+        /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: "\u2502" })
       ] }, i))
     ] });
   }
   return /* @__PURE__ */ jsxs2(Box2, { flexDirection: "column", children: [
-    /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: topLine }),
+    /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: topLine }),
     renderHeaderRow(columns),
-    /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: midLine }),
+    /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: midLine }),
     rows.map((row, i) => /* @__PURE__ */ jsxs2(Box2, { flexDirection: "column", children: [
       renderDataRow(row),
-      i < rows.length - 1 && /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: midLine })
+      i < rows.length - 1 && /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: midLine })
     ] }, i)),
-    /* @__PURE__ */ jsx2(Text2, { color: BORDER, children: botLine })
+    /* @__PURE__ */ jsx2(Text2, { color: BORDER2, children: botLine })
   ] });
 }
 
@@ -1857,7 +1956,7 @@ import { jsx as jsx5, jsxs as jsxs5 } from "react/jsx-runtime";
 var USAGE_WIDTH = 28;
 function HelpView({ data }) {
   if (data.mode === "detail" && data.entry) {
-    const { usage, description, psqlAlias, detail, example } = data.entry;
+    const { usage, description, psqlAlias, detail, example, examples } = data.entry;
     return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginTop: 1, children: [
       /* @__PURE__ */ jsxs5(Text5, { bold: true, color: "white", children: [
         usage,
@@ -1865,10 +1964,13 @@ function HelpView({ data }) {
       ] }),
       /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text5, { children: description }) }),
       detail && /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: detail }) }),
-      example && /* @__PURE__ */ jsxs5(Box5, { marginTop: 1, children: [
+      examples && examples.length > 0 ? /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginTop: 1, children: [
+        /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: "Examples:" }),
+        examples.map((ex, i) => /* @__PURE__ */ jsx5(Box5, { marginLeft: 2, children: /* @__PURE__ */ jsx5(Text5, { color: theme.accent, children: ex }) }, i))
+      ] }) : example ? /* @__PURE__ */ jsxs5(Box5, { marginTop: 1, children: [
         /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: "Example: " }),
         /* @__PURE__ */ jsx5(Text5, { color: theme.accent, children: example })
-      ] })
+      ] }) : null
     ] });
   }
   if (data.mode === "list" && data.groups) {
@@ -1894,8 +1996,137 @@ function HelpView({ data }) {
   return null;
 }
 
-// src/ui/components/App.tsx
+// src/ui/components/ErdView.tsx
+import { Box as Box6, Text as Text6 } from "ink";
 import { Fragment as Fragment3, jsx as jsx6, jsxs as jsxs6 } from "react/jsx-runtime";
+var TABLE_COLORS = [
+  "#f472b6",
+  "#34d399",
+  "#fb923c",
+  "#60a5fa",
+  "#a78bfa",
+  "#f87171",
+  "#fbbf24",
+  "#2dd4bf"
+];
+var PAD = 1;
+var GAP = 2;
+var FK_PREFIX = "FK \u2192 ";
+function computeMetrics(table) {
+  const nameW = Math.max(2, ...table.columns.map((c) => c.name.length));
+  const typeW = Math.max(4, ...table.columns.map((c) => c.type.length));
+  const keyW = Math.max(
+    2,
+    ...table.columns.map((c) => {
+      if (c.isPk) return 2;
+      if (c.fkTable) return FK_PREFIX.length + c.fkTable.length;
+      return 0;
+    })
+  );
+  const totalW = 4 + 3 * PAD * 2 + nameW + typeW + keyW;
+  return { nameW, typeW, keyW, totalW };
+}
+function groupIntoRows(metrics, termW) {
+  const rows = [];
+  let row = [];
+  let usedW = 0;
+  for (let i = 0; i < metrics.length; i++) {
+    const w = metrics[i].totalW;
+    if (row.length === 0) {
+      row.push(i);
+      usedW = w;
+    } else if (usedW + GAP + w <= termW) {
+      row.push(i);
+      usedW += GAP + w;
+    } else {
+      rows.push(row);
+      row = [i];
+      usedW = w;
+    }
+  }
+  if (row.length > 0) rows.push(row);
+  return rows;
+}
+function TableBox({ table, m, color, colorMap }) {
+  const { nameW, typeW, keyW, totalW } = m;
+  const sp = " ".repeat(PAD);
+  const p = (s, w) => s.slice(0, w).padEnd(w);
+  const top = "\u256D" + "\u2500".repeat(totalW - 2) + "\u256E";
+  const sep = "\u251C" + "\u2500".repeat(nameW + PAD * 2) + "\u252C" + "\u2500".repeat(typeW + PAD * 2) + "\u252C" + "\u2500".repeat(keyW + PAD * 2) + "\u2524";
+  const bot = "\u2570" + "\u2500".repeat(nameW + PAD * 2) + "\u2534" + "\u2500".repeat(typeW + PAD * 2) + "\u2534" + "\u2500".repeat(keyW + PAD * 2) + "\u256F";
+  const headerW = totalW - 4;
+  return /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsx6(Text6, { color, children: top }),
+    /* @__PURE__ */ jsxs6(Box6, { children: [
+      /* @__PURE__ */ jsx6(Text6, { color, children: "\u2502" }),
+      /* @__PURE__ */ jsxs6(Text6, { color, bold: true, children: [
+        sp,
+        p(table.name, headerW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color, children: "\u2502" })
+    ] }),
+    /* @__PURE__ */ jsx6(Text6, { color, children: sep }),
+    table.columns.map((col, i) => /* @__PURE__ */ jsxs6(Box6, { children: [
+      /* @__PURE__ */ jsx6(Text6, { color, children: "\u2502" }),
+      /* @__PURE__ */ jsxs6(Text6, { children: [
+        sp,
+        p(col.name, nameW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color, children: "\u2502" }),
+      /* @__PURE__ */ jsxs6(Text6, { dimColor: true, children: [
+        sp,
+        p(col.type, typeW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color, children: "\u2502" }),
+      col.isPk ? /* @__PURE__ */ jsxs6(Text6, { bold: true, children: [
+        sp,
+        p("PK", keyW),
+        sp
+      ] }) : col.fkTable ? /* @__PURE__ */ jsxs6(Fragment3, { children: [
+        /* @__PURE__ */ jsxs6(Text6, { dimColor: true, children: [
+          sp,
+          FK_PREFIX
+        ] }),
+        /* @__PURE__ */ jsxs6(Text6, { color: colorMap.get(col.fkTable) ?? color, children: [
+          p(col.fkTable, keyW - FK_PREFIX.length),
+          sp
+        ] })
+      ] }) : /* @__PURE__ */ jsxs6(Text6, { children: [
+        sp,
+        " ".repeat(keyW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color, children: "\u2502" })
+    ] }, i)),
+    /* @__PURE__ */ jsx6(Text6, { color, children: bot })
+  ] });
+}
+function ErdView({ data }) {
+  if (data.tables.length === 0) {
+    return /* @__PURE__ */ jsx6(Text6, { dimColor: true, children: "No tables found in the current schema." });
+  }
+  const termW = process.stdout.columns ?? 80;
+  const metrics = data.tables.map(computeMetrics);
+  const colorMap = new Map(
+    data.tables.map((t, i) => [t.name, TABLE_COLORS[i % TABLE_COLORS.length]])
+  );
+  const rows = groupIntoRows(metrics, termW);
+  return /* @__PURE__ */ jsx6(Box6, { flexDirection: "column", marginTop: 1, children: rows.map((row, ri) => /* @__PURE__ */ jsx6(Box6, { flexDirection: "row", marginBottom: ri < rows.length - 1 ? 1 : 0, children: row.map((ti, j) => /* @__PURE__ */ jsx6(Box6, { marginRight: j < row.length - 1 ? GAP : 0, children: /* @__PURE__ */ jsx6(
+    TableBox,
+    {
+      table: data.tables[ti],
+      m: metrics[ti],
+      color: colorMap.get(data.tables[ti].name) ?? BORDER,
+      colorMap
+    }
+  ) }, ti)) }, ri)) });
+}
+
+// src/ui/components/App.tsx
+import { Fragment as Fragment4, jsx as jsx7, jsxs as jsxs7 } from "react/jsx-runtime";
 var PAGE_SIZE = 50;
 var PLACEHOLDER2 = "#a5b4fc";
 function activePageSize() {
@@ -1909,26 +2140,27 @@ function limitLines(s, n) {
 }
 function EntryView({ entry }) {
   const showAi = entry.aiResponse !== "" || entry.aiError !== null;
+  const showErd = entry.erdData !== null;
   const isShell = entry.query.startsWith("!");
   const isCommand = !isShell && (entry.query.startsWith("/") || entry.query.startsWith("\\"));
   const label = isShell ? "Shell:" : isCommand ? "Command:" : "Query:";
-  return /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", marginBottom: 1, paddingX: 1, children: [
-    /* @__PURE__ */ jsxs6(Box6, { borderStyle: "round", borderColor: theme.accent, paddingX: 1, children: [
-      /* @__PURE__ */ jsxs6(Text6, { color: theme.accent, bold: true, children: [
+  return /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", marginBottom: 1, paddingX: 1, children: [
+    /* @__PURE__ */ jsxs7(Box7, { borderStyle: "round", borderColor: theme.accent, paddingX: 1, children: [
+      /* @__PURE__ */ jsxs7(Text7, { color: theme.accent, bold: true, children: [
         label,
         " "
       ] }),
-      /* @__PURE__ */ jsx6(Text6, { dimColor: true, children: entry.query })
+      /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: entry.query })
     ] }),
-    /* @__PURE__ */ jsx6(Box6, { marginTop: 1, flexDirection: "column", children: isShell ? entry.shellOutput !== null && /* @__PURE__ */ jsx6(Text6, { children: entry.shellOutput || "(no output)" }) : /* @__PURE__ */ jsxs6(Fragment3, { children: [
-      entry.commandMessage && (entry.commandMessage.helpData ? /* @__PURE__ */ jsx6(HelpView, { data: entry.commandMessage.helpData }) : entry.commandMessage.ok ? /* @__PURE__ */ jsxs6(Text6, { color: theme.accent, children: [
+    /* @__PURE__ */ jsx7(Box7, { marginTop: 1, flexDirection: "column", children: isShell ? entry.shellOutput !== null && /* @__PURE__ */ jsx7(Text7, { children: entry.shellOutput || "(no output)" }) : /* @__PURE__ */ jsxs7(Fragment4, { children: [
+      entry.commandMessage && (entry.commandMessage.helpData ? /* @__PURE__ */ jsx7(HelpView, { data: entry.commandMessage.helpData }) : entry.commandMessage.ok ? /* @__PURE__ */ jsxs7(Text7, { color: theme.accent, children: [
         "\u2713 ",
         entry.commandMessage.text
-      ] }) : /* @__PURE__ */ jsx6(ErrorBox, { message: entry.commandMessage.text })),
-      showAi ? /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", children: [
-        /* @__PURE__ */ jsx6(Text6, { color: theme.accent, bold: true, children: "Explanation:" }),
-        /* @__PURE__ */ jsx6(Box6, { flexDirection: "column", marginTop: 1, children: entry.aiError ? /* @__PURE__ */ jsx6(ErrorBox, { message: entry.aiError }) : /* @__PURE__ */ jsx6(Text6, { color: PLACEHOLDER2, children: entry.aiResponse }) })
-      ] }) : !entry.commandMessage && /* @__PURE__ */ jsx6(QueryResult, { state: entry.queryState, elapsed: entry.elapsed, page: entry.page, pageSize: PAGE_SIZE })
+      ] }) : /* @__PURE__ */ jsx7(ErrorBox, { message: entry.commandMessage.text })),
+      showErd ? /* @__PURE__ */ jsx7(ErdView, { data: entry.erdData }) : showAi ? /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", children: [
+        /* @__PURE__ */ jsx7(Text7, { color: theme.accent, bold: true, children: "Explanation:" }),
+        /* @__PURE__ */ jsx7(Box7, { flexDirection: "column", marginTop: 1, children: entry.aiError ? /* @__PURE__ */ jsx7(ErrorBox, { message: entry.aiError }) : /* @__PURE__ */ jsx7(Text7, { color: PLACEHOLDER2, children: entry.aiResponse }) })
+      ] }) : !entry.commandMessage && /* @__PURE__ */ jsx7(QueryResult, { state: entry.queryState, elapsed: entry.elapsed, page: entry.page, pageSize: PAGE_SIZE })
     ] }) })
   ] });
 }
@@ -1952,6 +2184,8 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
   const [aiError, setAiError] = useState3(null);
   const [shellOutput, setShellOutput] = useState3(null);
   const [isShellRunning, setIsShellRunning] = useState3(false);
+  const [erdData, setErdData] = useState3(null);
+  const [isErdLoading, setIsErdLoading] = useState3(false);
   const [completedEntries, setCompletedEntries] = useState3([]);
   const entryIdRef = useRef2(0);
   const aliasScope = connectionState.status === "connected" ? makeScope(connectionState.driver, connectionState.user, connectionState.host, connectionState.database) : "";
@@ -2017,8 +2251,29 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
       setIsStreaming(false);
     }
   }
+  async function handleErd() {
+    if (connectionState.status !== "connected") {
+      setCommandMessage({ ok: false, text: "Not connected to a database." });
+      return;
+    }
+    setErdData(null);
+    setIsErdLoading(true);
+    setCommandMessage(null);
+    setQueryState({ status: "idle" });
+    setAiResponse("");
+    setAiError(null);
+    try {
+      const data = await fetchErd(connectionState.client, connectionState.driver);
+      setErdData(data);
+    } catch (err) {
+      setCommandMessage({ ok: false, text: err instanceof Error ? err.message : String(err) });
+    } finally {
+      setIsErdLoading(false);
+    }
+  }
   async function handleQuery(sql) {
     if (connectionState.status !== "connected") return;
+    setErdData(null);
     setAiResponse("");
     setAiError(null);
     setCommandMessage(null);
@@ -2076,11 +2331,13 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
         page,
         aiResponse,
         aiError,
-        shellOutput
+        shellOutput,
+        erdData
       }
     ]);
   }
   async function handleShell(cmd) {
+    setErdData(null);
     setShellOutput(null);
     setIsShellRunning(true);
     setCommandMessage(null);
@@ -2126,6 +2383,9 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
           onChangeDatabase?.(db);
         },
         onExport: handleExport,
+        onErd: () => {
+          void handleErd();
+        },
         onClear: () => {
           process.stdout.write("\x1B[2J\x1B[3J\x1B[H");
           setCompletedEntries([]);
@@ -2144,6 +2404,7 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
       });
       if (result.cleared) return;
       setLastQuery(sql);
+      setErdData(null);
       setAiResponse("");
       setAiError(null);
       setElapsed(null);
@@ -2156,39 +2417,39 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
     }
     void handleQuery(sql);
   }
-  const isLoading = queryState.status === "running" || isStreaming || isShellRunning;
+  const isLoading = queryState.status === "running" || isStreaming || isShellRunning || isErdLoading;
   const isConnected = connectionState.status === "connected";
   const showAi = aiResponse !== "" || isStreaming || aiError !== null;
   const isShellEntry = lastQuery.startsWith("!");
   const isCommand = !isShellEntry && (lastQuery.startsWith("/") || lastQuery.startsWith("\\"));
   const activeLabel = isShellEntry ? "Shell:" : isCommand ? "Command:" : "Query:";
-  return /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", children: [
-    /* @__PURE__ */ jsx6(Static, { items: completedEntries, children: (entry) => /* @__PURE__ */ jsx6(EntryView, { entry }, entry.id) }),
-    /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", paddingX: 1, children: [
-      lastQuery === "" && /* @__PURE__ */ jsx6(Banner, { connectionState }),
-      lastQuery !== "" && /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", marginBottom: 2, children: [
-        /* @__PURE__ */ jsxs6(Box6, { borderStyle: "round", borderColor: theme.accent, paddingX: 1, children: [
-          /* @__PURE__ */ jsxs6(Text6, { color: theme.accent, bold: true, children: [
+  return /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", children: [
+    /* @__PURE__ */ jsx7(Static, { items: completedEntries, children: (entry) => /* @__PURE__ */ jsx7(EntryView, { entry }, entry.id) }),
+    /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", paddingX: 1, children: [
+      lastQuery === "" && /* @__PURE__ */ jsx7(Banner, { connectionState }),
+      lastQuery !== "" && /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", marginBottom: 2, children: [
+        /* @__PURE__ */ jsxs7(Box7, { borderStyle: "round", borderColor: theme.accent, paddingX: 1, children: [
+          /* @__PURE__ */ jsxs7(Text7, { color: theme.accent, bold: true, children: [
             activeLabel,
             " "
           ] }),
-          /* @__PURE__ */ jsx6(Text6, { dimColor: true, children: lastQuery })
+          /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: lastQuery })
         ] }),
-        /* @__PURE__ */ jsx6(Box6, { marginTop: 1, flexDirection: "column", children: isShellEntry ? isShellRunning ? /* @__PURE__ */ jsx6(Text6, { dimColor: true, children: "running\u2026" }) : /* @__PURE__ */ jsx6(Text6, { children: limitLines(shellOutput ?? "", activePageSize()) }) : /* @__PURE__ */ jsxs6(Fragment3, { children: [
-          commandMessage && (commandMessage.helpData ? /* @__PURE__ */ jsx6(HelpView, { data: commandMessage.helpData }) : commandMessage.ok ? /* @__PURE__ */ jsxs6(Text6, { color: theme.accent, children: [
+        /* @__PURE__ */ jsx7(Box7, { marginTop: 1, flexDirection: "column", children: isShellEntry ? isShellRunning ? /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: "running\u2026" }) : /* @__PURE__ */ jsx7(Text7, { children: limitLines(shellOutput ?? "", activePageSize()) }) : /* @__PURE__ */ jsxs7(Fragment4, { children: [
+          commandMessage && (commandMessage.helpData ? /* @__PURE__ */ jsx7(HelpView, { data: commandMessage.helpData }) : commandMessage.ok ? /* @__PURE__ */ jsxs7(Text7, { color: theme.accent, children: [
             "\u2713 ",
             commandMessage.text
-          ] }) : /* @__PURE__ */ jsx6(ErrorBox, { message: commandMessage.text })),
-          showAi ? /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", children: [
-            /* @__PURE__ */ jsx6(Text6, { color: theme.accent, bold: true, children: "Explanation:" }),
-            /* @__PURE__ */ jsx6(Box6, { flexDirection: "column", marginTop: 1, children: aiError ? /* @__PURE__ */ jsx6(ErrorBox, { message: aiError }) : /* @__PURE__ */ jsxs6(Text6, { color: PLACEHOLDER2, children: [
+          ] }) : /* @__PURE__ */ jsx7(ErrorBox, { message: commandMessage.text })),
+          isErdLoading ? /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: "Fetching schema\u2026" }) : erdData ? /* @__PURE__ */ jsx7(ErdView, { data: erdData }) : showAi ? /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", children: [
+            /* @__PURE__ */ jsx7(Text7, { color: theme.accent, bold: true, children: "Explanation:" }),
+            /* @__PURE__ */ jsx7(Box7, { flexDirection: "column", marginTop: 1, children: aiError ? /* @__PURE__ */ jsx7(ErrorBox, { message: aiError }) : /* @__PURE__ */ jsxs7(Text7, { color: PLACEHOLDER2, children: [
               aiResponse,
-              isStreaming && /* @__PURE__ */ jsx6(Text6, { color: PLACEHOLDER2, children: "\u258B" })
+              isStreaming && /* @__PURE__ */ jsx7(Text7, { color: PLACEHOLDER2, children: "\u258B" })
             ] }) })
-          ] }) : !commandMessage && /* @__PURE__ */ jsx6(QueryResult, { state: queryState, elapsed, page, pageSize: activePageSize() })
+          ] }) : !commandMessage && /* @__PURE__ */ jsx7(QueryResult, { state: queryState, elapsed, page, pageSize: activePageSize() })
         ] }) })
       ] }),
-      isConnected ? /* @__PURE__ */ jsx6(
+      isConnected ? /* @__PURE__ */ jsx7(
         QueryInput,
         {
           onSubmit: handleSubmit,
@@ -2200,16 +2461,16 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
           aliases,
           schema: schema ?? void 0
         }
-      ) : /* @__PURE__ */ jsx6(Text6, { dimColor: true, children: "Not connected. Press Ctrl+C to exit." }),
-      (vimEnabled || inputIsShell) && /* @__PURE__ */ jsx6(Box6, { marginTop: 1, children: /* @__PURE__ */ jsx6(Text6, { bold: true, color: inputIsShell ? theme.shellMode : vimMode === "NORMAL" ? theme.normalMode : theme.insertMode, children: isRawModeSupported ? inputIsShell ? "[SHELL]" : `[${vimMode}]` : "" }) })
+      ) : /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: "Not connected. Press Ctrl+C to exit." }),
+      (vimEnabled || inputIsShell) && /* @__PURE__ */ jsx7(Box7, { marginTop: 1, children: /* @__PURE__ */ jsx7(Text7, { bold: true, color: inputIsShell ? theme.shellMode : vimMode === "NORMAL" ? theme.normalMode : theme.insertMode, children: isRawModeSupported ? inputIsShell ? "[SHELL]" : `[${vimMode}]` : "" }) })
     ] })
   ] });
 }
 
 // src/ui/components/ConnectionWizard.tsx
 import { useState as useState4 } from "react";
-import { Box as Box7, Text as Text7, useInput as useInput3, useStdin as useStdin3 } from "ink";
-import { jsx as jsx7, jsxs as jsxs7 } from "react/jsx-runtime";
+import { Box as Box8, Text as Text8, useInput as useInput3, useStdin as useStdin3 } from "ink";
+import { jsx as jsx8, jsxs as jsxs8 } from "react/jsx-runtime";
 var DRIVERS = ["postgresql", "mysql", "sqlite"];
 var LABEL_WIDTH = 10;
 function fieldLabels(driver) {
@@ -2364,36 +2625,36 @@ function ConnectionWizard({ onConnect, initialError }) {
     { isActive: isRawModeSupported }
   );
   const isPassword = (idx) => fields.driver !== "sqlite" && idx === 5;
-  return /* @__PURE__ */ jsxs7(Box7, { flexDirection: "column", paddingX: 2, paddingTop: 2, paddingBottom: 1, children: [
-    /* @__PURE__ */ jsx7(Box7, { marginBottom: 1, children: /* @__PURE__ */ jsx7(Text7, { bold: true, color: theme.accent, children: "Connect to a database" }) }),
+  return /* @__PURE__ */ jsxs8(Box8, { flexDirection: "column", paddingX: 2, paddingTop: 2, paddingBottom: 1, children: [
+    /* @__PURE__ */ jsx8(Box8, { marginBottom: 1, children: /* @__PURE__ */ jsx8(Text8, { bold: true, color: theme.accent, children: "Connect to a database" }) }),
     labels.map((label, idx) => {
       const isFocused = focus === idx;
       const isDriver = idx === 0;
-      return /* @__PURE__ */ jsxs7(Box7, { children: [
-        /* @__PURE__ */ jsx7(Text7, { color: isFocused ? theme.accent : void 0, bold: isFocused, children: label.padEnd(LABEL_WIDTH) }),
-        isDriver ? /* @__PURE__ */ jsx7(Box7, { children: DRIVERS.map((d, i) => /* @__PURE__ */ jsxs7(Box7, { children: [
-          i > 0 && /* @__PURE__ */ jsx7(Text7, { children: "   " }),
-          /* @__PURE__ */ jsxs7(Text7, { color: fields.driver === d ? theme.accent : void 0, bold: fields.driver === d, children: [
+      return /* @__PURE__ */ jsxs8(Box8, { children: [
+        /* @__PURE__ */ jsx8(Text8, { color: isFocused ? theme.accent : void 0, bold: isFocused, children: label.padEnd(LABEL_WIDTH) }),
+        isDriver ? /* @__PURE__ */ jsx8(Box8, { children: DRIVERS.map((d, i) => /* @__PURE__ */ jsxs8(Box8, { children: [
+          i > 0 && /* @__PURE__ */ jsx8(Text8, { children: "   " }),
+          /* @__PURE__ */ jsxs8(Text8, { color: fields.driver === d ? theme.accent : void 0, bold: fields.driver === d, children: [
             fields.driver === d ? "\u25CF " : "\u25CB ",
             d.charAt(0).toUpperCase() + d.slice(1)
           ] })
-        ] }, d)) }) : /* @__PURE__ */ jsxs7(Box7, { children: [
-          /* @__PURE__ */ jsx7(Text7, { children: isPassword(idx) ? "\u2022".repeat(getTextValue(idx).length) : getTextValue(idx) }),
-          isFocused && /* @__PURE__ */ jsx7(Text7, { color: theme.accent, bold: true, children: "\u258C" }),
-          isFocused && isPassword(idx) && keychainHint && /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: "  (from keychain)" })
+        ] }, d)) }) : /* @__PURE__ */ jsxs8(Box8, { children: [
+          /* @__PURE__ */ jsx8(Text8, { children: isPassword(idx) ? "\u2022".repeat(getTextValue(idx).length) : getTextValue(idx) }),
+          isFocused && /* @__PURE__ */ jsx8(Text8, { color: theme.accent, bold: true, children: "\u258C" }),
+          isFocused && isPassword(idx) && keychainHint && /* @__PURE__ */ jsx8(Text8, { dimColor: true, children: "  (from keychain)" })
         ] })
       ] }, label);
     }),
-    /* @__PURE__ */ jsx7(Box7, { marginTop: 1, children: connecting ? /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: "Connecting\u2026" }) : error ? /* @__PURE__ */ jsxs7(Text7, { color: theme.error, children: [
+    /* @__PURE__ */ jsx8(Box8, { marginTop: 1, children: connecting ? /* @__PURE__ */ jsx8(Text8, { dimColor: true, children: "Connecting\u2026" }) : error ? /* @__PURE__ */ jsxs8(Text8, { color: theme.error, children: [
       "\u2717 ",
       error
     ] }) : null }),
-    /* @__PURE__ */ jsx7(Box7, { marginTop: 1, children: /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: "Tab \xB7 \u2191\u2193 navigate   Enter connect   \u2190\u2192 cycle driver" }) })
+    /* @__PURE__ */ jsx8(Box8, { marginTop: 1, children: /* @__PURE__ */ jsx8(Text8, { dimColor: true, children: "Tab \xB7 \u2191\u2193 navigate   Enter connect   \u2190\u2192 cycle driver" }) })
   ] });
 }
 
 // src/index.tsx
-import { jsx as jsx8 } from "react/jsx-runtime";
+import { jsx as jsx9 } from "react/jsx-runtime";
 var { values } = parseArgs({
   args: process.argv.slice(2).filter((a) => a !== "--"),
   options: {
@@ -2423,10 +2684,10 @@ function Root({ initialDsn: initialDsn2, aiUrl: aiUrl2, aiModel: aiModel2, aiKey
     }
   }, []);
   if (initialDsn2 && !connectionState && !dsnError) {
-    return /* @__PURE__ */ jsx8(Box8, { paddingX: 2, paddingTop: 1, children: /* @__PURE__ */ jsx8(Text8, { dimColor: true, children: "Connecting\u2026" }) });
+    return /* @__PURE__ */ jsx9(Box9, { paddingX: 2, paddingTop: 1, children: /* @__PURE__ */ jsx9(Text9, { dimColor: true, children: "Connecting\u2026" }) });
   }
   if (!connectionState) {
-    return /* @__PURE__ */ jsx8(ConnectionWizard, { onConnect: setConnectionState, initialError: dsnError ?? void 0 });
+    return /* @__PURE__ */ jsx9(ConnectionWizard, { onConnect: setConnectionState, initialError: dsnError ?? void 0 });
   }
   async function handleChangeDatabase(database) {
     if (connectionState.status !== "connected") return;
@@ -2436,7 +2697,7 @@ function Root({ initialDsn: initialDsn2, aiUrl: aiUrl2, aiModel: aiModel2, aiKey
     const next = await connectParams({ ...current.params, database });
     setConnectionState(next);
   }
-  return /* @__PURE__ */ jsx8(
+  return /* @__PURE__ */ jsx9(
     App,
     {
       connectionState,
@@ -2450,6 +2711,6 @@ function Root({ initialDsn: initialDsn2, aiUrl: aiUrl2, aiModel: aiModel2, aiKey
   );
 }
 render(
-  /* @__PURE__ */ jsx8(Root, { initialDsn, aiUrl, aiModel, aiKey }),
+  /* @__PURE__ */ jsx9(Root, { initialDsn, aiUrl, aiModel, aiKey }),
   { exitOnCtrlC: true }
 );

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@deaquinodev/querky",
-  "version": "0.4.3",
+  "version": "0.4.5",
   "description": "A quirky terminal SQL client with vim mode, AI features, and schema-aware autocomplete",
   "main": "dist/index.js",
   "files": [