storage-map-mcp-server 0.1.0

package/build/server.js

@@ -0,0 +1,172 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+import { parseStorageConfigs } from "./config.js";
+import { registry } from "./mcp-connectors/index.js";
+import { MySQLConnector, AthenaConnector } from "./clients/index.js";
+let storageConfigs = [];
+const mcpServer = new McpServer({
+    name: "storage-map",
+    version: "0.0.1",
+});
+mcpServer.registerTool("list_storages", {
+    description: "List all configured storage connections (RDB, NoSQL, Athena, etc.)",
+    inputSchema: {},
+}, () => {
+    const result = {
+        storages: storageConfigs.map(config => ({
+            id: config.id,
+            type: config.type,
+            writeMode: config.writeMode || false,
+            status: "configured",
+        }))
+    };
+    return {
+        content: [
+            {
+                type: "text",
+                text: JSON.stringify(result, null, 2),
+            },
+        ],
+    };
+});
+mcpServer.registerTool("query", {
+    description: "Execute a query on any storage (SQL for RDB/Athena, NoSQL query for MongoDB, etc.)",
+    inputSchema: {
+        storage_id: z.string().describe("Storage connection ID to use"),
+        query: z.string().describe("Query to execute (SQL, MongoDB query JSON, etc.)"),
+        parameters: z.any().optional().describe("Optional query parameters"),
+    },
+}, async ({ storage_id, query, parameters }) => {
+    const config = storageConfigs.find(c => c.id === storage_id);
+    if (!config) {
+        throw new Error(`Storage ${storage_id} not found`);
+    }
+    const connector = registry.getConnector(storage_id, config);
+    const result = await connector.query(query, parameters);
+    return {
+        content: [{
+                type: "text",
+                text: result.text,
+            }],
+    };
+});
+mcpServer.registerTool("execute", {
+    description: "Execute write operations (INSERT, UPDATE, DELETE for SQL; insert, update for NoSQL)",
+    inputSchema: {
+        storage_id: z.string().describe("Storage connection ID to use"),
+        operation: z.string().describe("Operation to execute (SQL statement or NoSQL operation)"),
+        parameters: z.any().optional().describe("Optional operation parameters"),
+    },
+}, async ({ storage_id, operation, parameters }) => {
+    const config = storageConfigs.find(c => c.id === storage_id);
+    if (!config) {
+        throw new Error(`Storage ${storage_id} not found`);
+    }
+    const connector = registry.getConnector(storage_id, config);
+    const result = await connector.execute(operation, parameters);
+    return {
+        content: [{
+                type: "text",
+                text: result.text,
+            }],
+    };
+});
+mcpServer.registerTool("list_collections", {
+    description: "List collections/tables/datasets in a storage (tables for RDB, collections for NoSQL, datasets for Athena)",
+    inputSchema: {
+        storage_id: z.string().describe("Storage connection ID"),
+        schema: z.string().optional().describe("Schema/database name (optional, for RDB/Athena)"),
+    },
+}, async ({ storage_id, schema }) => {
+    const config = storageConfigs.find(c => c.id === storage_id);
+    if (!config) {
+        throw new Error(`Storage ${storage_id} not found`);
+    }
+    const connector = registry.getConnector(storage_id, config);
+    const result = await connector.listCollections(schema);
+    return {
+        content: [{
+                type: "text",
+                text: result.text,
+            }],
+    };
+});
+mcpServer.registerTool("describe_collection", {
+    description: "Get schema/structure information for a collection/table",
+    inputSchema: {
+        storage_id: z.string().describe("Storage connection ID"),
+        collection: z.string().describe("Collection/table name"),
+        schema: z.string().optional().describe("Schema/database name (optional)"),
+    },
+}, async ({ storage_id, collection, schema }) => {
+    const config = storageConfigs.find(c => c.id === storage_id);
+    if (!config) {
+        throw new Error(`Storage ${storage_id} not found`);
+    }
+    const connector = registry.getConnector(storage_id, config);
+    const result = await connector.describeCollection(collection, schema);
+    return {
+        content: [{
+                type: "text",
+                text: result.text,
+            }],
+    };
+});
+mcpServer.registerTool("get_storage_info", {
+    description: "Get detailed information about a specific storage connection",
+    inputSchema: {
+        storage_id: z.string().describe("Storage connection ID"),
+    },
+}, async ({ storage_id }) => {
+    const config = storageConfigs.find(c => c.id === storage_id);
+    if (!config) {
+        throw new Error(`Storage ${storage_id} not found`);
+    }
+    const result = {
+        id: config.id,
+        type: config.type,
+        writeMode: config.writeMode || false,
+        connection: {
+            host: config.connection?.host,
+            port: config.connection?.port,
+            database: config.connection?.database,
+        }
+    };
+    return {
+        content: [
+            {
+                type: "text",
+                text: JSON.stringify(result, null, 2),
+            },
+        ],
+    };
+});
+function initializeStorages() {
+    storageConfigs = parseStorageConfigs();
+    if (storageConfigs.length === 0) {
+        console.error("Warning: No storage configurations found.");
+        console.error("Configure using environment variables:");
+        console.error("  STORAGE_<ID>_TYPE, STORAGE_<ID>_HOST, etc.");
+        console.error("  Or STORAGE_CONFIG with JSON array");
+        console.error("  Or simple DB_TYPE, DB_HOST, etc.");
+        return;
+    }
+    registry.register('mysql', MySQLConnector);
+    registry.register('athena', AthenaConnector);
+    storageConfigs.forEach(config => {
+        console.error(`  - ${config.id} (${config.type}) writeMode=${config.writeMode || false}`);
+    });
+}
+export async function startServer() {
+    try {
+        initializeStorages();
+        const transport = new StdioServerTransport();
+        await mcpServer.connect(transport);
+    }
+    catch (error) {
+        console.error("Error starting server:", error);
+        throw error;
+    }
+}
+//# sourceMappingURL=server.js.map

package/build/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,mBAAmB,EAAiB,MAAM,aAAa,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErE,IAAI,cAAc,GAAoB,EAAE,CAAC;AAEzC,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC;IAC9B,IAAI,EAAE,aAAa;IACnB,OAAO,EAAE,OAAO;CACjB,CAAC,CAAC;AAEH,SAAS,CAAC,YAAY,CACpB,eAAe,EACf;IACE,WAAW,EAAE,oEAAoE;IACjF,WAAW,EAAE,EAAE;CAChB,EACD,GAAG,EAAE;IACH,MAAM,MAAM,GAAG;QACb,QAAQ,EAAE,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACtC,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,KAAK;YACpC,MAAM,EAAE,YAAY;SACrB,CAAC,CAAC;KACJ,CAAC;IAEF,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;aACtC;SACF;KACF,CAAC;AACJ,CAAC,CACF,CAAC;AAGF,SAAS,CAAC,YAAY,CACpB,OAAO,EACP;IACE,WAAW,EAAE,oFAAoF;IACjG,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;QAC/D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;QAC9E,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;KACrE;CACF,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE;IAC1C,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;IAC7D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,YAAY,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAExD,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CAAC;KACH,CAAC;AACJ,CAAC,CACF,CAAC;AAGF,SAAS,CAAC,YAAY,CACpB,SAAS,EACT;IACE,WAAW,EAAE,qFAAqF;IAClG,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;QAC/D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yDAAyD,CAAC;QACzF,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+BAA+B,CAAC;KACzE;CACF,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE;IAC9C,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;IAC7D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,YAAY,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;IAE9D,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CAAC;KACH,CAAC;AACJ,CAAC,CACF,CAAC;AAGF,SAAS,CAAC,YAAY,CACpB,kBAAkB,EAClB;IACE,WAAW,EAAE,4GAA4G;IACzH,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACxD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;KAC1F;CACF,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE;IAC/B,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;IAC7D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,YAAY,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IAEvD,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CAAC;KACH,CAAC;AACJ,CAAC,CACF,CAAC;AAGF,SAAS,CAAC,YAAY,CACpB,qBAAqB,EACrB;IACE,WAAW,EAAE,yDAAyD;IACtE,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACxD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACxD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;KAC1E;CACF,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE;IAC3C,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;IAC7D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,YAAY,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,kBAAkB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAEtE,OAAO;QACL,OAAO,EAAE,CAAC;gBACR,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CAAC;KACH,CAAC;AACJ,CAAC,CACF,CAAC;AAGF,SAAS,CAAC,YAAY,CACpB,kBAAkB,EAClB;IACE,WAAW,EAAE,8DAA8D;IAC3E,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;KACzD;CACF,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;IACvB,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;IAC7D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,YAAY,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,MAAM,GAAG;QACb,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,KAAK;QACpC,UAAU,EAAE;YACV,IAAI,EAAE,MAAM,CAAC,UAAU,EAAE,IAAI;YAC7B,IAAI,EAAE,MAAM,CAAC,UAAU,EAAE,IAAI;YAC7B,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,QAAQ;SACtC;KACF,CAAC;IAEF,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;aACtC;SACF;KACF,CAAC;AACJ,CAAC,CACF,CAAC;AAEF,SAAS,kBAAkB;IACzB,cAAc,GAAG,mBAAmB,EAAE,CAAC;IAEvC,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC3D,OAAO,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,OAAO,CAAC,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAC9D,OAAO,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACrD,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACpD,OAAO;IACT,CAAC;IAED,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;IAC3C,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;IAG7C,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;QAC9B,OAAO,CAAC,KAAK,CAAC,OAAO,MAAM,CAAC,EAAE,KAAK,MAAM,CAAC,IAAI,eAAe,MAAM,CAAC,SAAS,IAAI,KAAK,EAAE,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,IAAI,CAAC;QACH,kBAAkB,EAAE,CAAC;QAErB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAC/C,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/package.json

@@ -0,0 +1,73 @@
+{
+  "name": "storage-map-mcp-server",
+  "version": "0.1.0",
+  "description": "MCP server for unified database and storage access",
+  "main": "build/index.js",
+  "types": "build/index.d.ts",
+  "type": "module",
+  "bin": {
+    "storage-map-server": "build/index.js"
+  },
+  "files": [
+    "build",
+    "README.md",
+    "LICENSE"
+  ],
+  "packageManager": "pnpm@10.18.0",
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx watch src/index.ts",
+    "start": "tsx src/index.ts",
+    "lint": "eslint src --ext .ts",
+    "format": "prettier --write \"src/**/*.ts\"",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf build"
+  },
+  "keywords": [
+    "mcp",
+    "database",
+    "mysql",
+    "athena",
+    "storage",
+    "query"
+  ],
+  "author": "Hoon <ihello0720@gmail.com>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/cyhoon/mcp-storage-map.git"
+  },
+  "bugs": {
+    "url": "https://github.com/cyhoon/mcp-storage-map/issues"
+  },
+  "homepage": "https://github.com/cyhoon/mcp-storage-map#readme",
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@aws-sdk/client-athena": "^3.873.0",
+    "@aws-sdk/client-s3": "^3.873.0",
+    "@modelcontextprotocol/sdk": "^1.17.2",
+    "dotenv": "^17.2.1",
+    "joi": "^17.11.0",
+    "lru-cache": "^10.1.0",
+    "mysql2": "^3.14.3",
+    "p-queue": "^8.0.1",
+    "winston": "^3.11.0",
+    "zod": "^3.25.76"
+  },
+  "devDependencies": {
+    "@types/node": "^24.2.1",
+    "@typescript-eslint/eslint-plugin": "^6.16.0",
+    "@typescript-eslint/parser": "^6.16.0",
+    "@vitest/ui": "^1.1.0",
+    "eslint": "^8.56.0",
+    "prettier": "^3.1.1",
+    "tsx": "^4.20.3",
+    "typescript": "^5.9.2",
+    "vitest": "^1.1.0"
+  },
+  "engines": {
+    "node": ">=24.13.1"
+  }
+}