wechaty-web-panel 1.6.112 → 1.6.113

package/dist/mcp/src/mcp/server.js

@@ -0,0 +1,421 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ErrorCode, ListToolsRequestSchema, McpError, } from '@modelcontextprotocol/sdk/types.js';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+// 导入我们的业务服务
+import ChatMCP from '../index.js';
+// 导入数据验证模式
+import { RobotIdSchema, SearchFriendsSchema, SearchGroupsSchema, SearchMessagesSchema, FullTextSearchSchema, ConversationDetailsSchema, GroupMembersDetailSchema, MessageHeatmapSchema, QuickFindSchema, RelationshipAnalysisSchema, ActivityAnalysisSchema, ContentAnalysisSchema, DataExportSchema, BatchCreateFriendsSchema, BatchCreateGroupsSchema, BatchCreateGroupMembersSchema, CreateChatMessageSchema } from './schemas.js';
+/**
+ * 微信聊天数据 MCP 服务器
+ */
+class ChatMcpServer {
+    constructor() {
+        this.server = new Server({
+            name: 'chat-mcp-server',
+            version: '1.0.0',
+            description: '聊天数据 MCP 服务器，提供查询对话记录，分析对话记录，导出对话记录，批量写入对话记录',
+        }, {
+            capabilities: {
+                tools: {},
+            },
+        });
+        this.chatMCP = null;
+        this.setupHandlers();
+    }
+    /**
+     * 初始化聊天MCP服务
+     */
+    async initializeChatMCP() {
+        if (!this.chatMCP) {
+            this.chatMCP = new ChatMCP();
+            await this.chatMCP.init();
+            console.log('✅ ChatMCP 服务已初始化');
+        }
+        return this.chatMCP;
+    }
+    /**
+     * 设置MCP服务器处理程序
+     */
+    setupHandlers() {
+        // 列出可用工具
+        this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+            return {
+                tools: [
+                    // === 查询工具 ===
+                    {
+                        name: 'search_friends',
+                        description: '搜索好友信息，支持关键词、标签等多种搜索条件',
+                        inputSchema: zodToJsonSchema(SearchFriendsSchema),
+                    },
+                    {
+                        name: 'search_groups',
+                        description: '搜索群组信息，支持群名、成员数等搜索条件',
+                        inputSchema: zodToJsonSchema(SearchGroupsSchema),
+                    },
+                    {
+                        name: 'search_messages',
+                        description: '搜索聊天消息，支持内容、时间范围等搜索条件',
+                        inputSchema: zodToJsonSchema(SearchMessagesSchema),
+                    },
+                    {
+                        name: 'full_text_search',
+                        description: '全文搜索，同时搜索好友、群组和消息',
+                        inputSchema: zodToJsonSchema(FullTextSearchSchema),
+                    },
+                    {
+                        name: 'get_robot_statistics',
+                        description: '获取机器人的统计信息，包括好友数、群组数、消息数等',
+                        inputSchema: zodToJsonSchema(RobotIdSchema),
+                    },
+                    {
+                        name: 'get_conversation_details',
+                        description: '获取会话的详细信息和统计数据',
+                        inputSchema: zodToJsonSchema(ConversationDetailsSchema),
+                    },
+                    {
+                        name: 'get_group_members_detail',
+                        description: '获取群组成员的详细信息和活跃度统计',
+                        inputSchema: zodToJsonSchema(GroupMembersDetailSchema),
+                    },
+                    {
+                        name: 'get_message_heatmap',
+                        description: '获取消息活动热力图数据，分析时间分布模式',
+                        inputSchema: zodToJsonSchema(MessageHeatmapSchema),
+                    },
+                    {
+                        name: 'get_system_info',
+                        description: '获取系统信息，包括数据库状态、系统性能等',
+                        inputSchema: zodToJsonSchema({}),
+                    },
+                    // === 分析工具 ===
+                    {
+                        name: 'quick_find',
+                        description: '快速查找工具，智能识别并搜索用户、群组或成员',
+                        inputSchema: zodToJsonSchema(QuickFindSchema),
+                    },
+                    {
+                        name: 'analyze_relationships',
+                        description: '分析用户关系，包括好友关系、共同群组等',
+                        inputSchema: zodToJsonSchema(RelationshipAnalysisSchema),
+                    },
+                    {
+                        name: 'analyze_activity',
+                        description: '分析用户或群组的活动模式和频率',
+                        inputSchema: zodToJsonSchema(ActivityAnalysisSchema),
+                    },
+                    {
+                        name: 'analyze_content',
+                        description: '分析聊天内容，包括关键词提取、情感分析等',
+                        inputSchema: zodToJsonSchema(ContentAnalysisSchema),
+                    },
+                    {
+                        name: 'export_data',
+                        description: '导出数据到指定格式（JSON、CSV、TXT）',
+                        inputSchema: zodToJsonSchema(DataExportSchema),
+                    },
+                    // === 数据写入工具 ===
+                    {
+                        name: 'batch_create_friends',
+                        description: '批量创建或更新好友信息',
+                        inputSchema: zodToJsonSchema(BatchCreateFriendsSchema),
+                    },
+                    {
+                        name: 'batch_create_groups',
+                        description: '批量创建或更新群组信息',
+                        inputSchema: zodToJsonSchema(BatchCreateGroupsSchema),
+                    },
+                    {
+                        name: 'batch_create_group_members',
+                        description: '批量创建或更新群成员信息',
+                        inputSchema: zodToJsonSchema(BatchCreateGroupMembersSchema),
+                    },
+                    {
+                        name: 'create_chat_message',
+                        description: '创建聊天消息记录',
+                        inputSchema: zodToJsonSchema(CreateChatMessageSchema),
+                    },
+                ],
+            };
+        });
+        // 执行工具调用
+        this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+            const { name, arguments: args } = request.params;
+            try {
+                // 确保ChatMCP服务已初始化
+                await this.initializeChatMCP();
+                switch (name) {
+                    // === 查询工具实现 ===
+                    case 'search_friends': {
+                        const params = SearchFriendsSchema.parse(args);
+                        const result = await this.chatMCP.searchFriends(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'search_groups': {
+                        const params = SearchGroupsSchema.parse(args);
+                        const result = await this.chatMCP.searchGroups(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'search_messages': {
+                        const params = SearchMessagesSchema.parse(args);
+                        const result = await this.chatMCP.searchChatMessages(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'full_text_search': {
+                        const params = FullTextSearchSchema.parse(args);
+                        const result = await this.chatMCP.fullTextSearch(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'get_robot_statistics': {
+                        const params = RobotIdSchema.parse(args);
+                        const result = await this.chatMCP.getRobotStatistics(params.robotId);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'get_conversation_details': {
+                        const params = ConversationDetailsSchema.parse(args);
+                        const result = await this.chatMCP.getConversationDetails(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'get_group_members_detail': {
+                        const params = GroupMembersDetailSchema.parse(args);
+                        const result = await this.chatMCP.getGroupMembersDetail(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'get_message_heatmap': {
+                        const params = MessageHeatmapSchema.parse(args);
+                        const result = await this.chatMCP.getMessageHeatmap(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'get_system_info': {
+                        const result = await this.chatMCP.getSystemInfo();
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    // === 分析工具实现 ===
+                    case 'quick_find': {
+                        const params = QuickFindSchema.parse(args);
+                        const result = await this.chatMCP.quickFind(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'analyze_relationships': {
+                        const params = RelationshipAnalysisSchema.parse(args);
+                        const result = await this.chatMCP.analyzeRelationships(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'analyze_activity': {
+                        const params = ActivityAnalysisSchema.parse(args);
+                        const result = await this.chatMCP.analyzeActivity(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'analyze_content': {
+                        const params = ContentAnalysisSchema.parse(args);
+                        const result = await this.chatMCP.analyzeContent(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'export_data': {
+                        const params = DataExportSchema.parse(args);
+                        const result = await this.chatMCP.exportData(params);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify(result, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    // === 数据写入工具实现 ===
+                    case 'batch_create_friends': {
+                        const params = BatchCreateFriendsSchema.parse(args);
+                        const result = await this.chatMCP.batchCreateFriends(params.friends, { updateOnDuplicate: params.updateOnDuplicate });
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify({
+                                        success: true,
+                                        message: `成功批量创建/更新 ${result.length} 条好友记录`,
+                                        recordCount: result.length
+                                    }, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'batch_create_groups': {
+                        const params = BatchCreateGroupsSchema.parse(args);
+                        const result = await this.chatMCP.batchCreateGroups(params.groups, { updateOnDuplicate: params.updateOnDuplicate });
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify({
+                                        success: true,
+                                        message: `成功批量创建/更新 ${result.length} 条群组记录`,
+                                        recordCount: result.length
+                                    }, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'batch_create_group_members': {
+                        const params = BatchCreateGroupMembersSchema.parse(args);
+                        const result = await this.chatMCP.batchCreateGroupMembers(params.members, { updateOnDuplicate: params.updateOnDuplicate });
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify({
+                                        success: true,
+                                        message: `成功批量创建/更新 ${result.length} 条群成员记录`,
+                                        recordCount: result.length
+                                    }, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    case 'create_chat_message': {
+                        const params = CreateChatMessageSchema.parse(args);
+                        const result = await this.chatMCP.createChatMessage(params.message);
+                        return {
+                            content: [
+                                {
+                                    type: 'text',
+                                    text: JSON.stringify({
+                                        success: true,
+                                        message: '成功创建聊天消息记录',
+                                        messageId: result.id
+                                    }, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    default:
+                        throw new McpError(ErrorCode.MethodNotFound, `未知工具: ${name}`);
+                }
+            }
+            catch (error) {
+                console.error(`执行工具 ${name} 时发生错误:`, error);
+                if (error instanceof McpError) {
+                    throw error;
+                }
+                throw new McpError(ErrorCode.InternalError, `执行工具失败: ${error.message}`);
+            }
+        });
+        // 错误处理
+        this.server.onerror = (error) => {
+            console.error('[MCP Server Error]', error);
+        };
+        process.on('SIGINT', async () => {
+            console.log('\n正在关闭 MCP 服务器...');
+            process.exit(0);
+        });
+    }
+    /**
+     * 启动MCP服务器
+     */
+    async run() {
+        const transport = new StdioServerTransport();
+        console.log('🚀 启动微信聊天数据 MCP 服务器...');
+        console.log('📋 可用工具:');
+        console.log('   查询工具: search_friends, search_groups, search_messages, full_text_search');
+        console.log('   统计工具: get_robot_statistics, get_conversation_details, get_group_members_detail');
+        console.log('   分析工具: quick_find, analyze_relationships, analyze_activity, analyze_content');
+        console.log('   数据工具: export_data, batch_create_friends, batch_create_groups');
+        console.log('   系统工具: get_system_info');
+        console.log('');
+        await this.server.connect(transport);
+        console.log('✅ MCP 服务器已启动，等待客户端连接...');
+    }
+}
+export default ChatMcpServer;
+//# sourceMappingURL=server.js.map