chaimi-keep-mcp 3.2.1-beta.4 → 3.2.1-beta.6

package/README.md

@@ -148,6 +148,23 @@ AI 会自动调用 `save_income` 工具记录收入。
 
 ## 更新日志
 
+### v3.2.1-beta.7 (2026-04-26)
+- **新增** 自动检测 Agent 类型
+  - 安装时根据配置文件路径自动识别 Agent 类型
+  - 自动设置 `AGENT_TYPE` 环境变量
+  - 支持：claude-desktop、cursor、windsurf、workbuddy、openclaw、mcporter
+
+### v3.2.1-beta.6 (2026-04-26)
+- **修复** agentType 设置问题
+  - 从环境变量 `AGENT_TYPE` 或 `MCP_AGENT_TYPE` 读取
+  - 支持用户在 `mcporter.json` 中配置 agentType
+
+### v3.2.1-beta.5 (2026-04-26)
+- **修复** 复用授权时 token 获取失败问题
+  - 云函数 `getDeviceToken` 支持 userCode 查询作为备选
+  - MCP Server 轮询时同时传入 userCode 和 deviceCode
+  - 彻底解决 deviceCode 不匹配导致的授权失败
+
 ### v3.2.1-beta.4 (2026-04-26)
 - **修复** deviceCode 保存不一致问题
   - 小程序端使用授权返回的实际 deviceCode，解决复用授权时 deviceCode 不匹配

package/bin/cli.js

@@ -125,6 +125,24 @@ function writeJSON(filePath, data) {
  * 为单个 Agent 安装配置
  * 原则：只添加缺失的配置，绝不覆盖用户已有配置
  */
+/**
+ * 根据配置文件路径自动检测 Agent 类型
+ * @param {string} configPath - 配置文件路径
+ * @returns {string} Agent 类型标识
+ */
+function detectAgentTypeFromPath(configPath) {
+  const lowerPath = configPath.toLowerCase();
+  
+  if (lowerPath.includes('claude')) return 'claude-desktop';
+  if (lowerPath.includes('cursor')) return 'cursor';
+  if (lowerPath.includes('windsurf') || lowerPath.includes('codeium')) return 'windsurf';
+  if (lowerPath.includes('workbuddy')) return 'workbuddy';
+  if (lowerPath.includes('openclaw')) return 'openclaw';
+  if (lowerPath.includes('mcporter') || lowerPath.includes('.mcporter')) return 'mcporter';
+  
+  return 'unknown';
+}
+
 function installToAgent(agent) {
   try {
     // 平台检查
@@ -160,7 +178,14 @@ function installToAgent(agent) {
     }
 
     // 添加柴米记账配置
-    config.mcpServers['柴米记账'] = JSON.parse(JSON.stringify(DEFAULT_CHAIMI_CONFIG));
+    const chaimiConfig = JSON.parse(JSON.stringify(DEFAULT_CHAIMI_CONFIG));
+    
+    // 【新增】自动检测并设置 Agent 类型
+    const agentType = detectAgentTypeFromPath(agent.configPath);
+    chaimiConfig.env.AGENT_TYPE = agentType;
+    console.log(`[配置] 自动设置 Agent 类型: ${agentType} (${agent.name})`);
+    
+    config.mcpServers['柴米记账'] = chaimiConfig;
 
     // 写入配置
     writeJSON(agent.configPath, config);

package/oauth.js

@@ -64,8 +64,10 @@ class OAuthManager {
       console.error('⏳ 等待用户授权，请勿关闭窗口...');
       console.error('   （请在手机微信中完成授权操作）');
       console.error('');
+      // 【新增】同时传入 userCode，用于复用授权场景
       const token = await this.pollForToken(
         deviceCodeRes.deviceCode,
+        deviceCodeRes.userCode,
         deviceCodeRes.interval * 1000 || 5000
       );
 
@@ -172,8 +174,14 @@ class OAuthManager {
   }
 
   // 单次轮询（用于后台轮询）
-  async pollForTokenOnce(deviceCode) {
+  async pollForTokenOnce(deviceCode, userCode) {
     try {
+      const params = { deviceCode };
+      // 【新增】如果有 userCode，一并传入（用于复用授权场景）
+      if (userCode) {
+        params.userCode = userCode;
+      }
+      
       const response = await fetch(this.mcpOAuthUrl, {
         method: 'POST',
         headers: {
@@ -181,9 +189,7 @@ class OAuthManager {
         },
         body: JSON.stringify({
           tool: 'deviceToken',
-          params: {
-            deviceCode
-          }
+          params
         })
       });
 
@@ -217,17 +223,17 @@ class OAuthManager {
     }
   }
 
-  async pollForToken(deviceCode, interval) {
+  async pollForToken(deviceCode, userCode, interval) {
     const maxAttempts = 60;
     let attempts = 0;
 
     while (attempts < maxAttempts) {
       attempts++;
-
       await this.delay(interval);
 
       try {
-        const token = await this.pollForTokenOnce(deviceCode);
+        // 【新增】传入 userCode，用于复用授权场景
+        const token = await this.pollForTokenOnce(deviceCode, userCode);
         if (token) {
           return token;
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chaimi-keep-mcp",
-  "version": "3.2.1-beta.4",
+  "version": "3.2.1-beta.6",
   "description": "柴米记账 MCP Server - 支持 Claude、Cursor、OpenClaw、WorkBuddy 等 AI 工具直接记账",
   "main": "server.js",
   "bin": {

package/server.js

@@ -482,7 +482,8 @@ async function getToken() {
       } else {
         // 验证码未过期，尝试直接获取token（用户可能已授权）
         try {
-          const token = await oauthManager.pollForTokenOnce(savedAuthState.deviceCode);
+          // 【新增】同时传入 userCode，用于复用授权场景
+          const token = await oauthManager.pollForTokenOnce(savedAuthState.deviceCode, savedAuthState.userCode);
           if (token) {
             // 授权成功
             cachedToken = token.accessToken;
@@ -2110,7 +2111,8 @@ async function startAuthFlow() {
         authState.isWaiting = true;
         console.error(`🔑 使用已保存的验证码：${savedAuthState.userCode}`);
         // 启动后台轮询
-        pollForAuthInBackground(savedAuthState.deviceCode, 5000);
+        // 【新增】同时传入 userCode，用于复用授权场景
+        pollForAuthInBackground(savedAuthState.deviceCode, savedAuthState.userCode, 5000);
         return;
       }
     }
@@ -2118,7 +2120,6 @@ async function startAuthFlow() {
     // 【新增】获取或设置Agent类型
     let agentType = await getAgentType();
     if (!agentType) {
-      // 首次使用，尝试从记账参数中获取，或使用默认值
       agentType = process.env.AGENT_TYPE || 'unknown';
       await saveAgentType(agentType);
       console.error(`[Auth] 首次使用，设置Agent类型: ${agentType}`);
@@ -2152,7 +2153,8 @@ async function startAuthFlow() {
     console.error('⏳ 请在小程序中完成授权，然后再次调用');
     
     // 启动后台轮询（3分钟）
-    pollForAuthInBackground(deviceCodeRes.deviceCode, 5000);
+    // 【新增】同时传入 userCode，用于复用授权场景
+    pollForAuthInBackground(deviceCodeRes.deviceCode, deviceCodeRes.userCode, 5000);
 
   } catch (err) {
     authState.error = err.message;
@@ -2162,7 +2164,7 @@ async function startAuthFlow() {
 }
 
 // 后台轮询授权状态（3分钟）
-async function pollForAuthInBackground(deviceCode, interval) {
+async function pollForAuthInBackground(deviceCode, userCode, interval) {
   const maxAttempts = 36; // 3分钟（36次 × 5秒）
   let attempts = 0;
 
@@ -2174,7 +2176,8 @@ async function pollForAuthInBackground(deviceCode, interval) {
 
     try {
       // 调用云函数检查授权状态
-      const token = await oauthManager.pollForTokenOnce(deviceCode);
+      // 【新增】同时传入 userCode，用于复用授权场景
+      const token = await oauthManager.pollForTokenOnce(deviceCode, userCode);
       
       if (token) {
         // 授权成功

package/utils/machine-id.js

@@ -156,10 +156,67 @@ async function getMacAddresses() {
   return macAddresses.sort();
 }
 
+/**
+ * 自动检测Agent类型
+ * 根据运行环境自动识别：trae、cursor、claude-desktop等
+ * @returns {string} 检测到的Agent类型
+ */
+function detectAgentType() {
+  // 检测环境变量
+  const envVars = {
+    TRAE: process.env.TRAE || process.env.TRAEPATH,
+    CURSOR: process.env.CURSOR || process.env.CURSOR_PATH,
+    CLAUDE: process.env.CLAUDE_DESKTOP || process.env.CLAUDE_APP,
+    VSCODE: process.env.VSCODE_CWD || process.env.VSCODE_PID,
+    WINDSURF: process.env.WINDSURF || process.env.WINDSURF_PATH
+  };
+  
+  // 根据环境变量判断
+  if (envVars.TRAEPATH || process.argv.some(arg => arg.includes('trae'))) {
+    return 'trae';
+  }
+  if (envVars.CURSOR || process.argv.some(arg => arg.includes('cursor'))) {
+    return 'cursor';
+  }
+  if (envVars.CLAUDE || process.argv.some(arg => arg.includes('claude'))) {
+    return 'claude-desktop';
+  }
+  if (envVars.WINDSURF) {
+    return 'windsurf';
+  }
+  if (envVars.VSCODE) {
+    return 'vscode';
+  }
+  
+  // 检测进程信息
+  try {
+    const ppid = process.ppid;
+    // 在 macOS/Linux 上可以通过父进程检测
+    if (process.platform === 'darwin' || process.platform === 'linux') {
+      try {
+        const { execSync } = require('child_process');
+        const parentCmd = execSync(`ps -p ${ppid} -o comm=`, { encoding: 'utf8' }).trim();
+        if (parentCmd.includes('trae')) return 'trae';
+        if (parentCmd.includes('cursor')) return 'cursor';
+        if (parentCmd.includes('claude')) return 'claude-desktop';
+        if (parentCmd.includes('windsurf')) return 'windsurf';
+      } catch (e) {
+        // 忽略错误
+      }
+    }
+  } catch (e) {
+    // 忽略错误
+  }
+  
+  // 默认返回 unknown
+  return 'unknown';
+}
+
 module.exports = {
   getMachineId,
   getDeviceFingerprint,
   getAgentType,
   saveAgentType,
-  getMacAddresses
+  getMacAddresses,
+  detectAgentType
 };