koishi-plugin-kkk 1.0.0

package/lib/karin/module/server/constants/routes.js

@@ -0,0 +1,48 @@
+"use strict";
+/**
+ * 路由常量定义
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ROUTES = exports.FULL_API_V1_PREFIX = exports.FULL_SSR_PREFIX = exports.FULL_ASSETS_PREFIX = exports.API_V1_PREFIX = exports.SSR_PREFIX = exports.ASSETS_PREFIX = exports.KKK_PREFIX = void 0;
+// 主路由前缀
+exports.KKK_PREFIX = '/kkk';
+// 子路由前缀
+/** 静态文件 */
+exports.ASSETS_PREFIX = '/assets';
+/** SSR 页面 */
+exports.SSR_PREFIX = '/ssr';
+/** API v1 */
+exports.API_V1_PREFIX = '/v1';
+// 完整路径
+/** 静态文件完整路径 */
+exports.FULL_ASSETS_PREFIX = `${exports.KKK_PREFIX}${exports.ASSETS_PREFIX}`;
+/** SSR 页面完整路径 */
+exports.FULL_SSR_PREFIX = `${exports.KKK_PREFIX}${exports.SSR_PREFIX}`;
+/** API v1 完整路径 */
+exports.FULL_API_V1_PREFIX = `${exports.KKK_PREFIX}${exports.API_V1_PREFIX}`;
+// API 路由
+exports.ROUTES = {
+    // Bot 管理
+    /** 获取所有 Bot */
+    BOTS: '/bots',
+    /** 获取指定 Bot 信息 */
+    BOT_INFO: '/bots/:botId',
+    /** 获取指定 Bot 的群列表 */
+    BOT_GROUPS: '/bots/:botId/groups',
+    /** 获取指定 Bot 群信息 */
+    BOT_GROUP_INFO: '/bots/:botId/groups/:groupId',
+    // 群组管理
+    /** 获取所有群组 */
+    GROUPS_BATCH: '/groups/batch',
+    // 配置管理
+    /** 获取插件所有配置 */
+    CONFIG: '/config',
+    // 视频流
+    /** 获取视频流 */
+    VIDEO_STREAM: '/stream/:filename',
+    /** 获取视频事件 */
+    VIDEO_EVENTS: '/video/:filename/events',
+    // SSR 页面
+    /** 视频播放页面 */
+    VIDEO_PAGE: '/video/:filename'
+};

package/lib/karin/module/server/controllers/bots.js

@@ -0,0 +1,230 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getGroupsBatch = exports.getBotGroupInfo = exports.getBotGroups = exports.getBotInfo = exports.getBots = void 0;
+/**
+ * Bot 管理 API
+ */
+const node_karin_1 = require("../../../../compat/node-karin.js");
+const getOnlineBotById = (botId) => {
+    return node_karin_1.karin.getAllBotList().find((item) => item.bot.account.selfId === botId)?.bot;
+};
+const getBotAvatar = async (bot, botId) => {
+    if (!bot)
+        return '';
+    try {
+        return (await bot.getAvatarUrl(botId)) || '';
+    }
+    catch (e) {
+        node_karin_1.logger.warn(`[BotsAPI] 获取 Bot 头像失败 ${botId}:`, e);
+        return '';
+    }
+};
+/**
+ * 获取所有在线 Bot 列表
+ * GET /api/v1/bots
+ */
+const getBots = async (_req, res) => {
+    try {
+        const botList = node_karin_1.karin.getAllBotList();
+        const bots = [];
+        for (const item of botList) {
+            const bot = item.bot;
+            // 跳过 console bot
+            if (bot.account.name === 'console')
+                continue;
+            let avatar = '';
+            try {
+                avatar = (await bot.getAvatarUrl(bot.account.selfId)) || '';
+            }
+            catch (e) {
+                node_karin_1.logger.warn(`[BotsAPI] 获取 Bot 头像失败 ${bot.account.selfId}:`, e);
+            }
+            bots.push({
+                id: bot.account.selfId,
+                name: bot.account.name || bot.account.selfId,
+                avatar,
+                isOnline: true
+            });
+        }
+        return (0, node_karin_1.createSuccessResponse)(res, bots);
+    }
+    catch (error) {
+        node_karin_1.logger.error('[BotsAPI] 获取 Bot 列表失败:', error);
+        return (0, node_karin_1.createServerErrorResponse)(res, '获取 Bot 列表失败');
+    }
+};
+exports.getBots = getBots;
+/**
+ * 获取指定 Bot 信息
+ * GET /api/v1/bots/:botId
+ */
+const getBotInfo = async (req, res) => {
+    try {
+        const botId = Array.isArray(req.params.botId) ? req.params.botId[0] : req.params.botId;
+        if (!botId) {
+            return (0, node_karin_1.createServerErrorResponse)(res, '缺少 botId 参数');
+        }
+        const bot = getOnlineBotById(botId);
+        if (!bot || bot.account.name === 'console') {
+            return (0, node_karin_1.createServerErrorResponse)(res, 'Bot 不存在或不在线');
+        }
+        const botInfo = {
+            id: bot.account.selfId,
+            name: bot.account.name || bot.account.selfId,
+            avatar: await getBotAvatar(bot, bot.account.selfId),
+            isOnline: true
+        };
+        return (0, node_karin_1.createSuccessResponse)(res, botInfo);
+    }
+    catch (error) {
+        node_karin_1.logger.error('[BotsAPI] 获取 Bot 信息失败:', error);
+        return (0, node_karin_1.createServerErrorResponse)(res, '获取 Bot 信息失败');
+    }
+};
+exports.getBotInfo = getBotInfo;
+/**
+ * 获取指定 Bot 的群列表
+ * GET /api/v1/bots/:botId/groups
+ */
+const getBotGroups = async (req, res) => {
+    try {
+        const botId = Array.isArray(req.params.botId) ? req.params.botId[0] : req.params.botId;
+        if (!botId) {
+            return (0, node_karin_1.createServerErrorResponse)(res, '缺少 botId 参数');
+        }
+        const botList = node_karin_1.karin.getAllBotList();
+        const botItem = botList.find((item) => item.bot.account.selfId === botId);
+        if (!botItem) {
+            return (0, node_karin_1.createServerErrorResponse)(res, 'Bot 不存在或不在线');
+        }
+        const bot = botItem.bot;
+        const groupList = await bot.getGroupList();
+        const groups = [];
+        for (const group of groupList) {
+            let avatar = '';
+            try {
+                avatar = (await bot.getGroupAvatarUrl(group.groupId)) || '';
+            }
+            catch {
+                // 忽略头像获取失败
+            }
+            groups.push({
+                id: group.groupId,
+                name: group.groupName || group.groupId,
+                avatar
+            });
+        }
+        return (0, node_karin_1.createSuccessResponse)(res, groups);
+    }
+    catch (error) {
+        node_karin_1.logger.error('[BotsAPI] 获取 Bot 群列表失败:', error);
+        return (0, node_karin_1.createServerErrorResponse)(res, '获取群列表失败');
+    }
+};
+exports.getBotGroups = getBotGroups;
+/**
+ * 获取指定 Bot 下的群信息
+ * GET /api/v1/bots/:botId/groups/:groupId
+ */
+const getBotGroupInfo = async (req, res) => {
+    try {
+        const botId = Array.isArray(req.params.botId) ? req.params.botId[0] : req.params.botId;
+        const groupId = Array.isArray(req.params.groupId) ? req.params.groupId[0] : req.params.groupId;
+        if (!botId || !groupId) {
+            return (0, node_karin_1.createServerErrorResponse)(res, '缺少 botId 或 groupId 参数');
+        }
+        const bot = getOnlineBotById(botId);
+        if (!bot) {
+            return (0, node_karin_1.createServerErrorResponse)(res, 'Bot 不存在或不在线');
+        }
+        let groupName = groupId;
+        let groupAvatar = '';
+        let memberCount;
+        try {
+            const groupInfo = await bot.getGroupInfo(groupId);
+            groupName = groupInfo?.groupName || groupName;
+            memberCount = groupInfo?.memberCount;
+            groupAvatar = (await bot.getGroupAvatarUrl(groupId)) || '';
+        }
+        catch (e) {
+            node_karin_1.logger.warn(`[BotsAPI] 获取群组信息失败 ${groupId}:`, e);
+        }
+        const groupDetail = {
+            id: groupId,
+            name: groupName,
+            avatar: groupAvatar,
+            botId: bot.account.selfId,
+            botName: bot.account.name || bot.account.selfId,
+            botAvatar: await getBotAvatar(bot, bot.account.selfId),
+            memberCount,
+            isOnline: true
+        };
+        return (0, node_karin_1.createSuccessResponse)(res, groupDetail);
+    }
+    catch (error) {
+        node_karin_1.logger.error('[BotsAPI] 获取 Bot 群信息失败:', error);
+        return (0, node_karin_1.createServerErrorResponse)(res, '获取群信息失败');
+    }
+};
+exports.getBotGroupInfo = getBotGroupInfo;
+/**
+ * 批量获取群组信息
+ * POST /api/v1/groups/batch
+ * Body: { groups: [{ groupId: string, botId: string }] }
+ */
+const getGroupsBatch = async (req, res) => {
+    try {
+        const { groups } = req.body;
+        if (!groups || !Array.isArray(groups)) {
+            return (0, node_karin_1.createServerErrorResponse)(res, '缺少 groups 参数');
+        }
+        const result = [];
+        const botList = node_karin_1.karin.getAllBotList();
+        for (const item of groups) {
+            const botItem = botList.find((b) => b.bot.account.selfId === item.botId);
+            if (!botItem) {
+                result.push({
+                    groupId: item.groupId,
+                    botId: item.botId,
+                    groupName: item.groupId,
+                    groupAvatar: '',
+                    botName: item.botId,
+                    botAvatar: '',
+                    isOnline: false
+                });
+                continue;
+            }
+            const bot = botItem.bot;
+            let groupName = item.groupId;
+            let groupAvatar = '';
+            let botName = bot.account.name || item.botId;
+            let botAvatar = '';
+            try {
+                const groupInfo = await bot.getGroupInfo(item.groupId);
+                if (groupInfo) {
+                    groupName = groupInfo.groupName || groupName;
+                }
+                groupAvatar = (await bot.getGroupAvatarUrl(item.groupId)) || '';
+                botAvatar = (await bot.getAvatarUrl(item.botId)) || '';
+            }
+            catch (e) {
+                node_karin_1.logger.warn(`[BotsAPI] 获取群组信息失败 ${item.groupId}:`, e);
+            }
+            result.push({
+                groupId: item.groupId,
+                botId: item.botId,
+                groupName,
+                groupAvatar,
+                botName,
+                botAvatar,
+                isOnline: true
+            });
+        }
+        return (0, node_karin_1.createSuccessResponse)(res, result);
+    }
+    catch (error) {
+        node_karin_1.logger.error('[BotsAPI] 批量获取群组信息失败:', error);
+        return (0, node_karin_1.createServerErrorResponse)(res, '批量获取群组信息失败');
+    }
+};
+exports.getGroupsBatch = getGroupsBatch;

package/lib/karin/module/server/controllers/config.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateAllConfig = exports.getAllConfig = void 0;
+/**
+ * 配置管理 API
+ * 提供配置的增删改查接口，供 APP 端使用
+ */
+const node_karin_1 = require("../../../../compat/node-karin.js");
+const amagiClient_1 = require("../../../module/utils/amagiClient");
+const Config_1 = require("../../../module/utils/Config");
+/**
+ * 获取所有配置
+ * GET /kkk/web/v1/config
+ */
+const getAllConfig = async (_req, res) => {
+    try {
+        const config = await Config_1.Config.All();
+        return (0, node_karin_1.createSuccessResponse)(res, config);
+    }
+    catch (error) {
+        return (0, node_karin_1.createServerErrorResponse)(res, `获取配置失败: ${error.message}`);
+    }
+};
+exports.getAllConfig = getAllConfig;
+/**
+ * 批量更新配置
+ * POST /kkk/web/v1/config
+ * Body: Partial<ConfigType>
+ */
+const updateAllConfig = async (req, res) => {
+    try {
+        const newConfig = req.body;
+        if (!newConfig || typeof newConfig !== 'object') {
+            return (0, node_karin_1.createBadRequestResponse)(res, '请求体必须是有效的配置对象');
+        }
+        const oldConfig = await Config_1.Config.All();
+        const results = [];
+        let needReloadAmagi = false;
+        for (const [module, config] of Object.entries(newConfig)) {
+            try {
+                if (!(module in oldConfig)) {
+                    results.push({ module, success: false, error: `配置模块 "${module}" 不存在` });
+                    continue;
+                }
+                const moduleName = module;
+                const success = await Config_1.Config.ModifyPro(moduleName, config);
+                results.push({ module, success });
+                // amagi 配置变化需要重载客户端
+                if (success && moduleName === 'amagi') {
+                    needReloadAmagi = true;
+                }
+            }
+            catch (error) {
+                results.push({ module, success: false, error: error.message });
+            }
+        }
+        // 同步数据库
+        if ('pushlist' in newConfig) {
+            await Config_1.Config.syncConfigToDatabase();
+        }
+        if (needReloadAmagi) {
+            (0, amagiClient_1.reloadAmagiConfig)();
+        }
+        const allSuccess = results.every((r) => r.success);
+        const updatedConfig = await Config_1.Config.All();
+        return (0, node_karin_1.createSuccessResponse)(res, {
+            config: updatedConfig,
+            results
+        }, allSuccess ? '所有配置更新成功' : '部分配置更新失败');
+    }
+    catch (error) {
+        return (0, node_karin_1.createServerErrorResponse)(res, `配置更新失败: ${error.message}`);
+    }
+};
+exports.updateAllConfig = updateAllConfig;

package/lib/karin/module/server/controllers/video.js

@@ -0,0 +1,208 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getVideoEvents = exports.getVideoPage = exports.getVideoStream = void 0;
+/**
+ * 视频控制器
+ */
+const node_fs_1 = __importDefault(require("node:fs"));
+const node_path_1 = __importDefault(require("node:path"));
+const render_1 = require("../../../../ktr/template/_preview/render");
+const node_karin_1 = require("../../../../compat/node-karin.js");
+const utils_1 = require("../../../module/utils");
+const Config_1 = require("../../../module/utils/Config");
+/**
+ * 视频文件流传输
+ */
+const getVideoStream = (req, res) => {
+    const filenameParam = req.params.filename;
+    const filename = Array.isArray(filenameParam) ? filenameParam[0] : filenameParam;
+    if (!filename) {
+        (0, node_karin_1.createNotFoundResponse)(res, '无效的文件名');
+        return;
+    }
+    const videoPath = utils_1.Common.validateVideoRequest(filename, res);
+    if (!videoPath) {
+        return;
+    }
+    try {
+        const stats = node_fs_1.default.statSync(videoPath);
+        const fileSize = stats.size;
+        const range = req.headers.range;
+        if (range) {
+            const parts = range.replace(/bytes=/, '').split('-');
+            const start = parseInt(parts[0], 10);
+            const end = parts[1] ? parseInt(parts[1], 10) : fileSize - 1;
+            if (start >= fileSize || end >= fileSize || start > end) {
+                res.status(416).send('请求范围不满足');
+                return;
+            }
+            const chunksize = end - start + 1;
+            const file = node_fs_1.default.createReadStream(videoPath, { start, end });
+            const head = {
+                'Content-Range': `bytes ${start}-${end}/${fileSize}`,
+                'Accept-Ranges': 'bytes',
+                'Content-Length': chunksize,
+                'Content-Type': 'video/mp4'
+            };
+            res.writeHead(206, head);
+            file.pipe(res);
+            file.on('error', (err) => {
+                node_karin_1.logger.error(`读取视频文件流时出错 (Range: ${start}-${end}): ${err.message}`);
+                if (!res.writableEnded) {
+                    res.end();
+                }
+            });
+        }
+        else {
+            const head = {
+                'Content-Length': fileSize,
+                'Content-Type': 'video/mp4',
+                'Accept-Ranges': 'bytes'
+            };
+            res.writeHead(200, head);
+            const file = node_fs_1.default.createReadStream(videoPath);
+            file.pipe(res);
+            file.on('error', (err) => {
+                node_karin_1.logger.error(`读取视频文件流时出错 (Full): ${err.message}`);
+                if (!res.headersSent) {
+                    try {
+                        (0, node_karin_1.createNotFoundResponse)(res, '读取视频文件失败');
+                    }
+                    catch (e) {
+                        node_karin_1.logger.error('发送读取错误响应失败:', e);
+                        if (!res.writableEnded) {
+                            res.end();
+                        }
+                    }
+                }
+                else if (!res.writableEnded) {
+                    res.end();
+                }
+            });
+        }
+    }
+    catch (error) {
+        if (error.code === 'ENOENT') {
+            if (!res.headersSent)
+                (0, node_karin_1.createNotFoundResponse)(res, '视频文件未找到');
+            else if (!res.writableEnded)
+                res.end();
+        }
+        else {
+            node_karin_1.logger.error(`处理视频数据请求时发生错误: ${error.message}`);
+            if (!res.headersSent) {
+                (0, node_karin_1.createNotFoundResponse)(res, '服务器错误');
+            }
+            else if (!res.writableEnded) {
+                res.end();
+            }
+        }
+    }
+};
+exports.getVideoStream = getVideoStream;
+/**
+ * 视频播放页面（SSR）
+ */
+const getVideoPage = (req, res) => {
+    const filenameParam = req.params.filename;
+    const filename = Array.isArray(filenameParam) ? filenameParam[0] : filenameParam;
+    if (!filename) {
+        (0, node_karin_1.createNotFoundResponse)(res, '无效的文件名');
+        return;
+    }
+    const videoPath = utils_1.Common.validateVideoRequest(filename, res);
+    if (!videoPath) {
+        return;
+    }
+    const videoDataUrl = `/kkk/v1/stream/${encodeURIComponent(filename)}`;
+    const previewInfo = utils_1.Common.getVideoPreview(filename);
+    const removeCache = previewInfo?.removeCache ?? Config_1.Config.app.removeCache;
+    const createdAt = previewInfo?.createdAt ?? Date.now();
+    const expireAt = previewInfo?.expireAt ?? (removeCache ? createdAt + 10 * 60 * 1000 : undefined);
+    const htmlContent = (0, render_1.renderVideoPreviewPage)({
+        filename,
+        filePath: previewInfo?.filePath ?? videoPath,
+        videoUrl: videoDataUrl,
+        removeCache,
+        createdAt,
+        expireAt,
+        eventsUrl: `/kkk/v1/video/${encodeURIComponent(filename)}/events`
+    });
+    res.setHeader('Cache-Control', 'no-cache');
+    res.setHeader('Content-Type', 'text/html; charset=utf-8');
+    res.send(htmlContent);
+};
+exports.getVideoPage = getVideoPage;
+/**
+ * 视频预览状态事件流（SSE）
+ */
+const getVideoEvents = (req, res) => {
+    const filenameParam = req.params.filename;
+    const filename = Array.isArray(filenameParam) ? filenameParam[0] : filenameParam;
+    if (!filename) {
+        (0, node_karin_1.createNotFoundResponse)(res, '无效的文件名');
+        return;
+    }
+    const safeName = node_path_1.default.basename(filename);
+    if (safeName !== filename || filename.includes('/') || filename.includes('\\')) {
+        (0, node_karin_1.createNotFoundResponse)(res, '无效的文件名');
+        return;
+    }
+    const previewInfo = utils_1.Common.getVideoPreview(filename);
+    if (!previewInfo) {
+        (0, node_karin_1.createNotFoundResponse)(res, '预览信息不存在');
+        return;
+    }
+    res.setHeader('Content-Type', 'text/event-stream');
+    res.setHeader('Cache-Control', 'no-cache');
+    res.setHeader('Connection', 'keep-alive');
+    res.flushHeaders?.();
+    let timer = null;
+    const sendPayload = () => {
+        const currentInfo = utils_1.Common.getVideoPreview(filename) ?? previewInfo;
+        const now = Date.now();
+        const remainingMs = currentInfo.expireAt ? Math.max(currentInfo.expireAt - now, 0) : null;
+        const fileMissing = currentInfo.filePath ? !node_fs_1.default.existsSync(currentInfo.filePath) : false;
+        const removed = Boolean(currentInfo.removedAt) || (currentInfo.removeCache && remainingMs === 0 && fileMissing);
+        if (removed && !currentInfo.removedAt) {
+            utils_1.Common.markVideoPreviewRemoved(currentInfo.filename);
+        }
+        const payload = {
+            filename: currentInfo.filename,
+            filePath: currentInfo.filePath,
+            removeCache: currentInfo.removeCache,
+            createdAt: currentInfo.createdAt,
+            expireAt: currentInfo.expireAt,
+            remainingMs,
+            removed,
+            serverNow: now
+        };
+        res.write(`data: ${JSON.stringify(payload)}\n\n`);
+        return removed;
+    };
+    const shouldCloseImmediately = sendPayload();
+    if (shouldCloseImmediately) {
+        res.end();
+        return;
+    }
+    timer = setInterval(() => {
+        const removed = sendPayload();
+        if (removed) {
+            if (timer) {
+                clearInterval(timer);
+                timer = null;
+            }
+            res.end();
+        }
+    }, 1000);
+    res.on('close', () => {
+        if (timer) {
+            clearInterval(timer);
+            timer = null;
+        }
+    });
+};
+exports.getVideoEvents = getVideoEvents;

package/lib/karin/module/server/index.js

@@ -0,0 +1,26 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * Server 模块导出
+ */
+__exportStar(require("./constants/routes"), exports);
+__exportStar(require("./controllers/bots"), exports);
+__exportStar(require("./controllers/config"), exports);
+__exportStar(require("./controllers/video"), exports);
+__exportStar(require("./middlewares/auth"), exports);
+// 初始化路由
+require("./routes");