chaimi-bookkeeping-mcp 3.1.13 → 3.1.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chaimi-bookkeeping-mcp",
-  "version": "3.1.13",
+  "version": "3.1.14",
   "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,25 @@ async function checkAuthInBackground() {
   }
 }
 
-// 启动后台授权流程（非阻塞）
-async function startBackgroundAuth() {
-  if (authState.isWaiting) {
-    return; // 已经在等待授权
-  }
-
+// 启动授权流程（获取验证码，不轮询，因为mcporter 30秒超时）
+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;
+      console.error(`🔑 使用已保存的验证码：${savedAuthState.userCode}`);
+      return;
+    }
 
-    // 获取设备码
+    // 获取新的设备码
     const deviceCodeRes = await oauthManager.requestDeviceCode(false);
     authState.deviceCode = deviceCodeRes.deviceCode;
     authState.userCode = deviceCodeRes.userCode;
 
-    // 保存授权状态
+    // 保存授权状态（30分钟有效）
     await oauthManager.saveAuthState({
       deviceCode: deviceCodeRes.deviceCode,
       userCode: deviceCodeRes.userCode,
@@ -828,56 +821,14 @@ async function startBackgroundAuth() {
     });
 
     console.error(`🔑 验证码：${deviceCodeRes.userCode}`);
-    console.error('⏳ 等待用户授权中...');
-
-    // 后台轮询（不阻塞）
-    pollForAuthInBackground(deviceCodeRes.deviceCode, deviceCodeRes.interval * 1000 || 5000);
+    console.error('⏳ 请在小程序中完成授权，然后再次调用');
 
   } catch (err) {
     authState.error = err.message;
-    authState.isWaiting = false;
     console.error('❌ 启动授权失败:', err.message);
   }
 }
 
-// 后台轮询授权状态（简化版，只轮询几次就停止，因为mcporter 30秒超时）
-async function pollForAuthInBackground(deviceCode, interval) {
-  const maxAttempts = 5; // 只轮询5次（25秒），因为mcporter 30秒超时
-  let attempts = 0;
-
-  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到文件
-        await oauthManager.clearAuthState();
-        console.error('✅ 授权成功！可以继续使用记账功能');
-        return;
-      }
-    } catch (err) {
-      if (err.message.includes('expired') || err.message.includes('invalid')) {
-        authState.error = err.message;
-        authState.isWaiting = false;
-        console.error('❌ 授权失败:', err.message);
-        return;
-      }
-      // 继续轮询
-    }
-  }
-
-  // 轮询结束（可能是超时或达到最大次数）
-  authState.isWaiting = false;
-  console.error('⏳ 授权检查完成，如果已授权请再次调用');
-}
-
 // 延迟函数
 function delay(ms) {
   return new Promise(resolve => setTimeout(resolve, ms));