@deaquinodev/querky 0.4.2 → 0.4.4

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 Box7, Text as Text7 } from "ink";
+import { render, Box as Box9, Text as Text9 } from "ink";
 
 // src/db/client.ts
 import { Client } from "pg";
@@ -196,7 +196,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 Box5, Text as Text5, 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) {
@@ -466,6 +466,9 @@ var PSQL_ALIASES = {
   du: "users",
   c: "changeDatabase"
 };
+var PSQL_REVERSE = Object.fromEntries(
+  Object.entries(PSQL_ALIASES).filter(([alias, name]) => alias !== name).map(([alias, name]) => [name, `\\${alias}`])
+);
 function describeBasicSql(driver, table) {
   const t = table.replace(/'/g, "''");
   if (driver === "postgresql") {
@@ -488,7 +491,10 @@ function describeFullSql(driver, table) {
 }
 var COMMANDS = {
   "clear": {
-    description: "Clear the scrollback history",
+    description: "Clear the terminal scrollback",
+    category: "Session",
+    usage: "/clear",
+    detail: "Erases the visible screen and scrollback buffer. The next query starts fresh from the top.",
     run: (ctx) => {
       ctx.onClear();
       return { ok: true, message: "", cleared: true };
@@ -496,6 +502,9 @@ var COMMANDS = {
   },
   "toggle-vim-mode": {
     description: "Toggle vim keybindings on/off",
+    category: "Session",
+    usage: "/toggle-vim-mode",
+    detail: "Switches the query input between standard editing and vim keybindings. In NORMAL mode use motions like w, b, 0, $, x, dd. Enter INSERT mode with i, a, A, I, o, O.",
     run: (ctx) => {
       const next = !ctx.vimEnabled;
       ctx.setVimEnabled(next);
@@ -503,7 +512,11 @@ var COMMANDS = {
     }
   },
   "d": {
-    description: "List tables, or describe a table: \\d [table]",
+    description: "List tables, or describe a table",
+    category: "Schema",
+    usage: "/d [table]",
+    detail: "Without an argument, lists all tables in the current database. With a table name, shows columns, types, nullability, and defaults.",
+    example: "/d users",
     run: (ctx) => {
       if (!ctx.args.trim()) {
         ctx.onQuery(DB_QUERIES[ctx.driver].tables);
@@ -514,7 +527,11 @@ var COMMANDS = {
     }
   },
   "describe": {
-    description: "Describe a table with constraints: /describe <table>",
+    description: "Describe a table with PK/FK/UQ constraints",
+    category: "Schema",
+    usage: "/describe <table>",
+    detail: "Like /d but includes constraint information \u2014 PK (primary key), FK (foreign key), UQ (unique) \u2014 in an extra key column. SQLite shows PK only.",
+    example: "/describe orders",
     run: (ctx) => {
       const table = ctx.args.trim();
       if (!table) return { ok: false, message: "Usage: /describe <table>" };
@@ -524,6 +541,9 @@ var COMMANDS = {
   },
   "databases": {
     description: "List available databases",
+    category: "Schema",
+    usage: "/databases",
+    detail: "Runs a driver-appropriate query to list all databases on the server.",
     run: (ctx) => {
       ctx.onQuery(DB_QUERIES[ctx.driver].databases);
       return { ok: true, message: "" };
@@ -531,6 +551,9 @@ var COMMANDS = {
   },
   "tables": {
     description: "List tables in the current database",
+    category: "Schema",
+    usage: "/tables",
+    detail: "Lists tables in the current database's public schema.",
     run: (ctx) => {
       ctx.onQuery(DB_QUERIES[ctx.driver].tables);
       return { ok: true, message: "" };
@@ -538,13 +561,20 @@ var COMMANDS = {
   },
   "users": {
     description: "List database users",
+    category: "Schema",
+    usage: "/users",
+    detail: "Lists users and their roles. On SQLite, shows a placeholder message since SQLite has no user system.",
     run: (ctx) => {
       ctx.onQuery(DB_QUERIES[ctx.driver].users);
       return { ok: true, message: "" };
     }
   },
   "changeDatabase": {
-    description: "Switch to a different database: \\c dbname",
+    description: "Switch to a different database",
+    category: "Session",
+    usage: "/changeDatabase <dbname>",
+    detail: "Switches the active connection to the specified database. Without an argument, shows the current database.",
+    example: "/changeDatabase myapp_prod",
     run: (ctx) => {
       if (!ctx.args) return { ok: true, message: `Connected to database: ${ctx.currentDatabase}` };
       ctx.onChangeDatabase(ctx.args.trim());
@@ -552,7 +582,11 @@ var COMMANDS = {
     }
   },
   "export": {
-    description: "Export last result to a file: /export csv or /export json",
+    description: "Export last result to a file",
+    category: "Data",
+    usage: "/export csv|json",
+    detail: "Writes the last query result to a timestamped file in your home directory. CSV files include a header row; JSON files are an array of row objects.",
+    example: "/export csv",
     run: (ctx) => {
       const fmt = ctx.args.trim().toLowerCase();
       if (fmt !== "csv" && fmt !== "json") {
@@ -563,7 +597,11 @@ var COMMANDS = {
     }
   },
   "explain": {
-    description: "Explain a SQL query using AI: /explain SELECT ...",
+    description: "Explain a SQL query using AI",
+    category: "AI",
+    usage: "/explain <SQL>",
+    detail: "Sends the query to your configured AI endpoint (Ollama by default, or any OpenAI-compatible API) and streams a plain-language explanation.",
+    example: "/explain SELECT * FROM orders WHERE status = 'pending'",
     run: (ctx) => {
       if (!ctx.args) return { ok: false, message: "Usage: /explain <SQL query>" };
       ctx.onExplain(ctx.args);
@@ -572,6 +610,9 @@ var COMMANDS = {
   },
   "explain-previous": {
     description: "Explain the last executed query using AI",
+    category: "AI",
+    usage: "/explain-previous",
+    detail: "Like /explain but uses the last SQL query you ran. Useful for quickly understanding a query after seeing its results.",
     run: (ctx) => {
       if (!ctx.lastSqlQuery) {
         return { ok: false, message: "No query to explain \u2014 run a SQL query first." };
@@ -581,7 +622,11 @@ var COMMANDS = {
     }
   },
   "save": {
-    description: "Save last query as an alias: /save <name>",
+    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",
     run: (ctx) => {
       const name = ctx.args.trim();
       if (!name) return { ok: false, message: "Usage: /save <name>" };
@@ -592,7 +637,11 @@ var COMMANDS = {
     }
   },
   "alias": {
-    description: "Define an alias inline: /alias <name> <SQL>",
+    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",
     run: (ctx) => {
       const [name, ...rest] = ctx.args.trim().split(/\s+/);
       if (!name || rest.length === 0) return { ok: false, message: "Usage: /alias <name> <SQL>" };
@@ -603,6 +652,9 @@ var COMMANDS = {
   },
   "aliases": {
     description: "List all saved aliases for this database",
+    category: "Aliases",
+    usage: "/aliases",
+    detail: "Prints all saved aliases for the current database connection, with their SQL template.",
     run: (ctx) => {
       const entries = Object.entries(ctx.aliases);
       if (entries.length === 0) return { ok: true, message: "No aliases saved. Use /save <name> or /alias <name> <SQL>." };
@@ -614,13 +666,62 @@ var COMMANDS = {
     }
   },
   "unalias": {
-    description: "Remove a saved alias: /unalias <name>",
+    description: "Remove a saved alias",
+    category: "Aliases",
+    usage: "/unalias <name>",
+    detail: "Removes a previously saved alias. Only affects aliases for the current database connection.",
+    example: "/unalias active-orders",
     run: (ctx) => {
       const name = ctx.args.trim();
       if (!name) return { ok: false, message: "Usage: /unalias <name>" };
       const removed = ctx.onDeleteAlias(name);
       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",
+    usage: "/help [command]",
+    detail: "Without arguments, lists all commands grouped by category. With a command name, shows detailed usage and an example.",
+    example: "/help explain",
+    run: (ctx) => {
+      const arg = ctx.args.trim();
+      if (arg) {
+        const name = PSQL_ALIASES[arg] ?? arg;
+        const cmd = COMMANDS[name];
+        if (!cmd) return { ok: false, message: `Unknown command: /${arg}` };
+        const entry = {
+          name,
+          usage: cmd.usage,
+          description: cmd.description,
+          psqlAlias: PSQL_REVERSE[name],
+          detail: cmd.detail,
+          example: cmd.example
+        };
+        return { ok: true, message: "", helpData: { mode: "detail", entry } };
+      }
+      const categoryOrder = ["AI", "Schema", "Data", "Aliases", "Session"];
+      const groups = categoryOrder.map((cat) => ({
+        category: cat,
+        entries: Object.entries(COMMANDS).filter(([, cmd]) => cmd.category === cat).map(([name, cmd]) => ({
+          name,
+          usage: cmd.usage,
+          description: cmd.description,
+          psqlAlias: PSQL_REVERSE[name]
+        }))
+      })).filter((g) => g.entries.length > 0);
+      return { ok: true, message: "", helpData: { mode: "list", groups } };
+    }
   }
 };
 var BUILTIN_COMMAND_LIST = Object.entries(COMMANDS).map(([name, cmd]) => ({
@@ -654,6 +755,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";
@@ -1760,8 +1938,181 @@ function Banner({ connectionState }) {
   ] });
 }
 
+// src/ui/components/HelpView.tsx
+import { Box as Box5, Text as Text5 } from "ink";
+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;
+    return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginTop: 1, children: [
+      /* @__PURE__ */ jsxs5(Text5, { bold: true, color: "white", children: [
+        usage,
+        psqlAlias ? `  ${psqlAlias}` : ""
+      ] }),
+      /* @__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: [
+        /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: "Example: " }),
+        /* @__PURE__ */ jsx5(Text5, { color: theme.accent, children: example })
+      ] })
+    ] });
+  }
+  if (data.mode === "list" && data.groups) {
+    return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginTop: 1, children: [
+      data.groups.map((group, gi) => /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginBottom: gi < data.groups.length - 1 ? 1 : 0, children: [
+        /* @__PURE__ */ jsx5(Text5, { bold: true, color: theme.accent, children: group.category }),
+        group.entries.map((entry) => /* @__PURE__ */ jsxs5(Box5, { marginLeft: 2, children: [
+          /* @__PURE__ */ jsx5(Text5, { color: "white", children: entry.usage.padEnd(USAGE_WIDTH) }),
+          /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: entry.description }),
+          entry.psqlAlias && /* @__PURE__ */ jsxs5(Text5, { dimColor: true, children: [
+            "  ",
+            entry.psqlAlias
+          ] })
+        ] }, entry.name))
+      ] }, group.category)),
+      /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsxs5(Text5, { dimColor: true, children: [
+        "/help ",
+        "<command>",
+        " for more details."
+      ] }) })
+    ] });
+  }
+  return null;
+}
+
+// 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 BORDER2 = "white";
+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: BORDER2, children: top }),
+    /* @__PURE__ */ jsxs6(Box6, { children: [
+      /* @__PURE__ */ jsx6(Text6, { color: BORDER2, children: "\u2502" }),
+      /* @__PURE__ */ jsxs6(Text6, { color, bold: true, children: [
+        sp,
+        p(table.name, headerW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color: BORDER2, children: "\u2502" })
+    ] }),
+    /* @__PURE__ */ jsx6(Text6, { color: BORDER2, children: sep }),
+    table.columns.map((col, i) => /* @__PURE__ */ jsxs6(Box6, { children: [
+      /* @__PURE__ */ jsx6(Text6, { color: BORDER2, children: "\u2502" }),
+      /* @__PURE__ */ jsxs6(Text6, { children: [
+        sp,
+        p(col.name, nameW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color: BORDER2, children: "\u2502" }),
+      /* @__PURE__ */ jsxs6(Text6, { dimColor: true, children: [
+        sp,
+        p(col.type, typeW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color: BORDER2, 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) ?? BORDER2, children: [
+          p(col.fkTable, keyW - FK_PREFIX.length),
+          sp
+        ] })
+      ] }) : /* @__PURE__ */ jsxs6(Text6, { children: [
+        sp,
+        " ".repeat(keyW),
+        sp
+      ] }),
+      /* @__PURE__ */ jsx6(Text6, { color: BORDER2, children: "\u2502" })
+    ] }, i)),
+    /* @__PURE__ */ jsx6(Text6, { color: BORDER2, 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) ?? BORDER2,
+      colorMap
+    }
+  ) }, ti)) }, ri)) });
+}
+
 // src/ui/components/App.tsx
-import { Fragment as Fragment3, jsx as jsx5, jsxs as jsxs5 } from "react/jsx-runtime";
+import { Fragment as Fragment4, jsx as jsx7, jsxs as jsxs7 } from "react/jsx-runtime";
 var PAGE_SIZE = 50;
 var PLACEHOLDER2 = "#a5b4fc";
 function activePageSize() {
@@ -1775,26 +2126,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__ */ jsxs5(Box5, { flexDirection: "column", marginBottom: 1, paddingX: 1, children: [
-    /* @__PURE__ */ jsxs5(Box5, { borderStyle: "round", borderColor: theme.accent, paddingX: 1, children: [
-      /* @__PURE__ */ jsxs5(Text5, { 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__ */ jsx5(Text5, { dimColor: true, children: entry.query })
+      /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: entry.query })
     ] }),
-    /* @__PURE__ */ jsx5(Box5, { marginTop: 1, flexDirection: "column", children: isShell ? entry.shellOutput !== null && /* @__PURE__ */ jsx5(Text5, { children: entry.shellOutput || "(no output)" }) : /* @__PURE__ */ jsxs5(Fragment3, { children: [
-      entry.commandMessage && (entry.commandMessage.ok ? /* @__PURE__ */ jsxs5(Text5, { 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__ */ jsx5(ErrorBox, { message: entry.commandMessage.text })),
-      showAi ? /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", children: [
-        /* @__PURE__ */ jsx5(Text5, { color: theme.accent, bold: true, children: "Explanation:" }),
-        /* @__PURE__ */ jsx5(Box5, { flexDirection: "column", marginTop: 1, children: entry.aiError ? /* @__PURE__ */ jsx5(ErrorBox, { message: entry.aiError }) : /* @__PURE__ */ jsx5(Text5, { color: PLACEHOLDER2, children: entry.aiResponse }) })
-      ] }) : !entry.commandMessage && /* @__PURE__ */ jsx5(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 })
     ] }) })
   ] });
 }
@@ -1818,6 +2170,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) : "";
@@ -1883,8 +2237,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);
@@ -1942,11 +2317,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);
@@ -1992,6 +2369,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([]);
@@ -2010,50 +2390,52 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
       });
       if (result.cleared) return;
       setLastQuery(sql);
+      setErdData(null);
       setAiResponse("");
       setAiError(null);
       setElapsed(null);
       setPage(0);
       setQueryState({ status: "idle" });
-      if (result.message) setCommandMessage({ ok: result.ok, text: result.message });
+      if (result.helpData) setCommandMessage({ ok: result.ok, text: "", helpData: result.helpData });
+      else if (result.message) setCommandMessage({ ok: result.ok, text: result.message });
       else setCommandMessage(null);
       return;
     }
     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__ */ jsxs5(Box5, { flexDirection: "column", children: [
-    /* @__PURE__ */ jsx5(Static, { items: completedEntries, children: (entry) => /* @__PURE__ */ jsx5(EntryView, { entry }, entry.id) }),
-    /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", paddingX: 1, children: [
-      lastQuery === "" && /* @__PURE__ */ jsx5(Banner, { connectionState }),
-      lastQuery !== "" && /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginBottom: 2, children: [
-        /* @__PURE__ */ jsxs5(Box5, { borderStyle: "round", borderColor: theme.accent, paddingX: 1, children: [
-          /* @__PURE__ */ jsxs5(Text5, { 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__ */ jsx5(Text5, { dimColor: true, children: lastQuery })
+          /* @__PURE__ */ jsx7(Text7, { dimColor: true, children: lastQuery })
         ] }),
-        /* @__PURE__ */ jsx5(Box5, { marginTop: 1, flexDirection: "column", children: isShellEntry ? isShellRunning ? /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: "running\u2026" }) : /* @__PURE__ */ jsx5(Text5, { children: limitLines(shellOutput ?? "", activePageSize()) }) : /* @__PURE__ */ jsxs5(Fragment3, { children: [
-          commandMessage && (commandMessage.ok ? /* @__PURE__ */ jsxs5(Text5, { 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__ */ jsx5(ErrorBox, { message: commandMessage.text })),
-          showAi ? /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", children: [
-            /* @__PURE__ */ jsx5(Text5, { color: theme.accent, bold: true, children: "Explanation:" }),
-            /* @__PURE__ */ jsx5(Box5, { flexDirection: "column", marginTop: 1, children: aiError ? /* @__PURE__ */ jsx5(ErrorBox, { message: aiError }) : /* @__PURE__ */ jsxs5(Text5, { 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__ */ jsx5(Text5, { color: PLACEHOLDER2, children: "\u258B" })
+              isStreaming && /* @__PURE__ */ jsx7(Text7, { color: PLACEHOLDER2, children: "\u258B" })
             ] }) })
-          ] }) : !commandMessage && /* @__PURE__ */ jsx5(QueryResult, { state: queryState, elapsed, page, pageSize: activePageSize() })
+          ] }) : !commandMessage && /* @__PURE__ */ jsx7(QueryResult, { state: queryState, elapsed, page, pageSize: activePageSize() })
         ] }) })
       ] }),
-      isConnected ? /* @__PURE__ */ jsx5(
+      isConnected ? /* @__PURE__ */ jsx7(
         QueryInput,
         {
           onSubmit: handleSubmit,
@@ -2065,16 +2447,16 @@ function App({ connectionState, aiUrl: aiUrl2, aiModel: aiModel2, aiKey: aiKey2,
           aliases,
           schema: schema ?? void 0
         }
-      ) : /* @__PURE__ */ jsx5(Text5, { dimColor: true, children: "Not connected. Press Ctrl+C to exit." }),
-      (vimEnabled || inputIsShell) && /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text5, { 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 Box6, Text as Text6, useInput as useInput3, useStdin as useStdin3 } from "ink";
-import { jsx as jsx6, jsxs as jsxs6 } 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) {
@@ -2229,36 +2611,36 @@ function ConnectionWizard({ onConnect, initialError }) {
     { isActive: isRawModeSupported }
   );
   const isPassword = (idx) => fields.driver !== "sqlite" && idx === 5;
-  return /* @__PURE__ */ jsxs6(Box6, { flexDirection: "column", paddingX: 2, paddingTop: 2, paddingBottom: 1, children: [
-    /* @__PURE__ */ jsx6(Box6, { marginBottom: 1, children: /* @__PURE__ */ jsx6(Text6, { 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__ */ jsxs6(Box6, { children: [
-        /* @__PURE__ */ jsx6(Text6, { color: isFocused ? theme.accent : void 0, bold: isFocused, children: label.padEnd(LABEL_WIDTH) }),
-        isDriver ? /* @__PURE__ */ jsx6(Box6, { children: DRIVERS.map((d, i) => /* @__PURE__ */ jsxs6(Box6, { children: [
-          i > 0 && /* @__PURE__ */ jsx6(Text6, { children: "   " }),
-          /* @__PURE__ */ jsxs6(Text6, { 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__ */ jsxs6(Box6, { children: [
-          /* @__PURE__ */ jsx6(Text6, { children: isPassword(idx) ? "\u2022".repeat(getTextValue(idx).length) : getTextValue(idx) }),
-          isFocused && /* @__PURE__ */ jsx6(Text6, { color: theme.accent, bold: true, children: "\u258C" }),
-          isFocused && isPassword(idx) && keychainHint && /* @__PURE__ */ jsx6(Text6, { 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__ */ jsx6(Box6, { marginTop: 1, children: connecting ? /* @__PURE__ */ jsx6(Text6, { dimColor: true, children: "Connecting\u2026" }) : error ? /* @__PURE__ */ jsxs6(Text6, { 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__ */ jsx6(Box6, { marginTop: 1, children: /* @__PURE__ */ jsx6(Text6, { 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 jsx7 } from "react/jsx-runtime";
+import { jsx as jsx9 } from "react/jsx-runtime";
 var { values } = parseArgs({
   args: process.argv.slice(2).filter((a) => a !== "--"),
   options: {
@@ -2288,10 +2670,10 @@ function Root({ initialDsn: initialDsn2, aiUrl: aiUrl2, aiModel: aiModel2, aiKey
     }
   }, []);
   if (initialDsn2 && !connectionState && !dsnError) {
-    return /* @__PURE__ */ jsx7(Box7, { paddingX: 2, paddingTop: 1, children: /* @__PURE__ */ jsx7(Text7, { 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__ */ jsx7(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;
@@ -2301,7 +2683,7 @@ function Root({ initialDsn: initialDsn2, aiUrl: aiUrl2, aiModel: aiModel2, aiKey
     const next = await connectParams({ ...current.params, database });
     setConnectionState(next);
   }
-  return /* @__PURE__ */ jsx7(
+  return /* @__PURE__ */ jsx9(
     App,
     {
       connectionState,
@@ -2315,6 +2697,6 @@ function Root({ initialDsn: initialDsn2, aiUrl: aiUrl2, aiModel: aiModel2, aiKey
   );
 }
 render(
-  /* @__PURE__ */ jsx7(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.2",
+  "version": "0.4.4",
   "description": "A quirky terminal SQL client with vim mode, AI features, and schema-aware autocomplete",
   "main": "dist/index.js",
   "files": [