chaimi-bookkeeping-mcp 2.1.9 → 2.2.1

package/README.md

@@ -111,8 +111,41 @@ A: `npm update -g chaimi-bookkeeping-mcp`
 - 邮箱：songyangx@gmail.com
 - npm 包地址：https://www.npmjs.com/package/chaimi-bookkeeping-mcp
 
+## 功能说明
+
+### 支持的记账类型
+
+| 工具 | 用途 | 示例 |
+|------|------|------|
+| `save_expense` | 单商品支出 | "午餐 35 元" |
+| `save_receipt` | 小票/多商品 | "超市购物：牛奶 10 元，面包 15 元" |
+| `save_income` | **收入记录** | "工资到账 8000 元" |
+| `get_expenses` | 查询记录 | "查看本月消费" |
+| `get_statistics` | 消费统计 | "本月花了多少" |
+
+### 收入记账示例
+
+在 WorkBuddy/OpenClaw 中输入：
+
+```
+记录工资收入 8000 元
+```
+
+或：
+
+```
+今天收到奖金 3000
+```
+
+AI 会自动调用 `save_income` 工具记录收入。
+
 ## 更新日志
 
+### v2.2.0 (2026-04-05)
+- **新增收入记账功能** (`save_income`)
+- 支持工资、奖金、红包、投资收益等收入类型
+- 支出和收入统一存储，便于统计分析
+
 ### v2.1.0 (2026-04-03)
 - 升级 OAuth 2.0 Device Flow 认证
 - 支持 npm 全局安装

package/oauth.js

@@ -69,10 +69,24 @@ class OAuthManager {
 
       console.log('✅ 获取设备码成功');
       console.log('');
-      console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
-      console.log(`  📱 验证码: ${deviceCodeRes.userCode}`);
-      console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
-      console.log('');
+      
+      // 【柴米记账授权】格式化输出，确保Agent能识别
+      const authMessage = `
+【柴米记账授权】
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+  📱 验证码: ${deviceCodeRes.userCode}
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+请在微信柴米记账小程序中输入此验证码完成授权
+`;
+      console.log(authMessage);
+      
+      // 保存授权状态，支持进程重启后复用
+      await this.saveAuthState({
+        deviceCode: deviceCodeRes.deviceCode,
+        userCode: deviceCodeRes.userCode,
+        timestamp: Date.now(),
+        status: 'pending'
+      });
 
       // 2. 根据环境选择交互方式
       if (useUrlScheme && deviceCodeRes.urlScheme) {
@@ -423,6 +437,77 @@ class OAuthManager {
   delay(ms) {
     return new Promise(resolve => setTimeout(resolve, ms));
   }
+
+  /**
+   * 保存授权状态到文件
+   * 支持进程重启后复用验证码
+   */
+  async saveAuthState(state) {
+    try {
+      const fs = require('fs').promises;
+      const path = require('path');
+      const stateFile = path.join(require('os').homedir(), '.mcporter', 'auth-state.json');
+      
+      // 确保目录存在
+      const dir = path.dirname(stateFile);
+      await fs.mkdir(dir, { recursive: true });
+      
+      // 保存状态
+      await fs.writeFile(
+        stateFile,
+        JSON.stringify(state, null, 2),
+        { mode: 0o600 }
+      );
+    } catch (err) {
+      // 保存失败不影响主流程
+      console.warn('保存授权状态失败:', err.message);
+    }
+  }
+
+  /**
+   * 加载授权状态
+   * 用于进程重启后恢复授权流程
+   */
+  async loadAuthState() {
+    try {
+      const fs = require('fs').promises;
+      const path = require('path');
+      const stateFile = path.join(require('os').homedir(), '.mcporter', 'auth-state.json');
+      
+      const data = await fs.readFile(stateFile, 'utf8');
+      return JSON.parse(data);
+    } catch (err) {
+      if (err.code === 'ENOENT') {
+        return null; // 文件不存在
+      }
+      console.warn('加载授权状态失败:', err.message);
+      return null;
+    }
+  }
+
+  /**
+   * 清除授权状态
+   * 授权成功或过期后清理
+   */
+  async clearAuthState() {
+    try {
+      const fs = require('fs').promises;
+      const path = require('path');
+      const stateFile = path.join(require('os').homedir(), '.mcporter', 'auth-state.json');
+      await fs.unlink(stateFile);
+    } catch (err) {
+      // 文件不存在或其他错误，忽略
+    }
+  }
+
+  /**
+   * 检查授权状态是否过期
+   * 默认10分钟有效期
+   */
+  isAuthStateExpired(state, maxAge = 10 * 60 * 1000) {
+    if (!state || !state.timestamp) return true;
+    return Date.now() - state.timestamp > maxAge;
+  }
 }
 
 /**

package/package.json

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

package/server.js

@@ -97,6 +97,9 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             unit: { type: 'string', description: '单位（如：斤、个、瓶）' },
             category: { type: 'string', description: '分类（如：餐饮、食品、交通）' },
             store: { type: 'string', description: '商家名称' },
+            agentType: { type: 'string', description: 'Agent类型（如：workbuddy、claude、cursor）' },
+            apiProvider: { type: 'string', description: 'AI提供商' },
+            rawInput: { type: 'string', description: '用户原始输入' },
           },
           required: ['name', 'amount', 'price', 'quantity'],
         },
@@ -134,6 +137,9 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
                 required: ['name', 'amount', 'price', 'quantity'],
               },
             },
+            agentType: { type: 'string', description: 'Agent类型（如：workbuddy、claude、cursor）' },
+            apiProvider: { type: 'string', description: 'AI提供商' },
+            rawInput: { type: 'string', description: '用户原始输入' },
           },
           required: ['store', 'items'],
         },
@@ -171,6 +177,24 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
           required: ['yearMonth'],
         },
       },
+      {
+        name: 'save_income',
+        description: '保存收入记录（工资、奖金、红包等）',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            name: { type: 'string', description: '收入来源（如：工资、奖金、红包）' },
+            amount: { type: 'number', description: '收入金额' },
+            category: { type: 'string', description: '收入分类（如：工资、奖金、投资）' },
+            store: { type: 'string', description: '付款方（如：公司名称）' },
+            note: { type: 'string', description: '备注' },
+            agentType: { type: 'string', description: 'Agent类型（如：workbuddy、claude、cursor）' },
+            apiProvider: { type: 'string', description: 'AI提供商' },
+            rawInput: { type: 'string', description: '用户原始输入' },
+          },
+          required: ['name', 'amount'],
+        },
+      },
     ],
   };
 });
@@ -179,6 +203,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
 const toolMapping = {
   'save_expense': 'addExpense',
   'save_receipt': 'addReceipt',
+  'save_income': 'addIncome',
   'get_expenses': 'getExpenses',
   'get_receipt_list': 'getExpenses',
   'get_statistics': 'getStatistics',
@@ -210,14 +235,30 @@ async function getToken() {
     cachedToken = oauthToken.accessToken;
     tokenExpireTime = oauthToken.expiresAt;
     lastRefreshTime = Date.now();
+    // 授权成功，清除授权状态
+    await oauthManager.clearAuthState();
     return cachedToken;
   } catch (err) {
-    // OAuth Token 无效，尝试重新授权
-    console.log('OAuth Token 无效，启动授权流程...');
+    // OAuth Token 无效，检查是否有未完成的授权
+    console.log('OAuth Token 无效，检查授权状态...');
+    
+    // 检查是否有未完成的授权
+    const authState = await oauthManager.loadAuthState();
+    if (authState && !oauthManager.isAuthStateExpired(authState)) {
+      console.log(`检测到未完成的授权，验证码: ${authState.userCode}`);
+      console.log('请在微信柴米记账小程序中输入此验证码完成授权');
+      console.log('授权完成后，请重新调用工具');
+      throw new Error(`授权进行中，请使用验证码 ${authState.userCode} 完成授权`);
+    }
+    
+    // 没有未完成的授权，启动新的授权流程
+    console.log('启动新的授权流程...');
     const newToken = await oauthManager.startAuthFlow();
     cachedToken = newToken.accessToken;
     tokenExpireTime = newToken.expiresAt;
     lastRefreshTime = Date.now();
+    // 授权成功，清除授权状态
+    await oauthManager.clearAuthState();
     return cachedToken;
   }
 }
@@ -270,6 +311,31 @@ async function callMcpHub(tool, params, token) {
 server.setRequestHandler(CallToolRequestSchema, async (request) => {
   const { name, arguments: args } = request.params;
 
+  // 【强制日志】测试日志是否输出 - 写入文件
+  const fs = require('fs');
+
+  // 从环境变量获取 agentType 和 apiProvider
+  const agentTypeFromEnv = process.env.AGENT_TYPE || process.env.MCP_AGENT_TYPE || '';
+  const apiProviderFromEnv = process.env.API_PROVIDER || process.env.MCP_API_PROVIDER || '';
+
+  // 如果参数中没有，使用环境变量的值
+  if (!args.agentType && agentTypeFromEnv) {
+    args.agentType = agentTypeFromEnv;
+  }
+  if (!args.apiProvider && apiProviderFromEnv) {
+    args.apiProvider = apiProviderFromEnv;
+  }
+
+  const logMsg = `
+[${new Date().toISOString()}] MCP 工具被调用
+工具名称: ${name}
+参数: ${JSON.stringify(args, null, 2)}
+环境变量 AGENT_TYPE: ${agentTypeFromEnv || '(未设置)'}
+环境变量 API_PROVIDER: ${apiProviderFromEnv || '(未设置)'}
+`;
+  fs.appendFileSync('/tmp/mcp-debug.log', logMsg);
+  console.error(logMsg);
+
   try {
     // 获取 Token
     const token = await getToken();
@@ -290,14 +356,14 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     let userMessage;
 
     // 【调试日志】打印原始传入参数
-    console.log('\n========== MCP 参数调试日志 ==========');
-    console.log('工具名称:', name);
-    console.log('原始参数 agentType:', args.agentType || '(未传入)');
-    console.log('原始参数 apiProvider:', args.apiProvider || '(未传入)');
-    console.log('原始参数 rawInput:', args.rawInput ? args.rawInput.substring(0, 50) + '...' : '(未传入)');
-    console.log('原始参数 llm_provider:', args.llm_provider || '(未传入)');
-    console.log('原始参数 store:', args.store || '(未传入)');
-    console.log('======================================\n');
+    console.error('\n========== MCP 参数调试日志 ==========');
+    console.error('工具名称:', name);
+    console.error('原始参数 agentType:', args.agentType || '(未传入)');
+    console.error('原始参数 apiProvider:', args.apiProvider || '(未传入)');
+    console.error('原始参数 rawInput:', args.rawInput ? args.rawInput.substring(0, 50) + '...' : '(未传入)');
+    console.error('原始参数 llm_provider:', args.llm_provider || '(未传入)');
+    console.error('原始参数 store:', args.store || '(未传入)');
+    console.error('======================================\n');
 
     // 根据工具类型选择处理流程
     switch (name) {
@@ -403,13 +469,25 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         const toolName = toolMapping[name];
         const mcpParams = convertParams(name, processedArgs);
         result = await callMcpHub(toolName, mcpParams, token);
-        
+
         if (result.success) {
           userMessage = `📈 统计查询成功`;
         }
         break;
       }
 
+      case 'save_income': {
+        // 收入记录直接调用 mcpHub
+        const mcpParams = convertParams('save_income', processedArgs);
+        result = await callMcpHub('addIncome', mcpParams, token);
+
+        // 格式化输出
+        if (result.success) {
+          userMessage = `✅ 收入记录成功\n\n| 来源 | 金额 | 分类 | 付款方 |\n|------|------|------|--------|\n| ${processedArgs.name} | ${processedArgs.amount}元 | ${processedArgs.category || '其他'} | ${processedArgs.store || '-'} |`;
+        }
+        break;
+      }
+
       default:
         throw new Error(`未知工具: ${name}`);
     }
@@ -583,6 +661,19 @@ function convertParams(toolName, args) {
         type: args.type,
       };
 
+    case 'save_income':
+      return {
+        name: sanitizeString(args.name, 100),
+        amount: validateAmount(args.amount),
+        category: sanitizeString(args.category, 50) || '其他',
+        store: sanitizeString(args.store, 100),
+        note: sanitizeString(args.note, 500),
+        transactionType: 'income',
+        rawInput: sanitizeString(args.rawInput, 1000),
+        agentType: args.agentType || '',
+        apiProvider: args.apiProvider || '',
+      };
+
     default:
       return args;
   }