@izhimu/qq 0.5.1 → 0.6.0

package/dist/src/core/dispatch.js

@@ -1,445 +1,257 @@
-/**
- * Message Dispatch Module
- * Handles routing and dispatching incoming messages to the AI
- */
-import { buildPendingHistoryContextFromMap, clearHistoryEntries, recordPendingHistoryEntry, resolveInboundRouteEnvelopeBuilderWithRuntime } from "openclaw/plugin-sdk";
-import { getRuntime, getContext, getSession, clearSession, updateSession, getLoginInfo, historyCache } from './runtime.js';
-import { getFile, sendMsg, setInputStatus } from './request.js';
-import { napCatToOpenClawMessage, openClawToNapCatMessage } from '../adapters/message.js';
-import { Logger as log, markdownToText, buildMediaMessage } from '../utils/index.js';
-import { CHANNEL_ID } from "./config.js";
-/**
- * Convert OpenClaw message content array to plain text
- * For images, includes the URL so AI models can access them
- * For replies, includes quoted message content if available
- */
-async function contentToPlainText(content) {
-    const results = await Promise.all(content.map(async (c) => {
-        switch (c.type) {
-            case 'text':
-                return c.text;
-            case 'at':
-                const target = c.isAll ? '@全体成员' : `@${c.userId}`;
-                return `[提及]${target}`;
-            case 'image':
-                return `[图片]${c.url}`;
-            case 'audio':
-                return `[音频]${c.path}`;
-            case 'video':
-                return `[视频]${c.url}`;
-            case 'file': {
-                const fileInfo = await getFile({ file_id: c.fileId });
-                if (!fileInfo.data?.file)
-                    return null;
-                return `[文件]${fileInfo.data.file}`;
-            }
-            case 'json':
-                return `[JSON]\n\n\`\`\`json\n${c.data}\n\`\`\``;
-            case 'reply': {
-                const senderInfo = c.sender && c.senderId ? `${c.sender}(${c.senderId})` : '(未知用户)';
-                const replyMsg = c.message ?? '(无法获取原消息)';
-                const quotedContent = `${senderInfo}:\n${replyMsg}`.replace(/^/gm, '> ');
-                return `[回复]\n\n${quotedContent}`;
-            }
-            default:
-                return null;
-        }
-    }));
-    return results.filter((v) => v !== null).join('\n');
-}
-async function contextToMedia(content) {
-    const hasMedia = content.some(c => c.type === 'image' || c.type === 'audio' || c.type === 'file');
-    if (!hasMedia) {
-        return;
-    }
-    const image = content.find(c => c.type === 'image');
-    if (image) {
-        return {
-            type: 'image/jpeg',
-            path: image.url,
-            url: image.url,
-        };
-    }
-    const audio = content.find(c => c.type === 'audio');
-    if (audio) {
-        return {
-            type: 'audio/amr',
-            path: audio.path,
-            url: audio.url,
-        };
-    }
-    const file = content.find(c => c.type === 'file');
-    if (file) {
-        const fileInfo = await getFile({ file_id: file.fileId });
-        if (fileInfo.data?.file == undefined) {
-            return;
-        }
-        return {
-            type: 'application/octet-stream',
-            path: fileInfo.data?.file,
-            url: fileInfo.data?.url,
-        };
-    }
-    return;
-}
-async function sendText(isGroup, chatId, text) {
-    const contextText = text.replace(/NO_REPLY\s*$/, '');
-    const context = getContext();
-    if (!context) {
-        log.warn('dispatch', `No gateway context`);
-        return;
-    }
-    const messageSegments = [{
-            type: 'text',
-            data: { text: context.account.markdownFormat ? markdownToText(contextText) : contextText }
-        }];
+import { createChannelInboundDebouncer, shouldDebounceTextInbound } from "openclaw/plugin-sdk/channel-inbound";
+import { createChannelReplyPipeline } from "openclaw/plugin-sdk/channel-reply-pipeline";
+import { resolveSendableOutboundReplyParts } from "openclaw/plugin-sdk/reply-payload";
+import { recordPendingHistoryEntry } from "openclaw/plugin-sdk/reply-history";
+import { getLoginInfo, historyCache } from './runtime.js';
+import { sendMsg, setInputStatus } from './request.js';
+import { outboundMessageAdapter } from '../adapters/message.js';
+import { Logger as log, buildMediaMessage } from '../utils/index.js';
+import { QQ_CHANNEL } from "./config.js";
+import { getQQConfigByChatType } from "./auth.js";
+async function send(account, isGroup, to, messageSegments) {
     try {
         await sendMsg({
             message_type: isGroup ? 'group' : 'private',
-            group_id: isGroup ? chatId : undefined,
-            user_id: !isGroup ? chatId : undefined,
-            message: messageSegments,
+            group_id: isGroup ? to : undefined,
+            user_id: !isGroup ? to : undefined,
+            message: await outboundMessageAdapter(messageSegments, account),
         });
-        log.info('dispatch', `Sent reply: ${text.slice(0, 100)}`);
+        log.debug('dispatch', `Sent reply success`);
     }
     catch (error) {
         log.error('dispatch', `Send failed: ${error}`);
     }
 }
-async function sendMedia(isGroup, chatId, mediaUrl) {
-    const content = [buildMediaMessage(mediaUrl)];
-    try {
-        await sendMsg({
-            message_type: isGroup ? 'group' : 'private',
-            group_id: isGroup ? chatId : undefined,
-            user_id: !isGroup ? chatId : undefined,
-            message: openClawToNapCatMessage(content),
-        });
-        log.info('dispatch', `Sent reply: ${mediaUrl.slice(0, 100)}`);
-    }
-    catch (error) {
-        log.error('dispatch', `Send failed: ${error}`);
+function mention(account, content, groupId, targetId, loginInfo) {
+    const config = getQQConfigByChatType(true, groupId, account);
+    const isMentionEnabled = !!config?.requireMention;
+    const isPokeEnabled = !!config?.requirePoke;
+    const isWakeEnabled = !!config?.wakeWord?.trim();
+    if (!isMentionEnabled && !isPokeEnabled && !isWakeEnabled) {
+        log.debug('dispatch', 'All requires are disabled, returning true by default.');
+        return true;
     }
+    const requireMention = isMentionEnabled &&
+        (content.includes('[提及]@全体成员') ||
+            (!!loginInfo?.userId && content.includes(`[提及]@${loginInfo.userId}`)));
+    const requirePoke = isPokeEnabled &&
+        (content.includes('[动作]') && targetId === loginInfo?.userId);
+    const requireWake = isWakeEnabled &&
+        content.includes(config.wakeWord ?? "");
+    log.debug('dispatch', `Require mention: ${requireMention}, require poke: ${requirePoke}, require wake: ${requireWake}`);
+    return requireMention || requirePoke || requireWake;
 }
 /**
- * Dispatch an incoming message to the AI for processing
+ * 防抖器
+ * @param params
  */
-export async function dispatchMessage(params) {
-    let { chatType, chatId, senderId, senderName, messageId, content, media, timestamp, targetId } = params;
-    const runtime = getRuntime();
-    if (!runtime) {
-        log.warn('dispatch', `Plugin runtime not available`);
-        return;
-    }
-    const context = getContext();
-    if (!context) {
-        log.warn('dispatch', `No gateway context`);
-        return;
-    }
-    const isGroup = chatType === 'group';
-    const config = context.account;
-    // At 模式处理
+function createDebouncer(params) {
+    return createChannelInboundDebouncer({
+        cfg: params.cfg,
+        channel: QQ_CHANNEL,
+        buildKey: (item) => {
+            const peerId = item.msg.isGroup
+                ? (item.msg.groupId ?? item.msg.senderId)
+                : item.msg.senderId;
+            return `qq:${item.account.accountId}:${peerId}`;
+        },
+        shouldDebounce: (item) => shouldDebounceTextInbound({
+            text: item.msg.text,
+            cfg: item.cfg,
+            hasMedia: item.msg.hasMedia,
+        }),
+        onFlush: async (items) => {
+            if (items.length === 0)
+                return;
+            const mergedText = items
+                .map((item) => item.msg.text)
+                .filter(Boolean)
+                .join("\n");
+            const first = items[0];
+            const mergedMsg = {
+                ...first.msg,
+                text: mergedText,
+            };
+            await processInboundMessage({
+                ...first,
+                msg: mergedMsg,
+            });
+        },
+        onError: (err, items) => {
+            log.error('dispatch', `debounce flush failed for ${items.length} items:`, err);
+        },
+    });
+}
+/**
+ * 入站消息解析
+ * @param params
+ */
+async function processInboundMessage(params) {
+    const { cfg, account, runtime, msg } = params;
+    const isGroup = msg.isGroup;
+    const loginInfo = getLoginInfo();
+    const peerId = isGroup ? (msg.groupId ?? msg.senderId) : msg.senderId;
+    // For group messages, check @mention requirement
     if (isGroup) {
-        const isMention = mention(content, chatId, targetId);
+        const isMention = mention(account, msg.text, peerId, msg.targetId, loginInfo);
         if (!isMention) {
             log.debug('dispatch', `Skipping group message (not mentioned)`);
-            const groupConfig = getGroupConfig(chatId, config);
+            const groupConfig = getQQConfigByChatType(true, peerId, account);
             recordPendingHistoryEntry({
                 historyMap: historyCache,
-                historyKey: chatId,
+                historyKey: peerId,
                 limit: groupConfig.historyLimit ?? 20,
                 entry: {
-                    sender: `${senderName}(${senderId})`,
-                    body: content,
-                    timestamp: timestamp,
-                    messageId: messageId,
+                    sender: `${msg.senderName}(${msg.senderId})`,
+                    body: msg.text,
+                    timestamp: msg.timestamp,
+                    messageId: msg.messageId,
                 },
             });
             return;
         }
     }
-    const { route, buildEnvelope } = resolveInboundRouteEnvelopeBuilderWithRuntime({
-        cfg: context.cfg,
-        channel: CHANNEL_ID,
-        accountId: context.accountId,
+    const { channel } = runtime;
+    // 1.解析路由
+    const route = channel.routing.resolveAgentRoute({
+        cfg,
+        channel: QQ_CHANNEL,
+        accountId: account.accountId,
         peer: {
             kind: isGroup ? "group" : "direct",
-            id: chatId,
+            id: peerId,
         },
-        runtime: runtime.channel,
-        sessionStore: context.cfg.session?.store
     });
-    // 终止信号
-    const session = getSession(route.sessionKey);
-    if (session.abortController) {
-        session.abortController.abort();
-        session.aborted = true;
-        log.info('dispatch', `Aborted previous session`);
-    }
-    if (isGroup) {
-        const groupConfig = getGroupConfig(chatId, config);
-        content = buildPendingHistoryContextFromMap({
-            historyMap: historyCache,
-            historyKey: chatId,
-            limit: groupConfig.historyLimit ?? 20,
-            currentMessage: content,
-            formatEntry: (e) => `${e.sender}: ${e.body}`,
-        });
-    }
-    const fromLabel = isGroup ? `group:${chatId}` : senderName || `user:${senderId}`;
-    const { storePath, body } = buildEnvelope({
-        channel: CHANNEL_ID,
+    // 2.构建 Envelope
+    const storePath = channel.session.resolveStorePath(cfg.session?.store, {
+        agentId: route.agentId
+    });
+    const previousTimestamp = channel.session.readSessionUpdatedAt({
+        storePath,
+        sessionKey: route.sessionKey,
+    });
+    const envelopeOptions = channel.reply.resolveEnvelopeFormatOptions(cfg);
+    const fromLabel = isGroup ? `group:${peerId}` : (msg.senderName ?? `user:${msg.senderId}`);
+    const body = channel.reply.formatAgentEnvelope({
+        channel: QQ_CHANNEL,
         from: fromLabel,
-        body: content,
-        timestamp,
+        timestamp: msg.timestamp,
+        previousTimestamp,
+        envelope: envelopeOptions,
+        body: msg.text,
     });
-    log.debug('dispatch', `Inbound envelope: ${body}`);
-    const fromAddress = `qq:${fromLabel}`;
-    const toAddress = `qq:${chatId}`;
-    const ctxPayload = runtime.channel.reply.finalizeInboundContext({
+    // 3.构建消息上下文
+    const ctxPayload = channel.reply.finalizeInboundContext({
         Body: body,
-        RawBody: content,
-        CommandBody: content,
-        From: fromAddress,
-        To: toAddress,
+        BodyForAgent: msg.text,
+        RawBody: msg.text,
+        CommandBody: msg.text,
+        From: `${QQ_CHANNEL}:${msg.senderId}`,
+        To: `${QQ_CHANNEL}:${peerId}`,
         SessionKey: route.sessionKey,
         AccountId: route.accountId,
-        ChatType: isGroup ? 'group' : 'direct',
+        ChatType: isGroup ? "channel" : "direct",
         ConversationLabel: fromLabel,
-        SenderId: senderId,
-        SenderName: senderName,
-        Provider: CHANNEL_ID,
-        Surface: CHANNEL_ID,
-        MessageSid: messageId,
-        Timestamp: timestamp,
-        MediaType: media?.type,
-        MediaPath: media?.path,
-        MediaUrl: media?.url,
-        OriginatingChannel: CHANNEL_ID,
-        OriginatingTo: toAddress,
+        SenderId: msg.senderId,
+        SenderName: msg.senderName,
+        WasMentioned: isGroup ? (msg.wasMentioned ?? false) : undefined,
+        Provider: QQ_CHANNEL,
+        Surface: QQ_CHANNEL,
+        MessageSid: msg.messageId,
+        MessageSidFull: msg.messageId,
+        Timestamp: msg.timestamp,
+        MediaType: msg.media?.type,
+        MediaPath: msg.media?.path,
+        MediaUrl: msg.media?.url,
+        ReplyToId: msg.replyToId,
+        OriginatingChannel: QQ_CHANNEL,
+        OriginatingTo: `${QQ_CHANNEL}:${peerId}`,
     });
-    log.info('dispatch', `Dispatching to agent ${route.agentId}, session: ${route.sessionKey}`);
-    await runtime.channel.session.recordInboundSession({
+    // 4.记录 Session
+    channel.session.recordSessionMetaFromInbound({
         storePath,
-        sessionKey: route.sessionKey,
+        sessionKey: ctxPayload.SessionKey ?? route.sessionKey,
         ctx: ctxPayload,
-        onRecordError(err) {
-            log.error('dispatch', `Failed to record inbound session: ${err}`);
-        },
+    }).catch((err) => {
+        log.error('dispatch', `session record failed: ${err}`);
     });
-    const messagesConfig = runtime.channel.reply.resolveEffectiveMessagesConfig(context.cfg, route.agentId);
-    try {
-        session.abortController = new AbortController();
-        updateSession(route.sessionKey, session);
-        await runtime.channel.reply.dispatchReplyWithBufferedBlockDispatcher({
-            ctx: ctxPayload,
-            cfg: context.cfg,
-            dispatcherOptions: {
-                humanDelay: {
-                    mode: "off"
-                },
-                responsePrefix: messagesConfig.responsePrefix,
-                onReplyStart: async () => {
-                    if (!isGroup) {
-                        // 输入状态
-                        await setInputStatus({
-                            user_id: senderId,
-                            event_type: 1
-                        });
-                    }
-                },
-                deliver: async (payload, info) => {
-                    if (session.aborted) {
-                        session.aborted = false;
-                        log.info('dispatch', `aborted skipping`);
-                        return;
-                    }
-                    if (isGroup) {
-                        clearHistoryEntries({ historyMap: historyCache, historyKey: chatId });
-                    }
-                    log.info('dispatch', `deliver(${info.kind}): ${JSON.stringify(payload)}`);
-                    if (payload.text && !payload.text.startsWith('MEDIA:')) {
-                        await sendText(isGroup, chatId, payload.text);
-                    }
-                    if (payload.text && payload.text.startsWith('MEDIA:')) {
-                        await sendMedia(isGroup, chatId, payload.text.replace('MEDIA:', ''));
-                    }
-                    if (payload.mediaUrl) {
-                        await sendMedia(isGroup, chatId, payload.mediaUrl);
-                    }
-                    if (payload.mediaUrls && payload.mediaUrls.length > 0) {
-                        for (const mediaUrl of payload.mediaUrls) {
-                            await sendMedia(isGroup, chatId, mediaUrl);
-                        }
+    // 5.创建 Reply Pipeline
+    const { onModelSelected, ...replyPipeline } = createChannelReplyPipeline({
+        cfg,
+        agentId: route.agentId,
+        channel: QQ_CHANNEL,
+        accountId: route.accountId,
+    });
+    // 6.分发 Reply
+    await channel.reply.dispatchReplyWithBufferedBlockDispatcher({
+        ctx: ctxPayload,
+        cfg,
+        dispatcherOptions: {
+            ...replyPipeline,
+            onReplyStart: async () => {
+                if (!isGroup) {
+                    // 输入状态
+                    await setInputStatus({
+                        user_id: msg.senderId,
+                        event_type: 1
+                    });
+                }
+            },
+            deliver: async (payload) => {
+                const reply = resolveSendableOutboundReplyParts(payload);
+                if (!reply.hasContent)
+                    return;
+                const chunkLimit = 4000;
+                const chunks = reply.trimmedText
+                    ? channel.text.chunkMarkdownText(reply.trimmedText, chunkLimit)
+                    : [];
+                const to = msg.replyToId ?? peerId;
+                const messageSegments = [];
+                for (const chunk of chunks) {
+                    messageSegments.push({ type: "text", text: chunk });
+                }
+                if (reply.hasMedia) {
+                    for (const mediaUrl of reply.mediaUrls) {
+                        messageSegments.push(buildMediaMessage(mediaUrl));
                     }
-                },
-                onError: async (err) => {
-                    log.error('dispatch', `Dispatch error: ${err}`);
-                    await sendText(isGroup, chatId, `[错误]\n${String(err)}`);
-                },
+                }
+                await send(account, isGroup, to, messageSegments);
             },
-            replyOptions: {
-                abortSignal: session.abortController?.signal,
+            onError: (err, info) => {
+                log.error('dispatch', `${info.kind} reply failed: ${err}`);
             },
+        },
+        replyOptions: {
+            onModelSelected,
+        },
+    });
+    // 7.结束输入状态
+    if (!isGroup) {
+        await setInputStatus({
+            user_id: msg.senderId,
+            event_type: 2
         });
-        log.info('dispatch', `Dispatch completed`);
-    }
-    catch (error) {
-        log.error('dispatch', `Message processing failed: ${error}`);
     }
-    finally {
-        if (!isGroup) {
-            // 输入状态
-            await setInputStatus({
-                user_id: senderId,
-                event_type: 2
+}
+export function createInboundHandler(params) {
+    const debouncer = createDebouncer(params);
+    return async (msg) => {
+        const canDebounce = shouldDebounceTextInbound({
+            text: msg.text,
+            cfg: params.cfg,
+            hasMedia: msg.hasMedia,
+        });
+        if (canDebounce && debouncer.debounceMs > 0) {
+            await debouncer.debouncer.enqueue({
+                ...params,
+                msg,
             });
         }
-        clearSession(route.sessionKey);
-    }
-}
-/**
- * Handle group message event
- */
-export async function handleGroupMessage(event) {
-    if (!allow(event.user_id.toString(), event.group_id.toString())) {
-        log.debug('dispatch', `Ignoring group message from ${event.user_id}`);
-        return;
-    }
-    const content = await napCatToOpenClawMessage(event.message);
-    const plainText = await contentToPlainText(content);
-    const media = await contextToMedia(content);
-    log.info('dispatch', `Group message from ${event.sender?.nickname || event.sender?.card || event.user_id}: ${plainText}, media: ${media != undefined}`);
-    await dispatchMessage({
-        chatType: 'group',
-        chatId: String(event.group_id),
-        senderId: String(event.user_id),
-        senderName: event.sender?.nickname || event.sender?.card,
-        messageId: String(event.message_id),
-        content: plainText,
-        media,
-        timestamp: event.time * 1000,
-    });
-}
-/**
- * Handle private message event
- */
-export async function handlePrivateMessage(event) {
-    if (!allow(event.user_id.toString())) {
-        log.debug('dispatch', `Ignoring message from ${event.user_id}`);
-        return;
-    }
-    const content = await napCatToOpenClawMessage(event.message);
-    const plainText = await contentToPlainText(content);
-    const media = await contextToMedia(content);
-    log.info('dispatch', `Private message from ${event.sender?.nickname || event.user_id}: ${plainText}, media: ${media != undefined}`);
-    await dispatchMessage({
-        chatType: 'direct',
-        chatId: String(event.user_id),
-        senderId: String(event.user_id),
-        senderName: event.sender?.nickname,
-        messageId: String(event.message_id),
-        content: plainText,
-        media,
-        timestamp: event.time * 1000,
-    });
-}
-/**
- * Handle poke event
- */
-function extractPokeActionText(rawInfo) {
-    if (!rawInfo)
-        return '戳了戳';
-    const actionItem = rawInfo.find(item => item.type === 'nor' && item.txt);
-    return actionItem?.txt || '戳了戳';
-}
-export async function handlePokeEvent(event) {
-    if (!allow(event.user_id.toString(), event.group_id?.toString())) {
-        log.debug('dispatch', `Poke from ${event.user_id} is not allowed`);
-        return;
-    }
-    const actionText = extractPokeActionText(event.raw_info);
-    log.info('dispatch', `Poke from ${event.user_id}: ${actionText}`);
-    const pokeMessage = actionText || '戳了戳';
-    const chatType = event.group_id ? 'group' : 'direct';
-    const chatId = String(event.group_id || event.user_id);
-    await dispatchMessage({
-        chatType,
-        chatId,
-        senderId: String(event.user_id),
-        senderName: String(event.user_id),
-        messageId: `poke_${event.user_id}_${Date.now()}`,
-        content: `[动作]${pokeMessage}`,
-        timestamp: Date.now(),
-        targetId: String(event.target_id),
-    });
-}
-function getGroupConfig(groupId, config) {
-    log.debug('dispatch', `All Custom config: ${JSON.stringify(config.messageGroupsCustom)}`);
-    let groupConfig = config.messageGroupsCustom[groupId];
-    if (!groupConfig) {
-        groupConfig = config.messageGroup;
-        log.debug('dispatch', `Use global config: ${JSON.stringify(groupConfig)}`);
-    }
-    else {
-        groupConfig = {
-            ...config.messageGroup,
-            ...groupConfig,
-        };
-    }
-    log.debug('dispatch', `Final config: ${JSON.stringify(groupConfig)}`);
-    return groupConfig;
-}
-function allow(userId, groupId) {
-    const context = getContext();
-    if (!context) {
-        log.warn('dispatch', `No gateway context`);
-        return false;
-    }
-    let config = groupId ? getGroupConfig(groupId, context.account) : context.account.messageDirect;
-    return allowJudgment(config, userId);
-}
-function allowJudgment(config, userId) {
-    if (config.denyFrom?.includes(userId)) {
-        log.debug('dispatch', `User ${userId} is denied`);
-        return false;
-    }
-    if (config.policy === 'allow') {
-        log.debug('dispatch', `User ${userId} is allowed`);
-        return true;
-    }
-    if (config.policy === 'deny') {
-        log.debug('dispatch', `User ${userId} is denied`);
-        return false;
-    }
-    if (config.allowFrom?.includes(userId)) {
-        log.debug('dispatch', `User ${userId} is allowed`);
-        return true;
-    }
-    return false;
-}
-function mention(content, groupId, targetId) {
-    const context = getContext();
-    if (!context) {
-        log.warn('dispatch', `No gateway context`);
-        return false;
-    }
-    let config = getGroupConfig(groupId, context.account);
-    const loginInfo = getLoginInfo();
-    const isMentionEnabled = !!config?.requireMention;
-    const isPokeEnabled = !!config?.requirePoke;
-    const isWakeEnabled = !!config?.wakeWord?.trim();
-    if (!isMentionEnabled && !isPokeEnabled && !isWakeEnabled) {
-        log.debug('dispatch', 'All requires are disabled, returning true by default.');
-        return true;
-    }
-    const requireMention = isMentionEnabled &&
-        (content.includes('[提及]@全体成员') ||
-            (!!loginInfo?.userId && content.includes(`[提及]@${loginInfo.userId}`)));
-    const requirePoke = isPokeEnabled &&
-        (content.includes('[动作]') && targetId === loginInfo?.userId);
-    const requireWake = isWakeEnabled &&
-        content.includes(config.wakeWord ?? "");
-    log.debug('dispatch', `Require mention: ${requireMention}, require poke: ${requirePoke}, require wake: ${requireWake}`);
-    return requireMention || requirePoke || requireWake;
+        else {
+            await processInboundMessage({
+                ...params,
+                msg,
+            });
+        }
+    };
 }