pg-admin-mcp 0.0.0-stage → 0.1.0

package/dist/tools/export.js

@@ -0,0 +1,215 @@
+import { Readable } from 'node:stream';
+import { pipeline } from 'node:stream/promises';
+import { z } from 'zod';
+import { loadProfiles } from '../config/profiles.js';
+import { planStatements } from '../core/classifier.js';
+import { toToolResult } from '../core/errors.js';
+import { formatOutcome } from '../core/format.js';
+import { columnIdentFromCatalog, resolveColumn, resolveRelation } from '../core/identifiers.js';
+import { renderHostPath, resolveWorkspacePath, writeFileAtomic } from '../core/paths.js';
+import { getColumns } from '../db/catalog.js';
+import { withTx } from '../db/session.js';
+import { assertNoObviousWrite } from './execute_sql.js';
+import { wherePredicate } from './select.js';
+import { createExportTransform } from './_export_format.js';
+import { defineTool, profileParam } from './_shared.js';
+/**
+ * `postgres_export` — streams a query result straight to a file (CSV/JSON/JSONL).
+ *
+ * **Response cap vs. file cap.** Every other read tool's *response* is bounded
+ * by `core/limits.ts#resolveLimits`/`truncateRows` — the row/byte cap that
+ * protects the MCP transport and the caller's context window. That module is
+ * deliberately never consulted here: the artifact this tool produces is a
+ * *file*, not a tool response (the response is a one-line "File written: …"
+ * confirmation), and a file the caller explicitly asked to export is not
+ * something this server should silently truncate. The only cap on the file
+ * itself is disk space and `timeoutMs`.
+ *
+ * **Streaming, not buffering.** The source is read through a server-side
+ * cursor (`DECLARE … CURSOR FOR <query>` + repeated `FETCH FORWARD n`) opened
+ * on the one client `withTx` checks out, so at most one `FETCH_BATCH_SIZE`
+ * batch of rows is ever in memory. That batch source feeds
+ * `tools/_export_format.ts#createExportTransform` and a destination file
+ * through `node:stream/promises#pipeline` — never a manual
+ * `while (true) { write(...) }` loop — so a slow disk (or a slow downstream
+ * consumer, for `postgres_copy`'s sibling design) applies real backpressure:
+ * `Readable.from`'s generator is not resumed, and so the next `FETCH` round
+ * trip is not issued, until the write side has drained.
+ *
+ * **`timeoutMs` — raises, then clamps.** Unlike `postgres_execute_sql` (which
+ * can only *tighten* the profile's `statementTimeoutMs`), an export is
+ * expected to legitimately run long (design.md risk 6), so a caller may raise
+ * it — but never past `Profile.exportTimeoutMaxMs`, the operator-set ceiling.
+ * Caveat worth knowing: PostgreSQL's `statement_timeout` bounds *each*
+ * statement, and every `FETCH` here is its own statement — so this bounds how
+ * long any single round trip (including the first, which is where the
+ * underlying query's execution actually starts) may take, not the export's
+ * total wall-clock time. A query that returns one slow-to-produce row per
+ * `FETCH` is not what this guards against; a stuck query or a stuck row is.
+ *
+ * **Two sources, one policy.** `source.kind === 'table'` is fully
+ * catalog-resolved (`resolveRelation`/`resolveColumn`, exactly like
+ * `postgres_select`) — never string-interpolated user identifiers.
+ * `source.kind === 'sql'` reuses `postgres_execute_sql`'s own
+ * `assertNoObviousWrite` + `planStatements` (single statement, params always
+ * bound) and, like that tool, ultimately relies on PostgreSQL's own
+ * `READ ONLY` transaction (SQLSTATE 25006) as the actual write fence.
+ */
+const FETCH_BATCH_SIZE = 1000;
+const CURSOR_NAME = 'pg_admin_mcp_export_cursor';
+const tableSource = z.object({
+    kind: z.literal('table'),
+    schema: z.string().min(1),
+    table: z.string().min(1),
+    columns: z.array(z.string().min(1)).min(1).optional().describe('Columns to export, in this order. Omit for every column, in ordinal order.'),
+    where: z.array(wherePredicate).optional().describe('Filter predicates, ANDed together — same shape as postgres_select.'),
+});
+const sqlSource = z.object({
+    kind: z.literal('sql'),
+    sql: z.string().min(1).describe('A single SELECT (or other row-returning) statement. Never string-interpolate values — use params.'),
+    params: z.array(z.unknown()).optional(),
+});
+const csvOptionsSchema = z.object({
+    delimiter: z.string().length(1).optional().describe('Defaults to ",".'),
+    header: z.boolean().optional().describe('Defaults to true.'),
+    nullValue: z
+        .string()
+        .optional()
+        .describe('Text written for SQL NULL, always unquoted — defaults to "". An actual empty string, and a real value whose text equals this sentinel, are both written quoted so none of the three collide.'),
+});
+const inputSchema = z.object({
+    ...profileParam,
+    source: z.discriminatedUnion('kind', [tableSource, sqlSource]).describe('A structured table+where selection, or a raw SQL source.'),
+    path: z.string().min(1).describe('Destination file path. Relative paths resolve against the profile\'s first allowed directory; absolute paths must still land inside an allowed directory.'),
+    format: z.enum(['csv', 'json', 'jsonl']),
+    overwrite: z.boolean().optional().describe('Required (true) to replace an existing file at path. Defaults to false.'),
+    timeoutMs: z
+        .number()
+        .int()
+        .min(1)
+        .optional()
+        .describe('Per-call statement timeout — may raise the profile\'s default, clamped to PG_<ALIAS>_EXPORT_TIMEOUT_MAX_MS.'),
+    csv: csvOptionsSchema.optional(),
+});
+async function prepareTableQuery(client, rel, source) {
+    const columnInfos = await getColumns(client, rel);
+    const knownColumns = columnInfos.map((c) => c.name);
+    const selectColumns = source.columns === undefined
+        ? columnInfos.map((c) => ({ name: c.name, ident: columnIdentFromCatalog(c.name) }))
+        : source.columns.map((name) => ({ name, ident: resolveColumn(source.schema, source.table, name, knownColumns) }));
+    let paramIndex = 1;
+    const whereParts = [];
+    const params = [];
+    for (const pred of source.where ?? []) {
+        const ident = resolveColumn(source.schema, source.table, pred.column, knownColumns);
+        if (pred.op === 'IS NULL' || pred.op === 'IS NOT NULL') {
+            whereParts.push(`${ident} ${pred.op}`);
+        }
+        else if (pred.op === 'IN' || pred.op === 'NOT IN') {
+            const placeholders = pred.value.map(() => `$${paramIndex++}`).join(', ');
+            whereParts.push(`${ident} ${pred.op} (${placeholders})`);
+            params.push(...pred.value);
+        }
+        else {
+            whereParts.push(`${ident} ${pred.op} $${paramIndex++}`);
+            params.push(pred.value);
+        }
+    }
+    const whereClause = whereParts.length === 0 ? '' : ` WHERE ${whereParts.join(' AND ')}`;
+    return {
+        sql: `SELECT ${selectColumns.map((c) => c.ident).join(', ')} FROM ${rel.schema}.${rel.name}${whereClause}`,
+        params,
+        columns: selectColumns.map((c) => c.name),
+    };
+}
+function prepareSqlQuery(profile, source) {
+    const params = source.params ?? [];
+    const statements = planStatements(source.sql, { allowMultiple: false, hasParams: params.length > 0 });
+    assertNoObviousWrite(statements, profile.alias);
+    return { sql: statements[0] ?? '', params };
+}
+/**
+ * Everything that needs its **own** pool checkout, done **before** `withTx`
+ * takes one — `resolveRelation` runs `withSession` internally, and a second
+ * checkout acquired while the first is still held is a deadlock, not a slow
+ * call: at `poolMax: 1` it can never be satisfied, and at any `poolMax` that
+ * many concurrent exports wedge the profile until
+ * `PG_MCP_CONNECT_TIMEOUT_MS` fires. `tools/select.ts` resolves before its own
+ * `withTx` for exactly this reason.
+ *
+ * What comes back is a closure over the already-resolved relation; only
+ * `getColumns`, which runs on the caller's client, stays inside the
+ * transaction.
+ */
+async function prepareSource(profile, source) {
+    if (source.kind === 'sql') {
+        const prepared = prepareSqlQuery(profile, source);
+        return () => Promise.resolve(prepared);
+    }
+    const rel = await resolveRelation(profile, source.schema, source.table);
+    return (client) => prepareTableQuery(client, rel, source);
+}
+export const exportTool = defineTool('postgres_export', {
+    title: 'Export a query result to a file',
+    description: 'Streams a table selection or a SELECT statement to a CSV/JSON/JSONL file on disk. Never buffers the full ' +
+        'result in memory (a server-side cursor feeds the file in batches), and the file itself is never truncated by ' +
+        'the row/byte caps that bound an ordinary tool response. Refuses to overwrite an existing file unless ' +
+        'overwrite:true. path resolves against the profile\'s allowed directories (see postgres_connection_status).',
+    inputSchema,
+    // Same annotation as `postgres_copy`, deliberately: both tools write a file
+    // and both replace an existing one under `overwrite: true`, so both are
+    // destructive and neither is idempotent (a second identical call either
+    // refuses, or replaces a file whose contents may differ because the table
+    // moved underneath it).
+    annotations: { readOnlyHint: false, destructiveHint: true, idempotentHint: false, openWorldHint: true },
+}, async (args) => {
+    try {
+        const registry = loadProfiles();
+        const profile = registry.get(args.profile);
+        const resolvedPath = await resolveWorkspacePath(args.path, profile.allowedDirs);
+        const timeoutMs = Math.min(args.timeoutMs ?? profile.statementTimeoutMs, profile.exportTimeoutMaxMs);
+        const csvOpts = {
+            delimiter: args.csv?.delimiter ?? ',',
+            header: args.csv?.header ?? true,
+            nullValue: args.csv?.nullValue ?? '',
+        };
+        const format = args.format;
+        const prepare = await prepareSource(profile, args.source);
+        const bytes = await withTx(profile, async (client) => {
+            const prepared = await prepare(client);
+            // `queryMode: 'extended'`, not a bare params array: `pg` only selects
+            // the extended protocol when `values.length > 0`, and a raw-SQL source
+            // with no params would otherwise DECLARE in simple-query mode. See
+            // `tools/execute_sql.ts`'s header. `planStatements` already refused a
+            // `;`-list; this is the structural half of the same fence.
+            await client.query({
+                text: `DECLARE ${CURSOR_NAME} NO SCROLL CURSOR FOR ${prepared.sql}`,
+                values: prepared.params,
+                queryMode: 'extended',
+            });
+            // The first FETCH is also where a raw-SQL source's column list comes
+            // from (its `RowDescription` is populated even when zero rows come
+            // back) — a table source already knows its columns from the catalog.
+            const first = await client.query(`FETCH FORWARD ${FETCH_BATCH_SIZE} FROM ${CURSOR_NAME}`);
+            const columns = prepared.columns ?? first.fields.map((f) => f.name);
+            async function* batches() {
+                let batch = first.rows;
+                for (;;) {
+                    if (batch.length > 0)
+                        yield batch;
+                    if (batch.length < FETCH_BATCH_SIZE)
+                        return;
+                    const next = await client.query(`FETCH FORWARD ${FETCH_BATCH_SIZE} FROM ${CURSOR_NAME}`);
+                    batch = next.rows;
+                }
+            }
+            return writeFileAtomic(resolvedPath, (dest) => pipeline(Readable.from(batches()), createExportTransform(format, columns, csvOpts), dest), { overwrite: args.overwrite === true });
+        }, { readOnly: true, operation: 'export', statementTimeoutMs: timeoutMs });
+        const hostPath = renderHostPath(resolvedPath, profile.allowedDirs);
+        return formatOutcome({ kind: 'file', path: resolvedPath, bytes, ...(hostPath === undefined ? {} : { hostPath }) });
+    }
+    catch (err) {
+        return toToolResult(err);
+    }
+});
+//# sourceMappingURL=export.js.map

package/dist/tools/export.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"export.js","sourceRoot":"","sources":["../../src/tools/export.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAgB,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,aAAa,EAAE,eAAe,EAAoB,MAAM,wBAAwB,CAAC;AAClH,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACzF,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,qBAAqB,EAAsC,MAAM,qBAAqB,CAAC;AAChG,OAAO,EAAE,UAAU,EAAE,YAAY,EAAY,MAAM,cAAc,CAAC;AAElE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AAEH,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAC9B,MAAM,WAAW,GAAG,4BAA4B,CAAC;AAEjD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IACxB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACzB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACxB,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4EAA4E,CAAC;IAC5I,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oEAAoE,CAAC;CACzH,CAAC,CAAC;AAEH,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC;IACzB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;IACtB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,mGAAmG,CAAC;IACpI,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CACxC,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;IACvE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IAC5D,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,8LAA8L,CAAC;CAC5M,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,GAAG,YAAY;IACf,MAAM,EAAE,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,0DAA0D,CAAC;IACnI,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,2JAA2J,CAAC;IAC7L,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACxC,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yEAAyE,CAAC;IACrH,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,EAAE;SACV,QAAQ,CAAC,6GAA6G,CAAC;IAC1H,GAAG,EAAE,gBAAgB,CAAC,QAAQ,EAAE;CACjC,CAAC,CAAC;AAWH,KAAK,UAAU,iBAAiB,CAAC,MAAkB,EAAE,GAAgB,EAAE,MAA0C;IAC/G,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAClD,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAEpD,MAAM,aAAa,GACjB,MAAM,CAAC,OAAO,KAAK,SAAS;QAC1B,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnF,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC;IAEtH,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACpF,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS,IAAI,IAAI,CAAC,EAAE,KAAK,aAAa,EAAE,CAAC;YACvD,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,UAAU,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzE,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,EAAE,KAAK,YAAY,GAAG,CAAC,CAAC;YACzD,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,EAAE,KAAK,UAAU,EAAE,EAAE,CAAC,CAAC;YACxD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;IAExF,OAAO;QACL,GAAG,EAAE,UAAU,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,IAAI,GAAG,WAAW,EAAE;QAC1G,MAAM;QACN,OAAO,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;KAC1C,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAgB,EAAE,MAAwC;IACjF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;IACnC,MAAM,UAAU,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,CAAC;IACtG,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAChD,OAAO,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC;AAC9C,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,KAAK,UAAU,aAAa,CAAC,OAAgB,EAAE,MAAc;IAC3D,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAClD,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IACxE,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,iBAAiB,CAAC,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,UAAU,CAClC,iBAAiB,EACjB;IACE,KAAK,EAAE,iCAAiC;IACxC,WAAW,EACT,2GAA2G;QAC3G,+GAA+G;QAC/G,uGAAuG;QACvG,4GAA4G;IAC9G,WAAW;IACX,4EAA4E;IAC5E,wEAAwE;IACxE,wEAAwE;IACxE,0EAA0E;IAC1E,wBAAwB;IACxB,WAAW,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE;CACxG,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;IACb,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,YAAY,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE3C,MAAM,YAAY,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAChF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACrG,MAAM,OAAO,GAAe;YAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,SAAS,IAAI,GAAG;YACrC,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,MAAM,IAAI,IAAI;YAChC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,SAAS,IAAI,EAAE;SACrC,CAAC;QACF,MAAM,MAAM,GAAiB,IAAI,CAAC,MAAM,CAAC;QACzC,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAE1D,MAAM,KAAK,GAAG,MAAM,MAAM,CACxB,OAAO,EACP,KAAK,EAAE,MAAM,EAAE,EAAE;YACf,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;YAEvC,sEAAsE;YACtE,uEAAuE;YACvE,mEAAmE;YACnE,sEAAsE;YACtE,2DAA2D;YAC3D,MAAM,MAAM,CAAC,KAAK,CAAC;gBACjB,IAAI,EAAE,WAAW,WAAW,yBAAyB,QAAQ,CAAC,GAAG,EAAE;gBACnE,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,SAAS,EAAE,UAAU;aACtB,CAAC,CAAC;YAEH,qEAAqE;YACrE,mEAAmE;YACnE,qEAAqE;YACrE,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAM,iBAAiB,gBAAgB,SAAS,WAAW,EAAE,CAAC,CAAC;YAC/F,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAEpE,KAAK,SAAS,CAAC,CAAC,OAAO;gBACrB,IAAI,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC;gBACvB,SAAS,CAAC;oBACR,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;wBAAE,MAAM,KAAK,CAAC;oBAClC,IAAI,KAAK,CAAC,MAAM,GAAG,gBAAgB;wBAAE,OAAO;oBAC5C,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,KAAK,CAAM,iBAAiB,gBAAgB,SAAS,WAAW,EAAE,CAAC,CAAC;oBAC9F,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC;gBACpB,CAAC;YACH,CAAC;YAED,OAAO,eAAe,CACpB,YAAY,EACZ,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,EACnG,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,KAAK,IAAI,EAAE,CACvC,CAAC;QACJ,CAAC,EACD,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,kBAAkB,EAAE,SAAS,EAAE,CACvE,CAAC;QAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACnE,OAAO,aAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;IACrH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;AACH,CAAC,CACF,CAAC"}

package/dist/tools/get_schema.js

@@ -0,0 +1,148 @@
+import { z } from 'zod';
+import { loadProfiles } from '../config/profiles.js';
+import { toToolResult } from '../core/errors.js';
+import { formatOutcome } from '../core/format.js';
+import { relationRefFromCatalog, resolveSchemaScope } from '../core/identifiers.js';
+import { resolveLimits, truncateRows } from '../core/limits.js';
+import { withSession } from '../db/session.js';
+import { gatherObjects, OBJECT_KINDS, RELATION_KINDS } from './_object_kinds.js';
+import { buildRelationOverview, buildRelationOverviewBatch } from './_relation_detail.js';
+import { defineTool, profileParam } from './_shared.js';
+/**
+ * `postgres_get_schema` — a structured snapshot of a database (or a filtered
+ * scope), composed from the same `db/catalog.ts` listings `list_objects`
+ * uses. Two detail levels, not a pagination cursor: this tool answers "what
+ * does this database look like", so the size control that matters is "start
+ * cheap, then narrow or drill down" rather than "walk every page."
+ *
+ *   - `summary` (default): name + kind only, one flat sorted list across every
+ *     schema in scope. Cheap enough to run first against an unknown, possibly
+ *     huge database.
+ *   - `full`: adds columns, primary key, outbound foreign keys, and indexes
+ *     for every table-like object (`RELATION_KINDS`) — the "full-columns"
+ *     level. Sequences/functions/enums already carry their compact
+ *     parameters/labels at the `details` field regardless of level, since
+ *     those come for free from the listing query with no extra round trip.
+ *
+ * Output size is the real risk here (a 500-table schema at `full` detail
+ * would blow context budget), so this routes through `core/limits.ts` twice:
+ * once as a row-count pre-cap *before* the (relatively expensive, one
+ * catalog round trip per relation) `full`-detail enrichment step, so an
+ * oversized request never pays to enrich objects it will just discard, and
+ * again as a byte-cap pass (`truncateRows`) over the already-enriched set.
+ * Either cap produces the same explicit, actionable truncation message:
+ * what was kept, what was cut, and how to narrow the scope.
+ */
+const RELATION_KIND_SET = new Set(RELATION_KINDS);
+const inputSchema = z.object({
+    ...profileParam,
+    schemas: z
+        .array(z.string().min(1))
+        .optional()
+        .describe('Restrict to these schemas. Omit for every schema in scope (Profile.schemas, or every non-system schema).'),
+    kinds: z
+        .array(z.enum(OBJECT_KINDS))
+        .optional()
+        .describe(`Restrict to these object kinds (${OBJECT_KINDS.join(', ')}). Omit for every kind.`),
+    namePattern: z
+        .string()
+        .max(200)
+        .optional()
+        .describe('SQL LIKE pattern (% and _ wildcards, \\ escapes a literal % or _), matched case-insensitively against the object name via ILIKE.'),
+    detail: z
+        .enum(['summary', 'full'])
+        .optional()
+        .describe('summary (default): name and kind only — cheap, start here on an unknown or large database. ' +
+        'full: adds columns, primary key, outbound foreign keys, and indexes for tables/views/materialized views.'),
+    maxObjects: z
+        .number()
+        .int()
+        .positive()
+        .optional()
+        .describe('Caps the number of objects included before truncation. Defaults to the profile row limit.'),
+});
+/** `summary` detail: names and kinds only, per the tool's own stated minimum. */
+export function toSummaryEntry(o) {
+    return { schema: o.schema, kind: o.kind, name: o.name };
+}
+/** `full` detail baseline (pre-enrichment): everything `summary` has plus owner/comment/kind-specific details. */
+export function toFullEntry(o) {
+    return { schema: o.schema, kind: o.kind, name: o.name, owner: o.owner, comment: o.comment, details: o.details };
+}
+/** An explicit, actionable truncation message: what was kept, what was cut, how to narrow. */
+export function truncationHint(kept, total, cutAt) {
+    const at = cutAt === undefined ? 'the cap' : `${cutAt.schema}.${cutAt.name} (${cutAt.kind})`;
+    return (`showing ${kept} of ${total} objects; truncated at ${at} — narrow the scope with schemas/kinds/namePattern, ` +
+        'or keep detail: "summary" for a cheaper listing of everything');
+}
+export const getSchemaTool = defineTool('postgres_get_schema', {
+    title: 'Get a PostgreSQL schema snapshot',
+    description: 'Structured snapshot of a database, or a filtered scope (schemas, kinds, name pattern), composed from the ' +
+        'same catalog listings postgres_list_objects uses. detail:"summary" (default) returns names and kinds only — ' +
+        'cheap enough for a large database. detail:"full" adds columns, primary key, outbound foreign keys, and ' +
+        'indexes for tables/views/materialized views. An oversized result is truncated with an explicit count and a ' +
+        'hint for narrowing scope, never silently cut. The "enum" kind lists enum types only — domain and composite ' +
+        'types are not covered by any kind here.',
+    inputSchema,
+    annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: true },
+}, async (args) => {
+    try {
+        const registry = loadProfiles();
+        const profile = registry.get(args.profile);
+        const detail = args.detail ?? 'summary';
+        // `gatherObjects` already returns a sorted, name-filtered (ILIKE, bound
+        // as `$N` — see `_object_kinds.ts`) set; no further JS-side filtering or
+        // re-sorting needed here.
+        const filtered = await withSession(profile, async (client) => {
+            const schemaNames = await resolveSchemaScope(client, profile, args.schemas);
+            return gatherObjects(client, schemaNames, args.kinds, args.namePattern);
+        }, { operation: 'get_schema' });
+        const limits = resolveLimits(profile.maxRows, args.maxObjects === undefined ? {} : { maxRows: args.maxObjects }, process.env);
+        // Pre-cap by object COUNT before the (per-relation catalog round trip)
+        // full-detail enrichment step, so a request that would enrich
+        // thousands of relations never pays for the objects it will just
+        // discard — this bounds real DB work, not only output size.
+        const preCapped = filtered.slice(0, limits.maxRows);
+        // Q4: one batched `oid = ANY($1)` pass (4 catalog queries total, via
+        // `buildRelationOverviewBatch`) for every table-like object in
+        // `preCapped`, instead of `buildRelationOverview` once per relation —
+        // the previous per-relation shape cost ~4 round trips per relation
+        // (166 queries measured for 40 objects; ~2000 at a default 500-row
+        // cap), unbounded by anything except `maxRows`.
+        const enrichedEntries = detail === 'summary'
+            ? preCapped.map((o) => ({ ...toSummaryEntry(o) }))
+            : await withSession(profile, async (client) => {
+                const baseline = preCapped.map(toFullEntry);
+                const rels = preCapped
+                    .filter((o) => o.oid !== null && RELATION_KIND_SET.has(o.kind))
+                    .map((o) => relationRefFromCatalog(o.oid, o.schema, o.name, o.kind));
+                const overviews = await buildRelationOverviewBatch(client, rels);
+                return baseline.map((entry, i) => {
+                    const source = preCapped[i];
+                    const overview = source?.oid === null || source?.oid === undefined ? undefined : overviews.get(source.oid);
+                    return overview === undefined ? entry : { ...entry, overview };
+                });
+            }, { operation: 'get_schema' });
+        const byBytes = truncateRows(enrichedEntries, limits, (entry) => JSON.stringify(entry));
+        const kept = byBytes.rows;
+        const totalCutByCount = filtered.length - preCapped.length;
+        const totalCutByBytes = byBytes.truncation.truncatedRows;
+        const truncatedCount = totalCutByCount + totalCutByBytes;
+        const cutAt = truncatedCount > 0 ? filtered[kept.length] : undefined;
+        const data = {
+            profile: profile.alias,
+            detail,
+            schemaCount: new Set(filtered.map((o) => o.schema)).size,
+            objectCount: kept.length,
+            totalMatched: filtered.length,
+            objects: kept,
+            truncated: truncatedCount > 0,
+            ...(truncatedCount > 0 ? { truncationMessage: truncationHint(kept.length, filtered.length, cutAt) } : {}),
+        };
+        return formatOutcome({ kind: 'report', data });
+    }
+    catch (err) {
+        return toToolResult(err);
+    }
+});
+//# sourceMappingURL=get_schema.js.map

package/dist/tools/get_schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get_schema.js","sourceRoot":"","sources":["../../src/tools/get_schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AACpF,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAsB,MAAM,oBAAoB,CAAC;AACrG,OAAO,EAAE,qBAAqB,EAAE,0BAA0B,EAAE,MAAM,uBAAuB,CAAC;AAC1F,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAExD;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAEH,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAS,cAAc,CAAC,CAAC;AAE1D,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,GAAG,YAAY;IACf,OAAO,EAAE,CAAC;SACP,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;SACxB,QAAQ,EAAE;SACV,QAAQ,CAAC,0GAA0G,CAAC;IACvH,KAAK,EAAE,CAAC;SACL,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;SAC3B,QAAQ,EAAE;SACV,QAAQ,CAAC,mCAAmC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,yBAAyB,CAAC;IAChG,WAAW,EAAE,CAAC;SACX,MAAM,EAAE;SACR,GAAG,CAAC,GAAG,CAAC;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,kIAAkI,CAAC;IAC/I,MAAM,EAAE,CAAC;SACN,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;SACzB,QAAQ,EAAE;SACV,QAAQ,CACP,6FAA6F;QAC3F,0GAA0G,CAC7G;IACH,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,GAAG,EAAE;SACL,QAAQ,EAAE;SACV,QAAQ,EAAE;SACV,QAAQ,CAAC,2FAA2F,CAAC;CACzG,CAAC,CAAC;AAgBH,iFAAiF;AACjF,MAAM,UAAU,cAAc,CAAC,CAAgB;IAC7C,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AAC1D,CAAC;AAED,kHAAkH;AAClH,MAAM,UAAU,WAAW,CAAC,CAAgB;IAC1C,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;AAClH,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,cAAc,CAAC,IAAY,EAAE,KAAa,EAAE,KAAgC;IAC1F,MAAM,EAAE,GAAG,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,GAAG,CAAC;IAC7F,OAAO,CACL,WAAW,IAAI,OAAO,KAAK,0BAA0B,EAAE,sDAAsD;QAC7G,+DAA+D,CAChE,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CACrC,qBAAqB,EACrB;IACE,KAAK,EAAE,kCAAkC;IACzC,WAAW,EACT,2GAA2G;QAC3G,8GAA8G;QAC9G,yGAAyG;QACzG,6GAA6G;QAC7G,6GAA6G;QAC7G,yCAAyC;IAC3C,WAAW;IACX,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE;CACvG,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;IACb,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,YAAY,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAW,IAAI,CAAC,MAAM,IAAI,SAAS,CAAC;QAEhD,wEAAwE;QACxE,yEAAyE;QACzE,0BAA0B;QAC1B,MAAM,QAAQ,GAAG,MAAM,WAAW,CAChC,OAAO,EACP,KAAK,EAAE,MAAM,EAAE,EAAE;YACf,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5E,OAAO,aAAa,CAAC,MAAM,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1E,CAAC,EACD,EAAE,SAAS,EAAE,YAAY,EAAE,CAC5B,CAAC;QAEF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;QAE9H,uEAAuE;QACvE,8DAA8D;QAC9D,iEAAiE;QACjE,4DAA4D;QAC5D,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAEpD,qEAAqE;QACrE,+DAA+D;QAC/D,sEAAsE;QACtE,mEAAmE;QACnE,mEAAmE;QACnE,gDAAgD;QAChD,MAAM,eAAe,GACnB,MAAM,KAAK,SAAS;YAClB,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAClD,CAAC,CAAC,MAAM,WAAW,CACf,OAAO,EACP,KAAK,EAAE,MAAM,EAAE,EAAE;gBACf,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC5C,MAAM,IAAI,GAAG,SAAS;qBACnB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;qBAC9D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,CAAC,GAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAuC,CAAC,CAAC,CAAC;gBACpH,MAAM,SAAS,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBACjE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;oBAC/B,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;oBAC5B,MAAM,QAAQ,GAAG,MAAM,EAAE,GAAG,KAAK,IAAI,IAAI,MAAM,EAAE,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBAC3G,OAAO,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,CAAC;gBACjE,CAAC,CAAC,CAAC;YACL,CAAC,EACD,EAAE,SAAS,EAAE,YAAY,EAAE,CAC5B,CAAC;QAER,MAAM,OAAO,GAAG,YAAY,CAAC,eAAe,EAAE,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QACxF,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAE1B,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;QAC3D,MAAM,eAAe,GAAG,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;QACzD,MAAM,cAAc,GAAG,eAAe,GAAG,eAAe,CAAC;QACzD,MAAM,KAAK,GAAG,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAErE,MAAM,IAAI,GAAG;YACX,OAAO,EAAE,OAAO,CAAC,KAAK;YACtB,MAAM;YACN,WAAW,EAAE,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;YACxD,WAAW,EAAE,IAAI,CAAC,MAAM;YACxB,YAAY,EAAE,QAAQ,CAAC,MAAM;YAC7B,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,cAAc,GAAG,CAAC;YAC7B,GAAG,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1G,CAAC;QAEF,OAAO,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;AACH,CAAC,CACF,CAAC"}

package/dist/tools/index.js

@@ -0,0 +1,40 @@
+import { adminTool } from './admin/index.js';
+import { backupTool } from './backup.js';
+import { connectionStatusTool } from './connection_status.js';
+import { copyTool } from './copy.js';
+import { deleteBatchTool } from './delete_batch.js';
+import { describeObjectTool } from './describe_object.js';
+import { executeSqlTool } from './execute_sql.js';
+import { explainTool } from './explain.js';
+import { exportTool } from './export.js';
+import { getSchemaTool } from './get_schema.js';
+import { insertBatchTool } from './insert_batch.js';
+import { listObjectsTool } from './list_objects.js';
+import { prepareMigrationTool } from './prepare_migration.js';
+import { restoreTool } from './restore.js';
+import { schemaDiffTool } from './schema_diff.js';
+import { schemaSnapshotTool } from './schema_snapshot.js';
+import { selectTool } from './select.js';
+import { updateBatchTool } from './update_batch.js';
+/** The tool registry. Each phase appends its `defineTool(...)` entries here. */
+export const tools = [
+    connectionStatusTool,
+    listObjectsTool,
+    describeObjectTool,
+    getSchemaTool,
+    selectTool,
+    explainTool,
+    executeSqlTool,
+    exportTool,
+    copyTool,
+    insertBatchTool,
+    updateBatchTool,
+    deleteBatchTool,
+    schemaSnapshotTool,
+    schemaDiffTool,
+    prepareMigrationTool,
+    adminTool,
+    backupTool,
+    restoreTool,
+];
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGpD,gFAAgF;AAChF,MAAM,CAAC,MAAM,KAAK,GAAc;IAC9B,oBAAoB;IACpB,eAAe;IACf,kBAAkB;IAClB,aAAa;IACb,UAAU;IACV,WAAW;IACX,cAAc;IACd,UAAU;IACV,QAAQ;IACR,eAAe;IACf,eAAe;IACf,eAAe;IACf,kBAAkB;IAClB,cAAc;IACd,oBAAoB;IACpB,SAAS;IACT,UAAU;IACV,WAAW;CACZ,CAAC"}