@hoststack.dev/mcp 0.7.0 → 0.9.0

package/dist/hoststack-mcp.js

@@ -680,6 +680,99 @@ defineTool({
     return respond({ summary: `Database ${args2.database_id} (${type}) is ${status}.`, data });
   }
 });
+defineTool({
+  name: "query_database",
+  category: "databases",
+  description: [
+    "Run a single READ-ONLY SQL statement against a managed Postgres database.",
+    "",
+    "When to use: ad-hoc data inspection \u2014 checking row counts, sampling rows, debugging a constraint. For schema changes or seed data, ship a Drizzle migration instead.",
+    "",
+    "Safety model (server-side, not bypassable):",
+    '  - Wrapped in `BEGIN TRANSACTION READ ONLY` \u2014 INSERT/UPDATE/DELETE/DDL fail with a clear "cannot execute \u2026 in a read-only transaction" error.',
+    "  - 30s statement_timeout \u2014 runaway queries are killed.",
+    "  - 1000-row cap; result CSV is also size-capped at 1 MiB. Both surface via `truncated: true`.",
+    "  - Postgres engine only (v1). Other engines return a 400.",
+    "  - Single statement only; no embedded `;` and no psql meta-commands (`\\\u2026`).",
+    '  - Every call is audit-logged (success or failure) as `database.query` so an admin can answer "who queried this DB, when, with what SQL".',
+    "",
+    "Inputs:",
+    `  - database_id: publicId of the postgres database (e.g. "db_\u2026"). Resolved via the SDK's publicId helper.`,
+    "  - sql: a single SELECT/WITH/SHOW/EXPLAIN statement, no trailing `;`. Max 16 KiB.",
+    "",
+    "Returns: { columns: string[], rows: string[][], rowCount: number, truncated: boolean, durationMs: number }. Values come back as strings (psql --csv); cast on the caller side when needed.",
+    "",
+    "Example:",
+    '  - query_database({ database_id: "db_abc", sql: "SELECT count(*) FROM users" }) \u2192 { columns: ["count"], rows: [["1234"]], \u2026 }',
+    `  - query_database({ database_id: "db_abc", sql: "SELECT id, email FROM users WHERE created_at > now() - interval '1 day' LIMIT 50" })`
+  ].join("\n"),
+  input: {
+    database_id: z5.string().describe("Database publicId (e.g. db_abc)."),
+    sql: z5.string().min(1).max(16384).describe(
+      "A single READ-ONLY SQL statement. No trailing `;`, no embedded `;`, no psql meta-commands."
+    )
+  },
+  handler: async (args2, ctx) => {
+    const teamId = await ctx.resolveTeamId();
+    const result = await ctx.hoststack.databases.query(teamId, args2.database_id, args2.sql);
+    const summary = result.truncated ? `Query returned ${result.rowCount} row${result.rowCount === 1 ? "" : "s"} (truncated; raise LIMIT if needed) in ${result.durationMs}ms.` : `Query returned ${result.rowCount} row${result.rowCount === 1 ? "" : "s"} in ${result.durationMs}ms.`;
+    return respond({ summary, data: result });
+  }
+});
+defineTool({
+  name: "upgrade_database_to_ha",
+  category: "databases",
+  description: [
+    "Trigger a single-node \u2192 HA migration on a managed Postgres database. The agent runs pg_dump \u2192 bootstrap 3-node Patroni cluster (1 leader + 2 sync replicas + HAProxy) \u2192 pg_restore. Customer connections to this database are briefly read-only during the cutover; the standalone container stays running (read-only) for 24 h as a rollback target before automatic decommission.",
+    "",
+    "Requires: PATRONI_ENABLED env on the HostStack deployment + `teams.ha_beta = true` on the calling team. Returns 403 otherwise. Returns 400 if the database is already HA, not Postgres, not on the standard or pro tier, or has a pg_database_size over 1 GB (in-memory archive transfer cap).",
+    "",
+    "When to use: a customer needs failover protection on an existing standalone Postgres. For new databases on an HA-eligible team, just call create_database \u2014 the cluster path is automatic.",
+    "",
+    "Inputs:",
+    "  - database_id: publicId of the postgres database to upgrade.",
+    "",
+    'Returns: 202 Accepted. The upgrade is async \u2014 poll get_database until `pg_engine_type === "patroni"` and `status === "available"` to confirm completion. The Cluster tab on the dashboard mirrors the same data.',
+    "",
+    'Example: upgrade_database_to_ha({ database_id: "db_abc" }) \u2192 { ok: true }'
+  ].join("\n"),
+  input: {
+    database_id: z5.string().describe("Database publicId (e.g. db_abc) to upgrade to HA.")
+  },
+  handler: async (args2, ctx) => {
+    const teamId = await ctx.resolveTeamId();
+    await ctx.hoststack.databases.upgradeToHa(teamId, args2.database_id);
+    return respond({
+      summary: `HA migration started for ${args2.database_id}. Poll get_database until pg_engine_type=patroni to confirm.`
+    });
+  }
+});
+defineTool({
+  name: "get_database_cluster",
+  category: "databases",
+  description: [
+    "Get the live + retired cluster members and failover history for a Patroni HA Postgres database.",
+    "",
+    "When to use: confirm an HA cluster is healthy (1 leader + 2 sync replicas + 1 proxy + 1 etcd live) or investigate a failover that fired. For standalone databases this returns 400 \u2014 call get_database first if you need to branch.",
+    "",
+    "Inputs:",
+    "  - database_id: publicId of the postgres database (must be HA).",
+    "",
+    "Returns: { members: { id, memberRole: 'etcd' | 'primary-candidate' | 'replica' | 'proxy', containerId, workerHostId, joinedAt, leftAt }[], failovers: { id, createdAt, oldLeader, newLeader }[] }.",
+    "",
+    'Example: get_database_cluster({ database_id: "db_abc" }) \u2192 { members: [...5 live rows], failovers: [] }'
+  ].join("\n"),
+  input: {
+    database_id: z5.string().describe("Database publicId for the HA cluster.")
+  },
+  handler: async (args2, ctx) => {
+    const teamId = await ctx.resolveTeamId();
+    const result = await ctx.hoststack.databases.getCluster(teamId, args2.database_id);
+    const live = result.members.filter((m) => m.leftAt === null);
+    const summary = `${live.length} live members (${live.map((m) => m.memberRole).join(", ")}); ${result.failovers.length} historical failovers.`;
+    return respond({ summary, data: result });
+  }
+});
 
 // src/tools/deploys.ts
 import { z as z6 } from "zod";