koishi-plugin-douyin-local-parser 0.1.0 → 0.1.1

package/README.md

@@ -14,6 +14,7 @@
 - 解析 `https://jingxuan.douyin.com/m/video|note|slides/...`
 - 从 Koishi 卡片元素的 `data` 字段、转义 JSON、普通文本中提取抖音链接
 - 支持图片、动图、视频、原文链接和作者信息返回
+- 支持视频直链发送，或先下载首个视频后以 Buffer / base64 / file URL 模式发送
 - 支持 OneBot / Red 适配器的合并转发元素
 
 ## 本地测试
@@ -39,3 +40,24 @@ plugins:
   /absolute/path/to/douyin-parser/lib:
     enabled: true
 ```
+
+常用视频发送配置：
+
+```yaml
+plugins:
+  /absolute/path/to/douyin-parser/lib:
+    enabled: true
+    showVideo: true
+    downloadVideoAsFile: false
+    videoDownloadMode: buffer
+```
+
+说明：
+
+- `downloadVideoAsFile: false`：默认行为，不下载视频，直接发送抖音视频直链。
+- `downloadVideoAsFile: true`：先用 Koishi 的 `ctx.http.file()` 下载首个视频，下载失败会自动回退直链。
+- `videoDownloadMode: buffer`：以二进制 Buffer 传给 `h.video()`，通常最推荐。
+- `videoDownloadMode: base64`：转换成 `base64://...` 发送，适合部分 OneBot 实现。
+- `videoDownloadMode: file`：写入系统临时目录并以 `file://` URL 发送，适合部分 Napcat / 本地网关场景。
+
+下载模式只处理每个作品的首个视频，避免多视频或动态图集一次性消耗过多带宽。

package/lib/index.d.ts

@@ -17,6 +17,8 @@ export interface Config {
     maxDescLength: number;
     descTruncateSuffix: string;
     showVideo: boolean;
+    downloadVideoAsFile: boolean;
+    videoDownloadMode: 'buffer' | 'file' | 'base64';
     showAuthor: boolean;
     showLink: boolean;
     showError: boolean;

package/lib/index.js

@@ -1,8 +1,15 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.usage = exports.Config = exports.name = void 0;
 exports.apply = apply;
 const koishi_1 = require("koishi");
+const node_os_1 = __importDefault(require("node:os"));
+const node_path_1 = __importDefault(require("node:path"));
+const node_url_1 = __importDefault(require("node:url"));
+const node_fs_1 = require("node:fs");
 const parser_1 = require("./parser");
 exports.name = 'douyin-parser';
 const logger = new koishi_1.Logger(exports.name);
@@ -31,6 +38,12 @@ exports.Config = koishi_1.Schema.intersect([
         maxDescLength: koishi_1.Schema.number().min(0).max(2000).step(10).default(160).description('描述最大字数。设为 0 时不展示描述。'),
         descTruncateSuffix: koishi_1.Schema.string().default('...(已截断)').description('描述超出最大字数时追加的截断标志。'),
         showVideo: koishi_1.Schema.boolean().default(true).description('返回视频元素。'),
+        downloadVideoAsFile: koishi_1.Schema.boolean().default(false).description('尝试先下载首个视频再发送，缓解 QQ / OneBot 等平台直链“资源已过期”的问题。会增加带宽消耗。'),
+        videoDownloadMode: koishi_1.Schema.union([
+            koishi_1.Schema.const('buffer').description('使用二进制 Buffer 方式发送视频（推荐）'),
+            koishi_1.Schema.const('base64').description('使用 base64:// 段发送视频（OneBot 常用格式，buffer 失败时可尝试）'),
+            koishi_1.Schema.const('file').description('写入临时文件并通过 file:// URL 发送（Napcat 等特殊环境）'),
+        ]).default('buffer').description('下载视频后的发送方式。'),
         showAuthor: koishi_1.Schema.boolean().default(true).description('展示作者。'),
         showLink: koishi_1.Schema.boolean().default(true).description('展示原文链接。'),
     }).description('内容设置'),
@@ -69,13 +82,13 @@ function apply(ctx, config) {
         const targets = links.filter((link) => shouldProcess(recent, session.channelId || session.guildId || 'private', link, config.minimumInterval));
         if (!targets.length)
             return next();
-        handleLinks(session, targets, config).catch((error) => {
+        handleLinks(ctx, session, targets, config).catch((error) => {
             logger.warn(error);
         });
         return next();
     }, config.middleware);
 }
-async function handleLinks(session, links, config) {
+async function handleLinks(ctx, session, links, config) {
     let waitTipMessageId;
     if (config.waitTip) {
         const result = await session.send(`${koishi_1.h.quote(session.messageId)}${config.waitTip}`);
@@ -86,7 +99,7 @@ async function handleLinks(session, links, config) {
         for (const link of links) {
             if (config.loggerinfo)
                 logger.info(`parse ${link}`);
-            const post = await (0, parser_1.fetchDouyinPost)(link, config);
+            const post = await preparePostVideo(ctx, await (0, parser_1.fetchDouyinPost)(link, config), config);
             allMessages.push(...(0, parser_1.buildDouyinMessages)(post, config, session));
         }
         if (!allMessages.length)
@@ -115,6 +128,72 @@ async function handleLinks(session, links, config) {
         }
     }
 }
+async function preparePostVideo(ctx, post, config) {
+    if (!config.downloadVideoAsFile || !config.showVideo)
+        return post;
+    const videoUrls = [...post.dynamicImageUrls, ...post.videoUrls];
+    const firstVideo = videoUrls[0];
+    if (!firstVideo || !/^https?:\/\//i.test(firstVideo))
+        return post;
+    try {
+        const file = await ctx.http.file(firstVideo);
+        if (!file?.data) {
+            if (config.loggerinfo)
+                logger.info('download video failed: empty response data');
+            return post;
+        }
+        const buffer = Buffer.from(file.data);
+        const mimeType = file.type || file.mime || 'video/mp4';
+        const mode = config.videoDownloadMode || 'buffer';
+        if (config.loggerinfo) {
+            logger.info(`downloaded first video (${Math.round(buffer.length / 1024)} KB), send mode: ${mode}`);
+        }
+        const remainingDynamicUrls = post.dynamicImageUrls.slice(firstVideo === post.dynamicImageUrls[0] ? 1 : 0);
+        const remainingVideoUrls = firstVideo === post.dynamicImageUrls[0] ? post.videoUrls : post.videoUrls.slice(1);
+        if (mode === 'buffer') {
+            return {
+                ...post,
+                videoBuffer: buffer,
+                videoMimeType: mimeType,
+                dynamicImageUrls: remainingDynamicUrls,
+                videoUrls: remainingVideoUrls,
+            };
+        }
+        const replacement = mode === 'file'
+            ? await createTempVideoFile(buffer, mimeType)
+            : `base64://${buffer.toString('base64')}`;
+        return {
+            ...post,
+            videoBuffer: undefined,
+            videoMimeType: mimeType,
+            dynamicImageUrls: firstVideo === post.dynamicImageUrls[0] ? [replacement, ...remainingDynamicUrls] : remainingDynamicUrls,
+            videoUrls: firstVideo === post.dynamicImageUrls[0] ? remainingVideoUrls : [replacement, ...remainingVideoUrls],
+        };
+    }
+    catch (error) {
+        if (config.loggerinfo)
+            logger.info(`download video failed: ${error instanceof Error ? error.message : String(error)}`);
+        return post;
+    }
+}
+async function createTempVideoFile(buffer, mimeType) {
+    const fileName = `douyin-video-${Date.now()}-${Math.random().toString(16).slice(2)}${getVideoFileExtension(mimeType)}`;
+    const filePath = node_path_1.default.join(node_os_1.default.tmpdir(), fileName);
+    await node_fs_1.promises.writeFile(filePath, buffer);
+    return node_url_1.default.pathToFileURL(filePath).href;
+}
+function getVideoFileExtension(mimeType) {
+    const lower = mimeType.toLowerCase();
+    if (lower.includes('mp4'))
+        return '.mp4';
+    if (lower.includes('webm'))
+        return '.webm';
+    if (lower.includes('ogg') || lower.includes('ogv'))
+        return '.ogv';
+    if (lower.includes('flv'))
+        return '.flv';
+    return '.mp4';
+}
 function shouldProcess(recent, channelId, link, seconds) {
     if (seconds <= 0)
         return true;

package/lib/parser.d.ts

@@ -25,6 +25,8 @@ export interface DouyinPost {
     imageUrls: string[];
     dynamicImageUrls: string[];
     videoUrls: string[];
+    videoBuffer?: Buffer;
+    videoMimeType?: string;
 }
 interface ParsedDouyinUrl {
     url: string;

package/lib/parser.js

@@ -93,6 +93,9 @@ function buildDouyinMessages(post, config, session) {
         }
     }
     if (config.showVideo) {
+        if (post.videoBuffer) {
+            messages.push((0, koishi_1.h)('message', attrs, koishi_1.h.video(post.videoBuffer, post.videoMimeType || 'video/mp4')));
+        }
         for (const videoUrl of [...post.dynamicImageUrls, ...post.videoUrls].slice(0, 1)) {
             messages.push((0, koishi_1.h)('message', attrs, koishi_1.h.video(videoUrl)));
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-douyin-local-parser",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Parse Douyin links and cards for Koishi.",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",

package/src/index.ts

@@ -1,5 +1,9 @@
 import { Context, Logger, Schema, h } from 'koishi'
-import { buildDouyinMessages, extractDouyinLinks, fetchDouyinPost } from './parser'
+import os from 'node:os'
+import path from 'node:path'
+import nodeUrl from 'node:url'
+import { promises as fs } from 'node:fs'
+import { buildDouyinMessages, extractDouyinLinks, fetchDouyinPost, DouyinPost } from './parser'
 
 export const name = 'douyin-parser'
 
@@ -22,6 +26,8 @@ export interface Config {
   maxDescLength: number
   descTruncateSuffix: string
   showVideo: boolean
+  downloadVideoAsFile: boolean
+  videoDownloadMode: 'buffer' | 'file' | 'base64'
   showAuthor: boolean
   showLink: boolean
   showError: boolean
@@ -53,6 +59,12 @@ export const Config: Schema<Config> = Schema.intersect([
     maxDescLength: Schema.number().min(0).max(2000).step(10).default(160).description('描述最大字数。设为 0 时不展示描述。'),
     descTruncateSuffix: Schema.string().default('...(已截断)').description('描述超出最大字数时追加的截断标志。'),
     showVideo: Schema.boolean().default(true).description('返回视频元素。'),
+    downloadVideoAsFile: Schema.boolean().default(false).description('尝试先下载首个视频再发送，缓解 QQ / OneBot 等平台直链“资源已过期”的问题。会增加带宽消耗。'),
+    videoDownloadMode: Schema.union([
+      Schema.const('buffer').description('使用二进制 Buffer 方式发送视频（推荐）'),
+      Schema.const('base64').description('使用 base64:// 段发送视频（OneBot 常用格式，buffer 失败时可尝试）'),
+      Schema.const('file').description('写入临时文件并通过 file:// URL 发送（Napcat 等特殊环境）'),
+    ]).default('buffer').description('下载视频后的发送方式。'),
     showAuthor: Schema.boolean().default(true).description('展示作者。'),
     showLink: Schema.boolean().default(true).description('展示原文链接。'),
   }).description('内容设置'),
@@ -94,7 +106,7 @@ export function apply(ctx: Context, config: Config) {
     const targets = links.filter((link) => shouldProcess(recent, session.channelId || session.guildId || 'private', link, config.minimumInterval))
     if (!targets.length) return next()
 
-    handleLinks(session, targets, config).catch((error) => {
+    handleLinks(ctx, session, targets, config).catch((error) => {
       logger.warn(error)
     })
 
@@ -102,7 +114,7 @@ export function apply(ctx: Context, config: Config) {
   }, config.middleware)
 }
 
-async function handleLinks(session: any, links: string[], config: Config) {
+async function handleLinks(ctx: Context, session: any, links: string[], config: Config) {
   let waitTipMessageId: string | undefined
 
   if (config.waitTip) {
@@ -115,7 +127,7 @@ async function handleLinks(session: any, links: string[], config: Config) {
 
     for (const link of links) {
       if (config.loggerinfo) logger.info(`parse ${link}`)
-      const post = await fetchDouyinPost(link, config)
+      const post = await preparePostVideo(ctx, await fetchDouyinPost(link, config), config)
       allMessages.push(...buildDouyinMessages(post, config, session))
     }
 
@@ -143,6 +155,74 @@ async function handleLinks(session: any, links: string[], config: Config) {
   }
 }
 
+async function preparePostVideo(ctx: Context, post: DouyinPost, config: Config): Promise<DouyinPost> {
+  if (!config.downloadVideoAsFile || !config.showVideo) return post
+
+  const videoUrls = [...post.dynamicImageUrls, ...post.videoUrls]
+  const firstVideo = videoUrls[0]
+  if (!firstVideo || !/^https?:\/\//i.test(firstVideo)) return post
+
+  try {
+    const file = await ctx.http.file(firstVideo)
+    if (!file?.data) {
+      if (config.loggerinfo) logger.info('download video failed: empty response data')
+      return post
+    }
+
+    const buffer = Buffer.from(file.data)
+    const mimeType = (file as any).type || (file as any).mime || 'video/mp4'
+    const mode = config.videoDownloadMode || 'buffer'
+
+    if (config.loggerinfo) {
+      logger.info(`downloaded first video (${Math.round(buffer.length / 1024)} KB), send mode: ${mode}`)
+    }
+
+    const remainingDynamicUrls = post.dynamicImageUrls.slice(firstVideo === post.dynamicImageUrls[0] ? 1 : 0)
+    const remainingVideoUrls = firstVideo === post.dynamicImageUrls[0] ? post.videoUrls : post.videoUrls.slice(1)
+
+    if (mode === 'buffer') {
+      return {
+        ...post,
+        videoBuffer: buffer,
+        videoMimeType: mimeType,
+        dynamicImageUrls: remainingDynamicUrls,
+        videoUrls: remainingVideoUrls,
+      }
+    }
+
+    const replacement = mode === 'file'
+      ? await createTempVideoFile(buffer, mimeType)
+      : `base64://${buffer.toString('base64')}`
+
+    return {
+      ...post,
+      videoBuffer: undefined,
+      videoMimeType: mimeType,
+      dynamicImageUrls: firstVideo === post.dynamicImageUrls[0] ? [replacement, ...remainingDynamicUrls] : remainingDynamicUrls,
+      videoUrls: firstVideo === post.dynamicImageUrls[0] ? remainingVideoUrls : [replacement, ...remainingVideoUrls],
+    }
+  } catch (error) {
+    if (config.loggerinfo) logger.info(`download video failed: ${error instanceof Error ? error.message : String(error)}`)
+    return post
+  }
+}
+
+async function createTempVideoFile(buffer: Buffer, mimeType: string): Promise<string> {
+  const fileName = `douyin-video-${Date.now()}-${Math.random().toString(16).slice(2)}${getVideoFileExtension(mimeType)}`
+  const filePath = path.join(os.tmpdir(), fileName)
+  await fs.writeFile(filePath, buffer)
+  return nodeUrl.pathToFileURL(filePath).href
+}
+
+function getVideoFileExtension(mimeType: string) {
+  const lower = mimeType.toLowerCase()
+  if (lower.includes('mp4')) return '.mp4'
+  if (lower.includes('webm')) return '.webm'
+  if (lower.includes('ogg') || lower.includes('ogv')) return '.ogv'
+  if (lower.includes('flv')) return '.flv'
+  return '.mp4'
+}
+
 function shouldProcess(recent: Map<string, number>, channelId: string, link: string, seconds: number) {
   if (seconds <= 0) return true
 

package/src/parser.ts

@@ -27,6 +27,8 @@ export interface DouyinPost {
   imageUrls: string[]
   dynamicImageUrls: string[]
   videoUrls: string[]
+  videoBuffer?: Buffer
+  videoMimeType?: string
 }
 
 interface ParsedDouyinUrl {
@@ -132,6 +134,9 @@ export function buildDouyinMessages(post: DouyinPost, config: DouyinConfigLike,
   }
 
   if (config.showVideo) {
+    if (post.videoBuffer) {
+      messages.push(h('message', attrs, h.video(post.videoBuffer, post.videoMimeType || 'video/mp4')))
+    }
     for (const videoUrl of [...post.dynamicImageUrls, ...post.videoUrls].slice(0, 1)) {
       messages.push(h('message', attrs, h.video(videoUrl)))
     }