@yawlabs/postgres-mcp 0.3.0 → 0.3.2

package/dist/index.js

@@ -35361,6 +35361,30 @@ async function runReadWrite(sql, params = []) {
     client.release();
   }
 }
+async function runReadWriteRollback(sql, params = []) {
+  if (!isWritesAllowed()) {
+    return {
+      ok: false,
+      error: "Write blocked: ALLOW_WRITES is not set. Set ALLOW_WRITES=1 in the MCP server env to enable DML/DDL."
+    };
+  }
+  const client = await getPool().connect();
+  const maxRows = getMaxRows();
+  try {
+    await client.query("BEGIN");
+    const result = await client.query(sql, params);
+    await client.query("ROLLBACK");
+    return { ok: true, data: toQueryResult(result, maxRows) };
+  } catch (err) {
+    try {
+      await client.query("ROLLBACK");
+    } catch {
+    }
+    return { ok: false, error: formatPgError(err) };
+  } finally {
+    client.release();
+  }
+}
 async function runInternal(sql, params = []) {
   try {
     const result = await getPool().query(sql, params);
@@ -35439,8 +35463,10 @@ var adminTools = [
     }),
     handler: async (input) => {
       const { includeSystem } = input;
-      const filter = includeSystem ? "" : "WHERE r.rolname NOT LIKE 'pg\\_%' ESCAPE '\\\\'";
+      const filter = includeSystem ? "" : "WHERE NOT starts_with(r.rolname, 'pg_')";
       return runInternal(
+        // Cast member_of to text[] so node-pg parses it into a JS array.
+        // Without the cast, it comes back as the postgres text form `{a,b}`.
         `SELECT
            r.rolname AS name,
            r.rolcanlogin AS can_login,
@@ -35450,11 +35476,11 @@ var adminTools = [
            r.rolreplication AS replication,
            r.rolbypassrls AS bypass_rls,
            COALESCE(
-             (SELECT array_agg(g.rolname ORDER BY g.rolname)
+             (SELECT array_agg(g.rolname::text ORDER BY g.rolname)
               FROM pg_catalog.pg_auth_members m
               JOIN pg_catalog.pg_roles g ON g.oid = m.roleid
               WHERE m.member = r.oid),
-             ARRAY[]::name[]
+             ARRAY[]::text[]
            ) AS member_of
          FROM pg_catalog.pg_roles r
          ${filter}
@@ -35635,14 +35661,16 @@ var adminTools = [
   }
 ];
 
-// src/tools/explain.ts
+// src/tools/params.ts
 var paramValue = external_exports3.lazy(
   () => external_exports3.union([external_exports3.string(), external_exports3.number(), external_exports3.boolean(), external_exports3.null(), external_exports3.array(paramValue), external_exports3.record(external_exports3.string(), paramValue)])
 );
+
+// src/tools/explain.ts
 var explainTools = [
   {
     name: "pg_explain",
-    description: "Get the query plan for a SQL statement. By default, this uses plain EXPLAIN (no execution). Set `analyze: true` to run the query with EXPLAIN ANALYZE \u2014 for non-SELECT statements, ALLOW_WRITES=1 is required (since ANALYZE actually executes the statement). Format is `text` (default) or `json`. Pass the raw SQL (not an EXPLAIN-prefixed statement).",
+    description: "Get the query plan for a SQL statement. By default, this uses plain EXPLAIN (no execution). Set `analyze: true` to run the query with EXPLAIN ANALYZE \u2014 for non-SELECT statements, ALLOW_WRITES=1 is required (since ANALYZE actually executes the statement). Writes executed during EXPLAIN ANALYZE are always rolled back, so you can inspect a plan for an INSERT/UPDATE/DELETE without persisting the mutation. Format is `text` (default) or `json`. Pass the raw SQL (not an EXPLAIN-prefixed statement).",
     annotations: {
       title: "Explain query plan",
       readOnlyHint: false,
@@ -35668,7 +35696,7 @@ var explainTools = [
       if (analyze) flags.push("ANALYZE");
       if (format === "json") flags.push("FORMAT JSON");
       const explainSql = flags.length > 0 ? `EXPLAIN (${flags.join(", ")}) ${sql}` : `EXPLAIN ${sql}`;
-      const result = analyze && isWritesAllowed() ? await runReadWrite(explainSql, params ?? []) : await runReadOnly(explainSql, params ?? []);
+      const result = analyze && isWritesAllowed() ? await runReadWriteRollback(explainSql, params ?? []) : await runReadOnly(explainSql, params ?? []);
       if (!result.ok) return result;
       const data = result.data;
       const rows = data?.rows ?? [];
@@ -35738,7 +35766,8 @@ var healthTools = [
            JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
            WHERE c.relkind IN ('r', 'p')
              AND n.nspname NOT IN ('pg_catalog', 'information_schema')
-             AND n.nspname NOT LIKE 'pg_toast%'`
+             AND n.nspname NOT LIKE 'pg_toast%'
+             AND n.nspname NOT LIKE 'pg_temp_%'`
         )
       ]);
       if (!versionRes.ok) return versionRes;
@@ -35758,9 +35787,6 @@ var healthTools = [
 ];
 
 // src/tools/query.ts
-var paramValue2 = external_exports3.lazy(
-  () => external_exports3.union([external_exports3.string(), external_exports3.number(), external_exports3.boolean(), external_exports3.null(), external_exports3.array(paramValue2), external_exports3.record(external_exports3.string(), paramValue2)])
-);
 var queryTools = [
   {
     name: "pg_query",
@@ -35775,7 +35801,7 @@ var queryTools = [
     },
     inputSchema: external_exports3.object({
       sql: external_exports3.string().min(1).max(1e6).describe("The SQL statement to execute. Hard cap of 1 MB."),
-      params: external_exports3.array(paramValue2).optional().describe("Positional parameters referenced as $1, $2, ... in the SQL.")
+      params: external_exports3.array(paramValue).optional().describe("Positional parameters referenced as $1, $2, ... in the SQL.")
     }),
     handler: async (input) => {
       const { sql, params } = input;
@@ -35902,10 +35928,10 @@ var schemaTools = [
       const foreignKeysQuery = `
         SELECT
           con.conname AS constraint_name,
-          array_agg(att.attname ORDER BY u.attposition) AS columns,
+          array_agg(att.attname::text ORDER BY u.attposition) AS columns,
           cl.relname AS foreign_table,
           fn.nspname AS foreign_schema,
-          array_agg(fatt.attname ORDER BY u.attposition) AS foreign_columns
+          array_agg(fatt.attname::text ORDER BY u.attposition) AS foreign_columns
         FROM pg_catalog.pg_constraint con
         JOIN pg_catalog.pg_class c ON c.oid = con.conrelid
         JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
@@ -36190,9 +36216,8 @@ var statsTools = [
            n_live_tup::text AS live_tuples,
            seq_tup_read::text AS seq_tup_read,
            CASE
-             WHEN COALESCE(idx_scan, 0) = 0 AND seq_scan > 0 THEN NULL
-             WHEN COALESCE(idx_scan, 0) = 0 THEN 0
-             ELSE (seq_scan::numeric / NULLIF(idx_scan, 0))::numeric(10, 2)::float8
+             WHEN COALESCE(idx_scan, 0) = 0 THEN NULL
+             ELSE (seq_scan::numeric / idx_scan)::numeric(10, 2)::float8
            END AS ratio
          FROM pg_catalog.pg_stat_user_tables
          WHERE n_live_tup >= $1
@@ -36258,7 +36283,7 @@ function compareVersions(a, b) {
 }
 
 // src/index.ts
-var version2 = true ? "0.3.0" : (await null).createRequire(import.meta.url)("../package.json").version;
+var version2 = true ? "0.3.2" : (await null).createRequire(import.meta.url)("../package.json").version;
 var subcommand = process.argv[2];
 if (subcommand === "version" || subcommand === "--version") {
   console.log(version2);
@@ -36320,4 +36345,7 @@ process.on("SIGINT", () => {
 process.on("SIGTERM", () => {
   void cleanup().finally(() => process.exit(0));
 });
+process.stdin.on("end", () => {
+  void cleanup().finally(() => process.exit(0));
+});
 //# sourceMappingURL=index.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yawlabs/postgres-mcp",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "PostgreSQL MCP server — query, schema introspection, explain, and health checks for AI assistants",
   "license": "MIT",
   "author": "YawLabs <contact@yaw.sh>",
@@ -31,9 +31,9 @@
     "build": "tsc && node build.mjs",
     "dev": "tsc --watch",
     "start": "node dist/index.js",
-    "test": "npm run build && node --test dist/**/*.test.js",
+    "test": "npm run build && node scripts/run-tests.mjs dist",
     "test:ci": "npm run test",
-    "test:integration": "npm run build && node --test dist/**/*.integration.test.js",
+    "test:integration": "npm run build && node scripts/run-tests.mjs dist --integration",
     "lint": "biome check src/",
     "lint:fix": "biome check --write src/",
     "prepublishOnly": "npm run build"