koishi-plugin-video-parser-all 0.1.4 → 0.1.6

package/lib/index.d.ts

@@ -3,6 +3,7 @@ export declare const name = "video-parser-all";
 export interface Config {
     enable: boolean;
     showWaitingTip: boolean;
+    showEstimatedTime: boolean;
     revokeWaitingTip: boolean;
     waitingTipText: string;
     sameLinkInterval: number;

package/lib/index.js

@@ -15,6 +15,7 @@ exports.name = 'video-parser-all';
 exports.Config = koishi_1.Schema.object({
     enable: koishi_1.Schema.boolean().default(true).description('是否启用插件'),
     showWaitingTip: koishi_1.Schema.boolean().default(true).description('是否显示解析等待提示'),
+    showEstimatedTime: koishi_1.Schema.boolean().default(true).description('是否显示预计解析完成时间'),
     revokeWaitingTip: koishi_1.Schema.boolean().default(true).description('是否撤回等待提示文本'),
     waitingTipText: koishi_1.Schema.string().default('正在解析视频…').description('等待提示文本'),
     sameLinkInterval: koishi_1.Schema.number().default(180).description('相同链接重复间隔秒'),
@@ -32,7 +33,7 @@ exports.Config = koishi_1.Schema.object({
     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('消息缓冲延迟秒'),
+    messageBufferDelay: koishi_1.Schema.number().default(1).description('消息接收缓冲延迟秒'),
     commonApi: koishi_1.Schema.string().default('https://api.bugpk.com/api/short_videos').description('通用解析API'),
     douyin: koishi_1.Schema.object({
         mode: koishi_1.Schema.union([
@@ -64,6 +65,7 @@ exports.Config = koishi_1.Schema.object({
 });
 const processed = new Map();
 const linkBuffer = new Map();
+const parseTimeRecords = new Map();
 const PLATFORM_KEYWORDS = {
     bilibili: ['bilibili', 'b23', 'B站'],
     kuaishou: ['kuaishou', '快手'],
@@ -90,6 +92,33 @@ function getPlatformType(url) {
         return 'bilibili';
     return null;
 }
+function formatTime(date) {
+    const hours = date.getHours().toString().padStart(2, '0');
+    const minutes = date.getMinutes().toString().padStart(2, '0');
+    const seconds = date.getSeconds().toString().padStart(2, '0');
+    return `${hours}:${minutes}:${seconds}`;
+}
+function getEstimatedParseTime(urls) {
+    let totalSeconds = 0;
+    urls.forEach(url => {
+        const platform = getPlatformType(url);
+        if (platform) {
+            const records = parseTimeRecords.get(platform) || [];
+            if (records.length > 0) {
+                const avgTime = records.reduce((sum, time) => sum + time, 0) / records.length;
+                totalSeconds += avgTime / 1000;
+            }
+            else {
+                totalSeconds += 3;
+            }
+        }
+        else {
+            totalSeconds += 3;
+        }
+    });
+    totalSeconds += 1;
+    return Math.ceil(totalSeconds);
+}
 async function downloadVideo(url, filename) {
     const dir = path_1.default.join(process.cwd(), 'temp_videos');
     if (!fs_1.default.existsSync(dir))
@@ -125,6 +154,7 @@ function apply(ctx, config) {
         return config.commonApi;
     }
     async function parse(url) {
+        const startTime = Date.now();
         const pt = getPlatformType(url);
         if (!pt)
             return null;
@@ -133,8 +163,15 @@ function apply(ctx, config) {
             return null;
         try {
             const { data } = await http.get(api, { params: { url } });
-            if (data.code === 200 && data.data)
+            if (data.code === 200 && data.data) {
+                const parseTime = Date.now() - startTime;
+                const records = parseTimeRecords.get(pt) || [];
+                records.push(parseTime);
+                if (records.length > 10)
+                    records.shift();
+                parseTimeRecords.set(pt, records);
                 return parseData(data.data);
+            }
         }
         catch { }
         return null;
@@ -219,15 +256,30 @@ function apply(ctx, config) {
         if (!items.length)
             return;
         if (config.enableForward && session.platform === 'onebot') {
-            const msgs = items.map(it => [
-                it.content ? (0, koishi_1.h)('message', { user_id: session.selfId }, [it.content]) : null,
-                it.video ? (0, koishi_1.h)('message', { user_id: session.selfId }, [it.video]) : null,
-            ].filter(Boolean)).flat();
+            const nodeList = items.map(it => {
+                const content = [];
+                if (it.content)
+                    content.push(it.content);
+                if (it.video)
+                    content.push(it.video);
+                return {
+                    user_id: session.selfId,
+                    time: Math.floor(Date.now() / 1000),
+                    content: content.join('\n')
+                };
+            });
             try {
-                await session.send((0, koishi_1.h)('forward', { messages: msgs }));
+                await session.send((0, koishi_1.h)('forward', {
+                    content: '群聊的聊天记录',
+                    brief: '[聊天记录]',
+                    data: nodeList
+                }));
                 return;
             }
-            catch { }
+            catch (e) {
+                if (!config.ignoreSendError)
+                    ctx.logger.warn(`合并转发发送失败: ${e.message}`);
+            }
         }
         for (const it of items) {
             if (it.content)
@@ -246,30 +298,24 @@ function apply(ctx, config) {
         if (!urls.length)
             return;
         const key = `${session.platform}:${session.userId}:${session.channelId}`;
-        if (config.showWaitingTip && !linkBuffer.has(key)) {
-            const tip = await session.send(config.waitingTipText).catch(() => null);
-            const mid = tip?.messageId || tip?.id;
-            linkBuffer.set(key, {
-                urls,
-                timer: setTimeout(() => flush(session), config.messageBufferDelay * 1000),
-                tipMsgId: mid,
-            });
-        }
-        else {
-            const ex = linkBuffer.get(key);
-            if (ex) {
-                clearTimeout(ex.timer);
-                ex.urls.push(...urls);
-                ex.timer = setTimeout(() => flush(session), config.messageBufferDelay * 1000);
-                linkBuffer.set(key, ex);
-            }
-            else {
-                linkBuffer.set(key, {
-                    urls,
-                    timer: setTimeout(() => flush(session), config.messageBufferDelay * 1000),
-                });
+        if (linkBuffer.has(key))
+            return;
+        let mid;
+        if (config.showWaitingTip) {
+            let tipText = config.waitingTipText;
+            if (config.showEstimatedTime) {
+                const estimatedSeconds = getEstimatedParseTime(urls);
+                const estimatedTime = new Date(Date.now() + estimatedSeconds * 1000);
+                tipText += ` 预计${estimatedSeconds}秒后完成（${formatTime(estimatedTime)}）`;
             }
+            const tip = await session.send(tipText).catch(() => null);
+            mid = tip?.messageId || tip?.id;
         }
+        linkBuffer.set(key, {
+            urls,
+            timer: setTimeout(() => flush(session), config.messageBufferDelay * 1000),
+            tipMsgId: mid,
+        });
     });
     setInterval(() => {
         const now = Date.now();

package/package.json

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