chaimi-keep-mcp 3.3.0-beta.0 → 3.3.0-beta.2

package/oauth.js

@@ -431,6 +431,10 @@ class FileTokenStorage extends TokenStorage {
     const dir = this.path.dirname(this.filePath);
     await this.fs.mkdir(dir, { recursive: true });
 
+    console.error('🔍 [DEBUG] 开始保存 Token');
+    console.error('🔍 [DEBUG] 加密密钥 (前8位):', this.key.substring(0, 8));
+    console.error('🔍 [DEBUG] Token 过期时间:', token.expiresAt);
+
     // 加密 Token 后保存
     const encrypted = encryptToken(token, this.key);
     const data = {
@@ -445,6 +449,7 @@ class FileTokenStorage extends TokenStorage {
       JSON.stringify(data, null, 2),
       { mode: 0o600 }
     );
+    console.error('🔍 [DEBUG] Token 保存成功:', this.filePath);
   }
 
   async load() {
@@ -452,6 +457,10 @@ class FileTokenStorage extends TokenStorage {
       const data = await this.fs.readFile(this.filePath, 'utf8');
       const parsed = JSON.parse(data);
 
+      console.error('🔍 [DEBUG] 开始加载 Token 文件');
+      console.error('🔍 [DEBUG] 文件版本:', parsed.version);
+      console.error('🔍 [DEBUG] 加密密钥 (前8位):', this.key.substring(0, 8));
+
       // 向后兼容：检测旧版明文格式
       if (!parsed.version || parsed.version === '1.0') {
         console.error('检测到旧版 Token 格式，自动升级...');
@@ -466,11 +475,16 @@ class FileTokenStorage extends TokenStorage {
       }
 
       // 新版加密格式，解密后返回
-      return decryptToken(parsed.encrypted, this.key);
+      console.error('🔍 [DEBUG] 开始解密 Token...');
+      const decrypted = decryptToken(parsed.encrypted, this.key);
+      console.error('🔍 [DEBUG] 解密成功，Token 过期时间:', decrypted.expiresAt);
+      return decrypted;
     } catch (err) {
       if (err.code === 'ENOENT') {
+        console.error('🔍 [DEBUG] Token 文件不存在');
         return null;
       }
+      console.error('🔍 [DEBUG] 加载 Token 失败:', err.message);
       throw err;
     }
   }

package/package.json

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

package/server.js

@@ -466,7 +466,35 @@ const TOKEN_REFRESH_INTERVAL = 2 * 60 * 60 * 1000;
 let lastRefreshTime = 0;
 
 async function getToken() {
-  // 检查是否已授权
+  // 【优化1】优先检查内存中的 token 是否有效
+  if (cachedToken && tokenExpireTime > Date.now() + 5 * 60 * 1000) {
+    if (Date.now() - lastRefreshTime < TOKEN_REFRESH_INTERVAL) {
+      return cachedToken;
+    }
+  }
+
+  // 【优化2】尝试从文件加载 token（优先级高于授权状态检查）
+  try {
+    const existingToken = await oauthManager.tokenStorage.load();
+    if (existingToken && existingToken.accessToken && existingToken.expiresAt) {
+      const expiresAt = new Date(existingToken.expiresAt).getTime();
+      if (expiresAt > Date.now() + 5 * 60 * 1000) {
+        // Token 有效，直接使用
+        cachedToken = existingToken.accessToken;
+        tokenExpireTime = expiresAt;
+        authState.isAuthorized = true;
+        lastRefreshTime = Date.now();
+        console.error('✅ 已从文件加载有效 Token，无需重新授权');
+        return cachedToken;
+      } else {
+        console.error('⏰ 保存的 Token 已过期，需要重新授权');
+      }
+    }
+  } catch (err) {
+    console.error('⚠️ 加载保存的 Token 失败:', err.message);
+  }
+
+  // 【优化3】Token 不存在或已过期，检查是否需要授权
   if (!authState.isAuthorized) {
     // 先检查是否有保存的授权状态（从文件加载）
     const savedAuthState = await oauthManager.loadAuthState();
@@ -507,20 +535,15 @@ async function getToken() {
         }
       }
     }
-    
+
     // 没有保存的授权状态或已过期，启动新的授权流程
     await startAuthFlow();
-    
+
     // 返回验证码给Agent
     throw new Error(`NEED_AUTH:${authState.userCode || '等待生成验证码'}`);
   }
 
-  if (cachedToken && tokenExpireTime > Date.now() + 5 * 60 * 1000) {
-    if (Date.now() - lastRefreshTime < TOKEN_REFRESH_INTERVAL) {
-      return cachedToken;
-    }
-  }
-
+  // 【优化4】已授权，尝试刷新 token
   if (!oauthManager) {
     throw new Error('OAuth 管理器未初始化，请检查 MCP_OAUTH_URL 配置');
   }
@@ -2158,21 +2181,23 @@ async function pollForAuthInBackground(deviceCode, interval) {
         tokenExpireTime = token.expiresAt;
         authState.isAuthorized = true;
         authState.isWaiting = false;
-        
+
         // 保存token到文件
         await oauthManager.tokenStorage.save(token);
-        
+
         // 【新增】保存默认 Agent 名称和 deviceCode（首次记账时会获取真实名称）
         await oauthManager.tokenStorage.saveAgentName('柴米AI助手');
         await oauthManager.tokenStorage.saveDeviceCode(authState.deviceCode);
-        
+
         // 清除授权状态
         await oauthManager.clearAuthState();
-        
-        console.error('✅ 授权成功！下次调用将快速启动');
-        
-        // 主动退出进程，让 mcporter 下次重新启动
-        process.exit(0);
+
+        console.error('✅ 授权成功！可以继续使用记账功能');
+
+        // 【修复】不退出进程，让 Agent 可以在同一会话中继续执行记账操作
+        // 这样 Agent 调用 get_skill 授权后，可以立即调用 save_expense 记账
+        // process.exit(0);  // 注释掉，避免授权成功后进程退出
+        return; // 直接返回，停止轮询
       }
     } catch (err) {
       if (err.message.includes('expired') || err.message.includes('invalid')) {
@@ -2187,8 +2212,9 @@ async function pollForAuthInBackground(deviceCode, interval) {
 
   // 3分钟超时
   authState.isWaiting = false;
-  console.error('⏰ 授权超时，请再次调用获取新验证码');
-  process.exit(0);
+  console.error('⏰ 授权超时（3分钟），请重新调用获取新验证码');
+  // 【修复】不退出进程，让用户可以重新尝试
+  // process.exit(0);
 }
 
 // 延迟函数