@sliverp/qqbot 1.3.0

package/dist/src/image-server.d.ts

@@ -0,0 +1,62 @@
+/**
+ * 本地图床服务器
+ * 提供安全的图片存储和访问服务
+ */
+export interface ImageServerConfig {
+    /** 监听端口 */
+    port: number;
+    /** 图片存储目录 */
+    storageDir: string;
+    /** 外部访问的基础 URL（如 http://your-server:port），留空则自动生成 */
+    baseUrl?: string;
+    /** 图片过期时间（秒），0 表示不过期 */
+    ttlSeconds?: number;
+    /** 允许的图片格式 */
+    allowedFormats?: string[];
+}
+/**
+ * 启动图床服务器
+ */
+export declare function startImageServer(config?: Partial<ImageServerConfig>): Promise<string>;
+/**
+ * 停止图床服务器
+ */
+export declare function stopImageServer(): Promise<void>;
+/**
+ * 保存图片并返回访问 URL
+ * @param imageData 图片数据（Buffer 或 base64 字符串）
+ * @param mimeType 图片 MIME 类型
+ * @param ttlSeconds 过期时间（秒），默认使用配置值
+ * @returns 图片访问 URL
+ */
+export declare function saveImage(imageData: Buffer | string, mimeType?: string, ttlSeconds?: number): string;
+/**
+ * 从本地文件路径保存图片到图床
+ * @param filePath 本地文件路径
+ * @param ttlSeconds 过期时间（秒），默认使用配置值
+ * @returns 图片访问 URL，如果文件不存在或不是图片则返回 null
+ */
+export declare function saveImageFromPath(filePath: string, ttlSeconds?: number): string | null;
+/**
+ * 检查图床服务器是否运行中
+ */
+export declare function isImageServerRunning(): boolean;
+/**
+ * 确保图床服务器正在运行
+ * 如果未运行，则自动启动
+ * @param publicBaseUrl 公网访问的基础 URL（如 http://your-server:18765）
+ * @returns 基础 URL，启动失败返回 null
+ */
+export declare function ensureImageServer(publicBaseUrl?: string): Promise<string | null>;
+/**
+ * 下载远程文件并保存到本地
+ * @param url 远程文件 URL
+ * @param destDir 目标目录
+ * @param originalFilename 原始文件名（可选，完整文件名包含扩展名）
+ * @returns 本地文件路径，失败返回 null
+ */
+export declare function downloadFile(url: string, destDir: string, originalFilename?: string): Promise<string | null>;
+/**
+ * 获取图床服务器配置
+ */
+export declare function getImageServerConfig(): Required<ImageServerConfig>;

package/dist/src/image-server.js

@@ -0,0 +1,401 @@
+/**
+ * 本地图床服务器
+ * 提供安全的图片存储和访问服务
+ */
+import http from "node:http";
+import fs from "node:fs";
+import path from "node:path";
+import crypto from "node:crypto";
+const DEFAULT_CONFIG = {
+    port: 18765,
+    storageDir: "./qqbot-images",
+    baseUrl: "",
+    ttlSeconds: 3600, // 默认 1 小时过期
+    allowedFormats: ["png", "jpg", "jpeg", "gif", "webp"],
+};
+let serverInstance = null;
+let currentConfig = { ...DEFAULT_CONFIG };
+let imageIndex = new Map();
+/**
+ * 生成安全的随机 ID
+ */
+function generateImageId() {
+    return crypto.randomBytes(16).toString("hex");
+}
+/**
+ * 验证请求路径是否安全（防止目录遍历攻击）
+ */
+function isPathSafe(requestPath, baseDir) {
+    const normalizedBase = path.resolve(baseDir);
+    const normalizedPath = path.resolve(baseDir, requestPath);
+    return normalizedPath.startsWith(normalizedBase + path.sep) || normalizedPath === normalizedBase;
+}
+/**
+ * 获取 MIME 类型
+ */
+function getMimeType(ext) {
+    const mimeTypes = {
+        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) {
+    const extMap = {
+        "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() {
+    const now = Date.now();
+    const expiredIds = [];
+    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() {
+    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() {
+    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, res) {
+    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) {
+    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() {
+    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, mimeType = "image/png", ttlSeconds) {
+    // 转换 base64 为 Buffer
+    let 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 = {
+        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, ttlSeconds) {
+    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() {
+    return serverInstance !== null;
+}
+/**
+ * 确保图床服务器正在运行
+ * 如果未运行，则自动启动
+ * @param publicBaseUrl 公网访问的基础 URL（如 http://your-server:18765）
+ * @returns 基础 URL，启动失败返回 null
+ */
+export async function ensureImageServer(publicBaseUrl) {
+    if (isImageServerRunning()) {
+        return publicBaseUrl || currentConfig.baseUrl || `http://0.0.0.0:${currentConfig.port}`;
+    }
+    try {
+        const config = {
+            port: DEFAULT_CONFIG.port,
+            storageDir: DEFAULT_CONFIG.storageDir,
+            // 使用用户配置的公网地址
+            baseUrl: publicBaseUrl || `http://0.0.0.0:${DEFAULT_CONFIG.port}`,
+            ttlSeconds: 3600, // 1 小时过期
+        };
+        await startImageServer(config);
+        console.log(`[image-server] Auto-started on port ${config.port}, baseUrl: ${config.baseUrl}`);
+        return config.baseUrl;
+    }
+    catch (err) {
+        console.error(`[image-server] Failed to auto-start: ${err}`);
+        return null;
+    }
+}
+/**
+ * 下载远程文件并保存到本地
+ * @param url 远程文件 URL
+ * @param destDir 目标目录
+ * @param originalFilename 原始文件名（可选，完整文件名包含扩展名）
+ * @returns 本地文件路径，失败返回 null
+ */
+export async function downloadFile(url, destDir, originalFilename) {
+    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;
+        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() {
+    return { ...currentConfig };
+}

package/dist/src/known-users.d.ts

@@ -0,0 +1,100 @@
+/**
+ * 已知用户存储
+ * 记录与机器人交互过的所有用户
+ * 支持主动消息和批量通知功能
+ */
+export interface KnownUser {
+    /** 用户 openid（唯一标识） */
+    openid: string;
+    /** 消息类型：私聊用户 / 群组 */
+    type: "c2c" | "group";
+    /** 用户昵称（如有） */
+    nickname?: string;
+    /** 群组 openid（如果是群消息） */
+    groupOpenid?: string;
+    /** 关联的机器人账户 ID */
+    accountId: string;
+    /** 首次交互时间戳 */
+    firstSeenAt: number;
+    /** 最后交互时间戳 */
+    lastSeenAt: number;
+    /** 交互次数 */
+    interactionCount: number;
+}
+/**
+ * 强制立即保存（用于进程退出前）
+ */
+export declare function flushKnownUsers(): void;
+/**
+ * 记录已知用户（收到消息时调用）
+ * @param user 用户信息（部分字段）
+ */
+export declare function recordKnownUser(user: {
+    openid: string;
+    type: "c2c" | "group";
+    nickname?: string;
+    groupOpenid?: string;
+    accountId: string;
+}): void;
+/**
+ * 获取单个用户信息
+ * @param accountId 机器人账户 ID
+ * @param openid 用户 openid
+ * @param type 消息类型
+ * @param groupOpenid 群组 openid（可选）
+ */
+export declare function getKnownUser(accountId: string, openid: string, type?: "c2c" | "group", groupOpenid?: string): KnownUser | undefined;
+/**
+ * 列出所有已知用户
+ * @param options 筛选选项
+ */
+export declare function listKnownUsers(options?: {
+    /** 筛选特定机器人账户的用户 */
+    accountId?: string;
+    /** 筛选消息类型 */
+    type?: "c2c" | "group";
+    /** 最近活跃时间（毫秒，如 86400000 表示最近 24 小时） */
+    activeWithin?: number;
+    /** 返回数量限制 */
+    limit?: number;
+    /** 排序方式 */
+    sortBy?: "lastSeenAt" | "firstSeenAt" | "interactionCount";
+    /** 排序方向 */
+    sortOrder?: "asc" | "desc";
+}): KnownUser[];
+/**
+ * 获取用户统计信息
+ * @param accountId 机器人账户 ID（可选，不传则返回所有账户的统计）
+ */
+export declare function getKnownUsersStats(accountId?: string): {
+    totalUsers: number;
+    c2cUsers: number;
+    groupUsers: number;
+    activeIn24h: number;
+    activeIn7d: number;
+};
+/**
+ * 删除用户记录
+ * @param accountId 机器人账户 ID
+ * @param openid 用户 openid
+ * @param type 消息类型
+ * @param groupOpenid 群组 openid（可选）
+ */
+export declare function removeKnownUser(accountId: string, openid: string, type?: "c2c" | "group", groupOpenid?: string): boolean;
+/**
+ * 清除所有用户记录
+ * @param accountId 机器人账户 ID（可选，不传则清除所有）
+ */
+export declare function clearKnownUsers(accountId?: string): number;
+/**
+ * 获取用户的所有群组（某用户在哪些群里交互过）
+ * @param accountId 机器人账户 ID
+ * @param openid 用户 openid
+ */
+export declare function getUserGroups(accountId: string, openid: string): string[];
+/**
+ * 获取群组的所有成员
+ * @param accountId 机器人账户 ID
+ * @param groupOpenid 群组 openid
+ */
+export declare function getGroupMembers(accountId: string, groupOpenid: string): KnownUser[];