koishi-plugin-video-parser-all 0.1.2 → 0.1.3

package/lib/index.d.ts

@@ -15,6 +15,9 @@ export interface Config {
     bugpkBilibiliApi: string;
     timeout: number;
     ignoreSendError: boolean;
+    enableForward: boolean;
+    downloadVideoBeforeSend: boolean;
+    messageBufferDelay: number;
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -8,37 +8,48 @@ exports.apply = apply;
 const koishi_1 = require("koishi");
 const axios_1 = __importDefault(require("axios"));
 const crypto_1 = __importDefault(require("crypto"));
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const promises_1 = require("stream/promises");
 exports.name = 'video-parser-all';
 exports.Config = koishi_1.Schema.object({
-    enable: koishi_1.Schema.boolean().default(true),
-    showWaitingTip: koishi_1.Schema.boolean().default(true),
-    waitingTipText: koishi_1.Schema.string().default('正在解析视频…'),
-    sameLinkInterval: koishi_1.Schema.number().default(180),
+    enable: koishi_1.Schema.boolean().default(true).description('是否启用插件'),
+    showWaitingTip: koishi_1.Schema.boolean().default(true).description('是否显示解析等待提示'),
+    waitingTipText: koishi_1.Schema.string().default('正在解析视频…').description('解析等待提示的文本内容'),
+    sameLinkInterval: koishi_1.Schema.number().default(180).description('相同链接重复解析间隔（秒），避免频繁解析同一链接'),
     imageParseFormat: koishi_1.Schema.string()
         .role('textarea')
         .default(`\${标题} \${tab} \${UP主}
 \${简介}
 \${~~~}
-\${封面}`),
-    showVideoUrl: koishi_1.Schema.boolean().default(false),
-    maxDescLength: koishi_1.Schema.number().default(200),
-    bugpkUniversalApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/short_videos'),
-    bugpkDouyinMainApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/douyin'),
-    bugpkDouyinBackupApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/dyjx'),
-    bugpkKuaishouApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/ksjx'),
-    bugpkBilibiliApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/bilibili'),
-    timeout: koishi_1.Schema.number().default(15000),
-    ignoreSendError: koishi_1.Schema.boolean().default(true).description('忽略消息发送失败的错误（避免日志刷屏）'),
+\${封面}`)
+        .description('解析结果的输出格式，支持占位符：${标题}、${UP主}、${简介}、${点赞}、${投币}、${收藏}、${转发}、${观看}、${弹幕}、${tab}、${~~~}、${封面}'),
+    showVideoUrl: koishi_1.Schema.boolean().default(false).description('是否在消息中显示无水印视频链接（false则直接发送视频文件）'),
+    maxDescLength: koishi_1.Schema.number().default(200).description('简介内容的最大长度，超出部分会被截断'),
+    bugpkUniversalApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/short_videos').description('通用视频解析API地址（优先调用）'),
+    bugpkDouyinMainApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/douyin').description('抖音主解析API地址（备用）'),
+    bugpkDouyinBackupApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/dyjx').description('抖音备用解析API地址（备用）'),
+    bugpkKuaishouApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/ksjx').description('快手解析API地址（备用）'),
+    bugpkBilibiliApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/bilibili').description('B站解析API地址（备用）'),
+    timeout: koishi_1.Schema.number().default(15000).description('API请求超时时间（毫秒）'),
+    ignoreSendError: koishi_1.Schema.boolean().default(true).description('忽略消息发送失败的错误，避免日志刷屏'),
+    enableForward: koishi_1.Schema.boolean().default(false).description('是否开启合并转发 仅支持 onebot 适配器 其他平台开启 无效'),
+    downloadVideoBeforeSend: koishi_1.Schema.boolean().default(false).description('是否将视频链接下载后再发送 （以解决部分onebot协议端的问题）否则使用视频直链发送'),
+    messageBufferDelay: koishi_1.Schema.number().default(1).description('消息接收缓冲延迟（秒）收到链接后等待指定时间，收集同时发送的多个链接后再逐个处理'),
 });
 const processed = new Map();
+const linkBuffer = new Map();
 const PLATFORM_KEYWORDS = {
     bilibili: ['bilibili', 'b23', 'B站'],
     kuaishou: ['kuaishou', '快手'],
     douyin: ['douyin', '抖音']
 };
 function extractUrl(content) {
-    const urlMatch = content.match(/https?:\/\/[^\s]+/i);
-    return urlMatch ? urlMatch[0] : null;
+    const urlMatches = content.match(/https?:\/\/[^\s]+/gi) || [];
+    return urlMatches.filter(url => {
+        const lowerUrl = url.toLowerCase();
+        return Object.values(PLATFORM_KEYWORDS).some(keywords => keywords.some(keyword => lowerUrl.includes(keyword.toLowerCase())));
+    });
 }
 function hasPlatformKeyword(content) {
     const lowerContent = content.toLowerCase();
@@ -57,6 +68,21 @@ function getPlatformType(url) {
         return 'bilibili';
     return null;
 }
+async function downloadVideo(url, filename) {
+    const downloadPath = path_1.default.join(process.cwd(), 'temp_videos');
+    if (!fs_1.default.existsSync(downloadPath)) {
+        fs_1.default.mkdirSync(downloadPath, { recursive: true });
+    }
+    const filePath = path_1.default.join(downloadPath, `${filename}.mp4`);
+    const response = await (0, axios_1.default)({
+        url,
+        method: 'GET',
+        responseType: 'stream',
+        timeout: 30000
+    });
+    await (0, promises_1.pipeline)(response.data, fs_1.default.createWriteStream(filePath));
+    return filePath;
+}
 function parseUniversalApiData(data) {
     return {
         title: data.title || '无标题',
@@ -112,6 +138,10 @@ function parseBilibili(data) {
         videoUrl = data.play;
     else if (data.durl && Array.isArray(data.durl))
         videoUrl = data.durl[0]?.url || '';
+    else if (data.hd_url)
+        videoUrl = data.hd_url;
+    else if (data.sd_url)
+        videoUrl = data.sd_url;
     const digg = data.like || data.digg || 0;
     const coin = data.coin || 0;
     const collect = data.favorite || data.collect || 0;
@@ -200,50 +230,112 @@ function apply(ctx, config) {
             cover: '', video: ''
         };
     }
-    async function sendResult(session, data) {
-        try {
-            let text = config.imageParseFormat
-                .replace(/\${标题}/g, data.title)
-                .replace(/\${UP主}/g, data.author)
-                .replace(/\${简介}/g, data.desc.slice(0, config.maxDescLength))
-                .replace(/\${点赞}/g, data.digg.toString())
-                .replace(/\${投币}/g, data.coin.toString())
-                .replace(/\${收藏}/g, data.collect.toString())
-                .replace(/\${转发}/g, data.share.toString())
-                .replace(/\${观看}/g, data.play.toString())
-                .replace(/\${弹幕}/g, data.danmaku.toString())
-                .replace(/\${tab}/g, '\t')
-                .replace(/\${~~~}/g, '——————————————');
-            const [beforeCover, afterCover] = text.split('\${封面}');
-            const msgParts = [];
-            if (beforeCover && beforeCover.trim())
-                msgParts.push(beforeCover.trim());
-            if (data.cover)
-                msgParts.push(koishi_1.h.image(data.cover));
-            if (afterCover && afterCover.trim())
-                msgParts.push(afterCover.trim());
-            if (data.video && config.showVideoUrl)
-                msgParts.push(`🔗 无水印链接：${data.video}`);
-            if (msgParts.length > 0) {
-                await session.send(msgParts.join('\n')).catch(e => {
-                    if (!config.ignoreSendError)
-                        ctx.logger.warn(`发送合并消息失败: ${e.message}`);
-                });
-            }
-            if (data.video && !config.showVideoUrl) {
+    async function processSingleUrl(session, url) {
+        const hash = crypto_1.default.createHash('md5').update(url).digest('hex');
+        const now = Date.now();
+        if (processed.get(hash) && now - processed.get(hash) < config.sameLinkInterval * 1000)
+            return;
+        processed.set(hash, now);
+        const data = await parseVideo(url);
+        let text = config.imageParseFormat
+            .replace(/\${标题}/g, data.title)
+            .replace(/\${UP主}/g, data.author)
+            .replace(/\${简介}/g, data.desc.slice(0, config.maxDescLength))
+            .replace(/\${点赞}/g, data.digg.toString())
+            .replace(/\${投币}/g, data.coin.toString())
+            .replace(/\${收藏}/g, data.collect.toString())
+            .replace(/\${转发}/g, data.share.toString())
+            .replace(/\${观看}/g, data.play.toString())
+            .replace(/\${弹幕}/g, data.danmaku.toString())
+            .replace(/\${tab}/g, '\t')
+            .replace(/\${~~~}/g, '——————————————');
+        const [beforeCover, afterCover] = text.split('\${封面}');
+        const msgParts = [];
+        if (beforeCover && beforeCover.trim())
+            msgParts.push(beforeCover.trim());
+        if (data.cover)
+            msgParts.push(koishi_1.h.image(data.cover));
+        if (afterCover && afterCover.trim())
+            msgParts.push(afterCover.trim());
+        if (data.video && config.showVideoUrl)
+            msgParts.push(`🔗 无水印链接：${data.video}`);
+        let videoMsg = '';
+        if (data.video && !config.showVideoUrl) {
+            if (config.downloadVideoBeforeSend && data.video) {
                 try {
-                    await session.send(koishi_1.h.video(data.video)).catch(e => {
-                        session.send(`📥 无水印视频：${data.video}`).catch(() => { });
-                    });
+                    const filename = crypto_1.default.createHash('md5').update(data.video).digest('hex');
+                    const filePath = await downloadVideo(data.video, filename);
+                    videoMsg = koishi_1.h.video(`file://${filePath}`);
                 }
                 catch (e) {
-                    await session.send(`📥 无水印视频：${data.video}`).catch(() => { });
+                    ctx.logger.error(`视频下载失败: ${e.message}`);
+                    videoMsg = `📥 无水印视频：${data.video}`;
                 }
             }
+            else {
+                videoMsg = koishi_1.h.video(data.video);
+            }
         }
-        catch (e) {
-            if (!config.ignoreSendError)
-                ctx.logger.error(`发送结果异常: ${e.message}`);
+        return {
+            content: msgParts.join('\n'),
+            video: videoMsg,
+            data
+        };
+    }
+    async function processBufferedUrls(session) {
+        const sessionKey = `${session.platform}:${session.userId}:${session.channelId}`;
+        const bufferData = linkBuffer.get(sessionKey);
+        if (!bufferData)
+            return;
+        clearTimeout(bufferData.timer);
+        linkBuffer.delete(sessionKey);
+        const results = [];
+        for (const url of bufferData.urls) {
+            const result = await processSingleUrl(session, url);
+            if (result)
+                results.push(result);
+        }
+        if (results.length === 0)
+            return;
+        // 修复：通过平台判断是否为onebot，而非直接访问session.adapter
+        if (config.enableForward && session.platform === 'onebot') {
+            const forwardMessages = results.map(result => {
+                const messages = [
+                    (0, koishi_1.h)('message', { user_id: session.selfId }, [result.content]),
+                ];
+                if (result.video)
+                    messages.push((0, koishi_1.h)('message', { user_id: session.selfId }, [result.video]));
+                return messages;
+            }).flat();
+            await session.send((0, koishi_1.h)('forward', {
+                messages: forwardMessages,
+                title: '视频解析结果',
+                brief: `共解析${results.length}个视频链接`,
+                source: '视频解析插件',
+                preview: [results[0].content.substring(0, 10) + '...'],
+                summary: `查看${results.length}条解析结果`
+            })).catch(e => {
+                if (!config.ignoreSendError)
+                    ctx.logger.warn(`合并转发发送失败: ${e.message}`);
+                sendIndividualMessages(session, results);
+            });
+        }
+        else {
+            sendIndividualMessages(session, results);
+        }
+    }
+    async function sendIndividualMessages(session, results) {
+        for (const result of results) {
+            try {
+                if (result.content)
+                    await session.send(result.content);
+                if (result.video)
+                    await session.send(result.video);
+            }
+            catch (e) {
+                if (!config.ignoreSendError)
+                    ctx.logger.warn(`单条消息发送失败: ${e.message}`);
+            }
         }
     }
     ctx.on('message', async (session) => {
@@ -253,22 +345,27 @@ function apply(ctx, config) {
             const content = session.content.trim();
             if (!hasPlatformKeyword(content))
                 return;
-            const url = extractUrl(content);
-            if (!url)
-                return;
-            const hash = crypto_1.default.createHash('md5').update(url).digest('hex');
-            const now = Date.now();
-            if (processed.get(hash) && now - processed.get(hash) < config.sameLinkInterval * 1000)
+            const urls = extractUrl(content);
+            if (urls.length === 0)
                 return;
-            processed.set(hash, now);
-            if (config.showWaitingTip) {
+            if (config.showWaitingTip && !linkBuffer.has(`${session.platform}:${session.userId}:${session.channelId}`)) {
                 await session.send(config.waitingTipText).catch(e => {
                     if (!config.ignoreSendError)
                         ctx.logger.warn(`发送等待提示失败: ${e.message}`);
                 });
             }
-            const data = await parseVideo(url);
-            await sendResult(session, data);
+            const sessionKey = `${session.platform}:${session.userId}:${session.channelId}`;
+            if (linkBuffer.has(sessionKey)) {
+                const bufferData = linkBuffer.get(sessionKey);
+                bufferData.urls.push(...urls);
+                clearTimeout(bufferData.timer);
+                bufferData.timer = setTimeout(() => processBufferedUrls(session), config.messageBufferDelay * 1000);
+                linkBuffer.set(sessionKey, bufferData);
+            }
+            else {
+                const timer = setTimeout(() => processBufferedUrls(session), config.messageBufferDelay * 1000);
+                linkBuffer.set(sessionKey, { urls, timer });
+            }
         }
         catch (e) {
             ctx.logger.error(`消息处理异常: ${e.message}`);
@@ -277,6 +374,16 @@ function apply(ctx, config) {
     setInterval(() => {
         const now = Date.now();
         processed.forEach((t, k) => now - t > 86400000 && processed.delete(k));
+        const tempPath = path_1.default.join(process.cwd(), 'temp_videos');
+        if (fs_1.default.existsSync(tempPath)) {
+            fs_1.default.readdirSync(tempPath).forEach(file => {
+                const filePath = path_1.default.join(tempPath, file);
+                const stat = fs_1.default.statSync(filePath);
+                if (now - stat.ctimeMs > 3600000) {
+                    fs_1.default.unlinkSync(filePath);
+                }
+            });
+        }
     }, 3600000);
     ctx.logger.info('视频解析插件加载完成');
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-video-parser-all",
   "description": "Koishi 视频解析插件，支持抖音/快手/B站链接解析，可自定义API和解析规则",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [