storage-map-mcp-server 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Hoon
+
+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

@@ -0,0 +1,265 @@
+# MCP Storage Map
+
+![License](https://img.shields.io/github/license/cyhoon/mcp-storage-map)
+![Node Version](https://img.shields.io/badge/node-%3E%3D24.13.1-brightgreen)
+
+Model Context Protocol (MCP) server for unified database and storage access. Query multiple databases (MySQL, AWS Athena, and more) through a single, consistent interface.
+
+## Quick Start
+
+### 1. Installation
+
+```bash
+pnpm add -g @storage-map/mcp-server
+```
+
+Or run directly with pnpm dlx:
+```bash
+pnpm dlx @storage-map/mcp-server
+```
+
+### 2. Configure Your IDE
+
+Add to your Claude Desktop, Claude Code, or Cursor MCP configuration.
+
+**Single Database Example:**
+
+```json
+{
+  "mcpServers": {
+    "storage-map": {
+      "command": "pnpm",
+      "args": ["dlx", "@storage-map/mcp-server"],
+      "env": {
+        "STORAGE_MYSQL_TYPE": "mysql",
+        "STORAGE_MYSQL_HOST": "localhost",
+        "STORAGE_MYSQL_PORT": "3306",
+        "STORAGE_MYSQL_USER": "root",
+        "STORAGE_MYSQL_PASSWORD": "password",
+        "STORAGE_MYSQL_DATABASE": "myapp",
+        "STORAGE_MYSQL_WRITE_MODE": "false"
+      }
+    }
+  }
+}
+```
+
+**Multiple Databases Example:**
+
+```json
+{
+  "mcpServers": {
+    "storage-map": {
+      "command": "pnpm",
+      "args": ["dlx", "@storage-map/mcp-server"],
+      "env": {
+        "STORAGE_PROD_TYPE": "mysql",
+        "STORAGE_PROD_HOST": "prod-db.example.com",
+        "STORAGE_PROD_PORT": "3306",
+        "STORAGE_PROD_USER": "readonly",
+        "STORAGE_PROD_PASSWORD": "secret",
+        "STORAGE_PROD_DATABASE": "production",
+
+        "STORAGE_STAGING_TYPE": "mysql",
+        "STORAGE_STAGING_HOST": "staging-db.example.com",
+        "STORAGE_STAGING_PORT": "3306",
+        "STORAGE_STAGING_USER": "admin",
+        "STORAGE_STAGING_PASSWORD": "secret",
+        "STORAGE_STAGING_DATABASE": "staging",
+        "STORAGE_STAGING_WRITE_MODE": "true",
+
+        "STORAGE_ANALYTICS_TYPE": "athena",
+        "STORAGE_ANALYTICS_REGION": "us-west-2",
+        "STORAGE_ANALYTICS_DATABASE": "analytics",
+        "STORAGE_ANALYTICS_S3_OUTPUT_LOCATION": "s3://analytics-results/"
+      }
+    }
+  }
+}
+```
+
+> **Tip**: Each database connection needs a unique ID (the part after `STORAGE_` and before the property name). In the examples above: `MYSQL`, `PROD`, `STAGING`, and `ANALYTICS` are the IDs.
+
+### 3. Start Using
+
+Ask your AI assistant to:
+- **"List all my configured databases"**
+- **"Query the users table from my MySQL database"**
+- **"Show me tables in my Athena database"**
+- **"Get the schema for the orders table"**
+- **"Execute SELECT * FROM products LIMIT 10 on mysql storage"**
+
+## Compatibility
+
+| Database | Status | Version |
+|----------|--------|---------|
+| MySQL | ✅ Fully supported | 5.7+ |
+| AWS Athena | ✅ Fully supported | All versions |
+
+## Key Tools
+
+| Tool | Description |
+|------|-------------|
+| `list_storages` | List all configured storage connections |
+| `query` | Execute read queries (SELECT, etc.) |
+| `execute` | Execute write operations (INSERT, UPDATE, DELETE) |
+| `list_collections` | List tables/collections in a storage |
+| `describe_collection` | Get schema information for a table/collection |
+| `get_storage_info` | Get detailed connection information |
+
+## Configuration
+
+### Environment Variables
+
+Storage Map uses a flexible configuration system with environment variables:
+
+**Pattern**: `STORAGE_<ID>_<PROPERTY>`
+
+- `<ID>`: Unique identifier for your storage connection (e.g., `MYSQL`, `PROD`, `STAGING`, `ANALYTICS`)
+  - Use descriptive names to identify each connection
+  - Same database type can have multiple IDs (e.g., `PROD_DB` and `STAGING_DB` for different MySQL servers)
+- `<PROPERTY>`: Configuration property (e.g., `TYPE`, `HOST`, `PORT`)
+
+**Common Properties**:
+- `TYPE`: Database type (`mysql`, `athena`) - **Required**
+- `WRITE_MODE`: Enable write operations (`true` or `false`, default: `false`)
+
+**MySQL**:
+- `HOST`: Database host - **Required**
+- `PORT`: Database port (default: `3306`)
+- `USER`: Username - **Required**
+- `PASSWORD`: Password - **Required**
+- `DATABASE`: Database name - **Required**
+
+**AWS Athena**:
+- `REGION`: AWS region (default: `us-east-1`)
+- `DATABASE`: Athena database/catalog (default: `default`)
+- `S3_OUTPUT_LOCATION`: S3 path for query results - **Required**
+- `WORKGROUP`: Athena workgroup (default: `primary`)
+
+### Configuration Examples
+
+**Example 1: Single MySQL Database**
+
+```bash
+STORAGE_MYSQL_TYPE=mysql
+STORAGE_MYSQL_HOST=localhost
+STORAGE_MYSQL_PORT=3306
+STORAGE_MYSQL_USER=root
+STORAGE_MYSQL_PASSWORD=password
+STORAGE_MYSQL_DATABASE=myapp
+```
+
+**Example 2: Multiple MySQL Databases (Prod + Staging)**
+
+```bash
+# Production (read-only)
+STORAGE_PROD_TYPE=mysql
+STORAGE_PROD_HOST=prod-db.example.com
+STORAGE_PROD_PORT=3306
+STORAGE_PROD_USER=readonly
+STORAGE_PROD_PASSWORD=secret
+STORAGE_PROD_DATABASE=production
+
+# Staging (read-write)
+STORAGE_STAGING_TYPE=mysql
+STORAGE_STAGING_HOST=staging-db.example.com
+STORAGE_STAGING_PORT=3306
+STORAGE_STAGING_USER=admin
+STORAGE_STAGING_PASSWORD=secret
+STORAGE_STAGING_DATABASE=staging
+STORAGE_STAGING_WRITE_MODE=true
+```
+
+**Example 3: MySQL + Athena**
+
+```bash
+# MySQL for operational data
+STORAGE_DB_TYPE=mysql
+STORAGE_DB_HOST=localhost
+STORAGE_DB_PORT=3306
+STORAGE_DB_USER=admin
+STORAGE_DB_PASSWORD=password
+STORAGE_DB_DATABASE=myapp
+
+# Athena for analytics
+STORAGE_ANALYTICS_TYPE=athena
+STORAGE_ANALYTICS_REGION=us-west-2
+STORAGE_ANALYTICS_DATABASE=analytics
+STORAGE_ANALYTICS_S3_OUTPUT_LOCATION=s3://my-bucket/athena-results/
+```
+
+## Features
+
+### Unified Interface
+Query different databases using the same MCP tools, regardless of the underlying technology.
+
+### Read-Only by Default
+All connections are read-only by default. Explicitly enable `WRITE_MODE` for write operations.
+
+### Multiple Connections
+Configure multiple databases of the same or different types simultaneously.
+
+### Extensible Architecture
+Easy to add new database connectors with the `McpConnector` interface.
+
+## Architecture
+
+```
+┌─────────────────────────────────────────┐
+│         MCP Server (server.ts)          │
+├─────────────────────────────────────────┤
+│  MCP Tools: query, execute, list, etc.  │
+└─────────────┬───────────────────────────┘
+              │
+      ┌───────┴────────┐
+      │   Registry     │
+      └───────┬────────┘
+              │
+    ┌─────────┴──────────┐
+    │                    │
+┌───▼──────┐      ┌─────▼─────┐
+│  MySQL   │      │  Athena   │
+│Connector │      │ Connector │
+└──────────┘      └───────────┘
+```
+
+Each connector implements the `McpConnector` interface:
+- `query()` - Execute read queries
+- `execute()` - Execute write operations
+- `listCollections()` - List tables/collections
+- `describeCollection()` - Get schema information
+
+## Security
+
+- Never commit credentials or API tokens to version control
+- Use environment variables for sensitive configuration
+- Keep `WRITE_MODE` disabled unless absolutely necessary
+- Review queries before execution, especially with write access enabled
+
+## Development
+
+```bash
+# Clone the repository
+git clone https://github.com/cyhoon/mcp-storage-map.git
+cd mcp-storage-map
+
+# Install dependencies
+pnpm install
+
+# Run in development mode
+pnpm dev
+
+# Build
+pnpm build
+
+# Run tests
+pnpm test
+
+# Lint
+pnpm lint
+```
+
+## License
+
+MIT - See [LICENSE](LICENSE)

package/build/clients/MySqlClient.d.ts

@@ -0,0 +1,14 @@
+import { StorageConfig } from '../config.js';
+import { McpConnector, McpConnectorResult } from '../mcp-connectors/index.js';
+export declare class MySQLConnector implements McpConnector {
+    private pool;
+    private config;
+    constructor(config: StorageConfig);
+    private connect;
+    query(query: string, params?: any): Promise<McpConnectorResult>;
+    execute(operation: string, params?: any): Promise<McpConnectorResult>;
+    listCollections(schema?: string): Promise<McpConnectorResult>;
+    describeCollection(collection: string, schema?: string): Promise<McpConnectorResult>;
+    disconnect(): Promise<void>;
+}
+//# sourceMappingURL=MySqlClient.d.ts.map

package/build/clients/MySqlClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MySqlClient.d.ts","sourceRoot":"","sources":["../../src/clients/MySqlClient.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAE9E,qBAAa,cAAe,YAAW,YAAY;IACjD,OAAO,CAAC,IAAI,CAA2B;IACvC,OAAO,CAAC,MAAM,CAAgB;gBAElB,MAAM,EAAE,aAAa;IAIjC,OAAO,CAAC,OAAO;IAiBT,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAyB/D,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA8BrE,eAAe,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAiC7D,kBAAkB,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA+BpF,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAQlC"}

package/build/clients/MySqlClient.js

@@ -0,0 +1,141 @@
+import mysql from 'mysql2/promise';
+export class MySQLConnector {
+    pool = null;
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    connect() {
+        if (!this.config.connection) {
+            throw new Error('MySQL connection config is missing');
+        }
+        this.pool = mysql.createPool({
+            host: this.config.connection.host,
+            port: this.config.connection.port || 3306,
+            user: this.config.connection.user,
+            password: this.config.connection.password,
+            database: this.config.connection.database,
+            waitForConnections: true,
+            connectionLimit: 10,
+            queueLimit: 0
+        });
+    }
+    async query(query, params) {
+        try {
+            if (!this.pool) {
+                this.connect();
+            }
+            const [rows] = await this.pool.execute(query, params);
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    rows,
+                    rowCount: Array.isArray(rows) ? rows.length : 0
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    query
+                }, null, 2)
+            };
+        }
+    }
+    async execute(operation, params) {
+        try {
+            if (!this.config.writeMode) {
+                throw new Error('Write operations are not allowed for this connection');
+            }
+            if (!this.pool) {
+                this.connect();
+            }
+            const [result] = await this.pool.execute(operation, params);
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    affectedRows: result.affectedRows || 0,
+                    insertId: result.insertId,
+                    message: result.message
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    operation
+                }, null, 2)
+            };
+        }
+    }
+    async listCollections(schema) {
+        try {
+            if (!this.pool) {
+                this.connect();
+            }
+            const database = schema || this.config.connection?.database;
+            const sqlQuery = database
+                ? 'SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ?'
+                : 'SHOW TABLES';
+            const [rows] = await this.pool.execute(sqlQuery, database ? [database] : undefined);
+            const tables = rows.map((row) => Object.values(row)[0]);
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    tables,
+                    count: tables.length,
+                    database: database
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    schema
+                }, null, 2)
+            };
+        }
+    }
+    async describeCollection(collection, schema) {
+        try {
+            if (!this.pool) {
+                this.connect();
+            }
+            const database = schema || this.config.connection?.database;
+            const fullTableName = database ? `${database}.${collection}` : collection;
+            const [columns] = await this.pool.execute(`DESCRIBE ${fullTableName}`);
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    table: collection,
+                    columns,
+                    database: database
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    table: collection,
+                    schema
+                }, null, 2)
+            };
+        }
+    }
+    async disconnect() {
+        if (!this.pool) {
+            return;
+        }
+        await this.pool.end();
+        this.pool = null;
+    }
+}
+//# sourceMappingURL=MySqlClient.js.map

package/build/clients/MySqlClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MySqlClient.js","sourceRoot":"","sources":["../../src/clients/MySqlClient.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,MAAM,gBAAgB,CAAC;AAInC,MAAM,OAAO,cAAc;IACjB,IAAI,GAAsB,IAAI,CAAC;IAC/B,MAAM,CAAgB;IAE9B,YAAY,MAAqB;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC;YAC3B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI;YACjC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,IAAI,IAAI;YACzC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI;YACjC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ;YACzC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ;YACzC,kBAAkB,EAAE,IAAI;YACxB,eAAe,EAAE,EAAE;YACnB,UAAU,EAAE,CAAC;SACd,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,MAAY;QACrC,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACvD,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI;oBACJ,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;iBAChD,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,wBAAwB;oBAChD,KAAK;iBACN,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB,EAAE,MAAY;QAC3C,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBAC3B,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;YAC1E,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAQ,CAAC;YACpE,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,CAAC;oBACtC,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,OAAO,EAAE,MAAM,CAAC,OAAO;iBACxB,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,wBAAwB;oBAChD,SAAS;iBACV,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,MAAe;QACnC,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC;YAC5D,MAAM,QAAQ,GAAG,QAAQ;gBACvB,CAAC,CAAC,yEAAyE;gBAC3E,CAAC,CAAC,aAAa,CAAC;YAElB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YACrF,MAAM,MAAM,GAAI,IAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAW,CAAC,CAAC;YAE7E,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,MAAM;oBACN,KAAK,EAAE,MAAM,CAAC,MAAM;oBACpB,QAAQ,EAAE,QAAQ;iBACnB,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,wBAAwB;oBAChD,MAAM;iBACP,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAkB,EAAE,MAAe;QAC1D,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC;YAC5D,MAAM,aAAa,GAAG,QAAQ,CAAC,CAAC,CAAC,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;YAE1E,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,IAAI,CAAC,IAAK,CAAC,OAAO,CAAC,YAAY,aAAa,EAAE,CAAC,CAAC;YAExE,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,UAAU;oBACjB,OAAO;oBACP,QAAQ,EAAE,QAAQ;iBACnB,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,wBAAwB;oBAChD,KAAK,EAAE,UAAU;oBACjB,MAAM;iBACP,EAAE,IAAI,EAAE,CAAC,CAAC;aACZ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;CACF"}

package/build/clients/athena-client/AthenaClient.d.ts

@@ -0,0 +1,17 @@
+import { StorageConfig } from '../../config.js';
+import { McpConnector, McpConnectorResult } from '../../mcp-connectors/index.js';
+export declare class AthenaConnector implements McpConnector {
+    private client;
+    private config;
+    private outputLocation;
+    private database;
+    private workGroup;
+    constructor(config: StorageConfig);
+    query(query: string, _params?: any): Promise<McpConnectorResult>;
+    execute(operation: string, params?: any): Promise<McpConnectorResult>;
+    listCollections(schema?: string): Promise<McpConnectorResult>;
+    describeCollection(collection: string, schema?: string): Promise<McpConnectorResult>;
+    private waitForQueryCompletion;
+    private formatResults;
+}
+//# sourceMappingURL=AthenaClient.d.ts.map

package/build/clients/athena-client/AthenaClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AthenaClient.d.ts","sourceRoot":"","sources":["../../../src/clients/athena-client/AthenaClient.ts"],"names":[],"mappings":"AAcA,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AAEjF,qBAAa,eAAgB,YAAW,YAAY;IAClD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,aAAa;IAiB3B,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAoDhE,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAoBrE,eAAe,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAgC7D,kBAAkB,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC;YA8C5E,sBAAsB;IA4BpC,OAAO,CAAC,aAAa;CAkCtB"}

package/build/clients/athena-client/AthenaClient.js

@@ -0,0 +1,199 @@
+import { AthenaClient, StartQueryExecutionCommand, GetQueryExecutionCommand, GetQueryResultsCommand, QueryExecutionState, ListTableMetadataCommand, GetTableMetadataCommand, } from '@aws-sdk/client-athena';
+export class AthenaConnector {
+    client;
+    config;
+    outputLocation;
+    database;
+    workGroup;
+    constructor(config) {
+        this.config = config;
+        const region = config.connection?.region || process.env.AWS_REGION || 'us-east-1';
+        this.outputLocation = config.connection?.s3OutputLocation || process.env.ATHENA_S3_OUTPUT || '';
+        this.database = config.connection?.database || process.env.ATHENA_DATABASE || 'default';
+        this.workGroup = config.connection?.workgroup || process.env.ATHENA_WORKGROUP || 'primary';
+        if (!this.outputLocation) {
+            throw new Error('Athena S3 output location is required (s3OutputLocation or ATHENA_S3_OUTPUT)');
+        }
+        this.client = new AthenaClient({
+            region,
+        });
+    }
+    async query(query, _params) {
+        try {
+            const startCommand = new StartQueryExecutionCommand({
+                QueryString: query,
+                QueryExecutionContext: {
+                    Database: this.database,
+                },
+                ResultConfiguration: {
+                    OutputLocation: this.outputLocation,
+                },
+                WorkGroup: this.workGroup,
+            });
+            const { QueryExecutionId } = await this.client.send(startCommand);
+            if (!QueryExecutionId) {
+                throw new Error('Failed to start query execution');
+            }
+            await this.waitForQueryCompletion(QueryExecutionId);
+            const resultsCommand = new GetQueryResultsCommand({
+                QueryExecutionId,
+            });
+            const results = await this.client.send(resultsCommand);
+            const rows = this.formatResults(results.ResultSet);
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    rows,
+                    rowCount: rows.length,
+                    queryExecutionId: QueryExecutionId,
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    query,
+                    database: this.database,
+                }, null, 2)
+            };
+        }
+    }
+    async execute(operation, params) {
+        try {
+            if (!this.config.writeMode) {
+                throw new Error('Write operations are not allowed for this connection');
+            }
+            return await this.query(operation, params);
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    operation,
+                }, null, 2)
+            };
+        }
+    }
+    async listCollections(schema) {
+        try {
+            const targetDatabase = schema || this.database;
+            const command = new ListTableMetadataCommand({
+                CatalogName: 'AwsDataCatalog',
+                DatabaseName: targetDatabase,
+            });
+            const response = await this.client.send(command);
+            const tables = response.TableMetadataList?.map(table => table.Name) || [];
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    tables,
+                    count: tables.length,
+                    database: targetDatabase,
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    database: schema || this.database,
+                }, null, 2)
+            };
+        }
+    }
+    async describeCollection(collection, schema) {
+        try {
+            const targetDatabase = schema || this.database;
+            const command = new GetTableMetadataCommand({
+                CatalogName: 'AwsDataCatalog',
+                DatabaseName: targetDatabase,
+                TableName: collection,
+            });
+            const response = await this.client.send(command);
+            const columns = response.TableMetadata?.Columns?.map(col => ({
+                name: col.Name,
+                type: col.Type,
+                comment: col.Comment,
+            })) || [];
+            const partitionKeys = response.TableMetadata?.PartitionKeys?.map(key => ({
+                name: key.Name,
+                type: key.Type,
+            })) || [];
+            return {
+                type: 'text',
+                text: JSON.stringify({
+                    table: collection,
+                    database: targetDatabase,
+                    columns,
+                    partitionKeys,
+                    tableType: response.TableMetadata?.TableType,
+                    location: response.TableMetadata?.Parameters?.['location'],
+                }, null, 2)
+            };
+        }
+        catch (error) {
+            return {
+                type: 'error',
+                text: JSON.stringify({
+                    error: error.message || 'Unknown error occurred',
+                    table: collection,
+                    database: schema || this.database,
+                }, null, 2)
+            };
+        }
+    }
+    async waitForQueryCompletion(queryExecutionId) {
+        const maxAttempts = 100;
+        const delayMs = 1000;
+        for (let i = 0; i < maxAttempts; i++) {
+            const command = new GetQueryExecutionCommand({
+                QueryExecutionId: queryExecutionId,
+            });
+            const response = await this.client.send(command);
+            const state = response.QueryExecution?.Status?.State;
+            switch (state) {
+                case QueryExecutionState.SUCCEEDED:
+                    return;
+                case QueryExecutionState.FAILED:
+                    throw new Error(`Query failed: ${response.QueryExecution?.Status?.StateChangeReason}`);
+                case QueryExecutionState.CANCELLED:
+                    throw new Error('Query was cancelled');
+                default:
+                    await new Promise(resolve => setTimeout(resolve, delayMs));
+            }
+        }
+        throw new Error('Query timeout - exceeded maximum wait time');
+    }
+    formatResults(resultSet) {
+        if (!resultSet?.Rows || resultSet.Rows.length === 0) {
+            return [];
+        }
+        const firstRow = resultSet.Rows[0];
+        const headers = [];
+        if (firstRow?.Data && Array.isArray(firstRow.Data)) {
+            firstRow.Data.forEach((col) => {
+                headers.push(col.VarCharValue);
+            });
+        }
+        const rows = [];
+        for (let i = 1; i < resultSet.Rows.length; i++) {
+            const row = resultSet.Rows[i];
+            const obj = {};
+            if (row.Data && Array.isArray(row.Data)) {
+                row.Data.forEach((cell, index) => {
+                    const header = headers[index];
+                    if (header) {
+                        obj[header] = cell.VarCharValue || null;
+                    }
+                });
+            }
+            rows.push(obj);
+        }
+        return rows;
+    }
+}
+//# sourceMappingURL=AthenaClient.js.map