@neverinfamous/postgres-mcp 1.0.1

package/dist/__tests__/mocks/adapter.js

@@ -0,0 +1,225 @@
+/**
+ * postgres-mcp - PostgreSQL Adapter Mock
+ *
+ * Provides mock implementation of PostgresAdapter for testing
+ * tools, resources, and prompts without database dependency.
+ */
+import { vi } from "vitest";
+/**
+ * Create a mock query result
+ */
+export function createMockQueryResult(rows = [], affectedRows = 0) {
+    return {
+        rows,
+        rowsAffected: affectedRows,
+        executionTimeMs: 5,
+    };
+}
+/**
+ * Create a mock column info
+ */
+export function createMockColumnInfo(name, type, nullable = true, primaryKey = false) {
+    return {
+        name,
+        type,
+        nullable,
+        primaryKey,
+        defaultValue: null,
+    };
+}
+/**
+ * Create a mock table info
+ */
+export function createMockTableInfo(name, schemaName = "public", rowCount = 100) {
+    return {
+        name,
+        schema: schemaName,
+        type: "table",
+        columns: [
+            createMockColumnInfo("id", "integer", false, true),
+            createMockColumnInfo("name", "varchar(255)", true, false),
+            createMockColumnInfo("created_at", "timestamp with time zone", false, false),
+        ],
+        rowCount,
+    };
+}
+/**
+ * Create mock index info
+ */
+export function createMockIndexInfo(tableName, indexName, schemaName = "public") {
+    return {
+        name: indexName,
+        tableName,
+        schemaName,
+        columns: ["id"],
+        unique: indexName.endsWith("_pkey"),
+        type: "btree",
+    };
+}
+/**
+ * Create mock schema info
+ */
+export function createMockSchemaInfo() {
+    return {
+        tables: [createMockTableInfo("users"), createMockTableInfo("products")],
+        views: [],
+        indexes: [createMockIndexInfo("users", "users_pkey")],
+    };
+}
+/**
+ * Create mock health status
+ */
+export function createMockHealthStatus(connected = true) {
+    return {
+        connected,
+        latencyMs: 5,
+        version: "16.1",
+        poolStats: {
+            total: 10,
+            active: 2,
+            idle: 8,
+            waiting: 0,
+            totalQueries: 100,
+        },
+    };
+}
+/**
+ * Create a mock PostgresAdapter
+ */
+export function createMockPostgresAdapter() {
+    const mockQueryResult = createMockQueryResult([{ id: 1, name: "test" }]);
+    return {
+        type: "postgresql",
+        name: "PostgreSQL Adapter",
+        version: "0.1.0",
+        // Connection methods
+        connect: vi.fn().mockResolvedValue(undefined),
+        disconnect: vi.fn().mockResolvedValue(undefined),
+        getHealth: vi.fn().mockResolvedValue(createMockHealthStatus()),
+        // Query execution
+        executeQuery: vi.fn().mockResolvedValue(mockQueryResult),
+        executeReadQuery: vi.fn().mockResolvedValue(mockQueryResult),
+        executeWriteQuery: vi.fn().mockResolvedValue(mockQueryResult),
+        // Transaction methods
+        beginTransaction: vi.fn().mockResolvedValue("txn-123"),
+        commitTransaction: vi.fn().mockResolvedValue(undefined),
+        rollbackTransaction: vi.fn().mockResolvedValue(undefined),
+        createSavepoint: vi.fn().mockResolvedValue(undefined),
+        releaseSavepoint: vi.fn().mockResolvedValue(undefined),
+        rollbackToSavepoint: vi.fn().mockResolvedValue(undefined),
+        getTransactionConnection: vi.fn().mockReturnValue(undefined),
+        executeOnConnection: vi.fn().mockResolvedValue(createMockQueryResult()),
+        // Schema methods
+        getSchema: vi.fn().mockResolvedValue(createMockSchemaInfo()),
+        listTables: vi.fn().mockResolvedValue([createMockTableInfo("users")]),
+        describeTable: vi.fn().mockResolvedValue(createMockTableInfo("users")),
+        listSchemas: vi.fn().mockResolvedValue(["public", "pg_catalog"]),
+        getTableIndexes: vi
+            .fn()
+            .mockResolvedValue([createMockIndexInfo("users", "users_pkey")]),
+        getAllIndexes: vi
+            .fn()
+            .mockResolvedValue([createMockIndexInfo("users", "users_pkey")]),
+        // Capabilities
+        getCapabilities: vi.fn().mockReturnValue({
+            json: true,
+            fullTextSearch: true,
+            vector: true,
+            geospatial: true,
+            transactions: true,
+            preparedStatements: true,
+            connectionPooling: true,
+            partitioning: true,
+            ltree: true,
+            cron: true,
+        }),
+        getSupportedToolGroups: vi
+            .fn()
+            .mockReturnValue([
+            "core",
+            "transactions",
+            "jsonb",
+            "text",
+            "fulltext",
+            "performance",
+            "admin",
+            "monitoring",
+            "backup",
+            "vector",
+            "postgis",
+            "partitioning",
+        ]),
+        // Definition getters
+        getToolDefinitions: vi.fn().mockReturnValue([]),
+        getResourceDefinitions: vi.fn().mockReturnValue([]),
+        getPromptDefinitions: vi.fn(),
+        // Pool access
+        getPool: vi.fn().mockReturnValue(null),
+    };
+}
+/**
+ * Create a mock RequestContext for handler testing
+ */
+export function createMockRequestContext() {
+    return {
+        timestamp: new Date(),
+        requestId: "test-request-" + Math.random().toString(36).slice(2, 9),
+    };
+}
+/**
+ * Helper to configure mock adapter response for specific queries
+ */
+export function configureMockAdapterQuery(adapter, pattern, result) {
+    const originalImpl = adapter.executeQuery.getMockImplementation();
+    const impl = (sql) => {
+        if (sql.includes(pattern)) {
+            return Promise.resolve(result);
+        }
+        if (typeof originalImpl === "function") {
+            return originalImpl(sql);
+        }
+        return Promise.resolve(createMockQueryResult());
+    };
+    // eslint-disable-next-line @typescript-eslint/no-misused-promises
+    adapter.executeQuery.mockImplementation(impl);
+}
+/**
+ * Create a mock PostgresAdapter that returns empty results
+ */
+export function createMockPostgresAdapterEmpty() {
+    const adapter = createMockPostgresAdapter();
+    const emptyResult = createMockQueryResult([]);
+    adapter.executeQuery.mockResolvedValue(emptyResult);
+    adapter.executeReadQuery.mockResolvedValue(emptyResult);
+    adapter.executeWriteQuery.mockResolvedValue({
+        rows: [],
+        rowsAffected: 0,
+        executionTimeMs: 1,
+    });
+    return adapter;
+}
+/**
+ * Create a mock PostgresAdapter that throws on query execution
+ */
+export function createMockPostgresAdapterWithError(errorMessage = "Database connection failed") {
+    const adapter = createMockPostgresAdapter();
+    const dbError = new Error(errorMessage);
+    adapter.executeQuery.mockRejectedValue(dbError);
+    adapter.executeReadQuery.mockRejectedValue(dbError);
+    adapter.executeWriteQuery.mockRejectedValue(dbError);
+    adapter.getHealth.mockResolvedValue(createMockHealthStatus(false));
+    return adapter;
+}
+/**
+ * Create a mock adapter for transaction testing with a mock connection
+ */
+export function createMockPostgresAdapterWithTransaction() {
+    const adapter = createMockPostgresAdapter();
+    const mockConnection = {
+        query: vi.fn().mockResolvedValue({ rows: [], rowCount: 0 }),
+        release: vi.fn(),
+    };
+    adapter.getTransactionConnection.mockReturnValue(mockConnection);
+    return { ...adapter, mockConnection };
+}
+//# sourceMappingURL=adapter.js.map

package/dist/__tests__/mocks/adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter.js","sourceRoot":"","sources":["../../../src/__tests__/mocks/adapter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAW5B;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,OAAkC,EAAE,EACpC,YAAY,GAAG,CAAC;IAEhB,OAAO;QACL,IAAI;QACJ,YAAY,EAAE,YAAY;QAC1B,eAAe,EAAE,CAAC;KACnB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,IAAY,EACZ,IAAY,EACZ,QAAQ,GAAG,IAAI,EACf,UAAU,GAAG,KAAK;IAElB,OAAO;QACL,IAAI;QACJ,IAAI;QACJ,QAAQ;QACR,UAAU;QACV,YAAY,EAAE,IAAI;KACnB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,IAAY,EACZ,UAAU,GAAG,QAAQ,EACrB,QAAQ,GAAG,GAAG;IAEd,OAAO;QACL,IAAI;QACJ,MAAM,EAAE,UAAU;QAClB,IAAI,EAAE,OAAO;QACb,OAAO,EAAE;YACP,oBAAoB,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC;YAClD,oBAAoB,CAAC,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,KAAK,CAAC;YACzD,oBAAoB,CAClB,YAAY,EACZ,0BAA0B,EAC1B,KAAK,EACL,KAAK,CACN;SACF;QACD,QAAQ;KACT,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,SAAiB,EACjB,SAAiB,EACjB,UAAU,GAAG,QAAQ;IAErB,OAAO;QACL,IAAI,EAAE,SAAS;QACf,SAAS;QACT,UAAU;QACV,OAAO,EAAE,CAAC,IAAI,CAAC;QACf,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;QACnC,IAAI,EAAE,OAAO;KACd,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,OAAO;QACL,MAAM,EAAE,CAAC,mBAAmB,CAAC,OAAO,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,CAAC;QACvE,KAAK,EAAE,EAAE;QACT,OAAO,EAAE,CAAC,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;KACtD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,SAAS,GAAG,IAAI;IACrD,OAAO;QACL,SAAS;QACT,SAAS,EAAE,CAAC;QACZ,OAAO,EAAE,MAAM;QACf,SAAS,EAAE;YACT,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,CAAC;YACT,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,CAAC;YACV,YAAY,EAAE,GAAG;SAClB;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB;IAWvC,MAAM,eAAe,GAAG,qBAAqB,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAEzE,OAAO;QACL,IAAI,EAAE,YAAqB;QAC3B,IAAI,EAAE,oBAAoB;QAC1B,OAAO,EAAE,OAAO;QAEhB,qBAAqB;QACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC7C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAChD,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,CAAC;QAE9D,kBAAkB;QAClB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC;QACxD,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC;QAC5D,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC;QAE7D,sBAAsB;QACtB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACtD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACvD,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACzD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACrD,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACtD,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACzD,wBAAwB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;QAC5D,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,CAAC;QAEvE,iBAAiB;QACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,EAAE,CAAC;QAC5D,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;QACrE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACtE,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAChE,eAAe,EAAE,EAAE;aAChB,EAAE,EAAE;aACJ,iBAAiB,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;QAClE,aAAa,EAAE,EAAE;aACd,EAAE,EAAE;aACJ,iBAAiB,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;QAGlE,eAAe;QACf,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YACvC,IAAI,EAAE,IAAI;YACV,cAAc,EAAE,IAAI;YACpB,MAAM,EAAE,IAAI;YACZ,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,IAAI;YAClB,kBAAkB,EAAE,IAAI;YACxB,iBAAiB,EAAE,IAAI;YACvB,YAAY,EAAE,IAAI;YAClB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,IAAI;SACX,CAAC;QACF,sBAAsB,EAAE,EAAE;aACvB,EAAE,EAAE;aACJ,eAAe,CAAC;YACf,MAAM;YACN,cAAc;YACd,OAAO;YACP,MAAM;YACN,UAAU;YACV,aAAa;YACb,OAAO;YACP,YAAY;YACZ,QAAQ;YACR,QAAQ;YACR,SAAS;YACT,cAAc;SACf,CAAC;QAEJ,qBAAqB;QACrB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QAC/C,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QACnD,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;QAE7B,cAAc;QACd,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;KACvC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB;IAItC,OAAO;QACL,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,SAAS,EAAE,eAAe,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;KACpE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB,CACvC,OAAqD,EACrD,OAAe,EACf,MAAmB;IAEnB,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC,qBAAqB,EAElD,CAAC;IAEd,MAAM,IAAI,GAAG,CAAC,GAAW,EAAwB,EAAE;QACjD,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1B,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;YACvC,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC;IACF,kEAAkE;IAClE,OAAO,CAAC,YAAY,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,8BAA8B;IAG5C,MAAM,OAAO,GAAG,yBAAyB,EAAE,CAAC;IAC5C,MAAM,WAAW,GAAG,qBAAqB,CAAC,EAAE,CAAC,CAAC;IAE9C,OAAO,CAAC,YAAY,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;IACpD,OAAO,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;IACxD,OAAO,CAAC,iBAAiB,CAAC,iBAAiB,CAAC;QAC1C,IAAI,EAAE,EAAE;QACR,YAAY,EAAE,CAAC;QACf,eAAe,EAAE,CAAC;KACnB,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kCAAkC,CAChD,YAAY,GAAG,4BAA4B;IAE3C,MAAM,OAAO,GAAG,yBAAyB,EAAE,CAAC;IAC5C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;IAExC,OAAO,CAAC,YAAY,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAChD,OAAO,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACpD,OAAO,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACpD,OAAO,CAAC,SAAsC,CAAC,iBAAiB,CAC/D,sBAAsB,CAAC,KAAK,CAAC,CAC9B,CAAC;IAEF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wCAAwC;IAQtD,MAAM,OAAO,GAAG,yBAAyB,EAAE,CAAC;IAE5C,MAAM,cAAc,GAAG;QACrB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAC3D,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;IAGA,OAAO,CAAC,wBACT,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;IAElC,OAAO,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,CAAC;AACxC,CAAC"}

package/dist/__tests__/mocks/index.d.ts

@@ -0,0 +1,11 @@
+/**
+ * postgres-mcp - Test Mocks
+ *
+ * Centralized mock factories for testing. All tests should import
+ * mocks from this module for consistency.
+ */
+export { createMockQueryResult, createMockColumnInfo, createMockTableInfo, createMockIndexInfo, createMockSchemaInfo, createMockHealthStatus, createMockPostgresAdapter, createMockPostgresAdapterEmpty, createMockPostgresAdapterWithError, createMockPostgresAdapterWithTransaction, createMockRequestContext, configureMockAdapterQuery, } from "./adapter.js";
+export type { MockPostgresAdapter } from "./adapter.js";
+export { createMockPoolClient, createMockPool, createMockConnectionPool, configureMockPoolQuery, } from "./pool.js";
+export type { QueryResult, TableInfo, IndexInfo, SchemaInfo, HealthStatus, ColumnInfo, } from "../../types/index.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/__tests__/mocks/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/__tests__/mocks/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EACL,qBAAqB,EACrB,oBAAoB,EACpB,mBAAmB,EACnB,mBAAmB,EACnB,oBAAoB,EACpB,sBAAsB,EACtB,yBAAyB,EACzB,8BAA8B,EAC9B,kCAAkC,EAClC,wCAAwC,EACxC,wBAAwB,EACxB,yBAAyB,GAC1B,MAAM,cAAc,CAAC;AACtB,YAAY,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAGxD,OAAO,EACL,oBAAoB,EACpB,cAAc,EACd,wBAAwB,EACxB,sBAAsB,GACvB,MAAM,WAAW,CAAC;AAGnB,YAAY,EACV,WAAW,EACX,SAAS,EACT,SAAS,EACT,UAAU,EACV,YAAY,EACZ,UAAU,GACX,MAAM,sBAAsB,CAAC"}

package/dist/__tests__/mocks/index.js

@@ -0,0 +1,11 @@
+/**
+ * postgres-mcp - Test Mocks
+ *
+ * Centralized mock factories for testing. All tests should import
+ * mocks from this module for consistency.
+ */
+// PostgreSQL Adapter mocks
+export { createMockQueryResult, createMockColumnInfo, createMockTableInfo, createMockIndexInfo, createMockSchemaInfo, createMockHealthStatus, createMockPostgresAdapter, createMockPostgresAdapterEmpty, createMockPostgresAdapterWithError, createMockPostgresAdapterWithTransaction, createMockRequestContext, configureMockAdapterQuery, } from "./adapter.js";
+// Connection pool mocks
+export { createMockPoolClient, createMockPool, createMockConnectionPool, configureMockPoolQuery, } from "./pool.js";
+//# sourceMappingURL=index.js.map

package/dist/__tests__/mocks/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/__tests__/mocks/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,2BAA2B;AAC3B,OAAO,EACL,qBAAqB,EACrB,oBAAoB,EACpB,mBAAmB,EACnB,mBAAmB,EACnB,oBAAoB,EACpB,sBAAsB,EACtB,yBAAyB,EACzB,8BAA8B,EAC9B,kCAAkC,EAClC,wCAAwC,EACxC,wBAAwB,EACxB,yBAAyB,GAC1B,MAAM,cAAc,CAAC;AAGtB,wBAAwB;AACxB,OAAO,EACL,oBAAoB,EACpB,cAAc,EACd,wBAAwB,EACxB,sBAAsB,GACvB,MAAM,WAAW,CAAC"}

package/dist/__tests__/mocks/pool.d.ts

@@ -0,0 +1,43 @@
+/**
+ * postgres-mcp - Connection Pool Mock
+ *
+ * Provides mock implementation of pg Pool and PoolClient
+ * for testing connection management without real database.
+ */
+import { vi } from "vitest";
+import type { Pool } from "pg";
+/**
+ * Create a mock PoolClient
+ */
+export declare function createMockPoolClient(): {
+    query: ReturnType<typeof vi.fn>;
+    release: (err?: Error | boolean) => void;
+};
+/**
+ * Create a mock pg Pool
+ */
+export declare function createMockPool(): Partial<Pool> & {
+    connect: ReturnType<typeof vi.fn>;
+    query: ReturnType<typeof vi.fn>;
+    end: ReturnType<typeof vi.fn>;
+    on: ReturnType<typeof vi.fn>;
+};
+/**
+ * Create a mock ConnectionPool (our custom wrapper)
+ */
+export declare function createMockConnectionPool(): {
+    execute: ReturnType<typeof vi.fn>;
+    getStats: ReturnType<typeof vi.fn>;
+    healthCheck: ReturnType<typeof vi.fn>;
+    shutdown: ReturnType<typeof vi.fn>;
+    acquire: ReturnType<typeof vi.fn>;
+    release: ReturnType<typeof vi.fn>;
+};
+/**
+ * Configure mock pool query responses
+ */
+export declare function configureMockPoolQuery(pool: ReturnType<typeof createMockPool>, pattern: string, result: {
+    rows: Record<string, unknown>[];
+    rowCount: number;
+}): void;
+//# sourceMappingURL=pool.d.ts.map

package/dist/__tests__/mocks/pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pool.d.ts","sourceRoot":"","sources":["../../../src/__tests__/mocks/pool.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC5B,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAE/B;;GAEG;AACH,wBAAgB,oBAAoB,IAAI;IACtC,KAAK,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAChC,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,OAAO,KAAK,IAAI,CAAC;CAC1C,CAKA;AAED;;GAEG;AACH,wBAAgB,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG;IAChD,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAClC,KAAK,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAChC,GAAG,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC9B,EAAE,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;CAC9B,CAYA;AAED;;GAEG;AACH,wBAAgB,wBAAwB,IAAI;IAC1C,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACnC,WAAW,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACtC,QAAQ,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACnC,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;CACnC,CAmBA;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,IAAI,EAAE,UAAU,CAAC,OAAO,cAAc,CAAC,EACvC,OAAO,EAAE,MAAM,EACf,MAAM,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,GAC5D,IAAI,CAoBN"}

package/dist/__tests__/mocks/pool.js

@@ -0,0 +1,71 @@
+/**
+ * postgres-mcp - Connection Pool Mock
+ *
+ * Provides mock implementation of pg Pool and PoolClient
+ * for testing connection management without real database.
+ */
+import { vi } from "vitest";
+/**
+ * Create a mock PoolClient
+ */
+export function createMockPoolClient() {
+    return {
+        query: vi.fn().mockResolvedValue({ rows: [], rowCount: 0 }),
+        release: vi.fn(),
+    };
+}
+/**
+ * Create a mock pg Pool
+ */
+export function createMockPool() {
+    const mockClient = createMockPoolClient();
+    return {
+        connect: vi.fn().mockResolvedValue(mockClient),
+        query: vi.fn().mockResolvedValue({ rows: [], rowCount: 0 }),
+        end: vi.fn().mockResolvedValue(undefined),
+        on: vi.fn().mockReturnThis(),
+        totalCount: 10,
+        idleCount: 8,
+        waitingCount: 0,
+    };
+}
+/**
+ * Create a mock ConnectionPool (our custom wrapper)
+ */
+export function createMockConnectionPool() {
+    const mockClient = createMockPoolClient();
+    return {
+        execute: vi
+            .fn()
+            .mockResolvedValue({ rows: [], rowCount: 0, executionTimeMs: 5 }),
+        getStats: vi.fn().mockReturnValue({
+            total: 10,
+            active: 2,
+            idle: 8,
+            waiting: 0,
+            totalQueries: 100,
+        }),
+        healthCheck: vi.fn().mockResolvedValue({ connected: true, latencyMs: 5 }),
+        shutdown: vi.fn().mockResolvedValue(undefined),
+        acquire: vi.fn().mockResolvedValue(mockClient),
+        release: vi.fn(),
+    };
+}
+/**
+ * Configure mock pool query responses
+ */
+export function configureMockPoolQuery(pool, pattern, result) {
+    const originalImpl = pool.query.getMockImplementation();
+    const impl = (sql) => {
+        if (sql.includes(pattern)) {
+            return Promise.resolve(result);
+        }
+        if (typeof originalImpl === "function") {
+            return originalImpl(sql);
+        }
+        return Promise.resolve({ rows: [], rowCount: 0 });
+    };
+    // eslint-disable-next-line @typescript-eslint/no-misused-promises
+    pool.query.mockImplementation(impl);
+}
+//# sourceMappingURL=pool.js.map

package/dist/__tests__/mocks/pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pool.js","sourceRoot":"","sources":["../../../src/__tests__/mocks/pool.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG5B;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAIlC,OAAO;QACL,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAC3D,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc;IAM5B,MAAM,UAAU,GAAG,oBAAoB,EAAE,CAAC;IAE1C,OAAO;QACL,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC;QAC9C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAC3D,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACzC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;QAC5B,UAAU,EAAE,EAAE;QACd,SAAS,EAAE,CAAC;QACZ,YAAY,EAAE,CAAC;KAChB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB;IAQtC,MAAM,UAAU,GAAG,oBAAoB,EAAE,CAAC;IAE1C,OAAO;QACL,OAAO,EAAE,EAAE;aACR,EAAE,EAAE;aACJ,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC;QACnE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAChC,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,CAAC;YACT,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,CAAC;YACV,YAAY,EAAE,GAAG;SAClB,CAAC;QACF,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QACzE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC9C,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC;QAC9C,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CACpC,IAAuC,EACvC,OAAe,EACf,MAA6D;IAE7D,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,qBAAqB,EAIxC,CAAC;IAEd,MAAM,IAAI,GAAG,CACX,GAAW,EACqD,EAAE;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1B,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;YACvC,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC;IACF,kEAAkE;IAClE,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;AACtC,CAAC"}

package/dist/adapters/DatabaseAdapter.d.ts

@@ -0,0 +1,139 @@
+/**
+ * postgres-mcp - Database Adapter Interface
+ *
+ * Abstract base class that all database adapters must implement.
+ * Provides a consistent interface for database operations.
+ */
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { DatabaseType, DatabaseConfig, QueryResult, SchemaInfo, TableInfo, HealthStatus, AdapterCapabilities, ToolDefinition, ResourceDefinition, PromptDefinition, RequestContext, ToolGroup } from "../types/index.js";
+/**
+ * Abstract base class for database adapters
+ */
+export declare abstract class DatabaseAdapter {
+    /** Database type identifier */
+    abstract readonly type: DatabaseType;
+    /** Human-readable adapter name */
+    abstract readonly name: string;
+    /** Adapter version */
+    abstract readonly version: string;
+    /** Connection state */
+    protected connected: boolean;
+    /**
+     * Connect to the database
+     * @param config - Database connection configuration
+     */
+    abstract connect(config: DatabaseConfig): Promise<void>;
+    /**
+     * Disconnect from the database
+     */
+    abstract disconnect(): Promise<void>;
+    /**
+     * Check if connected to the database
+     */
+    isConnected(): boolean;
+    /**
+     * Get health status of the database connection
+     */
+    abstract getHealth(): Promise<HealthStatus>;
+    /**
+     * Execute a read-only query (SELECT)
+     * @param sql - SQL query string
+     * @param params - Query parameters for prepared statements
+     */
+    abstract executeReadQuery(sql: string, params?: unknown[]): Promise<QueryResult>;
+    /**
+     * Execute a write query (INSERT, UPDATE, DELETE)
+     * @param sql - SQL query string
+     * @param params - Query parameters for prepared statements
+     */
+    abstract executeWriteQuery(sql: string, params?: unknown[]): Promise<QueryResult>;
+    /**
+     * Execute any query (for admin operations)
+     * @param sql - SQL query string
+     * @param params - Query parameters for prepared statements
+     */
+    abstract executeQuery(sql: string, params?: unknown[]): Promise<QueryResult>;
+    /**
+     * Get full database schema information
+     */
+    abstract getSchema(): Promise<SchemaInfo>;
+    /**
+     * List all tables in the database
+     */
+    abstract listTables(): Promise<TableInfo[]>;
+    /**
+     * Describe a specific table's structure
+     * @param tableName - Name of the table
+     */
+    abstract describeTable(tableName: string): Promise<TableInfo>;
+    /**
+     * List available schemas/databases
+     */
+    abstract listSchemas(): Promise<string[]>;
+    /**
+     * Get adapter capabilities
+     */
+    abstract getCapabilities(): AdapterCapabilities;
+    /**
+     * Get supported tool groups for this adapter
+     */
+    abstract getSupportedToolGroups(): ToolGroup[];
+    /**
+     * Get all tool definitions for this adapter
+     */
+    abstract getToolDefinitions(): ToolDefinition[];
+    /**
+     * Get all resource definitions for this adapter
+     */
+    abstract getResourceDefinitions(): ResourceDefinition[];
+    /**
+     * Get all prompt definitions for this adapter
+     */
+    abstract getPromptDefinitions(): PromptDefinition[];
+    /**
+     * Register tools with the MCP server
+     * @param server - MCP server instance
+     * @param enabledTools - Set of enabled tool names (from filtering)
+     */
+    registerTools(server: McpServer, enabledTools: Set<string>): void;
+    /**
+     * Register a single tool with the MCP server
+     */
+    protected registerTool(server: McpServer, tool: ToolDefinition): void;
+    /**
+     * Extract the Zod shape from a schema, handling complex pipelines
+     * Traverses through: preprocess, transform, effects, refine, pipe
+     */
+    private extractZodShape;
+    /**
+     * Register resources with the MCP server
+     */
+    registerResources(server: McpServer): void;
+    /**
+     * Register a single resource with the MCP server
+     */
+    protected registerResource(server: McpServer, resource: ResourceDefinition): void;
+    /**
+     * Register prompts with the MCP server
+     */
+    registerPrompts(server: McpServer): void;
+    /**
+     * Register a single prompt with the MCP server
+     */
+    protected registerPrompt(server: McpServer, prompt: PromptDefinition): void;
+    /**
+     * Validate query for safety (SQL injection prevention)
+     * @param sql - SQL query to validate
+     * @param isReadOnly - Whether to enforce read-only restrictions
+     */
+    validateQuery(sql: string, isReadOnly: boolean): void;
+    /**
+     * Create a request context for tool execution
+     */
+    createContext(requestId?: string): RequestContext;
+    /**
+     * Get adapter info for logging/debugging
+     */
+    getInfo(): Record<string, unknown>;
+}
+//# sourceMappingURL=DatabaseAdapter.d.ts.map

package/dist/adapters/DatabaseAdapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DatabaseAdapter.d.ts","sourceRoot":"","sources":["../../src/adapters/DatabaseAdapter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAGzE,OAAO,KAAK,EACV,YAAY,EACZ,cAAc,EACd,WAAW,EACX,UAAU,EACV,SAAS,EACT,YAAY,EACZ,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,SAAS,EACV,MAAM,mBAAmB,CAAC;AAE3B;;GAEG;AACH,8BAAsB,eAAe;IACnC,+BAA+B;IAC/B,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,YAAY,CAAC;IAErC,kCAAkC;IAClC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAE/B,sBAAsB;IACtB,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAElC,uBAAuB;IACvB,SAAS,CAAC,SAAS,UAAS;IAM5B;;;OAGG;IACH,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAEvD;;OAEG;IACH,QAAQ,CAAC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAEpC;;OAEG;IACH,WAAW,IAAI,OAAO;IAItB;;OAEG;IACH,QAAQ,CAAC,SAAS,IAAI,OAAO,CAAC,YAAY,CAAC;IAM3C;;;;OAIG;IACH,QAAQ,CAAC,gBAAgB,CACvB,GAAG,EAAE,MAAM,EACX,MAAM,CAAC,EAAE,OAAO,EAAE,GACjB,OAAO,CAAC,WAAW,CAAC;IAEvB;;;;OAIG;IACH,QAAQ,CAAC,iBAAiB,CACxB,GAAG,EAAE,MAAM,EACX,MAAM,CAAC,EAAE,OAAO,EAAE,GACjB,OAAO,CAAC,WAAW,CAAC;IAEvB;;;;OAIG;IACH,QAAQ,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC;IAM5E;;OAEG;IACH,QAAQ,CAAC,SAAS,IAAI,OAAO,CAAC,UAAU,CAAC;IAEzC;;OAEG;IACH,QAAQ,CAAC,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAE3C;;;OAGG;IACH,QAAQ,CAAC,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAE7D;;OAEG;IACH,QAAQ,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAMzC;;OAEG;IACH,QAAQ,CAAC,eAAe,IAAI,mBAAmB;IAE/C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,IAAI,SAAS,EAAE;IAM9C;;OAEG;IACH,QAAQ,CAAC,kBAAkB,IAAI,cAAc,EAAE;IAE/C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,IAAI,kBAAkB,EAAE;IAEvD;;OAEG;IACH,QAAQ,CAAC,oBAAoB,IAAI,gBAAgB,EAAE;IAEnD;;;;OAIG;IACH,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG,IAAI;IAiBjE;;OAEG;IACH,SAAS,CAAC,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,cAAc,GAAG,IAAI;IAwCrE;;;OAGG;IACH,OAAO,CAAC,eAAe;IAkCvB;;OAEG;IACH,iBAAiB,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI;IAW1C;;OAEG;IACH,SAAS,CAAC,gBAAgB,CACxB,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE,kBAAkB,GAC3B,IAAI;IAiCP;;OAEG;IACH,eAAe,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI;IAWxC;;OAEG;IACH,SAAS,CAAC,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI;IA0C3E;;;;OAIG;IACH,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,GAAG,IAAI;IA8CrD;;OAEG;IACH,aAAa,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,cAAc;IAOjD;;OAEG;IACH,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CAUnC"}