vmoo-mcp-database-server 1.2.1 → 1.3.0

package/bin/vmoo-mcp-merchant.js

@@ -0,0 +1,45 @@
+#!/usr/bin/env node
+
+// VMOO商户Cookie管理MCP服务器启动器
+
+import { spawn } from 'child_process';
+import { fileURLToPath } from 'url';
+import path from 'path';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+// 解析命令行参数
+const args = process.argv.slice(2);
+const configPath = args.find(arg => arg.startsWith('--config='))?.split('=')[1];
+
+console.log('🏪 启动VMOO商户Cookie管理MCP服务器...');
+
+// 服务器文件路径
+const serverPath = path.join(__dirname, '../vmoo-database-merchant/server.js');
+
+// 启动服务器
+const serverProcess = spawn('node', [serverPath], {
+  stdio: 'inherit',
+  env: {
+    ...process.env,
+    ...(configPath && { CONFIG_PATH: configPath })
+  }
+});
+
+// 处理进程退出
+serverProcess.on('close', (code) => {
+  console.log(`🏪 商户Cookie管理MCP服务器已退出，退出码: ${code}`);
+  process.exit(code);
+});
+
+// 处理中断信号
+process.on('SIGINT', () => {
+  console.log('\n🛑 正在关闭商户Cookie管理MCP服务器...');
+  serverProcess.kill('SIGINT');
+});
+
+process.on('SIGTERM', () => {
+  console.log('\n🛑 正在关闭商户Cookie管理MCP服务器...');
+  serverProcess.kill('SIGTERM');
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vmoo-mcp-database-server",
-  "version": "1.2.1",
+  "version": "1.3.0",
   "description": "VMOO数据库MCP服务器集合 - 支持开发和生产环境的安全数据库访问",
   "type": "module",
   "main": "index.js",
@@ -8,13 +8,16 @@
     "vmoo-mcp-database-server": "./bin/vmoo-mcp-server.js",
     "vmoo-mcp-dev": "./bin/vmoo-mcp-dev.js",
     "vmoo-mcp-prod": "./bin/vmoo-mcp-prod.js",
-    "vmoo-mcp-deliver": "./bin/vmoo-mcp-deliver.js"
+    "vmoo-mcp-deliver": "./bin/vmoo-mcp-deliver.js",
+    "vmoo-mcp-merchant": "./bin/vmoo-mcp-merchant.js"
   },
   "scripts": {
     "start:prod": "cd vmoo-database-prod && npm start",
     "start:dev": "cd vmoo-database-dev && npm start",
-    "install:all": "npm install && cd vmoo-database-prod && npm install && cd ../vmoo-database-dev && npm install",
-    "test:all": "cd vmoo-database-prod && npm test && cd ../vmoo-database-dev && npm test",
+    "start:deliver": "cd vmoo-database-deliver && npm start",
+    "start:merchant": "cd vmoo-database-merchant && npm start",
+    "install:all": "npm install && cd vmoo-database-prod && npm install && cd ../vmoo-database-dev && npm install && cd ../vmoo-database-deliver && npm install && cd ../vmoo-database-merchant && npm install",
+    "test:all": "cd vmoo-database-prod && npm test && cd ../vmoo-database-dev && npm test && cd ../vmoo-database-deliver && npm test && cd ../vmoo-database-merchant && npm test",
     "build": "echo '构建完成'",
     "deploy": "./scripts/deploy.sh"
   },
@@ -33,6 +36,9 @@
     "model-context-protocol",
     "production",
     "development",
+    "deliver",
+    "merchant",
+    "cookie",
     "privacy",
     "security"
   ],
@@ -47,6 +53,7 @@
     "vmoo-database-dev/",
     "vmoo-database-prod/",
     "vmoo-database-deliver/",
+    "vmoo-database-merchant/",
     "index.js",
     "README.md"
   ],

package/shared/database-utils.js

@@ -1,14 +1,34 @@
 // VMOO数据库工具函数
 import mysql from 'mysql2/promise';
 
+/**
+ * 替换配置中的环境变量
+ * @param {Object} config 配置对象
+ * @returns {Object} 替换后的配置对象
+ */
+function replaceEnvVariables(config) {
+  const configStr = JSON.stringify(config);
+  const replacedStr = configStr.replace(/\$\{([^}]+)\}/g, (_, envVar) => {
+    const value = process.env[envVar];
+    if (value === undefined) {
+      throw new Error(`环境变量 ${envVar} 未设置`);
+    }
+    return value;
+  });
+  return JSON.parse(replacedStr);
+}
+
 /**
  * 创建数据库连接池
  * @param {Object} config 数据库配置
  * @returns {Object} 连接池
  */
 function createConnectionPool(config) {
+  // 替换环境变量
+  const processedConfig = replaceEnvVariables(config);
+
   return mysql.createPool({
-    ...config,
+    ...processedConfig,
     waitForConnections: true,
     connectionLimit: 10,
     queueLimit: 0
@@ -92,6 +112,7 @@ async function listTables(pool, pattern = null) {
 }
 
 export {
+  replaceEnvVariables,
   createConnectionPool,
   normalizeTableName,
   executeQuery,

package/vmoo-database-merchant/config.json

@@ -0,0 +1,25 @@
+{
+  "database": {
+    "host": "43.142.37.34",
+    "port": 3306,
+    "user": "merchant_cookie_manager",
+    "password": "${MERCHANT_PASSWORD}",
+    "database": "merchant_cookie_manager",
+    "charset": "utf8mb4"
+  },
+  "security": {
+    "level": "development",
+    "description": "商户Cookie管理：允许查询、插入、更新，禁止删除操作",
+    "features": [
+      "SELECT、INSERT、UPDATE、ALTER TABLE、CREATE TABLE",
+      "禁止DELETE、DROP TABLE、DROP DATABASE",
+      "自动LIMIT限制",
+      "时区自动转换"
+    ]
+  },
+  "server": {
+    "name": "vmoo-database-merchant-server",
+    "version": "1.0.0",
+    "description": "商户Cookie管理数据库服务器"
+  }
+}

package/vmoo-database-merchant/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "vmoo-database-merchant-server",
+  "version": "1.0.0",
+  "description": "VMOO商户Cookie管理数据库MCP服务器",
+  "type": "module",
+  "main": "server.js",
+  "scripts": {
+    "start": "node server.js",
+    "dev": "node server.js",
+    "test": "echo \"商户Cookie管理MCP服务器测试通过\" && exit 0"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.17.2",
+    "mysql2": "^3.6.5"
+  },
+  "keywords": [
+    "mcp",
+    "database",
+    "mysql",
+    "vmoo",
+    "merchant",
+    "cookie",
+    "商户Cookie管理"
+  ],
+  "author": "VMOO Team",
+  "license": "MIT"
+}

package/vmoo-database-merchant/server.js

@@ -0,0 +1,297 @@
+#!/usr/bin/env node
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import mysql from 'mysql2/promise';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+// 导入共享模块
+const { createConnectionPool, executeQuery, normalizeTableName, getTableStructure, getTableCount, getTableSampleData, listTables } = await import('../shared/database-utils.js');
+const { SECURITY_LEVELS, performSecurityCheck, applyQueryLimits } = await import('../shared/security-utils.js');
+const { processTimeFields } = await import('../shared/time-utils.js');
+
+// 读取配置文件
+const config = JSON.parse(fs.readFileSync(path.join(__dirname, 'config.json'), 'utf8'));
+
+console.log('🏪 VMOO商户Cookie管理MCP服务器已启动 - vmoo-database-merchant-server v1.0.0');
+console.log('🔒 安全级别: development - 商户Cookie管理：允许查询、插入、更新，禁止删除操作');
+
+// 创建数据库连接池
+const pool = createConnectionPool(config.database);
+
+// 创建MCP服务器
+const server = new Server(
+  {
+    name: config.server.name,
+    version: config.server.version,
+  },
+  {
+    capabilities: {
+      tools: {},
+    },
+  }
+);
+
+// 注册工具
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+  return {
+    tools: [
+      {
+        name: 'query_database',
+        description: '执行SQL查询语句（开发环境：支持增删改查，禁止删除操作）',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            query: {
+              type: 'string',
+              description: 'SQL查询语句'
+            },
+            limit: {
+              type: 'number',
+              description: '限制返回结果数量，默认100',
+              default: 100
+            }
+          },
+          required: ['query']
+        }
+      },
+      {
+        name: 'query_with_time_format',
+        description: '执行SQL查询并自动格式化时间字段（推荐用于包含时间的查询）',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            query: {
+              type: 'string',
+              description: 'SQL查询语句，时间字段会自动转换为北京时间'
+            },
+            limit: {
+              type: 'number',
+              description: '限制返回结果数量，默认100',
+              default: 100
+            }
+          },
+          required: ['query']
+        }
+      },
+      {
+        name: 'describe_table',
+        description: '获取表结构信息',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            table_name: {
+              type: 'string',
+              description: '表名（可以包含或不包含fanwe_前缀）'
+            }
+          },
+          required: ['table_name']
+        }
+      },
+      {
+        name: 'list_tables',
+        description: '列出数据库中所有表',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            pattern: {
+              type: 'string',
+              description: '表名匹配模式（可选）'
+            }
+          }
+        }
+      },
+      {
+        name: 'table_sample_data',
+        description: '获取表的示例数据',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            table_name: {
+              type: 'string',
+              description: '表名'
+            },
+            limit: {
+              type: 'number',
+              description: '返回记录数量，默认10',
+              default: 10
+            }
+          },
+          required: ['table_name']
+        }
+      },
+      {
+        name: 'table_count',
+        description: '获取表的记录总数',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            table_name: {
+              type: 'string',
+              description: '表名'
+            }
+          },
+          required: ['table_name']
+        }
+      }
+    ]
+  };
+});
+
+// 处理工具调用
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+
+  try {
+    switch (name) {
+      case 'query_database':
+        return await handleQueryDatabase(args);
+      case 'query_with_time_format':
+        return await handleQueryWithTimeFormat(args);
+      case 'describe_table':
+        return await handleDescribeTable(args);
+      case 'list_tables':
+        return await handleListTables(args);
+      case 'table_sample_data':
+        return await handleTableSampleData(args);
+      case 'table_count':
+        return await handleTableCount(args);
+      default:
+        throw new Error(`未知工具: ${name}`);
+    }
+  } catch (error) {
+    return {
+      content: [
+        {
+          type: 'text',
+          text: `错误: ${error.message}`
+        }
+      ]
+    };
+  }
+});
+
+// 查询数据库
+async function handleQueryDatabase(args) {
+  const { query, limit = 100 } = args;
+  
+  // 安全检查
+  const securityCheck = performSecurityCheck(query, SECURITY_LEVELS.DEVELOPMENT);
+  if (!securityCheck.allowed) {
+    throw new Error(securityCheck.reason);
+  }
+  
+  // 应用查询限制
+  const finalQuery = applyQueryLimits(query, limit);
+  
+  const rows = await executeQuery(pool, finalQuery);
+  
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `🟡 开发环境查询结果 (${Array.isArray(rows) ? rows.length : 0} 条记录):\n\n${JSON.stringify(rows, null, 2)}`
+      }
+    ]
+  };
+}
+
+// 带时间格式化的查询
+async function handleQueryWithTimeFormat(args) {
+  const { query, limit = 100 } = args;
+  
+  // 安全检查
+  const securityCheck = performSecurityCheck(query, SECURITY_LEVELS.DEVELOPMENT);
+  if (!securityCheck.allowed) {
+    throw new Error(securityCheck.reason);
+  }
+  
+  // 应用查询限制
+  const finalQuery = applyQueryLimits(query, limit);
+  
+  const rows = await executeQuery(pool, finalQuery);
+  const processedRows = processTimeFields(rows);
+  
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `🟡 开发环境查询结果 (${Array.isArray(processedRows) ? processedRows.length : 0} 条记录):\n📅 时间字段已自动转换为北京时间\n\n${JSON.stringify(processedRows, null, 2)}`
+      }
+    ]
+  };
+}
+
+// 其他处理函数...
+async function handleDescribeTable(args) {
+  const { table_name } = args;
+  const rows = await getTableStructure(pool, table_name);
+  
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `表 ${normalizeTableName(table_name)} 的结构:\n\n${JSON.stringify(rows, null, 2)}`
+      }
+    ]
+  };
+}
+
+async function handleListTables(args) {
+  const { pattern } = args;
+  const rows = await listTables(pool, pattern);
+  
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `数据库表列表:\n\n${JSON.stringify(rows, null, 2)}`
+      }
+    ]
+  };
+}
+
+async function handleTableSampleData(args) {
+  const { table_name, limit = 10 } = args;
+  const rows = await getTableSampleData(pool, table_name, limit);
+  const processedRows = processTimeFields(rows);
+  
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `表 ${normalizeTableName(table_name)} 的示例数据 (前${limit}条):\n\n${JSON.stringify(processedRows, null, 2)}`
+      }
+    ]
+  };
+}
+
+async function handleTableCount(args) {
+  const { table_name } = args;
+  const count = await getTableCount(pool, table_name);
+  
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `表 ${normalizeTableName(table_name)} 共有 ${count} 条记录`
+      }
+    ]
+  };
+}
+
+// 启动服务器
+async function main() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error(`🟡 VMOO商户Cookie管理MCP服务器已启动 - ${config.server.name} v${config.server.version}`);
+  console.error(`🔒 安全级别: ${config.security.level} - ${config.security.description}`);
+}
+
+main().catch(console.error);