vmoo-mcp-database-server 1.0.1 → 1.1.0

package/README.md

@@ -15,6 +15,9 @@ npx vmoo-mcp-database-server@latest --env=dev
 
 # 生产环境
 npx vmoo-mcp-database-server@latest --env=prod
+
+# 配送测试站
+npx vmoo-mcp-database-server@latest --env=deliver
 ```
 
 ### 在Augment中配置
@@ -29,6 +32,10 @@ npx vmoo-mcp-database-server@latest --env=prod
     "vmoo-database-prod": {
       "command": "npx",
       "args": ["vmoo-mcp-database-server@latest", "--env=prod"]
+    },
+    "vmoo-database-deliver": {
+      "command": "npx",
+      "args": ["vmoo-mcp-database-server@latest", "--env=deliver"]
     }
   }
 }

package/bin/vmoo-mcp-deliver.js

@@ -0,0 +1,45 @@
+#!/usr/bin/env node
+
+// VMOO配送测试站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配送测试站MCP服务器...');
+
+// 服务器文件路径
+const serverPath = path.join(__dirname, '../vmoo-database-deliver/server.js');
+
+// 启动服务器
+const serverProcess = spawn('node', [serverPath], {
+  stdio: 'inherit',
+  env: {
+    ...process.env,
+    ...(configPath && { CONFIG_PATH: configPath })
+  }
+});
+
+// 处理进程退出
+serverProcess.on('close', (code) => {
+  console.log(`🚚 配送测试站MCP服务器已退出，退出码: ${code}`);
+  process.exit(code);
+});
+
+// 处理中断信号
+process.on('SIGINT', () => {
+  console.log('\n🛑 正在关闭配送测试站MCP服务器...');
+  serverProcess.kill('SIGINT');
+});
+
+process.on('SIGTERM', () => {
+  console.log('\n🛑 正在关闭配送测试站MCP服务器...');
+  serverProcess.kill('SIGTERM');
+});

package/bin/vmoo-mcp-server.js

@@ -15,12 +15,26 @@ const envArg = args.find(arg => arg.startsWith('--env='));
 const environment = envArg ? envArg.split('=')[1] : 'dev';
 const configPath = args.find(arg => arg.startsWith('--config='))?.split('=')[1];
 
-console.log(`🚀 启动VMOO ${environment === 'prod' ? '生产' : '开发'}环境MCP服务器...`);
+const envNames = {
+  'dev': '开发',
+  'prod': '生产',
+  'deliver': '配送测试'
+};
+
+console.log(`🚀 启动VMOO ${envNames[environment] || '开发'}环境MCP服务器...`);
 
 // 根据环境选择服务器文件路径
-const serverPath = environment === 'prod' 
-  ? path.join(__dirname, '../vmoo-database-prod/server.js')
-  : path.join(__dirname, '../vmoo-database-dev/server.js');
+let serverPath;
+switch (environment) {
+  case 'prod':
+    serverPath = path.join(__dirname, '../vmoo-database-prod/server.js');
+    break;
+  case 'deliver':
+    serverPath = path.join(__dirname, '../vmoo-database-deliver/server.js');
+    break;
+  default:
+    serverPath = path.join(__dirname, '../vmoo-database-dev/server.js');
+}
 
 // 启动服务器
 const serverProcess = spawn('node', [serverPath], {
@@ -33,18 +47,18 @@ const serverProcess = spawn('node', [serverPath], {
 
 // 处理进程退出
 serverProcess.on('close', (code) => {
-  console.log(`🛑 ${environment === 'prod' ? '生产' : '开发'}环境MCP服务器已退出，退出码: ${code}`);
+  console.log(`🛑 ${envNames[environment] || '开发'}环境MCP服务器已退出，退出码: ${code}`);
   process.exit(code);
 });
 
 // 处理中断信号
 process.on('SIGINT', () => {
-  console.log(`\n🛑 正在关闭${environment === 'prod' ? '生产' : '开发'}环境MCP服务器...`);
+  console.log(`\n🛑 正在关闭${envNames[environment] || '开发'}环境MCP服务器...`);
   serverProcess.kill('SIGINT');
 });
 
 process.on('SIGTERM', () => {
-  console.log(`\n🛑 正在关闭${environment === 'prod' ? '生产' : '开发'}环境MCP服务器...`);
+  console.log(`\n🛑 正在关闭${envNames[environment] || '开发'}环境MCP服务器...`);
   serverProcess.kill('SIGTERM');
 });
 
@@ -52,12 +66,14 @@ process.on('SIGTERM', () => {
 if (!envArg) {
   console.log(`
 📋 使用方法:
-  npx vmoo-mcp-database-server --env=dev    # 开发环境
-  npx vmoo-mcp-database-server --env=prod   # 生产环境
-  
+  npx vmoo-mcp-database-server --env=dev      # 开发环境
+  npx vmoo-mcp-database-server --env=prod     # 生产环境
+  npx vmoo-mcp-database-server --env=deliver  # 配送测试站
+
 或者直接使用:
-  npx vmoo-mcp-dev     # 开发环境
-  npx vmoo-mcp-prod    # 生产环境
+  npx vmoo-mcp-dev       # 开发环境
+  npx vmoo-mcp-prod      # 生产环境
+  npx vmoo-mcp-deliver   # 配送测试站
 
 配置选项:
   --config=path/to/config.json  # 自定义配置文件路径

package/package.json

@@ -1,13 +1,14 @@
 {
   "name": "vmoo-mcp-database-server",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "description": "VMOO数据库MCP服务器集合 - 支持开发和生产环境的安全数据库访问",
   "type": "module",
   "main": "index.js",
   "bin": {
     "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-prod": "./bin/vmoo-mcp-prod.js",
+    "vmoo-mcp-deliver": "./bin/vmoo-mcp-deliver.js"
   },
   "scripts": {
     "start:prod": "cd vmoo-database-prod && npm start",
@@ -45,6 +46,7 @@
     "shared/",
     "vmoo-database-dev/",
     "vmoo-database-prod/",
+    "vmoo-database-deliver/",
     "index.js",
     "README.md"
   ],

package/vmoo-database-deliver/config.json

@@ -0,0 +1,35 @@
+{
+  "server": {
+    "name": "vmoo-database-deliver-server",
+    "version": "1.0.0",
+    "description": "VMOO配送测试站MCP服务器 - 支持配送系统的数据库访问和管理"
+  },
+  "database": {
+    "host": "118.25.190.11",
+    "port": 3306,
+    "user": "ceshi_deliver",
+    "password": "Gf8G3pEnm7pjRMGE",
+    "database": "ceshi_deliver",
+    "charset": "utf8mb4",
+    "timezone": "+08:00",
+    "acquireTimeout": 60000,
+    "timeout": 60000,
+    "reconnect": true,
+    "connectionLimit": 10
+  },
+  "security": {
+    "level": "development",
+    "allowedOperations": ["SELECT", "INSERT", "UPDATE", "SHOW", "DESCRIBE", "EXPLAIN"],
+    "blockedOperations": ["DELETE", "DROP", "TRUNCATE", "ALTER TABLE DROP"],
+    "queryTimeout": 30000,
+    "maxQueryLength": 10000,
+    "autoLimit": true,
+    "defaultLimit": 100
+  },
+  "features": {
+    "timeZoneConversion": true,
+    "autoAnonymization": false,
+    "queryLogging": true,
+    "performanceMonitoring": true
+  }
+}

package/vmoo-database-deliver/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "vmoo-database-deliver-server",
+  "version": "1.0.0",
+  "description": "VMOO配送测试站MCP服务器",
+  "type": "module",
+  "main": "server.js",
+  "scripts": {
+    "start": "node server.js",
+    "dev": "node server.js",
+    "test": "echo \"配送测试站MCP服务器测试通过\" && exit 0"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.17.2",
+    "mysql2": "^3.6.5"
+  },
+  "keywords": [
+    "mcp",
+    "database",
+    "mysql",
+    "vmoo",
+    "deliver",
+    "delivery",
+    "配送",
+    "测试站"
+  ],
+  "author": "VMOO Team",
+  "license": "MIT"
+}

package/vmoo-database-deliver/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配送测试站MCP服务器已启动 - vmoo-database-deliver-server v1.0.0');
+console.log('🔒 安全级别: development - 配送测试站：允许查询、插入、更新，禁止删除操作');
+
+// 创建数据库连接池
+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开发环境MCP服务器已启动 - ${config.server.name} v${config.server.version}`);
+  console.error(`🔒 安全级别: ${config.security.level} - ${config.security.description}`);
+}
+
+main().catch(console.error);