chaimi-bookkeeping-mcp 2.2.4 → 2.2.6

package/package.json

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

package/server.js

@@ -84,9 +84,24 @@ const server = new Server(
 server.setRequestHandler(ListToolsRequestSchema, async () => {
   return {
     tools: [
+      {
+        name: 'quick_book',
+        description: '极简快捷记账 - 自动识别支出/收入，智能匹配分类。输入如："午餐 30 元"、"工资 8000 元"、"木屋烧烤 35 元"',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            name: { type: 'string', description: '商品/服务名称或自然语言描述（必填）' },
+            amount: { type: 'number', description: '金额（必填）' },
+            category: { type: 'string', description: '分类（可选，系统自动推荐）' },
+            store: { type: 'string', description: '商家名称（可选）' },
+            note: { type: 'string', description: '备注（可选）' },
+          },
+          required: ['name', 'amount'],
+        },
+      },
       {
         name: 'save_expense',
-        description: '保存单商品消费记录（AI文字记账）',
+        description: '保存单商品消费记录（AI 文字记账）',
         inputSchema: {
           type: 'object',
           properties: {
@@ -97,8 +112,8 @@ 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提供商' },
+            agentType: { type: 'string', description: 'Agent 类型（如：workbuddy、claude、cursor）' },
+            apiProvider: { type: 'string', description: 'AI 提供商' },
             rawInput: { type: 'string', description: '用户原始输入' },
           },
           required: ['name', 'amount', 'price', 'quantity'],
@@ -351,8 +366,56 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
 
     // 根据工具类型选择处理流程
     switch (name) {
+      case 'quick_book': {
+        // 极简快捷记账：自动识别支出/收入，智能匹配分类
+        console.log('处理极简快捷记账...');
+        
+        // 1. 判断是支出还是收入
+        const isIncome = isIncomeName(args.name);
+        
+        // 2. 智能匹配分类
+        const category = args.category || getCategory(args.name);
+        
+        // 3. 自动补全参数
+        const completedArgs = {
+          name: sanitizeString(args.name, 100),
+          amount: validateAmount(args.amount),
+          price: args.price || validateAmount(args.amount),
+          quantity: args.quantity || 1,
+          unit: args.unit || '个',
+          category: category,
+          store: sanitizeString(args.store, 100) || '未知商家',
+          note: sanitizeString(args.note, 500),
+          agentType: args.agentType || '',
+          apiProvider: args.apiProvider || '',
+          rawInput: args.rawInput || args.name,
+        };
+        
+        // 4. 根据类型选择接口
+        if (isIncome) {
+          // 收入记账
+          console.log('识别为收入，调用 save_income...');
+          const mcpParams = convertParams('save_income', completedArgs);
+          result = await callMcpHub('addIncome', mcpParams, token);
+          
+          if (result.success) {
+            userMessage = `✅ 记账成功\n\n| 收入来源 | 金额 | 分类 |\n|------|------|------|\n| ${completedArgs.name} | ${completedArgs.amount}元 | ${completedArgs.category || '其他收入'} |\n\n已保存到你的柴米记账小程序数据库。`;
+          }
+        } else {
+          // 支出记账
+          console.log('识别为支出，调用 save_expense...');
+          const mcpParams = convertParams('save_expense', completedArgs);
+          result = await callMcpHub('addExpense', mcpParams, token);
+          
+          if (result.success) {
+            userMessage = `✅ 记账成功\n\n| 商品 | 金额 | 分类 | 商家 |\n|------|------|------|------|\n| ${completedArgs.name} | ${completedArgs.amount}元 | ${completedArgs.category || '其他'} | ${completedArgs.store} |\n\n已保存到你的柴米记账小程序数据库。`;
+          }
+        }
+        break;
+      }
+      
       case 'save_expense': {
-        // 步骤1：校验数据完整性
+        // 步骤 1：校验数据完整性
         console.log('校验消费数据完整性...');
         const validationResult = await callMcpPrompt(
           'validateResult',
@@ -663,6 +726,40 @@ function convertParams(toolName, args) {
   }
 }
 
+// 智能分类：根据商品名称匹配分类
+function getCategory(name) {
+  if (!name) return '其他';
+  
+  const categoryMap = {
+    '餐饮': ['饭', '菜', '餐', '酒', '饮料', '咖啡', '茶', '早餐', '午餐', '晚餐', '夜宵', '烧烤', '火锅', '快餐', '外卖', '小吃', '奶茶', '面包', '蛋糕'],
+    '交通': ['车', '油', '票', '打车', '公交', '地铁', '加油', '停车', '租车', '火车', '飞机', '轮船', '客运'],
+    '购物': ['买', '购', '商场', '超市', '衣服', '鞋', '包', '化妆品', '护肤品', '数码', '电器', '家具', '日用品'],
+    '娱乐': ['玩', '游戏', '电影', '歌', 'KTV', '酒吧', '旅游', '景点', '门票', '健身', '运动', '书', '音乐'],
+    '医疗': ['药', '医', '病', '医院', '诊所', '体检', '疫苗', '牙科', '眼科', '门诊'],
+    '教育': ['学', '课', '培训', '考试', '学校', '学费', '教材', '文具', '培训费'],
+    '居住': ['房', '租', '物业', '水电', '燃气', '网络', '宽带', '话费', '维修', '装修'],
+    '工资': ['工资', '薪水', '奖金', '提成', '分红', '补贴', '津贴', '绩效'],
+    '其他收入': ['红包', '转账', '借款', '还款', '退款', '理赔', '中奖', '利息'],
+  };
+  
+  for (const [category, keywords] of Object.entries(categoryMap)) {
+    if (keywords.some(keyword => name.includes(keyword))) {
+      return category;
+    }
+  }
+  
+  return '其他';
+}
+
+// 判断是否为收入
+function isIncomeName(name) {
+  if (!name) return false;
+  
+  const incomeKeywords = ['工资', '薪水', '奖金', '提成', '分红', '补贴', '津贴', '绩效', '红包', '转账', '借款', '还款', '退款', '理赔', '中奖', '利息', '收入', '入账', '到账'];
+  
+  return incomeKeywords.some(keyword => name.includes(keyword));
+}
+
 // 获取月份最后一天
 function getMonthEndDate(yearMonth) {
   const [year, month] = yearMonth.split('-').map(Number);