@neverinfamous/postgres-mcp 1.0.1

package/dist/pool/ConnectionPool.js

@@ -0,0 +1,254 @@
+/**
+ * postgres-mcp - Connection Pool Manager
+ *
+ * Wraps pg connection pooling with health monitoring,
+ * statistics tracking, and graceful shutdown support.
+ */
+import pg from "pg";
+import { PoolError, ConnectionError } from "../types/index.js";
+import { logger } from "../utils/logger.js";
+/**
+ * Connection pool wrapper with statistics and health monitoring
+ */
+export class ConnectionPool {
+    pool = null;
+    config;
+    stats = {
+        total: 0,
+        active: 0,
+        idle: 0,
+        waiting: 0,
+        totalQueries: 0,
+    };
+    shuttingDown = false;
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Initialize the connection pool
+     */
+    async initialize() {
+        if (this.pool !== null) {
+            logger.warn("Connection pool already initialized");
+            return;
+        }
+        logger.info("Initializing PostgreSQL connection pool", {
+            host: this.config.host,
+            port: this.config.port,
+            database: this.config.database,
+        });
+        try {
+            const poolConfig = {
+                host: this.config.host,
+                port: this.config.port,
+                user: this.config.user,
+                password: this.config.password,
+                database: this.config.database,
+                max: this.config.pool?.max ?? 10,
+                min: this.config.pool?.min ?? 0,
+                idleTimeoutMillis: this.config.pool?.idleTimeoutMillis ?? 10000,
+                connectionTimeoutMillis: this.config.pool?.connectionTimeoutMillis ?? 10000,
+                allowExitOnIdle: this.config.pool?.allowExitOnIdle ?? true,
+                application_name: this.config.applicationName ?? "postgres-mcp",
+            };
+            if (this.config.ssl === true) {
+                poolConfig.ssl = { rejectUnauthorized: false };
+            }
+            else if (this.config.ssl !== undefined && this.config.ssl !== false) {
+                poolConfig.ssl = this.config.ssl;
+            }
+            if (this.config.statementTimeout !== undefined &&
+                this.config.statementTimeout > 0) {
+                poolConfig.statement_timeout = this.config.statementTimeout;
+            }
+            this.pool = new pg.Pool(poolConfig);
+            // Set up event handlers
+            this.pool.on("connect", () => {
+                this.stats.total++;
+                logger.debug("New connection established");
+            });
+            this.pool.on("acquire", () => {
+                this.stats.active++;
+                this.stats.idle = Math.max(0, this.stats.idle - 1);
+            });
+            this.pool.on("release", () => {
+                this.stats.active = Math.max(0, this.stats.active - 1);
+                this.stats.idle++;
+            });
+            this.pool.on("remove", () => {
+                this.stats.total = Math.max(0, this.stats.total - 1);
+                this.stats.idle = Math.max(0, this.stats.idle - 1);
+            });
+            this.pool.on("error", (err) => {
+                logger.error("Pool error", { error: err.message });
+            });
+            // Test connection
+            const client = await this.pool.connect();
+            const result = await client.query("SELECT version()");
+            client.release();
+            const version = result.rows[0];
+            logger.info("PostgreSQL connection pool initialized", {
+                version: version?.version ?? "unknown",
+            });
+        }
+        catch (error) {
+            // Clean up pool on initialization failure
+            if (this.pool !== null) {
+                try {
+                    await this.pool.end();
+                }
+                catch {
+                    // Ignore cleanup errors
+                }
+                this.pool = null;
+            }
+            const message = error instanceof Error ? error.message : "Unknown error";
+            logger.error("Failed to initialize connection pool", { error: message });
+            throw new ConnectionError(`Failed to connect to PostgreSQL: ${message}`);
+        }
+    }
+    /**
+     * Get a connection from the pool
+     */
+    async getConnection() {
+        if (this.pool === null) {
+            throw new PoolError("Connection pool not initialized");
+        }
+        if (this.shuttingDown) {
+            throw new PoolError("Connection pool is shutting down");
+        }
+        try {
+            this.stats.waiting++;
+            const client = await this.pool.connect();
+            this.stats.waiting = Math.max(0, this.stats.waiting - 1);
+            return client;
+        }
+        catch (error) {
+            this.stats.waiting = Math.max(0, this.stats.waiting - 1);
+            const message = error instanceof Error ? error.message : "Unknown error";
+            throw new PoolError(`Failed to acquire connection: ${message}`);
+        }
+    }
+    /**
+     * Release a connection back to the pool
+     */
+    releaseConnection(client) {
+        try {
+            client.release();
+        }
+        catch (error) {
+            logger.warn("Error releasing connection", {
+                error: error instanceof Error ? error.message : "Unknown error",
+            });
+        }
+    }
+    /**
+     * Execute a query using a pooled connection
+     */
+    async query(sql, params) {
+        if (this.pool === null) {
+            throw new PoolError("Connection pool not initialized");
+        }
+        const startTime = Date.now();
+        this.stats.totalQueries++;
+        try {
+            const result = await this.pool.query(sql, params);
+            logger.debug("Query executed", {
+                sql: sql.substring(0, 100),
+                rowCount: result.rowCount,
+                durationMs: Date.now() - startTime,
+            });
+            return result;
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : "Unknown error";
+            logger.error("Query failed", {
+                sql: sql.substring(0, 100),
+                error: message,
+            });
+            throw error;
+        }
+    }
+    /**
+     * Get pool statistics
+     */
+    getStats() {
+        if (this.pool !== null) {
+            // Update stats from pool if available
+            this.stats.total = this.pool.totalCount;
+            this.stats.idle = this.pool.idleCount;
+            this.stats.waiting = this.pool.waitingCount;
+            this.stats.active = this.stats.total - this.stats.idle;
+        }
+        return { ...this.stats };
+    }
+    /**
+     * Check pool health
+     */
+    async checkHealth() {
+        if (this.pool === null || this.shuttingDown) {
+            return {
+                connected: false,
+                error: this.shuttingDown
+                    ? "Pool is shutting down"
+                    : "Pool not initialized",
+            };
+        }
+        const startTime = Date.now();
+        try {
+            const result = await this.pool.query("SELECT version(), current_database()");
+            const latencyMs = Date.now() - startTime;
+            const row = result.rows[0];
+            return {
+                connected: true,
+                latencyMs,
+                version: row?.version ?? undefined,
+                poolStats: this.getStats(),
+                details: {
+                    database: row?.current_database,
+                },
+            };
+        }
+        catch (error) {
+            return {
+                connected: false,
+                error: error instanceof Error ? error.message : "Unknown error",
+                latencyMs: Date.now() - startTime,
+            };
+        }
+    }
+    /**
+     * Gracefully shutdown the pool
+     */
+    async shutdown() {
+        if (this.pool === null) {
+            return;
+        }
+        logger.info("Shutting down connection pool...");
+        this.shuttingDown = true;
+        try {
+            await this.pool.end();
+            this.pool = null;
+            logger.info("Connection pool shut down successfully");
+        }
+        catch (error) {
+            logger.error("Error during pool shutdown", {
+                error: error instanceof Error ? error.message : "Unknown error",
+            });
+            throw error;
+        }
+    }
+    /**
+     * Check if pool is initialized
+     */
+    isInitialized() {
+        return this.pool !== null && !this.shuttingDown;
+    }
+    /**
+     * Check if pool is shutting down
+     */
+    isClosing() {
+        return this.shuttingDown;
+    }
+}
+//# sourceMappingURL=ConnectionPool.js.map

package/dist/pool/ConnectionPool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConnectionPool.js","sourceRoot":"","sources":["../../src/pool/ConnectionPool.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AAGpB,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAiB5C;;GAEG;AACH,MAAM,OAAO,cAAc;IACjB,IAAI,GAAmB,IAAI,CAAC;IAC5B,MAAM,CAAuB;IAC7B,KAAK,GAAc;QACzB,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,IAAI,EAAE,CAAC;QACP,OAAO,EAAE,CAAC;QACV,YAAY,EAAE,CAAC;KAChB,CAAC;IACM,YAAY,GAAG,KAAK,CAAC;IAE7B,YAAY,MAA4B;QACtC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;YACnD,OAAO;QACT,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,yCAAyC,EAAE;YACrD,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,UAAU,GAAkB;gBAChC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,EAAE;gBAChC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC;gBAC/B,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,iBAAiB,IAAI,KAAK;gBAC/D,uBAAuB,EACrB,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,uBAAuB,IAAI,KAAK;gBACpD,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,eAAe,IAAI,IAAI;gBAC1D,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,cAAc;aAChE,CAAC;YAEF,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,KAAK,IAAI,EAAE,CAAC;gBAC7B,UAAU,CAAC,GAAG,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC;YACjD,CAAC;iBAAM,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,KAAK,KAAK,EAAE,CAAC;gBACtE,UAAU,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;YACnC,CAAC;YAED,IACE,IAAI,CAAC,MAAM,CAAC,gBAAgB,KAAK,SAAS;gBAC1C,IAAI,CAAC,MAAM,CAAC,gBAAgB,GAAG,CAAC,EAChC,CAAC;gBACD,UAAU,CAAC,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC9D,CAAC;YAED,IAAI,CAAC,IAAI,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEpC,wBAAwB;YACxB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBAC3B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACnB,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;gBACpB,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACvD,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACpB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;gBAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBACrD,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC5B,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtD,MAAM,CAAC,OAAO,EAAE,CAAC;YAEjB,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAqC,CAAC;YACnE,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE;gBACpD,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,SAAS;aACvC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,0CAA0C;YAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBACvB,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACxB,CAAC;gBAAC,MAAM,CAAC;oBACP,wBAAwB;gBAC1B,CAAC;gBACD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACnB,CAAC;YACD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACzE,MAAM,IAAI,eAAe,CAAC,oCAAoC,OAAO,EAAE,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa;QACjB,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,MAAM,IAAI,SAAS,CAAC,iCAAiC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,SAAS,CAAC,kCAAkC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;YACzD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,MAAM,IAAI,SAAS,CAAC,iCAAiC,OAAO,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,MAAkB;QAClC,IAAI,CAAC;YACH,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;gBACxC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CACT,GAAW,EACX,MAAkB;QAElB,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,MAAM,IAAI,SAAS,CAAC,iCAAiC,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;QAE1B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAY,GAAG,EAAE,MAAM,CAAC,CAAC;YAE7D,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE;gBAC7B,GAAG,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;gBAC1B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACnC,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;gBAC3B,GAAG,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;gBAC1B,KAAK,EAAE,OAAO;aACf,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,sCAAsC;YACtC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;YACxC,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;YACtC,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;QACzD,CAAC;QACD,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5C,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,IAAI,CAAC,YAAY;oBACtB,CAAC,CAAC,uBAAuB;oBACzB,CAAC,CAAC,sBAAsB;aAC3B,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,sCAAsC,CACvC,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAEZ,CAAC;YAEd,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,SAAS;gBACT,OAAO,EAAE,GAAG,EAAE,OAAO,IAAI,SAAS;gBAClC,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE;gBAC1B,OAAO,EAAE;oBACP,QAAQ,EAAE,GAAG,EAAE,gBAAgB;iBAChC;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;gBAC/D,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aAClC,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE;gBACzC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;CACF"}

package/dist/server/McpServer.d.ts

@@ -0,0 +1,50 @@
+/**
+ * postgres-mcp - MCP Server Wrapper
+ *
+ * Wraps the MCP SDK server with database adapter integration,
+ * tool filtering, logging capabilities, and graceful shutdown support.
+ */
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { DatabaseAdapter } from "../adapters/DatabaseAdapter.js";
+import type { ToolFilterConfig } from "../types/index.js";
+export interface ServerConfig {
+    name: string;
+    version: string;
+    adapter: DatabaseAdapter;
+    toolFilter?: string | undefined;
+}
+/**
+ * PostgreSQL MCP Server
+ */
+export declare class PostgresMcpServer {
+    private mcpServer;
+    private adapter;
+    private filterConfig;
+    private transport;
+    constructor(config: ServerConfig);
+    /**
+     * Register all tools, resources, and prompts
+     */
+    private registerComponents;
+    /**
+     * Start the server with stdio transport
+     */
+    start(): Promise<void>;
+    /**
+     * Gracefully stop the server
+     */
+    stop(): Promise<void>;
+    /**
+     * Get the underlying MCP server instance
+     */
+    getMcpServer(): McpServer;
+    /**
+     * Get the database adapter
+     */
+    getAdapter(): DatabaseAdapter;
+    /**
+     * Get filter configuration
+     */
+    getFilterConfig(): ToolFilterConfig;
+}
+//# sourceMappingURL=McpServer.d.ts.map

package/dist/server/McpServer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpServer.d.ts","sourceRoot":"","sources":["../../src/server/McpServer.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACtE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAK1D,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,eAAe,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACjC;AAED;;GAEG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,YAAY,CAAmB;IACvC,OAAO,CAAC,SAAS,CAAqC;gBAE1C,MAAM,EAAE,YAAY;IA+BhC;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAqB1B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAY5B;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAa3B;;OAEG;IACH,YAAY,IAAI,SAAS;IAIzB;;OAEG;IACH,UAAU,IAAI,eAAe;IAI7B;;OAEG;IACH,eAAe,IAAI,gBAAgB;CAGpC"}

package/dist/server/McpServer.js

@@ -0,0 +1,108 @@
+/**
+ * postgres-mcp - MCP Server Wrapper
+ *
+ * Wraps the MCP SDK server with database adapter integration,
+ * tool filtering, logging capabilities, and graceful shutdown support.
+ */
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { parseToolFilter } from "../filtering/ToolFilter.js";
+import { logger } from "../utils/logger.js";
+import { SERVER_INSTRUCTIONS } from "../constants/ServerInstructions.js";
+/**
+ * PostgreSQL MCP Server
+ */
+export class PostgresMcpServer {
+    mcpServer;
+    adapter;
+    filterConfig;
+    transport = null;
+    constructor(config) {
+        this.adapter = config.adapter;
+        this.filterConfig = parseToolFilter(config.toolFilter);
+        // Create MCP server with logging capability enabled and server instructions
+        this.mcpServer = new McpServer({
+            name: config.name,
+            version: config.version,
+        }, {
+            capabilities: {
+                logging: {},
+            },
+            instructions: SERVER_INSTRUCTIONS,
+        });
+        // Connect the logger to the underlying MCP server for protocol logging
+        // The McpServer.server property exposes the low-level Server instance
+        logger.setMcpServer(this.mcpServer.server);
+        logger.setLoggerName(config.name);
+        logger.info("MCP Server initialized", {
+            name: config.name,
+            version: config.version,
+            toolFilter: config.toolFilter ?? "none",
+            capabilities: ["logging"],
+        });
+    }
+    /**
+     * Register all tools, resources, and prompts
+     */
+    registerComponents() {
+        // Register tools (with filtering)
+        this.adapter.registerTools(this.mcpServer, this.filterConfig.enabledTools);
+        // Register resources
+        this.adapter.registerResources(this.mcpServer);
+        // Register prompts
+        this.adapter.registerPrompts(this.mcpServer);
+        const toolCount = this.filterConfig.enabledTools.size;
+        const resourceCount = this.adapter.getResourceDefinitions().length;
+        const promptCount = this.adapter.getPromptDefinitions().length;
+        logger.info("Components registered", {
+            tools: toolCount,
+            resources: resourceCount,
+            prompts: promptCount,
+        });
+    }
+    /**
+     * Start the server with stdio transport
+     */
+    async start() {
+        // Register all components
+        this.registerComponents();
+        // Create and connect transport
+        this.transport = new StdioServerTransport();
+        await this.mcpServer.connect(this.transport);
+        logger.info("MCP Server started with stdio transport");
+    }
+    /**
+     * Gracefully stop the server
+     */
+    async stop() {
+        logger.info("Stopping MCP Server...");
+        try {
+            await this.mcpServer.close();
+            logger.info("MCP Server stopped");
+        }
+        catch (error) {
+            logger.error("Error stopping server", {
+                error: error instanceof Error ? error.message : "Unknown error",
+            });
+        }
+    }
+    /**
+     * Get the underlying MCP server instance
+     */
+    getMcpServer() {
+        return this.mcpServer;
+    }
+    /**
+     * Get the database adapter
+     */
+    getAdapter() {
+        return this.adapter;
+    }
+    /**
+     * Get filter configuration
+     */
+    getFilterConfig() {
+        return this.filterConfig;
+    }
+}
+//# sourceMappingURL=McpServer.js.map

package/dist/server/McpServer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpServer.js","sourceRoot":"","sources":["../../src/server/McpServer.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAGjF,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AASzE;;GAEG;AACH,MAAM,OAAO,iBAAiB;IACpB,SAAS,CAAY;IACrB,OAAO,CAAkB;IACzB,YAAY,CAAmB;IAC/B,SAAS,GAAgC,IAAI,CAAC;IAEtD,YAAY,MAAoB;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAEvD,4EAA4E;QAC5E,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,CAC5B;YACE,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,EACD;YACE,YAAY,EAAE;gBACZ,OAAO,EAAE,EAAE;aACZ;YACD,YAAY,EAAE,mBAAmB;SAClC,CACF,CAAC;QAEF,uEAAuE;QACvE,sEAAsE;QACtE,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAElC,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;YACpC,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM;YACvC,YAAY,EAAE,CAAC,SAAS,CAAC;SAC1B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,kBAAkB;QACxB,kCAAkC;QAClC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;QAE3E,qBAAqB;QACrB,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAE/C,mBAAmB;QACnB,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAE7C,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,IAAI,CAAC;QACtD,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC,MAAM,CAAC;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,MAAM,CAAC;QAE/D,MAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE;YACnC,KAAK,EAAE,SAAS;YAChB,SAAS,EAAE,aAAa;YACxB,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,0BAA0B;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAE1B,+BAA+B;QAC/B,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAE5C,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAE7C,MAAM,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI;QACR,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAEtC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE;gBACpC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;CACF"}

package/dist/transports/http.d.ts

@@ -0,0 +1,126 @@
+/**
+ * postgres-mcp - HTTP Transport
+ *
+ * HTTP/SSE transport with OAuth 2.0 support.
+ */
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import type { OAuthResourceServer } from "../auth/OAuthResourceServer.js";
+import type { TokenValidator } from "../auth/TokenValidator.js";
+/**
+ * HTTP transport configuration
+ */
+export interface HttpTransportConfig {
+    /** Port to listen on */
+    port: number;
+    /** Host to bind to (default: localhost) */
+    host?: string;
+    /** OAuth resource server (optional) */
+    resourceServer?: OAuthResourceServer;
+    /** Token validator (optional, required if resourceServer is provided) */
+    tokenValidator?: TokenValidator;
+    /** CORS allowed origins (default: none) */
+    corsOrigins?: string[];
+    /** Allow credentials in CORS requests (default: false) */
+    corsAllowCredentials?: boolean;
+    /** Paths that bypass authentication */
+    publicPaths?: string[];
+    /**
+     * Enable rate limiting (default: true)
+     * Helps prevent DoS attacks and brute-force attempts
+     */
+    enableRateLimit?: boolean;
+    /**
+     * Rate limit window in milliseconds (default: 60000 = 1 minute)
+     */
+    rateLimitWindowMs?: number;
+    /**
+     * Maximum requests per window per IP (default: 100)
+     */
+    rateLimitMaxRequests?: number;
+    /**
+     * Maximum request body size in bytes (default: 1MB = 1048576)
+     * Prevents memory exhaustion from large payloads
+     */
+    maxBodySize?: number;
+    /**
+     * Enable HTTP Strict Transport Security header (default: false)
+     * Should only be enabled when running behind HTTPS
+     */
+    enableHSTS?: boolean;
+    /**
+     * HSTS max-age in seconds (default: 31536000 = 1 year)
+     */
+    hstsMaxAge?: number;
+}
+/**
+ * HTTP Transport for MCP
+ */
+export declare class HttpTransport {
+    private server;
+    private readonly config;
+    private transport;
+    private readonly onConnect?;
+    private readonly rateLimitMap;
+    private static readonly DEFAULT_RATE_LIMIT_WINDOW_MS;
+    private static readonly DEFAULT_RATE_LIMIT_MAX_REQUESTS;
+    private static readonly DEFAULT_MAX_BODY_SIZE;
+    private static readonly DEFAULT_HSTS_MAX_AGE;
+    constructor(config: HttpTransportConfig, onConnect?: (transport: StreamableHTTPServerTransport) => void);
+    /**
+     * Start the HTTP server
+     */
+    start(): Promise<void>;
+    /**
+     * Stop the HTTP server
+     */
+    stop(): Promise<void>;
+    /**
+     * Check if a path is public (bypasses authentication)
+     */
+    private isPublicPath;
+    /**
+     * Check rate limit for a request
+     * @returns true if request should be allowed, false if rate limited
+     */
+    private checkRateLimit;
+    /**
+     * Handle incoming HTTP request
+     */
+    private handleRequest;
+    /**
+     * Handle SSE connection request
+     */
+    private handleSSERequest;
+    /**
+     * Handle MCP message request
+     */
+    private handleMessageRequest;
+    /**
+     * Handle protected resource metadata endpoint
+     */
+    private handleProtectedResourceMetadata;
+    /**
+     * Handle health check endpoint
+     */
+    private handleHealthCheck;
+    /**
+     * Set security headers for all responses
+     */
+    private setSecurityHeaders;
+    /**
+     * Set CORS headers for browser-based MCP client support
+     *
+     * This implements the MCP SDK 1.25.1 recommendation of using external middleware
+     * for origin validation rather than the deprecated built-in options.
+     */
+    private setCorsHeaders;
+    /**
+     * Get the underlying transport
+     */
+    getTransport(): StreamableHTTPServerTransport | null;
+}
+/**
+ * Create an HTTP transport instance
+ */
+export declare function createHttpTransport(config: HttpTransportConfig, onConnect?: (transport: StreamableHTTPServerTransport) => void): HttpTransport;
+//# sourceMappingURL=http.d.ts.map

package/dist/transports/http.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../src/transports/http.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AAC1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAIhE;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,wBAAwB;IACxB,IAAI,EAAE,MAAM,CAAC;IAEb,2CAA2C;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,uCAAuC;IACvC,cAAc,CAAC,EAAE,mBAAmB,CAAC;IAErC,yEAAyE;IACzE,cAAc,CAAC,EAAE,cAAc,CAAC;IAEhC,2CAA2C;IAC3C,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IAEvB,0DAA0D;IAC1D,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAE/B,uCAAuC;IACvC,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IAMvB;;;OAGG;IACH,eAAe,CAAC,EAAE,OAAO,CAAC;IAE1B;;OAEG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAE3B;;OAEG;IACH,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAE9B;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;OAGG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IAErB;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAUD;;GAEG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAgD;IAC9D,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsB;IAC7C,OAAO,CAAC,SAAS,CAA8C;IAC/D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAEjB;IAGV,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAqC;IAGlE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAS;IAC7D,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,+BAA+B,CAAO;IAC9D,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAW;IACxD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAY;gBAGtD,MAAM,EAAE,mBAAmB,EAC3B,SAAS,CAAC,EAAE,CAAC,SAAS,EAAE,6BAA6B,KAAK,IAAI;IAqBhE;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAuB5B;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAa3B;;OAEG;IACH,OAAO,CAAC,YAAY;IAgBpB;;;OAGG;IACH,OAAO,CAAC,cAAc;IAuCtB;;OAEG;YACW,aAAa;IAiF3B;;OAEG;YACW,gBAAgB;IAkB9B;;OAEG;YACW,oBAAoB;IAalC;;OAEG;IACH,OAAO,CAAC,+BAA+B;IAYvC;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAUzB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IA0B1B;;;;;OAKG;IACH,OAAO,CAAC,cAAc;IA2BtB;;OAEG;IACH,YAAY,IAAI,6BAA6B,GAAG,IAAI;CAGrD;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,mBAAmB,EAC3B,SAAS,CAAC,EAAE,CAAC,SAAS,EAAE,6BAA6B,KAAK,IAAI,GAC7D,aAAa,CAEf"}