Merge pull request #113 from taptap/feat/inject-params-per-call

fix(proxy): inject private params on every tool call for MCP Server compatibility

Author: MikotoZero

docs/PROXY.md

@@ -1138,7 +1138,18 @@ cat config.json | node proxy.js
   "options": {
     "verbose": false,
     "reconnect_interval": 5000,
-    "request_timeout": 30000
+    "request_timeout": 30000,
+    "tool_call_timeout": 300000,
+    "reset_timeout_on_progress": true,
+    "health_check_interval": 30000,
+    "enable_cookie_sticky": true,
+    "inject_params_per_call": true,
+    "log": {
+      "root": "/tmp/taptap-mcp/logs",
+      "enabled": false,
+      "level": "info",
+      "max_days": 7
+    }
   }
 }
 ```
@@ -1166,7 +1177,17 @@ node proxy.js
 - `tenant.project_id` - 项目标识符（可选，仅用于日志）
 - `options.verbose` - 详细日志模式（默认 `false`）
 - `options.reconnect_interval` - 重连间隔（毫秒，默认 `5000`）
-- `options.request_timeout` - 请求超时（毫秒，默认 `30000`）
+- `options.request_timeout` - 请求队列超时（毫秒，默认 `30000`）
+- `options.tool_call_timeout` - 工具调用超时（毫秒，默认 `300000`，即 5 分钟）
+- `options.reset_timeout_on_progress` - 收到 progress 通知时重置超时计时器（默认 `true`）
+- `options.health_check_interval` - 健康检查间隔（毫秒，默认 `30000`）- 定期验证 Server 会话是否有效
+- `options.enable_cookie_sticky` - 启用 Cookie 会话粘性（默认 `true`）- 用于 K8s 多副本部署时的会话粘性
+- `options.inject_params_per_call` - 每次工具调用时注入私有参数（默认 `true`）- 为兼容不同 MCP Server 实现，默认每次调用都注入；如果目标 Server 支持从 Session 获取参数，可设为 `false` 以减少数据传输
+- `options.log` - 日志配置对象（可选）
+  - `options.log.root` - 日志根目录（默认 `/tmp/taptap-mcp/logs`）
+  - `options.log.enabled` - 是否启用文件日志（默认 `false`）
+  - `options.log.level` - 日志级别（默认 `info`）- 支持：emergency, alert, critical, error, warning, notice, info, debug
+  - `options.log.max_days` - 日志保留天数（默认 `7`）
 
 ### 6.5 部署场景
 
@@ -1384,6 +1405,15 @@ interface ProxyConfig {
     request_timeout?: number; // 请求队列超时（默认 30000ms）
     tool_call_timeout?: number; // Tool 调用超时（默认 300000ms，即 5 分钟）
     reset_timeout_on_progress?: boolean; // 收到 progress 通知时重置超时（默认 true）
+    health_check_interval?: number; // 健康检查间隔（默认 30000ms）
+    enable_cookie_sticky?: boolean; // 启用 Cookie 会话粘性（默认 true）
+    inject_params_per_call?: boolean; // 每次工具调用时注入私有参数（默认 true）
+    log?: {
+      root?: string; // 日志根目录（默认 /tmp/taptap-mcp/logs）
+      enabled?: boolean; // 是否启用文件日志（默认 false）
+      level?: string; // 日志级别（默认 info）
+      max_days?: number; // 日志保留天数（默认 7）
+    };
   };
 }
 ```
@@ -1419,6 +1449,15 @@ function generateProxyConfig(user: User, project: Project, macToken: MacToken):
       verbose: true, // 推荐开启详细日志
       tool_call_timeout: 300000, // Tool 调用超时 5 分钟
       reset_timeout_on_progress: true, // 收到 progress 通知时重置超时
+      health_check_interval: 30000, // 健康检查间隔 30 秒
+      enable_cookie_sticky: true, // 启用 Cookie 会话粘性
+      inject_params_per_call: true, // 每次调用都注入私有参数（推荐）
+      log: {
+        root: '/var/log/taptap-mcp', // TapCode 环境的日志目录
+        enabled: true, // 推荐开启文件日志
+        level: 'info',
+        max_days: 7,
+      },
     },
   };
 

package-lock.json

@@ -196,6 +196,38 @@ export class TapTapMCPProxy {
     return headers;
   }
 
+  /**
+   * 注入私有参数到工具调用参数中
+   *
+   * 私有参数已在初始化连接时通过 Headers 传递，此方法用于在每次工具调用时
+   * 也注入这些参数，以兼容不支持从 Session 获取参数的 MCP Server。
+   *
+   * 注入的参数（以下划线开头，表示私有）：
+   * - _mac_token: MAC 认证令牌
+   * - _user_id: 用户标识（可选）
+   * - _project_id: 项目标识（可选）
+   * - _project_path: 项目路径（可选）
+   */
+  private injectPrivateParams(args: Record<string, unknown> | undefined): Record<string, unknown> {
+    const injected: Record<string, unknown> = { ...(args || {}) };
+
+    // 注入 MAC Token（必需）
+    injected._mac_token = this.config.auth;
+
+    // 注入可选的会话上下文参数
+    if (this.config.tenant.user_id) {
+      injected._user_id = this.config.tenant.user_id;
+    }
+    if (this.config.tenant.project_id) {
+      injected._project_id = this.config.tenant.project_id;
+    }
+    if (this.config.tenant.project_path) {
+      injected._project_path = this.config.tenant.project_path;
+    }
+
+    return injected;
+  }
+
   /**
    * 连接到 TapTap MCP Server
    */
@@ -631,14 +663,16 @@ export class TapTapMCPProxy {
       return result;
     });
 
-    // 拦截 tools/call - 直接透传（参数已在 Session 创建时通过 Headers 传递）
+    // 拦截 tools/call - 注入私有参数后转发
     this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
       const { name, arguments: args } = request.params;
 
+      // 根据配置决定是否在每次调用时注入私有参数（默认注入，兼容不同 MCP Server）
+      const shouldInjectParams = this.config.options?.inject_params_per_call ?? true;
+      const finalArgs = shouldInjectParams ? this.injectPrivateParams(args) : args;
+
       if (this.config.options?.verbose) {
-        this.log('debug', `Tool call: ${name}`);
-        // 注意：MAC Token 和其他参数已在 Session 创建时通过 Headers 传递
-        // 无需每次工具调用都注入，减少数据传输量
+        this.log('debug', `Tool call: ${name} (inject_params_per_call: ${shouldInjectParams})`);
       }
 
       // 检查连接状态
@@ -650,7 +684,7 @@ export class TapTapMCPProxy {
           return new Promise((resolve, reject) => {
             this.pendingRequests.push({
               name,
-              arguments: args,
+              arguments: finalArgs,
               resolve,
               reject,
               timestamp: Date.now(),
@@ -665,12 +699,12 @@ export class TapTapMCPProxy {
         );
       }
 
-      // 透传到 TapTap Server（捕获网络错误并触发重连）
+      // 转发到 TapTap Server（捕获网络错误并触发重连）
       try {
         const result = await this.client.callTool(
           {
             name,
-            arguments: args,
+            arguments: finalArgs,
           },
           undefined, // resultSchema
           {
@@ -695,7 +729,7 @@ export class TapTapMCPProxy {
             return new Promise((resolve, reject) => {
               this.pendingRequests.push({
                 name,
-                arguments: args,
+                arguments: finalArgs,
                 resolve,
                 reject,
                 timestamp: Date.now(),

src/mcp-proxy/proxy.ts

@@ -98,6 +98,17 @@ export interface ProxyConfig {
      * 确保同一 Proxy 的所有请求被路由到同一个 MCP Server Pod
      */
     enable_cookie_sticky?: boolean;
+    /**
+     * 每次工具调用时也注入私有参数（默认 true）
+     *
+     * 私有参数（_mac_token, _user_id, _project_id, _project_path）会在两个时机传递：
+     * 1. 初始化连接时：始终通过 HTTP Headers 传递（不受此配置影响）
+     * 2. 每次工具调用时：通过工具参数注入（由此配置控制）
+     *
+     * 为兼容不同的 MCP Server 实现，默认每次调用都注入。
+     * 如果目标 Server 支持从 Session 获取这些参数，可设置为 false 以减少数据传输量。
+     */
+    inject_params_per_call?: boolean;
     /** 日志配置 */
     log?: LogConfig;
   };