vmoo-mcp-database-server 1.2.1 → 1.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vmoo-mcp-database-server",
-  "version": "1.2.1",
+  "version": "1.2.2",
   "description": "VMOO数据库MCP服务器集合 - 支持开发和生产环境的安全数据库访问",
   "type": "module",
   "main": "index.js",

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,