@marcelo-ochoa/server-postgres 1.0.0 → 1.0.2

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -40,7 +40,7 @@ The server provides schema information for each table in the database:
 
 ## Change Log
 
-See [Change Log](CHANGELOG.md) for the history of changes.
+See [Change Log](https://github.com/marcelo-ochoa/servers/blob/main/src/postgres/CHANGELOG.md) for the history of changes.
 
 ## Configuration
 
@@ -213,7 +213,7 @@ Optionally, you can add it to a file called `.vscode/mcp.json` in your workspace
 
 ## PostgreSQL AWR in action
 
-See [PostgreSQL AWR in action](AWR_example.md) for an example of a performance report generated by the `pg-awr` tool for a production Moodle database, highlighting critical performance issues and optimization opportunities.
+See [PostgreSQL AWR in action](https://github.com/marcelo-ochoa/servers/blob/main/src/postgres/AWR_example.md) for an example of a performance report generated by the `pg-awr` tool for a production Moodle database, highlighting critical performance issues and optimization opportunities.
 
 ## Building
 
@@ -229,4 +229,4 @@ As usual the code of this extension is at [GitHub](https://github.com/marcelo-oc
 
 ## License
 
-This MCP server is licensed under the MIT License. This means you are free to use, modify, and distribute the software, subject to the terms and conditions of the MIT License. For more details, please see the LICENSE file in the project repository.
+This MCP server is licensed under the [MIT License](https://github.com/marcelo-ochoa/servers/blob/main/src/postgres/LICENSE). This means you are free to use, modify, and distribute the software, subject to the terms and conditions of the MIT License. For more details, please see the LICENSE file in the project repository.

package/dist/handlers.js

@@ -1,4 +1,4 @@
-import { withConnection, getResourceBaseUrl } from "./db.js";
+import { withConnection } from "./db.js";
 import { queryHandler } from "./tools/query.js";
 import { statsHandler } from "./tools/stats.js";
 import { connectHandler } from "./tools/connect.js";
@@ -20,14 +20,13 @@ export const callToolHandler = async (request) => {
 };
 const SCHEMA_PATH = "schema";
 export const listResourcesHandler = async (request) => {
-    const resourceBaseUrl = getResourceBaseUrl();
     return await withConnection(async (client) => {
-        const result = await client.query("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'");
+        const result = await client.query("SELECT table_name, table_schema FROM information_schema.tables WHERE table_schema NOT IN ('information_schema', 'pg_catalog')");
         return {
             resources: result.rows.map((row) => ({
-                uri: new URL(`${row.table_name}/${SCHEMA_PATH}`, resourceBaseUrl).href,
+                uri: `postgres://${row.table_schema}/${row.table_name}/${SCHEMA_PATH}`,
                 mimeType: "application/json",
-                name: `"${row.table_name}" database schema`,
+                name: `"${row.table_schema}"."${row.table_name}" database schema`,
             })),
         };
     });
@@ -37,17 +36,52 @@ export const readResourceHandler = async (request) => {
     const pathComponents = resourceUrl.pathname.split("/");
     const schema = pathComponents.pop();
     const tableName = pathComponents.pop();
+    const schemaName = resourceUrl.hostname;
     if (schema !== SCHEMA_PATH) {
         throw new Error("Invalid resource URI");
     }
     return await withConnection(async (client) => {
-        const result = await client.query("SELECT column_name, data_type FROM information_schema.columns WHERE table_name = $1", [tableName]);
+        const columnsResult = await client.query(`SELECT 
+                column_name, 
+                data_type, 
+                is_nullable, 
+                column_default,
+                character_maximum_length,
+                numeric_precision,
+                numeric_scale 
+             FROM information_schema.columns 
+             WHERE table_name = $1 AND table_schema = $2
+             ORDER BY ordinal_position`, [tableName, schemaName]);
+        const indexesResult = await client.query(`SELECT
+                ix.relname as index_name,
+                i.indisunique as is_unique,
+                a.attname as column_name
+            FROM
+                pg_class t,
+                pg_class ix,
+                pg_index i,
+                pg_attribute a,
+                pg_namespace n
+            WHERE
+                t.oid = i.indrelid
+                AND ix.oid = i.indexrelid
+                AND a.attrelid = t.oid
+                AND a.attnum = ANY(i.indkey)
+                AND t.relkind = 'r'
+                AND t.relname = $1
+                AND n.oid = t.relnamespace
+                AND n.nspname = $2
+            ORDER BY
+                ix.relname`, [tableName, schemaName]);
         return {
             contents: [
                 {
                     uri: request.params.uri,
                     mimeType: "application/json",
-                    text: JSON.stringify(result.rows, null, 2),
+                    text: JSON.stringify({
+                        columns: columnsResult.rows,
+                        indexes: indexesResult.rows
+                    }, null, 2),
                 },
             ],
         };

package/dist/server.js

@@ -1,23 +1,41 @@
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { CallToolRequestSchema, ListResourcesRequestSchema, ListResourceTemplatesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { z } from "zod";
 import { initializePool } from "./db.js";
 import { listResourcesHandler, readResourceHandler, callToolHandler } from "./handlers.js";
 import { tools } from "./tools.js";
 const server = new Server({
     name: "postgres-server",
-    version: "1.0.0",
+    version: "1.0.2",
 }, {
     capabilities: {
         resources: {},
         tools: {},
+        prompts: {}, // Add this line to indicate support for prompts
     },
 });
+const prompts = [
+    { id: 1, text: "pg-query select * from test" },
+    { id: 2, text: "pg-explain select * from test" },
+    { id: 3, text: "pg-stats test" },
+    { id: 4, text: "pg-connect to PostgreSQL using a connection string like host.docker.internal:5432/dbname with user name and password" },
+    { id: 5, text: "pg-awr to generate a PostgreSQL performance report (requires pg_stat_statements extension)" }
+];
+const PromptsListRequestSchema = z.object({
+    method: z.literal("prompts/list"),
+    params: z.object({}),
+});
+server.setRequestHandler(PromptsListRequestSchema, async () => {
+    return {
+        prompts,
+    };
+});
 server.setRequestHandler(ListResourceTemplatesRequestSchema, async () => {
     return {
         resourceTemplates: [
             {
-                uriTemplate: "postgres://{database}/{table_name}/schema",
+                uriTemplate: "postgres://{schema}/{table_name}/schema",
                 name: "Table Schema",
                 description: "Schema information for a PostgreSQL database table including column names and data types",
             },

package/dist/tools/stats.js

@@ -1,6 +1,12 @@
 import { withConnection } from "../db.js";
 export const statsHandler = async (request) => {
-    const tableName = request.params.arguments?.name;
+    let schema = 'public';
+    let tableName = request.params.arguments?.name;
+    if (tableName.includes('.')) {
+        const parts = tableName.split('.');
+        schema = parts[0];
+        tableName = parts[1];
+    }
     return await withConnection(async (client) => {
         const result = await client.query(`
       SELECT json_build_object(
@@ -15,7 +21,7 @@ export const statsHandler = async (request) => {
           FROM pg_class c
           JOIN pg_namespace n ON n.oid = c.relnamespace
           LEFT JOIN pg_stat_user_tables s ON s.relid = c.oid
-          WHERE c.relname = $1 AND n.nspname = 'public'
+          WHERE c.relname = $1 AND n.nspname = $2
         ),
         'index_stats', (
           SELECT json_agg(
@@ -30,7 +36,7 @@ export const statsHandler = async (request) => {
           JOIN pg_class c ON c.oid = i.indrelid
           JOIN pg_class c2 ON c2.oid = i.indexrelid
           JOIN pg_namespace n ON n.oid = c.relnamespace
-          WHERE c.relname = $1 AND n.nspname = 'public'
+          WHERE c.relname = $1 AND n.nspname = $2
         ),
         'column_stats', (
           SELECT json_agg(
@@ -42,10 +48,10 @@ export const statsHandler = async (request) => {
             )
           )
           FROM pg_stats
-          WHERE tablename = $1 AND schemaname = 'public'
+          WHERE tablename = $1 AND schemaname = $2
         )
       ) as stats_json
-    `, [tableName]);
+    `, [tableName, schema]);
         return {
             content: [{ type: "text", text: JSON.stringify(result.rows[0].stats_json, null, 2), mimeType: "application/json" }],
             isError: false,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@marcelo-ochoa/server-postgres",
   "mcpName": "io.github.marcelo-ochoa/postgres",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "MCP server for interacting with PostgreSQL databases",
   "keywords": [
     "read-only-mcp",
@@ -27,7 +27,7 @@
     "watch": "tsc --watch"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "1.0.1",
+    "@modelcontextprotocol/sdk": "^1.24.2",
     "@toon-format/toon": "^1.0.0",
     "pg": "^8.13.0"
   },