@neverinfamous/postgres-mcp 1.0.1

package/dist/cli.js

@@ -0,0 +1,341 @@
+#!/usr/bin/env node
+/**
+ * postgres-mcp - CLI Entry Point
+ *
+ * Command-line interface for the PostgreSQL MCP server.
+ * Supports stdio, HTTP, and SSE transports with OAuth 2.0 authentication.
+ */
+import { Command } from "commander";
+import { PostgresAdapter } from "./adapters/postgresql/index.js";
+import { PostgresMcpServer } from "./server/McpServer.js";
+import { parseToolFilter, getFilterSummary } from "./filtering/ToolFilter.js";
+import { logger } from "./utils/logger.js";
+import { HttpTransport } from "./transports/http.js";
+import { OAuthResourceServer, TokenValidator, AuthorizationServerDiscovery, ALL_SCOPES, } from "./auth/index.js";
+const VERSION = "0.1.0";
+const program = new Command();
+program
+    .name("postgres-mcp")
+    .description("PostgreSQL MCP Server - Full-featured database tools for AI with OAuth 2.0")
+    .version(VERSION);
+program
+    // Connection options
+    .option("--postgres <url>", "PostgreSQL connection string (postgres://user:pass@host:port/database)")
+    .option("--host <host>", "PostgreSQL host (default: localhost)")
+    .option("--pg-port <port>", "PostgreSQL port (default: 5432)", parseInt)
+    .option("--user <user>", "PostgreSQL username")
+    .option("--password <password>", "PostgreSQL password")
+    .option("--database <database>", "PostgreSQL database name")
+    .option("--ssl", "Enable SSL connection")
+    .option("--pool-max <size>", "Maximum pool connections (default: 10)", parseInt)
+    // Server options
+    .option("--transport, -t <type>", "Transport type: stdio, http, sse (default: stdio)")
+    .option("--port, -p <port>", "HTTP port for http/sse transports (default: 3000)", parseInt)
+    .option("--tool-filter <filter>", 'Tool filter string (e.g., "-vector,-postgis")')
+    .option("--log-level <level>", "Log level: debug, info, notice, warning, error, critical, alert, emergency (default: info)")
+    // OAuth options
+    .option("--oauth-enabled, -o", "Enable OAuth 2.0 authentication")
+    .option("--oauth-issuer <url>", "Authorization server URL (issuer)")
+    .option("--oauth-audience <aud>", "Expected token audience")
+    .option("--oauth-jwks-uri <url>", "JWKS URI (auto-discovered from issuer if not set)")
+    .option("--oauth-clock-tolerance <seconds>", "Clock tolerance in seconds (default: 60)", parseInt)
+    .action(async (options) => {
+    // Set log level
+    const logLevel = options.logLevel ?? process.env["LOG_LEVEL"];
+    if (logLevel) {
+        logger.setLevel(logLevel);
+    }
+    // Build database config
+    const dbConfig = buildDatabaseConfig(options);
+    // Build OAuth config
+    const oauthConfig = await buildOAuthConfig(options);
+    // Create adapter and connect
+    const adapter = new PostgresAdapter();
+    try {
+        await adapter.connect(dbConfig);
+        // Get tool filter from option or environment
+        const toolFilter = options.toolFilter ??
+            process.env["POSTGRES_TOOL_FILTER"] ??
+            process.env["MCP_TOOL_FILTER"];
+        if (toolFilter) {
+            const filterConfig = parseToolFilter(toolFilter);
+            logger.info(getFilterSummary(filterConfig));
+        }
+        // Log OAuth status
+        if (oauthConfig?.enabled) {
+            logger.info("OAuth 2.0 authentication enabled", {
+                issuer: oauthConfig.issuer,
+            });
+        }
+        // Determine transport type
+        const transport = (options.transport ??
+            process.env["MCP_TRANSPORT"] ??
+            "stdio");
+        if (transport === "http" || transport === "sse") {
+            // Start with HTTP transport
+            await startHttpServer(adapter, toolFilter, oauthConfig, options);
+        }
+        else {
+            // Start with stdio transport (default)
+            await startStdioServer(adapter, toolFilter);
+        }
+    }
+    catch (error) {
+        logger.error("Failed to start server", {
+            error: error instanceof Error ? error.message : String(error),
+        });
+        await adapter.disconnect();
+        process.exit(1);
+    }
+});
+/**
+ * Build database configuration from CLI options and environment
+ */
+function buildDatabaseConfig(options) {
+    const config = {
+        type: "postgresql",
+    };
+    // Parse connection string or individual options
+    if (options.postgres) {
+        const url = new URL(options.postgres);
+        config.host = url.hostname;
+        config.port = parseInt(url.port, 10) || 5432;
+        config.username = url.username;
+        config.password = url.password;
+        config.database = url.pathname.slice(1); // Remove leading /
+        if (url.searchParams.get("ssl") === "true" ||
+            url.searchParams.get("sslmode") === "require") {
+            config.options = { ssl: true };
+        }
+    }
+    else {
+        config.host =
+            options.host ??
+                process.env["PGHOST"] ??
+                process.env["POSTGRES_HOST"] ??
+                "localhost";
+        config.port =
+            options.pgPort ??
+                parseInt(process.env["PGPORT"] ?? process.env["POSTGRES_PORT"] ?? "5432", 10);
+        config.username =
+            options.user ??
+                process.env["PGUSER"] ??
+                process.env["POSTGRES_USER"] ??
+                "postgres";
+        config.password =
+            options.password ??
+                process.env["PGPASSWORD"] ??
+                process.env["POSTGRES_PASSWORD"] ??
+                "";
+        config.database =
+            options.database ??
+                process.env["PGDATABASE"] ??
+                process.env["POSTGRES_DATABASE"] ??
+                "postgres";
+        if (options.ssl) {
+            config.options = { ssl: true };
+        }
+    }
+    // Pool configuration
+    if (options.poolMax !== undefined && options.poolMax > 0) {
+        config.pool = { max: options.poolMax };
+    }
+    return config;
+}
+/**
+ * Build OAuth configuration from CLI options and environment
+ */
+async function buildOAuthConfig(options) {
+    // Check if OAuth is enabled
+    const oauthEnabled = options.oauthEnabled ?? process.env["OAUTH_ENABLED"] === "true";
+    if (!oauthEnabled) {
+        return undefined;
+    }
+    const issuer = options.oauthIssuer ?? process.env["OAUTH_ISSUER"];
+    const audience = options.oauthAudience ?? process.env["OAUTH_AUDIENCE"];
+    let jwksUri = options.oauthJwksUri ?? process.env["OAUTH_JWKS_URI"];
+    const clockTolerance = options.oauthClockTolerance ??
+        (process.env["OAUTH_CLOCK_TOLERANCE"]
+            ? parseInt(process.env["OAUTH_CLOCK_TOLERANCE"], 10)
+            : 60);
+    // Auto-discover JWKS URI if not provided
+    if (!jwksUri && issuer) {
+        try {
+            const discovery = new AuthorizationServerDiscovery({
+                authServerUrl: issuer,
+            });
+            jwksUri = await discovery.getJwksUri();
+            logger.debug("JWKS URI discovered from issuer", { jwksUri });
+        }
+        catch (error) {
+            logger.warn("Failed to discover JWKS URI, OAuth may not work correctly", {
+                error: String(error),
+            });
+        }
+    }
+    // Build OAuth config (we already checked oauthEnabled at function start)
+    const oauthConfig = {
+        enabled: true,
+        clockTolerance,
+    };
+    if (issuer)
+        oauthConfig.authorizationServerUrl = issuer;
+    if (issuer)
+        oauthConfig.issuer = issuer;
+    if (audience)
+        oauthConfig.audience = audience;
+    if (jwksUri)
+        oauthConfig.jwksUri = jwksUri;
+    return oauthConfig;
+}
+/**
+ * Start the server with stdio transport
+ */
+async function startStdioServer(adapter, toolFilter) {
+    const server = new PostgresMcpServer({
+        name: "postgres-mcp",
+        version: VERSION,
+        adapter,
+        toolFilter,
+    });
+    // Handle shutdown
+    const shutdown = () => {
+        logger.info("Shutting down...");
+        void server
+            .stop()
+            .then(() => adapter.disconnect())
+            .then(() => process.exit(0));
+    };
+    process.on("SIGINT", shutdown);
+    process.on("SIGTERM", shutdown);
+    await server.start();
+}
+/**
+ * Start the server with HTTP transport
+ */
+async function startHttpServer(adapter, toolFilter, oauthConfig, options) {
+    const port = options.port ?? parseInt(process.env["PORT"] ?? "3000", 10);
+    const host = process.env["HOST"] ?? "localhost";
+    // Create OAuth components if enabled
+    let resourceServer;
+    let tokenValidator;
+    if (oauthConfig?.enabled &&
+        oauthConfig.issuer &&
+        oauthConfig.jwksUri &&
+        oauthConfig.audience) {
+        resourceServer = new OAuthResourceServer({
+            resource: `http://${host}:${String(port)}`,
+            authorizationServers: [oauthConfig.issuer],
+            scopesSupported: [...ALL_SCOPES],
+        });
+        tokenValidator = new TokenValidator({
+            jwksUri: oauthConfig.jwksUri,
+            issuer: oauthConfig.issuer,
+            audience: oauthConfig.audience,
+            clockTolerance: oauthConfig.clockTolerance,
+        });
+    }
+    // Create MCP server
+    const mcpServer = new PostgresMcpServer({
+        name: "postgres-mcp",
+        version: VERSION,
+        adapter,
+        toolFilter,
+    });
+    // Build HTTP transport config
+    const transportConfig = {
+        port,
+        host,
+        publicPaths: oauthConfig?.publicPaths ?? ["/health", "/.well-known/*"],
+    };
+    if (resourceServer)
+        transportConfig.resourceServer = resourceServer;
+    if (tokenValidator)
+        transportConfig.tokenValidator = tokenValidator;
+    // Create HTTP transport with OAuth
+    const httpTransport = new HttpTransport(transportConfig, (transport) => {
+        // Connect MCP server to the transport when client connects
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any
+        void mcpServer.getMcpServer().connect(transport);
+    });
+    // Handle shutdown
+    const shutdown = () => {
+        logger.info("Shutting down...");
+        void httpTransport
+            .stop()
+            .then(() => mcpServer.stop())
+            .then(() => adapter.disconnect())
+            .then(() => process.exit(0));
+    };
+    process.on("SIGINT", shutdown);
+    process.on("SIGTERM", shutdown);
+    // Start HTTP server
+    await httpTransport.start();
+    logger.info(`PostgreSQL MCP Server started on http://${host}:${String(port)}`);
+    if (oauthConfig?.enabled) {
+        logger.info("OAuth 2.0 protected resource metadata available at /.well-known/oauth-protected-resource");
+    }
+}
+// List tools command
+program
+    .command("list-tools")
+    .description("List all available tools")
+    .option("--filter <filter>", "Apply tool filter")
+    .option("--group <group>", "Filter by tool group")
+    // eslint-disable-next-line @typescript-eslint/require-await
+    .action(async (options) => {
+    const adapter = new PostgresAdapter();
+    const tools = adapter.getToolDefinitions();
+    const filterConfig = parseToolFilter(options.filter);
+    let filteredTools = tools;
+    if (options.group) {
+        filteredTools = tools.filter((t) => t.group === options.group);
+    }
+    filteredTools = filteredTools.filter((t) => filterConfig.enabledTools.has(t.name));
+    // Use stderr for all output - stdout is reserved for MCP protocol
+    console.error(`\nPostgreSQL MCP Tools (${String(filteredTools.length)}/${String(tools.length)}):\n`);
+    // Group by category
+    const grouped = new Map();
+    for (const tool of filteredTools) {
+        const groupTools = grouped.get(tool.group) ?? [];
+        groupTools.push(tool);
+        grouped.set(tool.group, groupTools);
+    }
+    for (const [group, groupTools] of grouped) {
+        console.error(`[${group}] (${String(groupTools.length)})`);
+        for (const tool of groupTools) {
+            const desc = tool.description.split(".")[0] ?? "";
+            console.error(`  - ${tool.name}: ${desc}`);
+        }
+        console.error("");
+    }
+});
+// Print tool count
+program
+    .command("info")
+    .description("Show server information")
+    // eslint-disable-next-line @typescript-eslint/require-await
+    .action(async () => {
+    const adapter = new PostgresAdapter();
+    const tools = adapter.getToolDefinitions();
+    const resources = adapter.getResourceDefinitions();
+    const prompts = adapter.getPromptDefinitions();
+    const groups = adapter.getSupportedToolGroups();
+    // Use stderr for all output - stdout is reserved for MCP protocol
+    console.error("\nPostgreSQL MCP Server");
+    console.error("=====================");
+    console.error(`Version: ${VERSION}`);
+    console.error(`Tools: ${String(tools.length)}`);
+    console.error(`Resources: ${String(resources.length)}`);
+    console.error(`Prompts: ${String(prompts.length)}`);
+    console.error(`Tool Groups: ${groups.join(", ")}`);
+    console.error("\nTransports: stdio (default), http, sse");
+    console.error("OAuth 2.0: Supported (RFC 9728/8414)");
+    console.error("\nCapabilities:");
+    const caps = adapter.getCapabilities();
+    for (const [cap, enabled] of Object.entries(caps)) {
+        console.error(`  ${cap}: ${enabled ? "✓" : "✗"}`);
+    }
+});
+program.parse();
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AACA;;;;;GAKG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC9E,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,aAAa,EAA4B,MAAM,sBAAsB,CAAC;AAC/E,OAAO,EACL,mBAAmB,EACnB,cAAc,EACd,4BAA4B,EAC5B,UAAU,GACX,MAAM,iBAAiB,CAAC;AAOzB,MAAM,OAAO,GAAG,OAAO,CAAC;AAmCxB,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,cAAc,CAAC;KACpB,WAAW,CACV,4EAA4E,CAC7E;KACA,OAAO,CAAC,OAAO,CAAC,CAAC;AAEpB,OAAO;IACL,qBAAqB;KACpB,MAAM,CACL,kBAAkB,EAClB,wEAAwE,CACzE;KACA,MAAM,CAAC,eAAe,EAAE,sCAAsC,CAAC;KAC/D,MAAM,CAAC,kBAAkB,EAAE,iCAAiC,EAAE,QAAQ,CAAC;KACvE,MAAM,CAAC,eAAe,EAAE,qBAAqB,CAAC;KAC9C,MAAM,CAAC,uBAAuB,EAAE,qBAAqB,CAAC;KACtD,MAAM,CAAC,uBAAuB,EAAE,0BAA0B,CAAC;KAC3D,MAAM,CAAC,OAAO,EAAE,uBAAuB,CAAC;KACxC,MAAM,CACL,mBAAmB,EACnB,wCAAwC,EACxC,QAAQ,CACT;IACD,iBAAiB;KAChB,MAAM,CACL,wBAAwB,EACxB,mDAAmD,CACpD;KACA,MAAM,CACL,mBAAmB,EACnB,mDAAmD,EACnD,QAAQ,CACT;KACA,MAAM,CACL,wBAAwB,EACxB,+CAA+C,CAChD;KACA,MAAM,CACL,qBAAqB,EACrB,4FAA4F,CAC7F;IACD,gBAAgB;KACf,MAAM,CAAC,qBAAqB,EAAE,iCAAiC,CAAC;KAChE,MAAM,CAAC,sBAAsB,EAAE,mCAAmC,CAAC;KACnE,MAAM,CAAC,wBAAwB,EAAE,yBAAyB,CAAC;KAC3D,MAAM,CACL,wBAAwB,EACxB,mDAAmD,CACpD;KACA,MAAM,CACL,mCAAmC,EACnC,0CAA0C,EAC1C,QAAQ,CACT;KACA,MAAM,CAAC,KAAK,EAAE,OAAmB,EAAE,EAAE;IACpC,gBAAgB;IAChB,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,IAAK,OAAO,CAAC,GAAG,CAAC,WAAW,CAA6B,CAAC;IAC5E,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC5B,CAAC;IAED,wBAAwB;IACxB,MAAM,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IAE9C,qBAAqB;IACrB,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEpD,6BAA6B;IAC7B,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;IAEtC,IAAI,CAAC;QACH,MAAM,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEhC,6CAA6C;QAC7C,MAAM,UAAU,GACd,OAAO,CAAC,UAAU;YAClB,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;QAEjC,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;YACjD,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC,CAAC;QAC9C,CAAC;QAED,mBAAmB;QACnB,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;gBAC9C,MAAM,EAAE,WAAW,CAAC,MAAM;aAC3B,CAAC,CAAC;QACL,CAAC;QAED,2BAA2B;QAC3B,MAAM,SAAS,GAAG,CAAC,OAAO,CAAC,SAAS;YAClC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;YAC5B,OAAO,CAAkB,CAAC;QAE5B,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;YAChD,4BAA4B;YAC5B,MAAM,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACN,uCAAuC;YACvC,MAAM,gBAAgB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;YACrC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC,CAAC;QACH,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;QAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL;;GAEG;AACH,SAAS,mBAAmB,CAAC,OAAmB;IAC9C,MAAM,MAAM,GAAmB;QAC7B,IAAI,EAAE,YAAY;KACnB,CAAC;IAEF,gDAAgD;IAChD,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC3B,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC;QAC7C,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC/B,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC/B,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB;QAE5D,IACE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,MAAM;YACtC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,SAAS,EAC7C,CAAC;YACD,MAAM,CAAC,OAAO,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,IAAI;YACT,OAAO,CAAC,IAAI;gBACZ,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;gBAC5B,WAAW,CAAC;QACd,MAAM,CAAC,IAAI;YACT,OAAO,CAAC,MAAM;gBACd,QAAQ,CACN,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,MAAM,EAC/D,EAAE,CACH,CAAC;QACJ,MAAM,CAAC,QAAQ;YACb,OAAO,CAAC,IAAI;gBACZ,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;gBAC5B,UAAU,CAAC;QACb,MAAM,CAAC,QAAQ;YACb,OAAO,CAAC,QAAQ;gBAChB,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;gBAChC,EAAE,CAAC;QACL,MAAM,CAAC,QAAQ;YACb,OAAO,CAAC,QAAQ;gBAChB,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;gBAChC,UAAU,CAAC;QAEb,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAED,qBAAqB;IACrB,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;QACzD,MAAM,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IACzC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAC7B,OAAmB;IAEnB,4BAA4B;IAC5B,MAAM,YAAY,GAChB,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,MAAM,CAAC;IAElE,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,QAAQ,GAAG,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACxE,IAAI,OAAO,GAAG,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACpE,MAAM,cAAc,GAClB,OAAO,CAAC,mBAAmB;QAC3B,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC;YACnC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,EAAE,EAAE,CAAC;YACpD,CAAC,CAAC,EAAE,CAAC,CAAC;IAEV,yCAAyC;IACzC,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,4BAA4B,CAAC;gBACjD,aAAa,EAAE,MAAM;aACtB,CAAC,CAAC;YACH,OAAO,GAAG,MAAM,SAAS,CAAC,UAAU,EAAE,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,2DAA2D,EAAE;gBACvE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;aACrB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,yEAAyE;IACzE,MAAM,WAAW,GAAgB;QAC/B,OAAO,EAAE,IAAI;QACb,cAAc;KACf,CAAC;IACF,IAAI,MAAM;QAAE,WAAW,CAAC,sBAAsB,GAAG,MAAM,CAAC;IACxD,IAAI,MAAM;QAAE,WAAW,CAAC,MAAM,GAAG,MAAM,CAAC;IACxC,IAAI,QAAQ;QAAE,WAAW,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC9C,IAAI,OAAO;QAAE,WAAW,CAAC,OAAO,GAAG,OAAO,CAAC;IAC3C,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAC7B,OAAwB,EACxB,UAAmB;IAEnB,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;QACnC,IAAI,EAAE,cAAc;QACpB,OAAO,EAAE,OAAO;QAChB,OAAO;QACP,UAAU;KACX,CAAC,CAAC;IAEH,kBAAkB;IAClB,MAAM,QAAQ,GAAG,GAAS,EAAE;QAC1B,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAChC,KAAK,MAAM;aACR,IAAI,EAAE;aACN,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;aAChC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAEhC,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAC5B,OAAwB,EACxB,UAA8B,EAC9B,WAAoC,EACpC,OAAmB;IAEnB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC;IACzE,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC;IAEhD,qCAAqC;IACrC,IAAI,cAA+C,CAAC;IACpD,IAAI,cAA0C,CAAC;IAE/C,IACE,WAAW,EAAE,OAAO;QACpB,WAAW,CAAC,MAAM;QAClB,WAAW,CAAC,OAAO;QACnB,WAAW,CAAC,QAAQ,EACpB,CAAC;QACD,cAAc,GAAG,IAAI,mBAAmB,CAAC;YACvC,QAAQ,EAAE,UAAU,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE;YAC1C,oBAAoB,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC;YAC1C,eAAe,EAAE,CAAC,GAAG,UAAU,CAAC;SACjC,CAAC,CAAC;QAEH,cAAc,GAAG,IAAI,cAAc,CAAC;YAClC,OAAO,EAAE,WAAW,CAAC,OAAO;YAC5B,MAAM,EAAE,WAAW,CAAC,MAAM;YAC1B,QAAQ,EAAE,WAAW,CAAC,QAAQ;YAC9B,cAAc,EAAE,WAAW,CAAC,cAAc;SAC3C,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB;IACpB,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC;QACtC,IAAI,EAAE,cAAc;QACpB,OAAO,EAAE,OAAO;QAChB,OAAO;QACP,UAAU;KACX,CAAC,CAAC;IAEH,8BAA8B;IAC9B,MAAM,eAAe,GAAwB;QAC3C,IAAI;QACJ,IAAI;QACJ,WAAW,EAAE,WAAW,EAAE,WAAW,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC;KACvE,CAAC;IACF,IAAI,cAAc;QAAE,eAAe,CAAC,cAAc,GAAG,cAAc,CAAC;IACpE,IAAI,cAAc;QAAE,eAAe,CAAC,cAAc,GAAG,cAAc,CAAC;IAEpE,mCAAmC;IACnC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,eAAe,EAAE,CAAC,SAAS,EAAE,EAAE;QACrE,2DAA2D;QAC3D,qGAAqG;QACrG,KAAK,SAAS,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,SAAgB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,kBAAkB;IAClB,MAAM,QAAQ,GAAG,GAAS,EAAE;QAC1B,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAChC,KAAK,aAAa;aACf,IAAI,EAAE;aACN,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;aAC5B,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;aAChC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAEhC,oBAAoB;IACpB,MAAM,aAAa,CAAC,KAAK,EAAE,CAAC;IAE5B,MAAM,CAAC,IAAI,CACT,2CAA2C,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE,CAClE,CAAC;IAEF,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;QACzB,MAAM,CAAC,IAAI,CACT,0FAA0F,CAC3F,CAAC;IACJ,CAAC;AACH,CAAC;AAED,qBAAqB;AACrB,OAAO;KACJ,OAAO,CAAC,YAAY,CAAC;KACrB,WAAW,CAAC,0BAA0B,CAAC;KACvC,MAAM,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;KAChD,MAAM,CAAC,iBAAiB,EAAE,sBAAsB,CAAC;IAClD,4DAA4D;KAC3D,MAAM,CAAC,KAAK,EAAE,OAAyB,EAAE,EAAE;IAC1C,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;IACtC,MAAM,KAAK,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAE3C,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAErD,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,CAAC,CAAC;IACjE,CAAC;IAED,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACzC,YAAY,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CACtC,CAAC;IAEF,kEAAkE;IAClE,OAAO,CAAC,KAAK,CACX,2BAA2B,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CACtF,CAAC;IAEF,oBAAoB;IACpB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAChD,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACjD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC;QAC1C,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3D,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAClD,OAAO,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACpB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,mBAAmB;AACnB,OAAO;KACJ,OAAO,CAAC,MAAM,CAAC;KACf,WAAW,CAAC,yBAAyB,CAAC;IACvC,4DAA4D;KAC3D,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;IACtC,MAAM,KAAK,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAC3C,MAAM,SAAS,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IACnD,MAAM,OAAO,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAC/C,MAAM,MAAM,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAEhD,kEAAkE;IAClE,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;IACzC,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACvC,OAAO,CAAC,KAAK,CAAC,YAAY,OAAO,EAAE,CAAC,CAAC;IACrC,OAAO,CAAC,KAAK,CAAC,UAAU,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAChD,OAAO,CAAC,KAAK,CAAC,cAAc,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACxD,OAAO,CAAC,KAAK,CAAC,YAAY,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACpD,OAAO,CAAC,KAAK,CAAC,gBAAgB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC1D,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;IACtD,OAAO,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAClD,OAAO,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;IACpD,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC"}

package/dist/codemode/api.d.ts

@@ -0,0 +1,62 @@
+/**
+ * postgres-mcp - Code Mode API
+ *
+ * Exposes all 194 PostgreSQL tools organized by their 19 groups
+ * for use within the sandboxed execution environment.
+ */
+import type { PostgresAdapter } from "../adapters/postgresql/PostgresAdapter.js";
+/**
+ * Main API class exposing all tool groups
+ */
+export declare class PgApi {
+    readonly core: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly transactions: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly jsonb: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly text: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly performance: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly admin: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly monitoring: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly backup: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly schema: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly vector: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly postgis: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly partitioning: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly stats: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly cron: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly partman: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly kcache: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly citext: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly ltree: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    readonly pgcrypto: Record<string, (...args: unknown[]) => Promise<unknown>>;
+    private readonly toolsByGroup;
+    constructor(adapter: PostgresAdapter);
+    /**
+     * Group tools by their tool group
+     */
+    private groupTools;
+    /**
+     * Get list of available groups and their method counts
+     */
+    getAvailableGroups(): Record<string, number>;
+    /**
+     * Get list of methods available in a group
+     */
+    getGroupMethods(groupName: string): string[];
+    /**
+     * Get help information listing all groups and their methods.
+     * Call pg.help() in code mode to discover available APIs.
+     *
+     * @returns Object with group names as keys and arrays of method names as values
+     */
+    help(): Record<string, string[]>;
+    /**
+     * Create a serializable API binding for the sandbox
+     * This creates references that can be called from isolated-vm
+     */
+    createSandboxBindings(): Record<string, unknown>;
+}
+/**
+ * Create a PgApi instance for an adapter
+ */
+export declare function createPgApi(adapter: PostgresAdapter): PgApi;
+//# sourceMappingURL=api.d.ts.map

package/dist/codemode/api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../src/codemode/api.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,2CAA2C,CAAC;AA0zBjF;;GAEG;AACH,qBAAa,KAAK;IAChB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACxE,QAAQ,CAAC,YAAY,EAAE,MAAM,CAC3B,MAAM,EACN,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CACzC,CAAC;IACF,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACzE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACxE,QAAQ,CAAC,WAAW,EAAE,MAAM,CAC1B,MAAM,EACN,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CACzC,CAAC;IACF,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACzE,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9E,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3E,QAAQ,CAAC,YAAY,EAAE,MAAM,CAC3B,MAAM,EACN,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CACzC,CAAC;IACF,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACzE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACxE,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3E,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACzE,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IAE5E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAgC;gBAEjD,OAAO,EAAE,eAAe;IAuGpC;;OAEG;IACH,OAAO,CAAC,UAAU;IAgBlB;;OAEG;IACH,kBAAkB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAQ5C;;OAEG;IACH,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,EAAE;IAQ5C;;;;;OAKG;IACH,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAUhC;;;OAGG;IACH,qBAAqB,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CAwqBjD;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,eAAe,GAAG,KAAK,CAE3D"}