chaimi-bookkeeping-mcp 3.1.13 → 3.1.15

package/bin/cli.js

@@ -25,7 +25,10 @@ const DEFAULT_CHAIMI_CONFIG = {
     MCP_OAUTH_URL: 'https://cloud1-2gfe5jhjef06b85d-1412172089.ap-shanghai.app.tcloudbase.com/mcpOAuth',
     MCP_HUB_URL: 'https://cloud1-2gfe5jhjef06b85d-1412172089.ap-shanghai.app.tcloudbase.com/mcpHub-mcp',
     MCP_PROMPT_URL: 'https://cloud1-2gfe5jhjef06b85d-1412172089.ap-shanghai.app.tcloudbase.com/mcpPrompt'
-  }
+  },
+  // 首次授权时需要常驻运行，等待用户授权（3分钟轮询）
+  // 授权成功后进程自动退出，下次调用时 mcporter 会重新启动
+  lifecycle: 'keep-alive'
 };
 
 // 支持的 Agent 列表

package/package.json

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

package/server.js

@@ -241,7 +241,7 @@ let lastRefreshTime = 0;
 async function getToken() {
   // 检查是否已授权
   if (!authState.isAuthorized) {
-    // 先检查是否有已保存的授权状态（从文件加载）
+    // 先检查是否有保存的授权状态（从文件加载）
     const savedAuthState = await oauthManager.loadAuthState();
     if (savedAuthState && savedAuthState.deviceCode) {
       // 有保存的授权状态，尝试直接获取token（用户可能已授权）
@@ -252,38 +252,28 @@ async function getToken() {
           cachedToken = token.accessToken;
           tokenExpireTime = token.expiresAt;
           authState.isAuthorized = true;
-          authState.isWaiting = false;
+          await oauthManager.tokenStorage.save(token);
           await oauthManager.clearAuthState();
           console.error('✅ 授权成功！可以继续使用记账功能');
           return cachedToken;
         }
+        // 用户还未授权，返回同一个验证码
+        authState.deviceCode = savedAuthState.deviceCode;
+        authState.userCode = savedAuthState.userCode;
+        throw new Error(`NEED_AUTH:${savedAuthState.userCode}`);
       } catch (err) {
-        // 获取失败，继续启动后台授权
+        if (err.message.startsWith('NEED_AUTH:')) {
+          throw err;
+        }
+        // 其他错误，继续获取新的验证码
         console.error('⚠️  检查授权状态失败:', err.message);
       }
     }
     
-    // 未授权，启动后台授权流程
-    if (!authState.isWaiting) {
-      await startBackgroundAuth();
-    }
-    
-    // 等待一段时间，让后台轮询有机会完成授权
-    // 最多等待10秒，每500ms检查一次
-    let waitAttempts = 0;
-    const maxWaitAttempts = 20; // 10秒
+    // 没有保存的授权状态，启动新的授权流程
+    await startAuthFlow();
     
-    while (waitAttempts < maxWaitAttempts && authState.isWaiting && !authState.isAuthorized) {
-      await delay(500);
-      waitAttempts++;
-    }
-    
-    // 再次检查是否已授权
-    if (authState.isAuthorized) {
-      return cachedToken;
-    }
-    
-    // 仍未授权，返回验证码给Agent
+    // 返回验证码给Agent
     throw new Error(`NEED_AUTH:${authState.userCode || '等待生成验证码'}`);
   }
 
@@ -805,22 +795,29 @@ async function checkAuthInBackground() {
   }
 }
 
-// 启动后台授权流程（非阻塞）
-async function startBackgroundAuth() {
-  if (authState.isWaiting) {
-    return; // 已经在等待授权
-  }
-
+// 启动授权流程（获取验证码，启动后台轮询）
+async function startAuthFlow() {
   try {
-    authState.isWaiting = true;
-    authState.error = null;
+    // 先检查是否有保存的授权状态
+    const savedAuthState = await oauthManager.loadAuthState();
+    if (savedAuthState && savedAuthState.userCode) {
+      // 有保存的验证码，直接使用
+      authState.deviceCode = savedAuthState.deviceCode;
+      authState.userCode = savedAuthState.userCode;
+      authState.isWaiting = true;
+      console.error(`🔑 使用已保存的验证码：${savedAuthState.userCode}`);
+      // 启动后台轮询
+      pollForAuthInBackground(savedAuthState.deviceCode, 5000);
+      return;
+    }
 
-    // 获取设备码
+    // 获取新的设备码
     const deviceCodeRes = await oauthManager.requestDeviceCode(false);
     authState.deviceCode = deviceCodeRes.deviceCode;
     authState.userCode = deviceCodeRes.userCode;
+    authState.isWaiting = true;
 
-    // 保存授权状态
+    // 保存授权状态（30分钟有效）
     await oauthManager.saveAuthState({
       deviceCode: deviceCodeRes.deviceCode,
       userCode: deviceCodeRes.userCode,
@@ -828,10 +825,10 @@ async function startBackgroundAuth() {
     });
 
     console.error(`🔑 验证码：${deviceCodeRes.userCode}`);
-    console.error('⏳ 等待用户授权中...');
-
-    // 后台轮询（不阻塞）
-    pollForAuthInBackground(deviceCodeRes.deviceCode, deviceCodeRes.interval * 1000 || 5000);
+    console.error('⏳ 请在小程序中完成授权，然后再次调用');
+    
+    // 启动后台轮询（3分钟）
+    pollForAuthInBackground(deviceCodeRes.deviceCode, 5000);
 
   } catch (err) {
     authState.error = err.message;
@@ -840,42 +837,53 @@ async function startBackgroundAuth() {
   }
 }
 
-// 后台轮询授权状态（简化版，只轮询几次就停止，因为mcporter 30秒超时）
+// 后台轮询授权状态（3分钟）
 async function pollForAuthInBackground(deviceCode, interval) {
-  const maxAttempts = 5; // 只轮询5次（25秒），因为mcporter 30秒超时
+  const maxAttempts = 36; // 3分钟（36次 × 5秒）
   let attempts = 0;
 
+  console.error('⏳ 等待用户授权中...（3分钟内有效）');
+
   while (attempts < maxAttempts && authState.isWaiting) {
     attempts++;
     await delay(interval);
 
     try {
+      // 调用云函数检查授权状态
       const token = await oauthManager.pollForTokenOnce(deviceCode);
+      
       if (token) {
         // 授权成功
         cachedToken = token.accessToken;
         tokenExpireTime = token.expiresAt;
         authState.isAuthorized = true;
         authState.isWaiting = false;
-        await oauthManager.tokenStorage.save(token); // 保存token到文件
+        
+        // 保存token到文件
+        await oauthManager.tokenStorage.save(token);
+        // 清除授权状态
         await oauthManager.clearAuthState();
-        console.error('✅ 授权成功！可以继续使用记账功能');
-        return;
+        
+        console.error('✅ 授权成功！下次调用将快速启动');
+        
+        // 主动退出进程，让 mcporter 下次重新启动
+        process.exit(0);
       }
     } catch (err) {
       if (err.message.includes('expired') || err.message.includes('invalid')) {
         authState.error = err.message;
         authState.isWaiting = false;
         console.error('❌ 授权失败:', err.message);
-        return;
+        process.exit(1);
       }
       // 继续轮询
     }
   }
 
-  // 轮询结束（可能是超时或达到最大次数）
+  // 3分钟超时
   authState.isWaiting = false;
-  console.error('⏳ 授权检查完成，如果已授权请再次调用');
+  console.error('⏰ 授权超时，请再次调用获取新验证码');
+  process.exit(0);
 }
 
 // 延迟函数