wechaty-web-panel 1.6.112 → 1.6.113

package/dist/mcp/src/config/database.js

@@ -0,0 +1,51 @@
+import { Sequelize } from 'sequelize';
+import { config } from 'dotenv';
+config();
+// 数据库配置
+const dbConfig = {
+    host: process.env.DB_HOST || '0.0.0.0',
+    port: parseInt(process.env.DB_PORT) || 5432,
+    database: process.env.DB_NAME || 'chat_mcp',
+    username: process.env.DB_USER || 'postgres',
+    password: process.env.DB_PASSWORD || 'postgres',
+    dialect: 'postgres',
+    logging: process.env.NODE_ENV === 'development' ? console.log : false,
+    pool: {
+        max: 5,
+        min: 0,
+        acquire: 30000,
+        idle: 10000
+    }
+};
+// 创建Sequelize实例
+export const sequelize = new Sequelize(dbConfig.database, dbConfig.username, dbConfig.password, {
+    host: dbConfig.host,
+    port: dbConfig.port,
+    dialect: dbConfig.dialect,
+    logging: dbConfig.logging,
+    pool: dbConfig.pool
+});
+// 测试数据库连接
+export const testConnection = async () => {
+    try {
+        await sequelize.authenticate();
+        console.log('✅ 数据库连接成功');
+    }
+    catch (error) {
+        console.error('❌ 数据库连接失败:', error);
+        throw error;
+    }
+};
+// 初始化数据库
+export const initDatabase = async () => {
+    try {
+        await sequelize.sync({ force: false });
+        console.log('✅ 数据库同步成功');
+    }
+    catch (error) {
+        console.error('❌ 数据库同步失败:', error);
+        throw error;
+    }
+};
+export default sequelize;
+//# sourceMappingURL=database.js.map

package/dist/mcp/src/index.js

@@ -0,0 +1,238 @@
+import { testConnection, initDatabase } from './config/database.js';
+import ChatDataService from './services/ChatDataService.js';
+import McpService from './services/McpService.js';
+import McpTools from './services/McpTools.js';
+/**
+ * 初始化微信聊天数据存储系统
+ */
+class ChatMCP {
+    constructor() {
+        this.chatDataService = new ChatDataService();
+        this.mcpService = new McpService();
+        this.mcpTools = new McpTools();
+    }
+    /**
+     * 初始化数据库连接和表结构
+     */
+    async init() {
+        try {
+            console.log('🚀 正在启动微信聊天数据存储系统...');
+            // 测试数据库连接
+            await testConnection();
+            // 初始化数据库表结构
+            await initDatabase();
+            console.log('🎉 微信聊天数据存储系统启动成功！');
+            return this;
+        }
+        catch (error) {
+            console.error('💥 系统启动失败:', error);
+            throw error;
+        }
+    }
+    /**
+     * 批量写入好友信息
+     * @param {Array} friendsData - 好友信息数组
+     * @param {Object} options - 配置选项
+     * @returns {Promise<Array>} 创建的好友记录
+     */
+    async batchCreateFriends(friendsData, options) {
+        return this.chatDataService.batchCreateFriends(friendsData, options);
+    }
+    /**
+     * 批量写入群信息
+     * @param {Array} groupsData - 群信息数组
+     * @param {Object} options - 配置选项
+     * @returns {Promise<Array>} 创建的群记录
+     */
+    async batchCreateGroups(groupsData, options) {
+        return this.chatDataService.batchCreateGroups(groupsData, options);
+    }
+    /**
+     * 批量写入群成员信息
+     * @param {Array} membersData - 群成员信息数组
+     * @param {Object} options - 配置选项
+     * @returns {Promise<Array>} 创建的群成员记录
+     */
+    async batchCreateGroupMembers(membersData, options) {
+        return this.chatDataService.batchCreateGroupMembers(membersData, options);
+    }
+    /**
+     * 写入对话记录
+     * @param {Object|Array} messageData - 消息数据，可以是单条或数组
+     * @returns {Promise<Object|Array>} 创建的消息记录
+     */
+    async createChatMessage(messageData) {
+        return this.chatDataService.createChatMessage(messageData);
+    }
+    /**
+     * 使用事务批量写入所有数据
+     * @param {Object} data - 包含所有类型数据的对象
+     * @returns {Promise<Object>} 写入结果
+     */
+    async batchCreateAllData(data) {
+        return this.chatDataService.batchCreateAllData(data);
+    }
+    /**
+     * 获取好友列表
+     * @param {string} robotId - 机器人ID
+     * @returns {Promise<Array>} 好友列表
+     */
+    async getFriends(robotId) {
+        return this.chatDataService.getFriendsByRobotId(robotId);
+    }
+    /**
+     * 获取群组列表
+     * @param {string} robotId - 机器人ID
+     * @returns {Promise<Array>} 群组列表
+     */
+    async getGroups(robotId) {
+        return this.chatDataService.getGroupsByRobotId(robotId);
+    }
+    /**
+     * 获取聊天记录
+     * @param {Object} query - 查询条件
+     * @returns {Promise<Array>} 聊天记录列表
+     */
+    async getChatMessages(query) {
+        return this.chatDataService.getChatMessages(query);
+    }
+    // =================== MCP 高级查询方法 ===================
+    /**
+     * 搜索好友
+     * @param {Object} params - 搜索参数
+     * @returns {Promise<Object>} 搜索结果
+     */
+    async searchFriends(params) {
+        return this.mcpService.searchFriends(params);
+    }
+    /**
+     * 搜索群组
+     * @param {Object} params - 搜索参数
+     * @returns {Promise<Object>} 搜索结果
+     */
+    async searchGroups(params) {
+        return this.mcpService.searchGroups(params);
+    }
+    /**
+     * 搜索聊天消息
+     * @param {Object} params - 搜索参数
+     * @returns {Promise<Object>} 搜索结果
+     */
+    async searchChatMessages(params) {
+        return this.mcpService.searchChatMessages(params);
+    }
+    /**
+     * 获取机器人统计信息
+     * @param {string} robotId - 机器人ID
+     * @returns {Promise<Object>} 统计信息
+     */
+    async getRobotStatistics(robotId) {
+        return this.mcpService.getRobotStatistics(robotId);
+    }
+    /**
+     * 获取会话详细信息
+     * @param {Object} params - 查询参数
+     * @returns {Promise<Object>} 会话信息
+     */
+    async getConversationDetails(params) {
+        return this.mcpService.getConversationDetails(params);
+    }
+    /**
+     * 获取群组成员详细信息
+     * @param {Object} params - 查询参数
+     * @returns {Promise<Object>} 群组成员信息
+     */
+    async getGroupMembersDetail(params) {
+        return this.mcpService.getGroupMembersDetail(params);
+    }
+    /**
+     * 获取消息热力图数据
+     * @param {Object} params - 查询参数
+     * @returns {Promise<Array>} 热力图数据
+     */
+    async getMessageHeatmap(params) {
+        return this.mcpService.getMessageHeatmap(params);
+    }
+    /**
+     * 全文搜索
+     * @param {Object} params - 搜索参数
+     * @returns {Promise<Object>} 搜索结果
+     */
+    async fullTextSearch(params) {
+        return this.mcpService.fullTextSearch(params);
+    }
+    /**
+     * 获取系统信息
+     * @returns {Promise<Object>} 系统信息
+     */
+    async getSystemInfo() {
+        return this.mcpService.getSystemInfo();
+    }
+    // =================== MCP 工具方法 ===================
+    /**
+     * 快速查找
+     * @param {Object} params - 查询参数
+     * @returns {Promise<Object>} 查找结果
+     */
+    async quickFind(params) {
+        return this.mcpTools.quickFind(params);
+    }
+    /**
+     * 关系分析
+     * @param {Object} params - 分析参数
+     * @returns {Promise<Object>} 关系分析结果
+     */
+    async analyzeRelationships(params) {
+        return this.mcpTools.analyzeRelationships(params);
+    }
+    /**
+     * 活动分析
+     * @param {Object} params - 分析参数
+     * @returns {Promise<Object>} 活动分析结果
+     */
+    async analyzeActivity(params) {
+        return this.mcpTools.analyzeActivity(params);
+    }
+    /**
+     * 内容分析
+     * @param {Object} params - 分析参数
+     * @returns {Promise<Object>} 内容分析结果
+     */
+    async analyzeContent(params) {
+        return this.mcpTools.analyzeContent(params);
+    }
+    /**
+     * 导出数据
+     * @param {Object} params - 导出参数
+     * @returns {Promise<Object>} 导出结果
+     */
+    async exportData(params) {
+        return this.mcpTools.exportData(params);
+    }
+}
+// 导出主类和服务类
+export default ChatMCP;
+export { ChatDataService, McpService, McpTools };
+// 如果直接运行此文件，则进行初始化演示
+if (import.meta.url === `file://${process.argv[1]}`) {
+    console.log('📋 微信聊天数据存储系统 - 使用示例');
+    console.log('');
+    console.log('// 创建实例');
+    console.log('const chatMCP = new ChatMCP();');
+    console.log('');
+    console.log('// 初始化系统');
+    console.log('await chatMCP.init();');
+    console.log('');
+    console.log('// 批量写入好友信息');
+    console.log('await chatMCP.batchCreateFriends([{ robotId: "robot1", wxid: "friend1", name: "张三" }]);');
+    console.log('');
+    console.log('// 批量写入群信息');
+    console.log('await chatMCP.batchCreateGroups([{ robotId: "robot1", wxid: "group1", name: "技术交流群" }]);');
+    console.log('');
+    console.log('// 批量写入群成员信息');
+    console.log('await chatMCP.batchCreateGroupMembers([{ robotId: "robot1", roomWxid: "group1", wxid: "member1", name: "李四" }]);');
+    console.log('');
+    console.log('// 写入对话记录');
+    console.log('await chatMCP.createChatMessage({ robotId: "robot1", conversationId: "friend1", content: "你好", recordType: "contact", timestamp: Date.now() });');
+}
+//# sourceMappingURL=index.js.map

package/dist/mcp/src/mcp/schemas.js

@@ -0,0 +1,178 @@
+import { z } from 'zod';
+/**
+ * MCP 工具的数据验证模式定义
+ */
+// 基础参数模式
+export const RobotIdSchema = z.object({
+    robotId: z.string().describe('机器人账号ID')
+});
+// 搜索好友参数模式
+export const SearchFriendsSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    keyword: z.string().optional().describe('搜索关键词（姓名、备注、别名、微信号）'),
+    tags: z.string().optional().describe('标签筛选'),
+    limit: z.number().default(50).describe('返回结果数量限制'),
+    offset: z.number().default(0).describe('结果偏移量'),
+    orderBy: z.string().default('createdAt').describe('排序字段'),
+    orderDirection: z.enum(['ASC', 'DESC']).default('DESC').describe('排序方向')
+});
+// 搜索群组参数模式
+export const SearchGroupsSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    keyword: z.string().optional().describe('群名关键词'),
+    includeMembers: z.boolean().default(false).describe('是否包含成员信息'),
+    minMemberCount: z.number().default(0).describe('最小成员数'),
+    maxMemberCount: z.number().optional().describe('最大成员数'),
+    limit: z.number().default(50).describe('返回结果数量限制'),
+    offset: z.number().default(0).describe('结果偏移量')
+});
+// 搜索聊天消息参数模式
+export const SearchMessagesSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    conversationId: z.string().optional().describe('会话ID'),
+    keyword: z.string().optional().describe('消息内容关键词'),
+    contentType: z.string().optional().describe('内容类型（文字、图片、文件等）'),
+    recordType: z.enum(['contact', 'room']).optional().describe('记录类型（contact私聊/room群聊）'),
+    chatUserId: z.string().optional().describe('聊天用户ID'),
+    isRobotAnswer: z.boolean().optional().describe('是否机器人回复'),
+    startTime: z.number().optional().describe('开始时间戳'),
+    endTime: z.number().optional().describe('结束时间戳'),
+    limit: z.number().default(100).describe('返回结果数量限制'),
+    offset: z.number().default(0).describe('结果偏移量')
+});
+// 全文搜索参数模式
+export const FullTextSearchSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    keyword: z.string().describe('搜索关键词'),
+    searchTypes: z.array(z.enum(['friends', 'groups', 'messages'])).default(['friends', 'groups', 'messages']).describe('搜索类型'),
+    limit: z.number().default(20).describe('每种类型的返回结果数量限制')
+});
+// 会话详情参数模式
+export const ConversationDetailsSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    conversationId: z.string().describe('会话ID')
+});
+// 群组成员详情参数模式
+export const GroupMembersDetailSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    roomWxid: z.string().describe('群微信ID'),
+    includeMessageStats: z.boolean().default(true).describe('是否包含消息统计')
+});
+// 消息热力图参数模式
+export const MessageHeatmapSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    conversationId: z.string().optional().describe('会话ID'),
+    startDate: z.string().optional().describe('开始日期（YYYY-MM-DD格式）'),
+    endDate: z.string().optional().describe('结束日期（YYYY-MM-DD格式）'),
+    granularity: z.enum(['hour', 'day', 'week', 'month']).default('hour').describe('时间粒度')
+});
+// 快速查找参数模式
+export const QuickFindSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    query: z.string().describe('查询内容'),
+    type: z.enum(['auto', 'friend', 'group', 'member']).default('auto').describe('查找类型')
+});
+// 关系分析参数模式
+export const RelationshipAnalysisSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    userId: z.string().describe('目标用户ID'),
+    analysisType: z.enum(['basic', 'comprehensive']).default('comprehensive').describe('分析类型')
+});
+// 活动分析参数模式
+export const ActivityAnalysisSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    targetId: z.string().describe('目标ID（用户或群组）'),
+    targetType: z.enum(['auto', 'user', 'group']).default('auto').describe('目标类型'),
+    timeRange: z.number().default(30).describe('分析时间范围（天数）'),
+    granularity: z.enum(['hour', 'day', 'week']).default('day').describe('时间粒度')
+});
+// 内容分析参数模式
+export const ContentAnalysisSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    conversationId: z.string().optional().describe('会话ID'),
+    timeRange: z.number().default(30).describe('分析时间范围（天数）'),
+    analysisTypes: z.array(z.enum(['contentType', 'keywords', 'sentiment'])).default(['contentType', 'keywords', 'sentiment']).describe('分析类型')
+});
+// 数据导出参数模式
+export const DataExportSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    exportType: z.enum(['messages', 'friends', 'groups', 'statistics']).default('messages').describe('导出类型'),
+    conversationId: z.string().optional().describe('会话ID（仅导出消息时需要）'),
+    format: z.enum(['json', 'csv', 'txt']).default('json').describe('导出格式'),
+    timeRange: z.number().default(30).describe('时间范围（天数）'),
+    includeMetadata: z.boolean().default(true).describe('是否包含元数据')
+});
+// 批量写入好友参数模式
+export const BatchCreateFriendsSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    friends: z.array(z.object({
+        robotId: z.string().describe('机器人账号ID'),
+        wxid: z.string().describe('好友微信ID'),
+        wxNumber: z.string().optional().describe('好友微信号'),
+        name: z.string().optional().describe('昵称'),
+        avatar: z.string().optional().describe('头像URL'),
+        alias: z.string().optional().describe('别名'),
+        remark: z.string().optional().describe('备注'),
+        tags: z.string().optional().describe('标签')
+    })).describe('好友信息数组'),
+    updateOnDuplicate: z.boolean().default(true).describe('重复时是否更新')
+});
+// 批量写入群组参数模式
+export const BatchCreateGroupsSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    groups: z.array(z.object({
+        robotId: z.string().describe('机器人账号ID'),
+        wxid: z.string().describe('群微信ID'),
+        adminId: z.string().optional().describe('群管理员微信ID'),
+        name: z.string().optional().describe('群名'),
+        avatar: z.string().optional().describe('群头像URL'),
+        memberCount: z.number().optional().describe('群成员数')
+    })).describe('群组信息数组'),
+    updateOnDuplicate: z.boolean().default(true).describe('重复时是否更新')
+});
+// 批量写入群成员参数模式
+export const BatchCreateGroupMembersSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    members: z.array(z.object({
+        robotId: z.string().describe('机器人账号ID'),
+        roomWxid: z.string().describe('群微信ID'),
+        wxid: z.string().describe('群成员微信ID'),
+        wxNumber: z.string().optional().describe('群成员微信号'),
+        name: z.string().optional().describe('群成员昵称'),
+        avatar: z.string().optional().describe('群成员头像URL'),
+        alias: z.string().optional().describe('群成员别名'),
+        remark: z.string().optional().describe('群成员备注')
+    })).describe('群成员信息数组'),
+    updateOnDuplicate: z.boolean().default(true).describe('重复时是否更新')
+});
+// 写入聊天消息参数模式
+export const CreateChatMessageSchema = z.object({
+    robotId: z.string().describe('机器人账号ID'),
+    message: z.object({
+        robotId: z.string().describe('机器人账号ID'),
+        robotName: z.string().optional().describe('机器人昵称'),
+        conversationId: z.string().describe('会话ID'),
+        conversationName: z.string().optional().describe('会话名'),
+        conversationAvatar: z.string().optional().describe('会话头像'),
+        isRobotAnswer: z.boolean().default(false).describe('是否机器人回复'),
+        realUser: z.number().default(0).describe('是否真人回复'),
+        msgId: z.string().optional().describe('消息ID'),
+        contentType: z.string().default('文字').describe('内容类型'),
+        recordType: z.enum(['contact', 'room']).describe('记录类型'),
+        chatUserName: z.string().optional().describe('聊天用户昵称'),
+        chatUserAlias: z.string().optional().describe('聊天用户备注'),
+        chatUserId: z.string().optional().describe('聊天用户ID'),
+        chatUserAvatar: z.string().optional().describe('聊天用户头像'),
+        roomName: z.string().optional().describe('群聊名称'),
+        roomId: z.string().optional().describe('群聊ID'),
+        content: z.string().optional().describe('聊天内容'),
+        url: z.string().optional().describe('H5链接'),
+        fileName: z.string().optional().describe('文件名称'),
+        isImage: z.boolean().default(false).describe('是否图片'),
+        description: z.string().optional().describe('H5描述'),
+        imageUrl: z.string().optional().describe('H5图片链接'),
+        title: z.string().optional().describe('H5标题'),
+        timestamp: z.number().describe('聊天时间戳')
+    }).describe('消息信息')
+});
+//# sourceMappingURL=schemas.js.map