chaimi-keep-mcp 3.1.24

package/server.js

@@ -0,0 +1,1064 @@
+#!/usr/bin/env node
+/**
+ * 柴米记账 MCP Server (Node.js 版本)
+ * 适配微信云函数 mcpHub
+ * 支持 Claude Desktop、Cursor、WorkBuddy、OpenClaw
+ */
+
+// 加载 .env 文件
+try {
+  require('dotenv').config();
+} catch (e) {
+  // dotenv 未安装，忽略
+}
+
+const { Server } = require('@modelcontextprotocol/sdk/server/index.js');
+const { StdioServerTransport } = require('@modelcontextprotocol/sdk/server/stdio.js');
+const {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+} = require('@modelcontextprotocol/sdk/types.js');
+const path = require('path');
+const os = require('os');
+const fs = require('fs');
+const crypto = require('crypto');
+
+// 读取 package.json 获取版本
+const packageJson = JSON.parse(fs.readFileSync(path.join(__dirname, 'package.json'), 'utf8'));
+const MCP_VERSION = packageJson.version;
+
+// 导入 OAuth 模块
+const { OAuthManager, FileTokenStorage } = require('./oauth.js');
+
+// API 签名密钥（用于验证请求来自合法 MCP Server）
+const CHAIMI_API_SECRET = 'chaimi-mcp-secret-2024';
+
+// URL 加密密钥
+const URL_ENCRYPT_KEY = 'chaimi-url-key-2024';
+
+// 加密的云函数 URL
+const ENCRYPTED_URLS = {
+  hub: 'enc:v1:0b1c15191e53025a1100421e0148000057545156020903080f1d431054180f484819030203035158595043085d5801044c0502114c5b1e53441346150a01065811100d5e0e4b1a425f1f5f571320140b40044e05',
+  oauth: 'enc:v1:0b1c15191e53025a1100421e0148000057545156020903080f1d431054180f484819030203035158595043085d5801044c0502114c5b1e53441346150a01065811100d5e0e4b1a425f1f5f571327201c1901',
+  prompt: 'enc:v1:0b1c15191e53025a1100421e0148000057545156020903080f1d431054180f484819030203035158595043085d5801044c0502114c5b1e53441346150a01065811100d5e0e4b1a425f1f5f5713381306001959'
+};
+
+// 解密 URL 函数
+function decryptUrl(encrypted, key) {
+  const hex = encrypted.replace('enc:v1:', '');
+  let result = '';
+  for (let i = 0; i < hex.length; i += 2) {
+    const byte = parseInt(hex.substr(i, 2), 16);
+    result += String.fromCharCode(byte ^ key.charCodeAt((i / 2) % key.length));
+  }
+  return result;
+}
+
+// 配置 - 微信云函数（运行时解密）
+const MCP_HUB_URL = process.env.MCP_HUB_URL || decryptUrl(ENCRYPTED_URLS.hub, URL_ENCRYPT_KEY);
+const MCP_PROMPT_URL = process.env.MCP_PROMPT_URL || decryptUrl(ENCRYPTED_URLS.prompt, URL_ENCRYPT_KEY);
+const MCP_OAUTH_URL = process.env.MCP_OAUTH_URL || decryptUrl(ENCRYPTED_URLS.oauth, URL_ENCRYPT_KEY);
+
+// Token 缓存
+let cachedToken = null;
+let tokenExpireTime = 0;
+
+// OAuth 管理器实例
+let oauthManager = null;
+
+// 初始化 OAuth 管理器
+function initOAuthManager() {
+  const tokenStorage = new FileTokenStorage(
+    path.join(os.homedir(), '.mcporter', 'oauth-token.json')
+  );
+
+  oauthManager = new OAuthManager({
+    mcpOAuthUrl: MCP_OAUTH_URL,
+    tokenStorage: tokenStorage,
+    // ⚠️ 重要：此回调在用户首次使用时显示验证码，切勿删除或清空！
+    // 如果删除，用户将无法看到授权验证码，导致无法使用 MCP Server
+    // 参见 FUNCTION_REGISTRY.md 中的 "验证码显示" 功能
+    // 注意：使用 console.error，避免污染 stdout（MCP 协议通信使用 stdout）
+    onQrCode: (qrData) => {
+      // 输出授权信息到控制台（用户可见）
+      console.error('\n' + '='.repeat(60));
+      console.error('  柴米AI记账 MCP Server - 首次使用需要授权');
+      console.error('='.repeat(60));
+      console.error('');
+      console.error(`  验证码：${qrData.userCode}`);
+      console.error('');
+      console.error('  请在"柴米AI记账"小程序中完成授权：');
+      console.error('');
+      console.error('  1. 打开"柴米AI记账"小程序');
+      console.error('  2. 点击"我的" → "🤖 Agent 授权"');
+      console.error(`  3. 输入验证码：${qrData.userCode}`);
+      console.error('');
+      console.error('='.repeat(60) + '\n');
+    },
+    // ⚠️ 重要：此回调在授权成功后通知用户，切勿删除！
+    onTokenReady: (token) => {
+      console.error('✅ 授权成功！Token 已保存。');
+    }
+  });
+
+  return oauthManager;
+}
+
+// 创建 MCP Server
+const server = new Server(
+  {
+    name: 'chaimi-keep-mcp',
+    version: MCP_VERSION,
+  },
+  {
+    capabilities: {
+      tools: {},
+    },
+  }
+);
+
+// 工具列表
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+  return {
+    tools: [
+      {
+        name: 'save_expense',
+        description: '保存单商品消费记录（AI文字记账）。只需提供商品名称和金额，其他参数自动填充。示例：name="午餐", amount=35',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            name: { type: 'string', description: '商品名称（必填）' },
+            amount: { type: 'number', description: '金额（必填）' },
+            category: { type: 'string', description: '分类（可选，如：餐饮、食品、交通）' },
+            store: { type: 'string', description: '商家名称（可选）' },
+            date: { type: 'string', description: '消费时间（ISO格式）（可选）' },
+            note: { type: 'string', description: '备注（可选）' },
+            agentType: { type: 'string', description: '【推荐自动填充】Agent类型，如：claude-desktop、cursor、openclaw、workbuddy、trae' },
+            apiProvider: { type: 'string', description: '【推荐自动填充】AI服务提供商，如：anthropic、openai、doubao、aliyun' },
+            rawInput: { type: 'string', description: '【推荐自动填充】用户的原始输入内容，用于记录用户原始请求' },
+            mcp_version: { type: 'string', description: 'MCP Server 版本号（自动填充）' },
+          },
+          required: ['name', 'amount'],
+        },
+      },
+      {
+        name: 'save_receipt',
+        description: '【v3.1.21】【图片小票专用】保存购物小票/发票/收据。⚠️ 重要：items数组中每个商品必须包含完整的4个字段：name（商品名称）、amount（金额=单价×数量）、price（单价）、quantity（数量）。示例：[{"name":"苹果","amount":5.5,"price":5.5,"quantity":1}]',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            store: { type: 'string', description: '商家名称（必填）' },
+            date: { type: 'string', description: '消费时间（ISO 8601 格式，必须包含日期和时间，如：2026-04-10T13:06:21）' },
+            totalAmount: { type: 'number', description: '总金额（所有商品amount之和）' },
+            originalAmount: { type: 'number', description: '应付金额（优惠前）' },
+            discountAmount: { type: 'number', description: '优惠金额' },
+            actualAmount: { type: 'number', description: '实付金额（优惠后）' },
+            paymentMethod: { type: 'string', description: '支付方式，如：微信支付、支付宝' },
+            receiptNo: { type: 'string', description: '小票编号' },
+            items: {
+              type: 'array',
+              description: '【必填】商品列表，必须是数组格式。每个商品必须包含：name、amount、price、quantity。⚠️ 注意：amount=price×quantity',
+              items: {
+                type: 'object',
+                properties: {
+                  name: { type: 'string', description: '【必填】商品名称' },
+                  originalName: { type: 'string', description: '原始商品名称（含规格）' },
+                  price: { type: 'number', description: '【必填】单价' },
+                  quantity: { type: 'number', description: '【必填】数量' },
+                  unit: { type: 'string', description: '单位，如：斤、个、份' },
+                  amount: { type: 'number', description: '【必填】金额，必须等于 price × quantity' },
+                  weight: { type: 'string', description: '重量' },
+                  marketPrice: { type: 'string', description: '市场单价（元/500g）' },
+                  category: { type: 'string', description: '分类，如：餐饮、食品、购物' },
+                },
+                required: ['name', 'amount', 'price', 'quantity'],
+              },
+            },
+            agentType: { type: 'string', description: '【自动填充】Agent类型：claude-desktop、cursor、openclaw、workbuddy、trae' },
+            apiProvider: { type: 'string', description: '【自动填充】AI服务提供商：anthropic、openai、doubao、aliyun' },
+            rawInput: { type: 'string', description: '【自动填充】用户的原始输入内容' },
+            mcp_version: { type: 'string', description: '【自动填充】MCP Server版本号' },
+          },
+          required: ['store', 'items'],
+        },
+      },
+      {
+        name: 'get_expenses',
+        description: '查询消费记录',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            limit: { type: 'number', description: '返回数量限制', default: 10 },
+            source: { type: 'string', description: '来源筛选' },
+          },
+        },
+      },
+      {
+        name: 'get_receipt_list',
+        description: '获取小票列表',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            limit: { type: 'number', description: '返回数量限制', default: 5 },
+          },
+        },
+      },
+      {
+        name: 'get_statistics',
+        description: '获取消费统计',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            yearMonth: { type: 'string', description: '年月（如：2026-03）' },
+            type: { type: 'string', description: '统计类型：item/receipt/category' },
+          },
+          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: '付款方（如：公司名称）' },
+            date: { type: 'string', description: '收入时间（ISO格式）（可选）' },
+            note: { type: 'string', description: '备注' },
+            agentType: { type: 'string', description: '【推荐自动填充】Agent类型，如：claude-desktop、cursor、openclaw、workbuddy、trae' },
+            apiProvider: { type: 'string', description: '【推荐自动填充】AI服务提供商，如：anthropic、openai、doubao、aliyun' },
+            rawInput: { type: 'string', description: '【推荐自动填充】用户的原始输入内容，用于记录用户原始请求' },
+            mcp_version: { type: 'string', description: 'MCP Server 版本号（自动填充）' },
+          },
+          required: ['name', 'amount'],
+        },
+      },
+      {
+        name: 'get_parse_prompt',
+        description: '获取小票图片解析的Prompt模板，用于指导大模型如何格式化输出小票信息。返回的Prompt应作为system message，配合小票图片作为user message调用大模型',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            type: { type: 'string', description: 'Prompt类型，如：parseReceipt', default: 'parseReceipt' },
+          },
+        },
+      },
+      {
+        name: 'submit_feedback',
+        description: '提交反馈、建议或问题报告。当用户遇到记账问题、有功能建议或发现异常时，可使用此工具提交反馈。反馈类型包括：bug(问题报告)、feature(功能建议)、improvement(优化建议)、other(其他)。提交成功后会返回反馈编号，可用于查询处理进度。注意：反馈内容不要超过150个字符，超过部分会被截断',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            content: { type: 'string', description: '反馈内容，详细描述问题或建议（必填，最多150个字符）' },
+            type: { type: 'string', description: '反馈类型：bug(问题报告)、feature(功能建议)、improvement(优化建议)、other(其他)，默认other', enum: ['bug', 'feature', 'improvement', 'other'], default: 'other' },
+            contact: { type: 'string', description: '联系方式（可选），如邮箱、微信等，方便后续沟通' },
+            context: { type: 'string', description: '上下文信息（可选），如相关记账记录ID、错误截图描述等' },
+            agentType: { type: 'string', description: '【推荐自动填充】Agent类型，如：claude-desktop、cursor、openclaw、workbuddy、trae' },
+            apiProvider: { type: 'string', description: '【推荐自动填充】AI服务提供商，如：anthropic、openai、doubao、aliyun' },
+          },
+          required: ['content'],
+        },
+      },
+    ],
+  };
+});
+
+// 工具名称映射：MCP Tool -> mcpHub Tool
+const toolMapping = {
+  'save_expense': 'addExpense',
+  'save_receipt': 'addReceipt',
+  'save_income': 'addIncome',
+  'get_expenses': 'getExpenses',
+  'get_receipt_list': 'getExpenses',
+  'get_statistics': 'getStatistics',
+  'submit_feedback': 'addFeedback',
+};
+
+// 获取或刷新 Token（OAuth 2.0 Device Flow）
+const TOKEN_REFRESH_INTERVAL = 2 * 60 * 60 * 1000;
+let lastRefreshTime = 0;
+
+async function getToken() {
+  // 检查是否已授权
+  if (!authState.isAuthorized) {
+    // 先检查是否有保存的授权状态（从文件加载）
+    const savedAuthState = await oauthManager.loadAuthState();
+    if (savedAuthState && savedAuthState.deviceCode) {
+      // 检查验证码是否已过期
+      if (oauthManager.isAuthStateExpired(savedAuthState)) {
+        console.error('⏰ 保存的验证码已过期，清除并生成新的验证码');
+        await oauthManager.clearAuthState();
+        // 继续执行，生成新的验证码
+      } else {
+        // 验证码未过期，尝试直接获取token（用户可能已授权）
+        try {
+          const token = await oauthManager.pollForTokenOnce(savedAuthState.deviceCode);
+          if (token) {
+            // 授权成功
+            cachedToken = token.accessToken;
+            tokenExpireTime = token.expiresAt;
+            authState.isAuthorized = true;
+            await oauthManager.tokenStorage.save(token);
+            await oauthManager.clearAuthState();
+            console.error('✅ 授权成功！可以继续使用记账功能');
+            // 返回特殊标记，让上层知道这是刚完成授权的情况
+            throw new Error(`AUTH_JUST_COMPLETED:${cachedToken}`);
+          }
+          // 用户还未授权，返回同一个验证码
+          authState.deviceCode = savedAuthState.deviceCode;
+          authState.userCode = savedAuthState.userCode;
+          throw new Error(`NEED_AUTH:${savedAuthState.userCode}`);
+        } catch (err) {
+          if (err.message.startsWith('NEED_AUTH:') || err.message.startsWith('AUTH_JUST_COMPLETED:')) {
+            throw err;
+          }
+          // 其他错误，继续获取新的验证码
+          console.error('⚠️  检查授权状态失败:', err.message);
+        }
+      }
+    }
+    
+    // 没有保存的授权状态或已过期，启动新的授权流程
+    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;
+    }
+  }
+
+  if (!oauthManager) {
+    throw new Error('OAuth 管理器未初始化，请检查 MCP_OAUTH_URL 配置');
+  }
+
+  try {
+    const oauthToken = await oauthManager.getValidToken();
+    cachedToken = oauthToken.accessToken;
+    tokenExpireTime = oauthToken.expiresAt;
+    lastRefreshTime = Date.now();
+    await oauthManager.clearAuthState();
+    return cachedToken;
+  } catch (err) {
+    // Token 获取失败，可能需要重新授权
+    authState.isAuthorized = false;
+    authState.isWaiting = false;
+    throw new Error(`NEED_AUTH:授权已过期，请重新授权`);
+  }
+}
+
+// 调用 mcpPrompt 云函数（获取 Prompt 或校验数据）
+async function callMcpPrompt(tool, params, token) {
+  const response = await fetch(MCP_PROMPT_URL, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${token}`,
+    },
+    body: JSON.stringify({
+      tool: tool,
+      params: params,
+    }),
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`mcpPrompt 调用失败: ${errorText}`);
+  }
+
+  return await response.json();
+}
+
+// 生成请求签名（用于验证请求来自合法 MCP Server）
+function generateSignature(body, timestamp, secret) {
+  const payload = JSON.stringify(body) + timestamp;
+  return crypto
+    .createHmac('sha256', secret)
+    .update(payload)
+    .digest('hex');
+}
+
+// 调用 mcpHub 云函数
+async function callMcpHub(tool, params, token) {
+  const body = {
+    tool: tool,
+    params: params,
+  };
+  const timestamp = Date.now().toString();
+  const signature = generateSignature(body, timestamp, CHAIMI_API_SECRET);
+
+  const response = await fetch(MCP_HUB_URL, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${token}`,
+      'X-Chaimi-Signature': signature,
+      'X-Chaimi-Timestamp': timestamp,
+    },
+    body: JSON.stringify(body),
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`mcpHub 调用失败: ${errorText}`);
+  }
+
+  return await response.json();
+}
+
+// 工具调用处理
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+
+  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;
+  }
+
+  try {
+    const token = await getToken();
+
+    let processedArgs = { ...args };
+    if (processedArgs.items && typeof processedArgs.items === 'string') {
+      try {
+        processedArgs.items = JSON.parse(processedArgs.items);
+      } catch (e) {
+      }
+    }
+
+    let result;
+    let userMessage;
+
+    switch (name) {
+      case 'save_expense': {
+        const mcpParams = convertParams('save_expense', processedArgs);
+        result = await callMcpHub('addExpense', mcpParams, token);
+        
+        if (result.success) {
+          const displayName = processedArgs.name || '未知商品';
+          const displayAmount = processedArgs.amount || 0;
+          const displayCategory = processedArgs.category || '其他';
+          const displayStore = processedArgs.store || '-';
+          userMessage = `✅ 记账成功\n\n| 商品 | 金额 | 分类 | 商家 |\n|------|------|------|------|\n| ${displayName} | ${displayAmount}元 | ${displayCategory} | ${displayStore} |\n\n已保存到你的柴米记账小程序数据库。`;
+          
+          if (!processedArgs.agentType || !processedArgs.apiProvider) {
+            userMessage += '\n\n💡 提示：传递agentType和apiProvider参数可解锁小程序记录来源统计功能，示例：agentType="openclaw", apiProvider="doubao"';
+          }
+        }
+        break;
+      }
+
+      case 'save_receipt': {
+        // 0. 参数格式强制检查 - 确保 items 中每个商品都有完整的字段
+        if (processedArgs.items && Array.isArray(processedArgs.items)) {
+          const invalidItems = [];
+          processedArgs.items.forEach((item, index) => {
+            const missingFields = [];
+            if (!item.hasOwnProperty('name')) missingFields.push('name');
+            if (!item.hasOwnProperty('amount')) missingFields.push('amount');
+            if (!item.hasOwnProperty('price')) missingFields.push('price');
+            if (!item.hasOwnProperty('quantity')) missingFields.push('quantity');
+            
+            if (missingFields.length > 0) {
+              invalidItems.push(`商品${index + 1}(${item.name || '未命名'})缺少字段: ${missingFields.join(', ')}`);
+            }
+          });
+          
+          if (invalidItems.length > 0) {
+            result = {
+              success: false,
+              error: `参数格式错误：${invalidItems.join('; ')}。请更新您的 MCP Skill 或检查 items 数组格式。每个商品必须包含：name, amount, price, quantity`,
+              code: 400
+            };
+            userMessage = `❌ 保存失败\n\n错误：商品数据格式不完整\n\n${invalidItems.join('\n')}\n\n💡 解决方案：\n1. 请更新您的柴米记账 MCP Skill\n2. 确保每个商品包含完整的4个字段：name, amount, price, quantity\n3. amount 必须等于 price × quantity`;
+            break;
+          }
+        }
+        
+        // 1. 调用 parseReceipt 重新提取小票信息（覆盖 Agent 传的数据）
+        if (processedArgs.rawInput) {
+          const parseResult = await callMcpPrompt(
+            'parseReceipt',
+            {
+              text: processedArgs.rawInput,
+              type: 'parseReceipt'
+            },
+            token
+          );
+          
+          if (parseResult.success && parseResult.data) {
+            // 用 parseReceipt 提取的信息覆盖 Agent 传的参数
+            processedArgs = {
+              ...processedArgs,
+              ...parseResult.data
+            };
+          }
+        }
+        
+        // 2. 调用 fillDefaults 补充默认值
+        const fillResult = await callMcpPrompt(
+          'fillDefaults',
+          {
+            data: processedArgs,
+            type: 'parseReceipt'
+          },
+          token
+        );
+        
+        if (fillResult.success) {
+          processedArgs = {
+            ...fillResult.data,
+            ...processedArgs
+          };
+        }
+        
+        const mcpParams = convertParams('save_receipt', processedArgs);
+        result = await callMcpHub('addReceipt', mcpParams, token);
+        
+        if (result.success) {
+          const totalAmount = processedArgs.totalAmount || processedArgs.items?.reduce((sum, item) => sum + (item.amount || 0), 0) || 0;
+          const itemsList = processedArgs.items?.map(item => `• ${item.name || '未知商品'} - ${item.amount || 0}元`).join('\n') || '暂无商品明细';
+          const storeName = processedArgs.store || '-';
+          const category = processedArgs.items?.[0]?.category || '其他';
+          userMessage = `✅ 小票记录成功\n\n| 项目 | 内容 |\n|------|------|\n| 商家 | ${storeName} |\n| 金额 | ${totalAmount}元 |\n| 分类 | ${category} |\n\n商品明细：\n${itemsList}\n\n已保存到你的柴米记账小程序数据库。`;
+          
+          if (!processedArgs.agentType || !processedArgs.apiProvider) {
+            userMessage += '\n\n💡 提示：传递agentType和apiProvider参数可解锁小程序记录来源统计功能，示例：agentType="openclaw", apiProvider="doubao"';
+          }
+        }
+        break;
+      }
+
+      case 'get_expenses':
+      case 'get_receipt_list': {
+        const toolName = toolMapping[name];
+        const mcpParams = convertParams(name, processedArgs);
+        result = await callMcpHub(toolName, mcpParams, token);
+        
+        if (result.success) {
+          userMessage = `📊 消费记录查询成功\n共找到 ${result.data?.length || 0} 条记录`;
+        }
+        break;
+      }
+
+      case 'get_statistics': {
+        const toolName = toolMapping[name];
+        const mcpParams = convertParams(name, processedArgs);
+        result = await callMcpHub(toolName, mcpParams, token);
+
+        if (result.success) {
+          userMessage = `📈 统计查询成功`;
+        }
+        break;
+      }
+
+      case 'save_income': {
+        const mcpParams = convertParams('save_income', processedArgs);
+        result = await callMcpHub('addIncome', mcpParams, token);
+
+        if (result.success) {
+          const displayName = processedArgs.name || '未知来源';
+          const displayAmount = processedArgs.amount || 0;
+          const displayCategory = processedArgs.category || '其他';
+          const displayStore = processedArgs.store || '-';
+          userMessage = `✅ 收入记录成功\n\n| 来源 | 金额 | 分类 | 付款方 |\n|------|------|------|--------|\n| ${displayName} | ${displayAmount}元 | ${displayCategory} | ${displayStore} |`;
+        }
+        break;
+      }
+
+      case 'get_parse_prompt': {
+        const promptType = args.type || 'parseReceipt';
+        const promptResult = await callMcpPrompt('getPrompt', { type: promptType }, token);
+
+        if (promptResult.success) {
+          result = {
+            success: true,
+            data: {
+              type: promptType,
+              version: promptResult.data.version,
+              systemPrompt: promptResult.data.systemPrompt,
+              userPromptTemplate: promptResult.data.userPromptTemplate,
+              examples: promptResult.data.examples,
+              instructions: '请将 systemPrompt 作为 system message，把小票图片作为 user message，调用你的大模型进行解析。解析完成后，调用 save_receipt 工具保存结果。'
+            }
+          };
+          userMessage = `✅ 获取解析Prompt成功\n\n版本: ${promptResult.data.version}\n\n## System Prompt\n\n请将以下内容作为 system message 发送给大模型：\n\n\`\`\`\n${promptResult.data.systemPrompt}\n\`\`\`\n\n## 使用说明\n\n1. 将上面的 System Prompt 作为 system message\n2. 将小票图片作为 user message\n3. 调用大模型解析\n4. 解析完成后，调用 save_receipt 工具保存结果\n\n## 示例\n\n${JSON.stringify(promptResult.data.examples, null, 2)}`;
+        } else {
+          result = { success: false, error: promptResult.error };
+          userMessage = `❌ 获取Prompt失败：${promptResult.error}`;
+        }
+        break;
+      }
+
+      case 'submit_feedback': {
+        const feedbackData = {
+          content: processedArgs.content,
+          feedType: processedArgs.type || 'other',
+          contact: processedArgs.contact || '',
+          context: processedArgs.context || '',
+          agentType: processedArgs.agentType || '',
+          apiProvider: processedArgs.apiProvider || '',
+          mcp_version: MCP_VERSION,
+          userAgent: request.headers?.['user-agent'] || ''
+        };
+
+        result = await callMcpHub('addFeedback', feedbackData, token);
+
+        if (result.success) {
+          const typeText = {
+            bug: '问题报告',
+            feature: '功能建议',
+            improvement: '优化建议',
+            other: '其他反馈'
+          }[feedbackData.feedType] || '其他反馈';
+
+          userMessage = `✅ 反馈提交成功！\n\n感谢您的反馈，我们会尽快处理。\n\n━━━━━━━━━━━━━━\n📋 反馈编号：${result.data.feedbackId}\n📂 类型：${typeText}\n⏰ 时间：${new Date().toLocaleString('zh-CN')}\n━━━━━━━━━━━━━━\n\n💡 提示：\n- 反馈编号可用于查询处理进度\n- 如需补充信息，可再次提交并备注原编号\n- 我们会在小程序客服中回复您`;
+        } else {
+          userMessage = `❌ 反馈提交失败\n\n错误信息：${result.error || '未知错误'}\n\n💡 建议：\n- 检查网络连接后重试\n- 如持续失败，请稍后重试或联系客服`;
+        }
+        break;
+      }
+
+      default:
+        throw new Error(`未知工具: ${name}`);
+    }
+
+    if (!result.success) {
+      userMessage = `❌ 操作失败：${result.error || '未知错误'}`;
+    }
+
+    userMessage += `\n\n---\n📦 柴米记账 MCP v${MCP_VERSION}`;
+
+    return {
+      content: [
+        {
+          type: 'text',
+          text: userMessage,
+        },
+      ],
+    };
+  } catch (error) {
+    // 处理授权错误
+    if (error.message.startsWith('NEED_AUTH:')) {
+      const userCode = error.message.replace('NEED_AUTH:', '');
+      
+      if (userCode && userCode !== '等待生成验证码') {
+        return {
+          content: [
+            {
+              type: 'text',
+              text: `🔐 首次使用需要授权
+
+━━━━━━━━━━━━━━
+📱 验证码：**${userCode}**
+━━━━━━━━━━━━━━
+
+请在"柴米AI记账"小程序中完成授权：
+
+1️⃣ 打开微信小程序"柴米AI记账"
+2️⃣ 点击"我的" → "🤖 Agent 授权"
+3️⃣ 输入验证码：**${userCode}**
+4️⃣ 点击确认授权
+
+⚠️ 重要提示：
+• 授权完成后，MCP Server 会自动退出（正常现象）
+• 请重新发送您的记账指令，我才能为您完成记账
+• 首次授权后，后续记账将无需再次授权
+
+⏳ 验证码有效期：30分钟`,
+            },
+          ],
+        };
+      } else {
+        return {
+          content: [
+            {
+              type: 'text',
+              text: `⏳ 正在准备授权，请稍等几秒后重试...`,
+            },
+          ],
+        };
+      }
+    }
+    
+    // 处理刚完成授权的情况
+    if (error.message.startsWith('AUTH_JUST_COMPLETED:')) {
+      return {
+        content: [
+          {
+            type: 'text',
+            text: `✅ 授权成功！
+
+━━━━━━━━━━━━━━
+🎉 恭喜！您已完成授权
+━━━━━━━━━━━━━━
+
+现在可以正常使用记账功能了！
+
+请重新发送您的记账指令，例如：
+• "记录一笔午餐 35元"
+• "保存小票，商家是沃尔玛，商品有..."
+
+首次授权后，后续记账将无需再次授权。`,
+          },
+        ],
+      };
+    }
+    
+    return {
+      content: [
+        {
+          type: 'text',
+          text: `Error: ${error.message}`,
+        },
+      ],
+      isError: true,
+    };
+  }
+});
+
+function sanitizeString(str, maxLength = 200) {
+  if (!str || typeof str !== 'string') return '';
+  return str.replace(/<[^>]*>/g, '').substring(0, maxLength);
+}
+
+function validateAmount(amount) {
+  const num = parseFloat(amount);
+  return isNaN(num) || num < 0 ? 0 : num;
+}
+
+function validateQuantity(quantity) {
+  const num = parseFloat(quantity);
+  return isNaN(num) || num <= 0 ? 1 : num;
+}
+
+function extractWeightInGrams(weightStr) {
+  if (!weightStr || typeof weightStr !== 'string') return 0;
+  
+  const match = weightStr.match(/(\d+\.?\d*)\s*(g|kg|斤|克|千克)/i);
+  if (!match) return 0;
+  
+  const value = parseFloat(match[1]);
+  const unit = match[2].toLowerCase();
+  
+  switch (unit) {
+    case 'g':
+    case '克':
+      return value;
+    case 'kg':
+    case '千克':
+      return value * 1000;
+    case '斤':
+      return value * 500;
+    default:
+      return value;
+  }
+}
+
+function calculateMarketPrice(amount, weightStr) {
+  const amountNum = parseFloat(amount);
+  const weightInGrams = extractWeightInGrams(weightStr);
+  
+  if (isNaN(amountNum) || amountNum <= 0 || weightInGrams <= 0) {
+    return '';
+  }
+  
+  const pricePer500g = (amountNum / weightInGrams) * 500;
+  return `${pricePer500g.toFixed(2)}元/500g`;
+}
+
+function convertParams(toolName, args) {
+  switch (toolName) {
+    case 'save_expense': {
+      if (!args.name || typeof args.name !== 'string' || args.name.trim() === '') {
+        throw new Error('缺少必填参数：name（商品名称）');
+      }
+      if (args.amount === undefined || args.amount === null) {
+        throw new Error('缺少必填参数：amount（金额）');
+      }
+      
+      const name = sanitizeString(args.name, 100);
+      const amount = validateAmount(args.amount);
+      
+      return {
+        name: name,
+        originalName: name,
+        amount: amount,
+        price: amount,
+        quantity: 1,
+        category: sanitizeString(args.category, 50) || '其他',
+        store: sanitizeString(args.store, 100) || '',
+        date: args.date,
+        transactionType: 'expense',
+        unit: '',
+        weight: '',
+        marketPrice: '',
+        note: sanitizeString(args.note, 500) || '',
+        rawInput: sanitizeString(args.rawInput, 1000) || '',
+        agentType: args.agentType || '',
+        apiProvider: args.apiProvider || '',
+        mcp_version: args.mcp_version || MCP_VERSION,
+        source: 'mcp_txt_expense',
+      };
+    }
+
+    case 'save_receipt': {
+      let items = args.items;
+      if (typeof items === 'string') {
+        try {
+          items = JSON.parse(items);
+        } catch (e) {
+          throw new Error('items 参数格式错误：必须是数组或 JSON 数组字符串');
+        }
+      }
+      if (!Array.isArray(items)) {
+        throw new Error('items 参数必须是数组');
+      }
+      
+      return {
+        items: items.map((item, index) => {
+          const weight = sanitizeString(item.weight, 50);
+          const amount = validateAmount(item.amount);
+          const marketPrice = sanitizeString(item.marketPrice, 50) || calculateMarketPrice(amount, weight);
+          
+          return {
+            itemIndex: index,
+            name: sanitizeString(item.name, 100),
+            originalName: sanitizeString(item.originalName, 200) || sanitizeString(item.name, 100),
+            amount: amount,
+            category: sanitizeString(item.category, 50) || '其他',
+            subCategory: sanitizeString(item.subCategory, 50) || '',
+            transactionType: item.transactionType || 'expense',
+            price: validateAmount(item.price),
+            quantity: item.quantity ? String(item.quantity).substring(0, 20) : '1',
+            unit: sanitizeString(item.unit, 20),
+            weight: weight,
+            marketPrice: marketPrice,
+            note: sanitizeString(item.note, 500),
+          };
+        }),
+        store: sanitizeString(args.store, 100),
+        receiptNo: sanitizeString(args.receiptNo, 50),
+        totalAmount: validateAmount(args.totalAmount),
+        originalAmount: validateAmount(args.originalAmount),
+        discountAmount: validateAmount(args.discountAmount),
+        actualAmount: validateAmount(args.actualAmount),
+        paymentMethod: sanitizeString(args.paymentMethod, 50),
+        memberCardNo: sanitizeString(args.memberCardNo, 50),
+        currentPoints: parseInt(args.currentPoints) || 0,
+        totalPoints: parseInt(args.totalPoints) || 0,
+        date: args.date,
+        rawInput: sanitizeString(args.rawInput, 2000),
+        agentType: args.agentType || '',
+        apiProvider: args.apiProvider || '',
+        mcp_version: args.mcp_version || MCP_VERSION,
+        source: 'mcp_receipt',
+      };
+    }
+
+    case 'get_expenses':
+      return {
+        limit: Math.min(parseInt(args.limit) || 10, 100),
+        source: args.source,
+      };
+
+    case 'get_statistics':
+      return {
+        startDate: args.yearMonth ? `${args.yearMonth}-01` : undefined,
+        endDate: args.yearMonth ? getMonthEndDate(args.yearMonth) : undefined,
+        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),
+        date: args.date,
+        note: sanitizeString(args.note, 500),
+        transactionType: 'income',
+        rawInput: sanitizeString(args.rawInput, 1000),
+        agentType: args.agentType || '',
+        apiProvider: args.apiProvider || '',
+        mcp_version: args.mcp_version || MCP_VERSION,
+        source: 'mcp_txt_income',
+      };
+
+    default:
+      return args;
+  }
+}
+
+function getMonthEndDate(yearMonth) {
+  const [year, month] = yearMonth.split('-').map(Number);
+  const lastDay = new Date(year, month, 0).getDate();
+  return `${yearMonth}-${lastDay}`;
+}
+
+// 授权状态管理
+let authState = {
+  isAuthorized: false,
+  isWaiting: false,
+  userCode: null,
+  deviceCode: null,
+  error: null
+};
+
+async function main() {
+  initOAuthManager();
+
+  // 快速检查授权状态，不阻塞 Server 启动
+  // 使用 Promise.race 确保最多等待 2 秒
+  Promise.race([
+    checkAuthInBackground(),
+    new Promise(resolve => setTimeout(() => {
+      console.error('⏳ 授权检查超时，继续启动 Server');
+      resolve();
+    }, 2000))
+  ]);
+
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+}
+
+// 后台检查授权状态
+async function checkAuthInBackground() {
+  try {
+    const existingToken = await oauthManager.tokenStorage.load();
+    if (existingToken) {
+      // 已有token，直接使用
+      const validToken = await oauthManager.getValidToken();
+      cachedToken = validToken.accessToken;
+      tokenExpireTime = validToken.expiresAt;
+      authState.isAuthorized = true;
+      console.error('✅ 已加载保存的授权信息');
+    } else {
+      // 没有token，等待首次调用时再处理
+      console.error('⏳ 首次使用需要授权，将在调用工具时提示');
+      authState.isAuthorized = false;
+    }
+  } catch (err) {
+    console.error('⚠️  授权检查失败:', err.message);
+    authState.isAuthorized = false;
+  }
+}
+
+// 启动授权流程（获取验证码，启动后台轮询）
+async function startAuthFlow() {
+  try {
+    // 先检查是否有保存的授权状态
+    const savedAuthState = await oauthManager.loadAuthState();
+    if (savedAuthState && savedAuthState.userCode) {
+      // 检查验证码是否过期
+      if (oauthManager.isAuthStateExpired(savedAuthState)) {
+        console.error('⏰ 验证码已过期，生成新的验证码');
+        await oauthManager.clearAuthState();
+      } else {
+        // 有保存的验证码，直接使用
+        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,
+      expiresAt: Date.now() + deviceCodeRes.expiresIn * 1000
+    });
+
+    console.error(`🔑 验证码：${deviceCodeRes.userCode}`);
+    console.error('⏳ 请在小程序中完成授权，然后再次调用');
+    
+    // 启动后台轮询（3分钟）
+    pollForAuthInBackground(deviceCodeRes.deviceCode, 5000);
+
+  } catch (err) {
+    authState.error = err.message;
+    authState.isWaiting = false;
+    console.error('❌ 启动授权失败:', err.message);
+  }
+}
+
+// 后台轮询授权状态（3分钟）
+async function pollForAuthInBackground(deviceCode, interval) {
+  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;
+        
+        // 保存token到文件
+        await oauthManager.tokenStorage.save(token);
+        // 清除授权状态
+        await oauthManager.clearAuthState();
+        
+        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);
+        process.exit(1);
+      }
+      // 继续轮询
+    }
+  }
+
+  // 3分钟超时
+  authState.isWaiting = false;
+  console.error('⏰ 授权超时，请再次调用获取新验证码');
+  process.exit(0);
+}
+
+// 延迟函数
+function delay(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+// 注意：使用 console.error，避免污染 stdout（MCP 协议通信使用 stdout）
+main().catch((err) => {
+  console.error('❌ MCP Server 启动失败:', err.message);
+  console.error('堆栈:', err.stack);
+  process.exit(1);
+});