Merge pull request #21 from sliverp/feature/group

Feature/group
This commit is contained in:
Bijin
2026-01-30 18:59:56 +08:00
committed by GitHub
12 changed files with 10832 additions and 79 deletions

2
.gitignore vendored Normal file
View File

@@ -0,0 +1,2 @@
node_modules
dist

9
moltbot.plugin.json Normal file
View File

@@ -0,0 +1,9 @@
{
"id": "qqbot",
"channels": ["qqbot"],
"configSchema": {
"type": "object",
"additionalProperties": false,
"properties": {}
}
}

9
openclaw.plugin.json Normal file
View File

@@ -0,0 +1,9 @@
{
"id": "qqbot",
"channels": ["qqbot"],
"configSchema": {
"type": "object",
"additionalProperties": false,
"properties": {}
}
}

9852
package-lock.json generated Normal file

File diff suppressed because it is too large Load Diff

View File

@@ -1,6 +1,6 @@
{
"name": "qqbot",
"version": "1.2.1",
"version": "1.2.3",
"type": "module",
"main": "dist/index.js",
"types": "dist/index.d.ts",
@@ -10,6 +10,9 @@
"moltbot": {
"extensions": ["./index.ts"]
},
"openclaw": {
"extensions": ["./index.ts"]
},
"scripts": {
"build": "tsc || true",
"dev": "tsc --watch",
@@ -27,6 +30,8 @@
"typescript": "^5.0.0"
},
"peerDependencies": {
"clawdbot": "*"
"clawdbot": "*",
"moltbot": "*",
"openclaw": "*"
}
}

View File

@@ -56,18 +56,21 @@ export function clearTokenCache(): void {
/**
* msg_seq 追踪器 - 用于对同一条消息的多次回复
* key: msg_id, value: 当前 seq 值
* 使用时间戳作为基础值,确保进程重启后不会重复
*/
const msgSeqTracker = new Map<string, number>();
const seqBaseTime = Math.floor(Date.now() / 1000) % 100000000; // 取秒级时间戳的后8位作为基础
/**
* 获取并递增消息序号
* 返回的 seq 会基于时间戳,避免进程重启后重复
*/
export function getNextMsgSeq(msgId: string): number {
const current = msgSeqTracker.get(msgId) ?? 0;
const next = current + 1;
msgSeqTracker.set(msgId, next);
// 清理过期的序号(超过 5 次或 60 分钟后无意义)
// 清理过期的序号
// 简单策略:保留最近 1000 条
if (msgSeqTracker.size > 1000) {
const keys = Array.from(msgSeqTracker.keys());
@@ -76,7 +79,8 @@ export function getNextMsgSeq(msgId: string): number {
}
}
return next;
// 结合时间戳基础值,确保唯一性
return seqBaseTime + next;
}
/**
@@ -209,3 +213,161 @@ export async function sendProactiveGroupMessage(
msg_type: 0,
});
}
// ============ 富媒体消息支持 ============
/**
* 媒体文件类型
*/
export enum MediaFileType {
IMAGE = 1,
VIDEO = 2,
VOICE = 3,
FILE = 4, // 暂未开放
}
/**
* 上传富媒体文件的响应
*/
export interface UploadMediaResponse {
file_uuid: string;
file_info: string;
ttl: number;
id?: string; // 仅当 srv_send_msg=true 时返回
}
/**
* 上传富媒体文件到 C2C 单聊
* @param accessToken 访问令牌
* @param openid 用户 openid
* @param fileType 文件类型
* @param url 媒体资源 URL
* @param srvSendMsg 是否直接发送(推荐 false获取 file_info 后再发送)
*/
export async function uploadC2CMedia(
accessToken: string,
openid: string,
fileType: MediaFileType,
url: string,
srvSendMsg = false
): Promise<UploadMediaResponse> {
return apiRequest(accessToken, "POST", `/v2/users/${openid}/files`, {
file_type: fileType,
url,
srv_send_msg: srvSendMsg,
});
}
/**
* 上传富媒体文件到群聊
* @param accessToken 访问令牌
* @param groupOpenid 群 openid
* @param fileType 文件类型
* @param url 媒体资源 URL
* @param srvSendMsg 是否直接发送(推荐 false获取 file_info 后再发送)
*/
export async function uploadGroupMedia(
accessToken: string,
groupOpenid: string,
fileType: MediaFileType,
url: string,
srvSendMsg = false
): Promise<UploadMediaResponse> {
return apiRequest(accessToken, "POST", `/v2/groups/${groupOpenid}/files`, {
file_type: fileType,
url,
srv_send_msg: srvSendMsg,
});
}
/**
* 发送 C2C 单聊富媒体消息
* @param accessToken 访问令牌
* @param openid 用户 openid
* @param fileInfo 从 uploadC2CMedia 获取的 file_info
* @param msgId 被动回复时需要的消息 ID
* @param content 可选的文字内容
*/
export async function sendC2CMediaMessage(
accessToken: string,
openid: string,
fileInfo: string,
msgId?: string,
content?: string
): Promise<{ id: string; timestamp: number }> {
const msgSeq = msgId ? getNextMsgSeq(msgId) : 1;
return apiRequest(accessToken, "POST", `/v2/users/${openid}/messages`, {
msg_type: 7, // 富媒体消息类型
media: { file_info: fileInfo },
msg_seq: msgSeq,
...(content ? { content } : {}),
...(msgId ? { msg_id: msgId } : {}),
});
}
/**
* 发送群聊富媒体消息
* @param accessToken 访问令牌
* @param groupOpenid 群 openid
* @param fileInfo 从 uploadGroupMedia 获取的 file_info
* @param msgId 被动回复时需要的消息 ID
* @param content 可选的文字内容
*/
export async function sendGroupMediaMessage(
accessToken: string,
groupOpenid: string,
fileInfo: string,
msgId?: string,
content?: string
): Promise<{ id: string; timestamp: string }> {
const msgSeq = msgId ? getNextMsgSeq(msgId) : 1;
return apiRequest(accessToken, "POST", `/v2/groups/${groupOpenid}/messages`, {
msg_type: 7, // 富媒体消息类型
media: { file_info: fileInfo },
msg_seq: msgSeq,
...(content ? { content } : {}),
...(msgId ? { msg_id: msgId } : {}),
});
}
/**
* 发送带图片的 C2C 单聊消息(封装上传+发送)
* @param accessToken 访问令牌
* @param openid 用户 openid
* @param imageUrl 图片 URL
* @param msgId 被动回复时需要的消息 ID
* @param content 可选的文字内容
*/
export async function sendC2CImageMessage(
accessToken: string,
openid: string,
imageUrl: string,
msgId?: string,
content?: string
): Promise<{ id: string; timestamp: number }> {
// 先上传图片获取 file_info
const uploadResult = await uploadC2CMedia(accessToken, openid, MediaFileType.IMAGE, imageUrl, false);
// 再发送富媒体消息
return sendC2CMediaMessage(accessToken, openid, uploadResult.file_info, msgId, content);
}
/**
* 发送带图片的群聊消息(封装上传+发送)
* @param accessToken 访问令牌
* @param groupOpenid 群 openid
* @param imageUrl 图片 URL
* @param msgId 被动回复时需要的消息 ID
* @param content 可选的文字内容
*/
export async function sendGroupImageMessage(
accessToken: string,
groupOpenid: string,
imageUrl: string,
msgId?: string,
content?: string
): Promise<{ id: string; timestamp: string }> {
// 先上传图片获取 file_info
const uploadResult = await uploadGroupMedia(accessToken, groupOpenid, MediaFileType.IMAGE, imageUrl, false);
// 再发送富媒体消息
return sendGroupMediaMessage(accessToken, groupOpenid, uploadResult.file_info, msgId, content);
}

View File

@@ -30,13 +30,13 @@ export const qqbotPlugin: ChannelPlugin<ResolvedQQBotAccount> = {
listAccountIds: (cfg) => listQQBotAccountIds(cfg),
resolveAccount: (cfg, accountId) => resolveQQBotAccount(cfg, accountId),
defaultAccountId: () => DEFAULT_ACCOUNT_ID,
isConfigured: (account) => Boolean(account.appId && account.clientSecret),
isConfigured: (account) => Boolean(account?.appId && account?.clientSecret),
describeAccount: (account) => ({
accountId: account.accountId,
name: account.name,
enabled: account.enabled,
configured: Boolean(account.appId && account.clientSecret),
tokenSource: account.secretSource,
accountId: account?.accountId ?? DEFAULT_ACCOUNT_ID,
name: account?.name,
enabled: account?.enabled ?? false,
configured: Boolean(account?.appId && account?.clientSecret),
tokenSource: account?.secretSource,
}),
},
setup: {
@@ -63,6 +63,7 @@ export const qqbotPlugin: ChannelPlugin<ResolvedQQBotAccount> = {
clientSecret,
clientSecretFile: input.tokenFile,
name: input.name,
imageServerBaseUrl: input.imageServerBaseUrl,
});
},
},
@@ -118,11 +119,11 @@ export const qqbotPlugin: ChannelPlugin<ResolvedQQBotAccount> = {
lastError: null,
},
buildAccountSnapshot: ({ account, runtime }) => ({
accountId: account.accountId,
name: account.name,
enabled: account.enabled,
configured: Boolean(account.appId && account.clientSecret),
tokenSource: account.secretSource,
accountId: account?.accountId ?? DEFAULT_ACCOUNT_ID,
name: account?.name,
enabled: account?.enabled ?? false,
configured: Boolean(account?.appId && account?.clientSecret),
tokenSource: account?.secretSource,
running: runtime?.running ?? false,
connected: runtime?.connected ?? false,
lastConnectedAt: runtime?.lastConnectedAt ?? null,

View File

@@ -63,6 +63,7 @@ export function resolveQQBotAccount(
dmPolicy: qqbot?.dmPolicy,
allowFrom: qqbot?.allowFrom,
systemPrompt: qqbot?.systemPrompt,
imageServerBaseUrl: qqbot?.imageServerBaseUrl,
};
appId = qqbot?.appId ?? "";
} else {
@@ -97,6 +98,7 @@ export function resolveQQBotAccount(
clientSecret,
secretSource,
systemPrompt: accountConfig.systemPrompt,
imageServerBaseUrl: accountConfig.imageServerBaseUrl || process.env.QQBOT_IMAGE_SERVER_BASE_URL,
config: accountConfig,
};
}
@@ -107,7 +109,7 @@ export function resolveQQBotAccount(
export function applyQQBotAccountConfig(
cfg: MoltbotConfig,
accountId: string,
input: { appId?: string; clientSecret?: string; clientSecretFile?: string; name?: string }
input: { appId?: string; clientSecret?: string; clientSecretFile?: string; name?: string; imageServerBaseUrl?: string }
): MoltbotConfig {
const next = { ...cfg };
@@ -124,6 +126,7 @@ export function applyQQBotAccountConfig(
? { clientSecretFile: input.clientSecretFile }
: {}),
...(input.name ? { name: input.name } : {}),
...(input.imageServerBaseUrl ? { imageServerBaseUrl: input.imageServerBaseUrl } : {}),
},
};
} else {
@@ -144,6 +147,7 @@ export function applyQQBotAccountConfig(
? { clientSecretFile: input.clientSecretFile }
: {}),
...(input.name ? { name: input.name } : {}),
...(input.imageServerBaseUrl ? { imageServerBaseUrl: input.imageServerBaseUrl } : {}),
},
},
},

View File

@@ -1,17 +1,43 @@
import WebSocket from "ws";
import path from "node:path";
import type { ResolvedQQBotAccount, WSPayload, C2CMessageEvent, GuildMessageEvent, GroupMessageEvent } from "./types.js";
import { getAccessToken, getGatewayUrl, sendC2CMessage, sendChannelMessage, sendGroupMessage, clearTokenCache } from "./api.js";
import { getAccessToken, getGatewayUrl, sendC2CMessage, sendChannelMessage, sendGroupMessage, clearTokenCache, sendC2CImageMessage, sendGroupImageMessage } from "./api.js";
import { getQQBotRuntime } from "./runtime.js";
import { startImageServer, saveImage, saveImageFromPath, isImageServerRunning, downloadFile, type ImageServerConfig } from "./image-server.js";
// QQ Bot intents
// QQ Bot intents - 按权限级别分组
const INTENTS = {
// 基础权限(默认有)
GUILDS: 1 << 0, // 频道相关
GUILD_MEMBERS: 1 << 1, // 频道成员
PUBLIC_GUILD_MESSAGES: 1 << 30, // 频道公开消息(公域)
// 需要申请的权限
DIRECT_MESSAGE: 1 << 12, // 频道私信
GROUP_AND_C2C: 1 << 25, // 群聊和 C2C 私聊(需申请)
};
// 权限级别:从高到低依次尝试
const INTENT_LEVELS = [
// Level 0: 完整权限(群聊 + 私信 + 频道)
{
name: "full",
intents: INTENTS.PUBLIC_GUILD_MESSAGES | INTENTS.DIRECT_MESSAGE | INTENTS.GROUP_AND_C2C,
description: "群聊+私信+频道",
},
// Level 1: 群聊 + 频道(无私信)
{
name: "group+channel",
intents: INTENTS.PUBLIC_GUILD_MESSAGES | INTENTS.GROUP_AND_C2C,
description: "群聊+频道",
},
// Level 2: 仅频道(基础权限)
{
name: "channel-only",
intents: INTENTS.PUBLIC_GUILD_MESSAGES | INTENTS.GUILD_MEMBERS,
description: "仅频道消息",
},
];
// 重连配置
const RECONNECT_DELAYS = [1000, 2000, 5000, 10000, 30000, 60000]; // 递增延迟
const RATE_LIMIT_DELAY = 60000; // 遇到频率限制时等待 60 秒
@@ -19,6 +45,11 @@ const MAX_RECONNECT_ATTEMPTS = 100;
const MAX_QUICK_DISCONNECT_COUNT = 3; // 连续快速断开次数阈值
const QUICK_DISCONNECT_THRESHOLD = 5000; // 5秒内断开视为快速断开
// 图床服务器配置(可通过环境变量覆盖)
const IMAGE_SERVER_PORT = parseInt(process.env.QQBOT_IMAGE_SERVER_PORT || "18765", 10);
// 使用绝对路径,确保文件保存和读取使用同一目录
const IMAGE_SERVER_DIR = process.env.QQBOT_IMAGE_SERVER_DIR || path.join(process.env.HOME || "/home/ubuntu", "clawd", "qqbot-images");
export interface GatewayContext {
account: ResolvedQQBotAccount;
abortSignal: AbortSignal;
@@ -32,6 +63,31 @@ export interface GatewayContext {
};
}
/**
* 启动图床服务器
*/
async function ensureImageServer(log?: GatewayContext["log"], publicBaseUrl?: string): Promise<string | null> {
if (isImageServerRunning()) {
return publicBaseUrl || `http://0.0.0.0:${IMAGE_SERVER_PORT}`;
}
try {
const config: Partial<ImageServerConfig> = {
port: IMAGE_SERVER_PORT,
storageDir: IMAGE_SERVER_DIR,
// 使用用户配置的公网地址,而不是 0.0.0.0
baseUrl: publicBaseUrl || `http://0.0.0.0:${IMAGE_SERVER_PORT}`,
ttlSeconds: 3600, // 1 小时过期
};
await startImageServer(config);
log?.info(`[qqbot] Image server started on port ${IMAGE_SERVER_PORT}, baseUrl: ${config.baseUrl}`);
return config.baseUrl!;
} catch (err) {
log?.error(`[qqbot] Failed to start image server: ${err}`);
return null;
}
}
/**
* 启动 Gateway WebSocket 连接(带自动重连)
*/
@@ -42,6 +98,17 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
throw new Error("QQBot not configured (missing appId or clientSecret)");
}
// 如果配置了公网 URL启动图床服务器
let imageServerBaseUrl: string | null = null;
if (account.imageServerBaseUrl) {
// 使用用户配置的公网地址作为 baseUrl
await ensureImageServer(log, account.imageServerBaseUrl);
imageServerBaseUrl = account.imageServerBaseUrl;
log?.info(`[qqbot:${account.accountId}] Image server enabled with URL: ${imageServerBaseUrl}`);
} else {
log?.info(`[qqbot:${account.accountId}] Image server disabled (no imageServerBaseUrl configured)`);
}
let reconnectAttempts = 0;
let isAborted = false;
let currentWs: WebSocket | null = null;
@@ -53,7 +120,8 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
let isConnecting = false; // 防止并发连接
let reconnectTimer: ReturnType<typeof setTimeout> | null = null; // 重连定时器
let shouldRefreshToken = false; // 下次连接是否需要刷新 token
let identifyFailCount = 0; // identify 失败次数
let intentLevelIndex = 0; // 当前尝试的权限级别索引
let lastSuccessfulIntentLevel = -1; // 上次成功的权限级别
abortSignal.addEventListener("abort", () => {
isAborted = true;
@@ -174,22 +242,49 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
const envelopeOptions = pluginRuntime.channel.reply.resolveEnvelopeFormatOptions(cfg);
// 组装消息体,添加系统提示词
const builtinPrompt = "由于平台限制你的回复中不可以包含任何URL";
let builtinPrompt = "由于平台限制你的回复中不可以包含任何URL";
// 只有配置了图床公网地址,才告诉 AI 可以发送图片
if (imageServerBaseUrl) {
builtinPrompt += `
【发送图片】
你可以发送本地图片文件给用户。只需在回复中直接引用图片的绝对路径即可,系统会自动处理。
支持 png、jpg、gif、webp 格式。`;
}
const systemPrompts = [builtinPrompt];
if (account.systemPrompt) {
systemPrompts.push(account.systemPrompt);
}
// 处理附件(图片等)
// 处理附件(图片等)- 下载到本地供 clawdbot 访问
let attachmentInfo = "";
const imageUrls: string[] = [];
// 存到 clawdbot 工作目录下的 downloads 文件夹
const downloadDir = path.join(process.env.HOME || "/home/ubuntu", "clawd", "downloads");
if (event.attachments?.length) {
for (const att of event.attachments) {
if (att.content_type?.startsWith("image/")) {
imageUrls.push(att.url);
attachmentInfo += `\n[图片: ${att.url}]`;
// 下载附件到本地,使用原始文件名
const localPath = await downloadFile(att.url, downloadDir, att.filename);
if (localPath) {
if (att.content_type?.startsWith("image/")) {
imageUrls.push(localPath);
attachmentInfo += `\n[图片: ${localPath}]`;
} else {
attachmentInfo += `\n[附件: ${localPath}]`;
}
log?.info(`[qqbot:${account.accountId}] Downloaded attachment to: ${localPath}`);
} else {
attachmentInfo += `\n[附件: ${att.filename ?? att.content_type}]`;
// 下载失败,提供原始 URL 作为后备
log?.error(`[qqbot:${account.accountId}] Failed to download attachment: ${att.url}`);
if (att.content_type?.startsWith("image/")) {
imageUrls.push(att.url);
attachmentInfo += `\n[图片: ${att.url}] (下载失败,可能无法访问)`;
} else {
attachmentInfo += `\n[附件: ${att.filename ?? att.content_type}] (下载失败)`;
}
}
}
}
@@ -296,39 +391,185 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
cfg,
dispatcherOptions: {
responsePrefix: messagesConfig.responsePrefix,
deliver: async (payload: { text?: string }) => {
deliver: async (payload: { text?: string; mediaUrls?: string[]; mediaUrl?: string }) => {
hasResponse = true;
if (timeoutId) {
clearTimeout(timeoutId);
timeoutId = null;
}
let replyText = payload.text ?? "";
if (!replyText.trim()) return;
log?.info(`[qqbot:${account.accountId}] deliver called, payload keys: ${Object.keys(payload).join(", ")}`);
// 处理回复内容,避免被 QQ 识别为 URL
const originalText = replyText;
let replyText = payload.text ?? "";
// 把所有可能被识别为 URL 的点替换为下划线
// 匹配:字母/数字.字母/数字 的模式
replyText = replyText.replace(/([a-zA-Z0-9])\.([a-zA-Z0-9])/g, "$1_$2");
// 收集所有图片路径
const imageUrls: string[] = [];
const hasReplacement = replyText !== originalText;
if (hasReplacement) {
replyText += "\n\n由于平台限制回复中的部分符号已被替换";
// 处理 mediaUrls 和 mediaUrl 字段(本地文件路径)
const mediaPaths: string[] = [];
if (payload.mediaUrls?.length) {
mediaPaths.push(...payload.mediaUrls);
}
if (payload.mediaUrl && !mediaPaths.includes(payload.mediaUrl)) {
mediaPaths.push(payload.mediaUrl);
}
for (const localPath of mediaPaths) {
if (localPath && imageServerBaseUrl) {
try {
const savedUrl = saveImageFromPath(localPath);
if (savedUrl) {
imageUrls.push(savedUrl);
log?.info(`[qqbot:${account.accountId}] Saved media to server: ${localPath}`);
} else {
log?.error(`[qqbot:${account.accountId}] Failed to save media (not found or not image): ${localPath}`);
}
} catch (err) {
log?.error(`[qqbot:${account.accountId}] Failed to save media: ${err}`);
}
}
}
// 如果没有文本也没有图片,跳过
if (!replyText.trim() && imageUrls.length === 0) {
log?.info(`[qqbot:${account.accountId}] Empty reply, skipping`);
return;
}
// 0. 提取 MEDIA: 前缀的本地文件路径(从文本中)
const mediaPathRegex = /MEDIA:([^\s\n]+)/gi;
const mediaMatches = [...replyText.matchAll(mediaPathRegex)];
for (const match of mediaMatches) {
const localPath = match[1];
if (localPath && imageServerBaseUrl) {
// 将本地文件复制到图床
try {
const savedUrl = saveImageFromPath(localPath);
if (savedUrl) {
imageUrls.push(savedUrl);
log?.info(`[qqbot:${account.accountId}] Saved local image to server: ${localPath}`);
} else {
log?.error(`[qqbot:${account.accountId}] Failed to save local image (not found or not image): ${localPath}`);
}
} catch (err) {
log?.error(`[qqbot:${account.accountId}] Failed to save local image: ${err}`);
}
}
// 从文本中移除 MEDIA: 行
replyText = replyText.replace(match[0], "").trim();
}
// 0.5. 提取本地绝对文件路径(/path/to/image.png 或 /path/to/image_123_png 格式)
// 支持标准扩展名和下划线替换后的扩展名
const localPathRegex = /(\/[^\s\n]+?(?:\.(?:png|jpg|jpeg|gif|webp)|_(?:png|jpg|jpeg|gif|webp)(?:\s|$)))/gi;
const localPathMatches = [...replyText.matchAll(localPathRegex)];
for (const match of localPathMatches) {
let localPath = match[1].trim();
if (localPath && imageServerBaseUrl) {
// 如果是下划线格式的扩展名,转换回点格式
localPath = localPath.replace(/_(?=(?:png|jpg|jpeg|gif|webp)$)/, ".");
try {
const savedUrl = saveImageFromPath(localPath);
if (savedUrl) {
imageUrls.push(savedUrl);
log?.info(`[qqbot:${account.accountId}] Saved local path image to server: ${localPath}`);
} else {
log?.error(`[qqbot:${account.accountId}] Local path not found or not image: ${localPath}`);
}
} catch (err) {
log?.error(`[qqbot:${account.accountId}] Failed to save local path image: ${err}`);
}
}
// 从文本中移除本地路径
replyText = replyText.replace(match[0], "").trim();
}
// 1. 提取 base64 图片data:image/xxx;base64,...
const base64ImageRegex = /!\[([^\]]*)\]\((data:image\/[^;]+;base64,[A-Za-z0-9+/=]+)\)|(?<![(\[])(data:image\/[^;]+;base64,[A-Za-z0-9+/=]+)/gi;
const base64Matches = [...replyText.matchAll(base64ImageRegex)];
for (const match of base64Matches) {
const dataUrl = match[2] || match[3];
if (dataUrl && imageServerBaseUrl) {
// 将 base64 保存到本地图床
try {
const savedUrl = saveImage(dataUrl);
imageUrls.push(savedUrl);
log?.info(`[qqbot:${account.accountId}] Saved base64 image to local server`);
} catch (err) {
log?.error(`[qqbot:${account.accountId}] Failed to save base64 image: ${err}`);
}
}
// 从文本中移除 base64
replyText = replyText.replace(match[0], "").trim();
}
// 2. 提取 URL 图片Markdown 格式或纯 URL
const imageUrlRegex = /!\[([^\]]*)\]\((https?:\/\/[^\s)]+\.(?:png|jpg|jpeg|gif|webp)(?:\?[^\s)]*)?)\)|(?<![(\[])(https?:\/\/[^\s)]+\.(?:png|jpg|jpeg|gif|webp)(?:\?[^\s]*)?)/gi;
const urlMatches = [...replyText.matchAll(imageUrlRegex)];
for (const match of urlMatches) {
// match[2] 是 Markdown 格式的 URLmatch[3] 是纯 URL
const url = match[2] || match[3];
if (url) {
imageUrls.push(url);
}
}
// 从文本中移除图片 URL避免被 QQ 拦截
let textWithoutImages = replyText;
for (const match of urlMatches) {
textWithoutImages = textWithoutImages.replace(match[0], "").trim();
}
// 处理剩余文本中的 URL 点号(只有在没有图片的情况下才替换,避免误伤)
const hasImages = imageUrls.length > 0;
let hasReplacement = false;
if (!hasImages) {
const originalText = textWithoutImages;
textWithoutImages = textWithoutImages.replace(/([a-zA-Z0-9])\.([a-zA-Z0-9])/g, "$1_$2");
hasReplacement = textWithoutImages !== originalText;
if (hasReplacement && textWithoutImages.trim()) {
textWithoutImages += "\n\n由于平台限制回复中的部分符号已被替换";
}
}
try {
await sendWithTokenRetry(async (token) => {
if (event.type === "c2c") {
await sendC2CMessage(token, event.senderId, replyText, event.messageId);
} else if (event.type === "group" && event.groupOpenid) {
await sendGroupMessage(token, event.groupOpenid, replyText, event.messageId);
} else if (event.channelId) {
await sendChannelMessage(token, event.channelId, replyText, event.messageId);
// 先发送图片(如果有)
for (const imageUrl of imageUrls) {
try {
await sendWithTokenRetry(async (token) => {
if (event.type === "c2c") {
await sendC2CImageMessage(token, event.senderId, imageUrl, event.messageId);
} else if (event.type === "group" && event.groupOpenid) {
await sendGroupImageMessage(token, event.groupOpenid, imageUrl, event.messageId);
}
// 频道消息暂不支持富媒体,跳过图片
});
log?.info(`[qqbot:${account.accountId}] Sent image: ${imageUrl.slice(0, 50)}...`);
} catch (imgErr) {
log?.error(`[qqbot:${account.accountId}] Failed to send image: ${imgErr}`);
// 图片发送失败时,显示错误信息而不是 URL
const errMsg = String(imgErr).slice(0, 200);
textWithoutImages = `[图片发送失败: ${errMsg}]\n${textWithoutImages}`;
}
});
log?.info(`[qqbot:${account.accountId}] Sent reply`);
}
// 再发送文本(如果有)
if (textWithoutImages.trim()) {
await sendWithTokenRetry(async (token) => {
if (event.type === "c2c") {
await sendC2CMessage(token, event.senderId, textWithoutImages, event.messageId);
} else if (event.type === "group" && event.groupOpenid) {
await sendGroupMessage(token, event.groupOpenid, textWithoutImages, event.messageId);
} else if (event.channelId) {
await sendChannelMessage(token, event.channelId, textWithoutImages, event.messageId);
}
});
log?.info(`[qqbot:${account.accountId}] Sent text reply`);
}
pluginRuntime.channel.activity.record({
channel: "qqbot",
@@ -346,12 +587,13 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
clearTimeout(timeoutId);
timeoutId = null;
}
// 发送错误提示给用户
// 发送错误提示给用户,显示完整错误信息
const errMsg = String(err);
if (errMsg.includes("401") || errMsg.includes("key") || errMsg.includes("auth")) {
await sendErrorMessage("[ClawdBot] 大模型 API Key 可能无效,请检查配置");
} else {
await sendErrorMessage(`[ClawdBot] 处理消息时出错: ${errMsg.slice(0, 100)}`);
// 显示完整错误信息,截取前 500 字符
await sendErrorMessage(`[ClawdBot] 出错: ${errMsg.slice(0, 500)}`);
}
},
},
@@ -372,7 +614,7 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
}
} catch (err) {
log?.error(`[qqbot:${account.accountId}] Message processing failed: ${err}`);
await sendErrorMessage(`[ClawdBot] 处理消息失败: ${String(err).slice(0, 100)}`);
await sendErrorMessage(`[ClawdBot] 处理失败: ${String(err).slice(0, 500)}`);
}
};
@@ -385,7 +627,8 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
ws.on("message", async (data) => {
try {
const payload = JSON.parse(data.toString()) as WSPayload;
const rawData = data.toString();
const payload = JSON.parse(rawData) as WSPayload;
const { op, d, s, t } = payload;
if (s) lastSeq = s;
@@ -409,22 +652,15 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
}));
} else {
// 新连接,发送 Identify
// 如果 identify 失败多次,尝试只使用基础权限
let intents: number;
if (identifyFailCount >= 3) {
// 只使用基础权限(频道消息)
intents = INTENTS.PUBLIC_GUILD_MESSAGES | INTENTS.GUILD_MEMBERS;
log?.info(`[qqbot:${account.accountId}] Using basic intents only (after ${identifyFailCount} failures): ${intents}`);
} else {
// 使用完整权限
intents = INTENTS.PUBLIC_GUILD_MESSAGES | INTENTS.DIRECT_MESSAGE | INTENTS.GROUP_AND_C2C;
log?.info(`[qqbot:${account.accountId}] Sending identify with intents: ${intents}`);
}
// 如果有上次成功的级别,直接使用;否则从当前级别开始尝试
const levelToUse = lastSuccessfulIntentLevel >= 0 ? lastSuccessfulIntentLevel : intentLevelIndex;
const intentLevel = INTENT_LEVELS[Math.min(levelToUse, INTENT_LEVELS.length - 1)];
log?.info(`[qqbot:${account.accountId}] Sending identify with intents: ${intentLevel.intents} (${intentLevel.description})`);
ws.send(JSON.stringify({
op: 2,
d: {
token: `QQBot ${accessToken}`,
intents: intents,
intents: intentLevel.intents,
shard: [0, 1],
},
}));
@@ -445,8 +681,10 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
if (t === "READY") {
const readyData = d as { session_id: string };
sessionId = readyData.session_id;
identifyFailCount = 0; // 连接成功,重置失败计数
log?.info(`[qqbot:${account.accountId}] Ready, session: ${sessionId}`);
// 记录成功的权限级别
lastSuccessfulIntentLevel = intentLevelIndex;
const successLevel = INTENT_LEVELS[intentLevelIndex];
log?.info(`[qqbot:${account.accountId}] Ready with ${successLevel.description}, session: ${sessionId}`);
onReady?.(d);
} else if (t === "RESUMED") {
log?.info(`[qqbot:${account.accountId}] Session resumed`);
@@ -511,22 +749,27 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
case 9: // Invalid Session
const canResume = d as boolean;
log?.error(`[qqbot:${account.accountId}] Invalid session, can resume: ${canResume}`);
const currentLevel = INTENT_LEVELS[intentLevelIndex];
log?.error(`[qqbot:${account.accountId}] Invalid session (${currentLevel.description}), can resume: ${canResume}, raw: ${rawData}`);
if (!canResume) {
sessionId = null;
lastSeq = null;
identifyFailCount++;
// 标记需要刷新 token可能是 token 过期导致的)
shouldRefreshToken = true;
if (identifyFailCount >= 3) {
log?.error(`[qqbot:${account.accountId}] Identify failed ${identifyFailCount} times. This may be a permission issue.`);
log?.error(`[qqbot:${account.accountId}] Please check: 1) AppID/Secret is correct 2) Bot has GROUP_AND_C2C permission on QQ Open Platform`);
// 尝试降级到下一个权限级别
if (intentLevelIndex < INTENT_LEVELS.length - 1) {
intentLevelIndex++;
const nextLevel = INTENT_LEVELS[intentLevelIndex];
log?.info(`[qqbot:${account.accountId}] Downgrading intents to: ${nextLevel.description}`);
} else {
// 已经是最低权限级别了
log?.error(`[qqbot:${account.accountId}] All intent levels failed. Please check AppID/Secret.`);
shouldRefreshToken = true;
}
}
cleanup();
// Invalid Session 后等待一段时间再重连
scheduleReconnect(5000);
scheduleReconnect(3000);
break;
}
} catch (err) {
@@ -538,12 +781,28 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
log?.info(`[qqbot:${account.accountId}] WebSocket closed: ${code} ${reason.toString()}`);
isConnecting = false; // 释放锁
// 4903 等错误码表示 session 创建失败,需要刷新 token
if (code === 4903 || code === 4009 || code === 4014) {
log?.info(`[qqbot:${account.accountId}] Session error (${code}), will refresh token`);
// 根据错误码处理
// 4009: 可以重新发起 resume
// 4900-4913: 内部错误,需要重新 identify
// 4914: 机器人已下架
// 4915: 机器人已封禁
if (code === 4914 || code === 4915) {
log?.error(`[qqbot:${account.accountId}] Bot is ${code === 4914 ? "offline/sandbox-only" : "banned"}. Please contact QQ platform.`);
cleanup();
// 不重连,直接退出
return;
}
if (code === 4009) {
// 4009 可以尝试 resume保留 session
log?.info(`[qqbot:${account.accountId}] Error 4009, will try resume`);
shouldRefreshToken = true;
} else if (code >= 4900 && code <= 4913) {
// 4900-4913 内部错误,清除 session 重新 identify
log?.info(`[qqbot:${account.accountId}] Internal error (${code}), will re-identify`);
sessionId = null;
lastSeq = null;
shouldRefreshToken = true;
}
// 检测是否是快速断开(连接后很快就断了)
@@ -552,12 +811,10 @@ export async function startGateway(ctx: GatewayContext): Promise<void> {
quickDisconnectCount++;
log?.info(`[qqbot:${account.accountId}] Quick disconnect detected (${connectionDuration}ms), count: ${quickDisconnectCount}`);
// 如果连续快速断开超过阈值,清除 session 并等待更长时间
// 如果连续快速断开超过阈值,等待更长时间
if (quickDisconnectCount >= MAX_QUICK_DISCONNECT_COUNT) {
log?.info(`[qqbot:${account.accountId}] Too many quick disconnects, clearing session and refreshing token`);
sessionId = null;
lastSeq = null;
shouldRefreshToken = true;
log?.error(`[qqbot:${account.accountId}] Too many quick disconnects. This may indicate a permission issue.`);
log?.error(`[qqbot:${account.accountId}] Please check: 1) AppID/Secret correct 2) Bot permissions on QQ Open Platform`);
quickDisconnectCount = 0;
cleanup();
// 快速断开太多次,等待更长时间再重连

446
src/image-server.ts Normal file
View File

@@ -0,0 +1,446 @@
/**
* 本地图床服务器
* 提供安全的图片存储和访问服务
*/
import http from "node:http";
import fs from "node:fs";
import path from "node:path";
import crypto from "node:crypto";
export interface ImageServerConfig {
/** 监听端口 */
port: number;
/** 图片存储目录 */
storageDir: string;
/** 外部访问的基础 URL如 http://your-server:port留空则自动生成 */
baseUrl?: string;
/** 图片过期时间0 表示不过期 */
ttlSeconds?: number;
/** 允许的图片格式 */
allowedFormats?: string[];
}
interface StoredImage {
id: string;
filename: string;
mimeType: string;
createdAt: number;
ttl: number;
}
const DEFAULT_CONFIG: Required<ImageServerConfig> = {
port: 18765,
storageDir: "./qqbot-images",
baseUrl: "",
ttlSeconds: 3600, // 默认 1 小时过期
allowedFormats: ["png", "jpg", "jpeg", "gif", "webp"],
};
let serverInstance: http.Server | null = null;
let currentConfig: Required<ImageServerConfig> = { ...DEFAULT_CONFIG };
let imageIndex = new Map<string, StoredImage>();
/**
* 生成安全的随机 ID
*/
function generateImageId(): string {
return crypto.randomBytes(16).toString("hex");
}
/**
* 验证请求路径是否安全(防止目录遍历攻击)
*/
function isPathSafe(requestPath: string, baseDir: string): boolean {
const normalizedBase = path.resolve(baseDir);
const normalizedPath = path.resolve(baseDir, requestPath);
return normalizedPath.startsWith(normalizedBase + path.sep) || normalizedPath === normalizedBase;
}
/**
* 获取 MIME 类型
*/
function getMimeType(ext: string): string {
const mimeTypes: Record<string, string> = {
png: "image/png",
jpg: "image/jpeg",
jpeg: "image/jpeg",
gif: "image/gif",
webp: "image/webp",
};
return mimeTypes[ext.toLowerCase()] || "application/octet-stream";
}
/**
* 从 MIME 类型获取扩展名
*/
function getExtFromMime(mimeType: string): string | null {
const extMap: Record<string, string> = {
"image/png": "png",
"image/jpeg": "jpg",
"image/gif": "gif",
"image/webp": "webp",
"application/pdf": "pdf",
"application/json": "json",
"text/plain": "txt",
"text/csv": "csv",
"application/msword": "doc",
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": "docx",
"application/vnd.ms-excel": "xls",
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": "xlsx",
};
return extMap[mimeType] || null;
}
/**
* 清理过期图片
*/
function cleanupExpiredImages(): void {
const now = Date.now();
const expiredIds: string[] = [];
for (const [id, image] of imageIndex) {
if (image.ttl > 0 && now - image.createdAt > image.ttl * 1000) {
expiredIds.push(id);
}
}
for (const id of expiredIds) {
const image = imageIndex.get(id);
if (image) {
const filePath = path.join(currentConfig.storageDir, image.filename);
try {
if (fs.existsSync(filePath)) {
fs.unlinkSync(filePath);
}
} catch {
// 忽略删除错误
}
imageIndex.delete(id);
}
}
}
/**
* 加载已有的图片索引
*/
function loadImageIndex(): void {
const indexPath = path.join(currentConfig.storageDir, ".index.json");
try {
if (fs.existsSync(indexPath)) {
const data = JSON.parse(fs.readFileSync(indexPath, "utf-8"));
imageIndex = new Map(Object.entries(data));
}
} catch {
imageIndex = new Map();
}
}
/**
* 保存图片索引
*/
function saveImageIndex(): void {
const indexPath = path.join(currentConfig.storageDir, ".index.json");
try {
const data = Object.fromEntries(imageIndex);
fs.writeFileSync(indexPath, JSON.stringify(data, null, 2));
} catch {
// 忽略保存错误
}
}
/**
* 处理 HTTP 请求
*/
function handleRequest(req: http.IncomingMessage, res: http.ServerResponse): void {
const url = new URL(req.url || "/", `http://localhost:${currentConfig.port}`);
const pathname = url.pathname;
// 设置 CORS 头(允许 QQ 服务器访问)
res.setHeader("Access-Control-Allow-Origin", "*");
res.setHeader("Access-Control-Allow-Methods", "GET, OPTIONS");
if (req.method === "OPTIONS") {
res.writeHead(204);
res.end();
return;
}
// 只允许 GET 请求访问图片
if (req.method !== "GET") {
res.writeHead(405, { "Content-Type": "text/plain" });
res.end("Method Not Allowed");
return;
}
// 解析图片 ID路径格式: /images/{id}.{ext}
const match = pathname.match(/^\/images\/([a-f0-9]{32})\.(\w+)$/);
if (!match) {
res.writeHead(404, { "Content-Type": "text/plain" });
res.end("Not Found");
return;
}
const [, imageId, requestedExt] = match;
const image = imageIndex.get(imageId);
if (!image) {
res.writeHead(404, { "Content-Type": "text/plain" });
res.end("Image Not Found");
return;
}
// 检查是否过期
if (image.ttl > 0 && Date.now() - image.createdAt > image.ttl * 1000) {
res.writeHead(410, { "Content-Type": "text/plain" });
res.end("Image Expired");
return;
}
// 安全检查:确保文件路径在存储目录内
const filePath = path.join(currentConfig.storageDir, image.filename);
if (!isPathSafe(image.filename, currentConfig.storageDir)) {
res.writeHead(403, { "Content-Type": "text/plain" });
res.end("Forbidden");
return;
}
// 读取并返回图片
try {
if (!fs.existsSync(filePath)) {
res.writeHead(404, { "Content-Type": "text/plain" });
res.end("File Not Found");
return;
}
const imageData = fs.readFileSync(filePath);
res.writeHead(200, {
"Content-Type": image.mimeType,
"Content-Length": imageData.length,
"Cache-Control": image.ttl > 0 ? `max-age=${image.ttl}` : "max-age=31536000",
});
res.end(imageData);
} catch (err) {
res.writeHead(500, { "Content-Type": "text/plain" });
res.end("Internal Server Error");
}
}
/**
* 启动图床服务器
*/
export function startImageServer(config?: Partial<ImageServerConfig>): Promise<string> {
return new Promise((resolve, reject) => {
if (serverInstance) {
const baseUrl = currentConfig.baseUrl || `http://localhost:${currentConfig.port}`;
resolve(baseUrl);
return;
}
currentConfig = { ...DEFAULT_CONFIG, ...config };
// 确保存储目录存在
if (!fs.existsSync(currentConfig.storageDir)) {
fs.mkdirSync(currentConfig.storageDir, { recursive: true });
}
// 加载图片索引
loadImageIndex();
// 启动定期清理
const cleanupInterval = setInterval(cleanupExpiredImages, 60000); // 每分钟清理一次
serverInstance = http.createServer(handleRequest);
serverInstance.on("error", (err) => {
clearInterval(cleanupInterval);
reject(err);
});
serverInstance.listen(currentConfig.port, () => {
const baseUrl = currentConfig.baseUrl || `http://localhost:${currentConfig.port}`;
resolve(baseUrl);
});
});
}
/**
* 停止图床服务器
*/
export function stopImageServer(): Promise<void> {
return new Promise((resolve) => {
if (serverInstance) {
serverInstance.close(() => {
serverInstance = null;
saveImageIndex();
resolve();
});
} else {
resolve();
}
});
}
/**
* 保存图片并返回访问 URL
* @param imageData 图片数据Buffer 或 base64 字符串)
* @param mimeType 图片 MIME 类型
* @param ttlSeconds 过期时间(秒),默认使用配置值
* @returns 图片访问 URL
*/
export function saveImage(
imageData: Buffer | string,
mimeType: string = "image/png",
ttlSeconds?: number
): string {
// 转换 base64 为 Buffer
let buffer: Buffer;
if (typeof imageData === "string") {
// 处理 data URL 格式
const base64Match = imageData.match(/^data:([^;]+);base64,(.+)$/);
if (base64Match) {
mimeType = base64Match[1];
buffer = Buffer.from(base64Match[2], "base64");
} else {
buffer = Buffer.from(imageData, "base64");
}
} else {
buffer = imageData;
}
// 生成唯一 ID 和文件名
const imageId = generateImageId();
const ext = getExtFromMime(mimeType) || "png";
const filename = `${imageId}.${ext}`;
// 确保存储目录存在
if (!fs.existsSync(currentConfig.storageDir)) {
fs.mkdirSync(currentConfig.storageDir, { recursive: true });
}
// 保存文件
const filePath = path.join(currentConfig.storageDir, filename);
fs.writeFileSync(filePath, buffer);
// 记录到索引
const image: StoredImage = {
id: imageId,
filename,
mimeType,
createdAt: Date.now(),
ttl: ttlSeconds ?? currentConfig.ttlSeconds,
};
imageIndex.set(imageId, image);
saveImageIndex();
// 返回访问 URL
const baseUrl = currentConfig.baseUrl || `http://localhost:${currentConfig.port}`;
return `${baseUrl}/images/${imageId}.${ext}`;
}
/**
* 从本地文件路径保存图片到图床
* @param filePath 本地文件路径
* @param ttlSeconds 过期时间(秒),默认使用配置值
* @returns 图片访问 URL如果文件不存在或不是图片则返回 null
*/
export function saveImageFromPath(filePath: string, ttlSeconds?: number): string | null {
try {
console.log(`[image-server] saveImageFromPath: ${filePath}`);
// 检查文件是否存在
if (!fs.existsSync(filePath)) {
console.log(`[image-server] File not found: ${filePath}`);
return null;
}
// 读取文件
const buffer = fs.readFileSync(filePath);
console.log(`[image-server] File size: ${buffer.length}`);
// 根据扩展名获取 MIME 类型
const ext = path.extname(filePath).toLowerCase().replace(".", "");
console.log(`[image-server] Extension: "${ext}"`);
const mimeType = getMimeType(ext);
console.log(`[image-server] MIME type: ${mimeType}`);
// 只处理图片文件
if (!mimeType.startsWith("image/")) {
console.log(`[image-server] Not an image file`);
return null;
}
// 使用 saveImage 保存
return saveImage(buffer, mimeType, ttlSeconds);
} catch (err) {
console.error(`[image-server] saveImageFromPath error:`, err);
return null;
}
}
/**
* 检查图床服务器是否运行中
*/
export function isImageServerRunning(): boolean {
return serverInstance !== null;
}
/**
* 下载远程文件并保存到本地
* @param url 远程文件 URL
* @param destDir 目标目录
* @param originalFilename 原始文件名(可选,完整文件名包含扩展名)
* @returns 本地文件路径,失败返回 null
*/
export async function downloadFile(
url: string,
destDir: string,
originalFilename?: string
): Promise<string | null> {
try {
// 确保目录存在
if (!fs.existsSync(destDir)) {
fs.mkdirSync(destDir, { recursive: true });
}
// 下载文件
const response = await fetch(url);
if (!response.ok) {
console.error(`[image-server] Download failed: ${response.status} ${response.statusText}`);
return null;
}
const buffer = Buffer.from(await response.arrayBuffer());
// 确定文件名
let finalFilename: string;
if (originalFilename) {
// 使用原始文件名,但添加时间戳避免冲突
const ext = path.extname(originalFilename);
const baseName = path.basename(originalFilename, ext);
const timestamp = Date.now();
finalFilename = `${baseName}_${timestamp}${ext}`;
} else {
// 没有原始文件名,生成随机名
finalFilename = `${generateImageId()}.bin`;
}
const filePath = path.join(destDir, finalFilename);
// 保存文件
fs.writeFileSync(filePath, buffer);
console.log(`[image-server] Downloaded file: ${filePath}`);
return filePath;
} catch (err) {
console.error(`[image-server] Download error:`, err);
return null;
}
}
/**
* 获取图床服务器配置
*/
export function getImageServerConfig(): Required<ImageServerConfig> {
return { ...currentConfig };
}

View File

@@ -29,12 +29,14 @@ interface QQBotChannelConfig {
clientSecret?: string;
clientSecretFile?: string;
name?: string;
imageServerBaseUrl?: string;
accounts?: Record<string, {
enabled?: boolean;
appId?: string;
clientSecret?: string;
clientSecretFile?: string;
name?: string;
imageServerBaseUrl?: string;
}>;
}

View File

@@ -19,6 +19,8 @@ export interface ResolvedQQBotAccount {
secretSource: "config" | "file" | "env" | "none";
/** 系统提示词 */
systemPrompt?: string;
/** 图床服务器公网地址 */
imageServerBaseUrl?: string;
config: QQBotAccountConfig;
}
@@ -35,6 +37,8 @@ export interface QQBotAccountConfig {
allowFrom?: string[];
/** 系统提示词,会添加在用户消息前面 */
systemPrompt?: string;
/** 图床服务器公网地址,用于发送图片,例如 http://your-ip:18765 */
imageServerBaseUrl?: string;
}
/**