chaimi-keep-mcp 3.3.0-beta.3 → 3.3.0-beta.5

package/oauth.js

@@ -428,28 +428,81 @@ class FileTokenStorage extends TokenStorage {
   }
 
   async save(token) {
-    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 = {
-      version: '2.0',
-      encrypted: encrypted,
-      algorithm: 'aes-256-cbc',
-      updatedAt: new Date().toISOString()
-    };
-
-    await this.fs.writeFile(
-      this.filePath,
-      JSON.stringify(data, null, 2),
-      { mode: 0o600 }
-    );
-    console.error('🔍 [DEBUG] Token 保存成功:', this.filePath);
+    try {
+      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 = {
+        version: '2.0',
+        encrypted: encrypted,
+        algorithm: 'aes-256-cbc',
+        updatedAt: new Date().toISOString()
+      };
+
+      const content = JSON.stringify(data, null, 2);
+      console.error('🔍 [DEBUG] 准备写入内容长度:', content.length, '字节');
+
+      // 【修复】使用原子写入：先写临时文件，再重命名
+      // 这样可以避免首次创建文件时的竞态条件
+      const tempFilePath = this.filePath + '.tmp';
+
+      // 写入临时文件
+      await this.fs.writeFile(
+        tempFilePath,
+        content,
+        { mode: 0o600 }
+      );
+
+      // 验证临时文件写入是否成功
+      const tempStats = await this.fs.stat(tempFilePath);
+      console.error('🔍 [DEBUG] 临时文件写入成功，大小:', tempStats.size, '字节');
+
+      if (tempStats.size === 0) {
+        throw new Error('临时文件写入后大小为 0，写入失败');
+      }
+
+      // 原子重命名（覆盖目标文件）
+      await this.fs.rename(tempFilePath, this.filePath);
+      console.error('🔍 [DEBUG] 文件重命名成功');
+
+      // 验证最终文件
+      const stats = await this.fs.stat(this.filePath);
+      console.error('🔍 [DEBUG] Token 文件保存成功');
+      console.error('🔍 [DEBUG] 文件路径:', this.filePath);
+      console.error('🔍 [DEBUG] 文件大小:', stats.size, '字节');
+
+      if (stats.size === 0) {
+        throw new Error('Token 文件大小为 0，写入失败');
+      }
+
+      // 立即读取验证
+      const readBack = await this.fs.readFile(this.filePath, 'utf8');
+      if (readBack.length === 0) {
+        throw new Error('Token 文件读取验证失败，内容为空');
+      }
+      console.error('🔍 [DEBUG] Token 文件验证成功，内容长度:', readBack.length, '字节');
+
+    } catch (error) {
+      console.error('❌ Token 保存失败:', error.message);
+      console.error('❌ 错误详情:', error);
+
+      // 清理临时文件
+      try {
+        const tempFilePath = this.filePath + '.tmp';
+        await this.fs.unlink(tempFilePath);
+        console.error('🔍 [DEBUG] 已清理临时文件');
+      } catch (cleanupError) {
+        // 忽略清理错误
+      }
+
+      throw error;
+    }
   }
 
   async load() {

package/package.json

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

package/server.js

@@ -2201,11 +2201,23 @@ async function pollForAuthInBackground(deviceCode, interval) {
         authState.isWaiting = false;
 
         // 保存token到文件
-        await oauthManager.tokenStorage.save(token);
+        try {
+          await oauthManager.tokenStorage.save(token);
+          console.error('🔍 [DEBUG] Token 保存完成');
+        } catch (error) {
+          console.error('❌ Token 保存失败:', error.message);
+          throw error;
+        }
 
         // 【新增】保存默认 Agent 名称和 deviceCode（首次记账时会获取真实名称）
-        await oauthManager.tokenStorage.saveAgentName('柴米AI助手');
-        await oauthManager.tokenStorage.saveDeviceCode(authState.deviceCode);
+        try {
+          await oauthManager.tokenStorage.saveAgentName('柴米AI助手');
+          await oauthManager.tokenStorage.saveDeviceCode(authState.deviceCode);
+          console.error('🔍 [DEBUG] Agent 名称和 deviceCode 保存完成');
+        } catch (error) {
+          console.error('❌ Agent 信息保存失败:', error.message);
+          // 这个失败不影响授权，只记录日志
+        }
 
         // 清除授权状态
         await oauthManager.clearAuthState();