@bililive-tools/douyin-recorder 1.7.1 → 1.9.0

package/README.md

@@ -45,10 +45,12 @@ interface Options {
   videoFormat?: "auto"; // 视频格式： "auto", "ts", "mkv" ，auto模式下, 分段使用 "ts"，不分段使用 "mp4"
   useServerTimestamp?: boolean; // 控制弹幕是否使用服务端时间戳，默认为true
   doubleScreen?: boolean; // 是否使用双屏直播流，开启后如果是双屏直播，那么就使用拼接的流，默认为true
-  recorderType?: "auto" | "ffmpeg" | "mesio"; // 底层录制器，使用mesio时videoFormat参数无效
+  recorderType?: "auto" | "ffmpeg" | "mesio" | "bililive"; // 底层录制器，使用mesio和bililive时videoFormat参数无效
   auth?: string; // 传递cookie
   uid?: string; // 参数为 sec_user_uid 参数
   api?: "web" | "webHTML" | "mobile" | "userHTML" | "balance" | "random"; // 使用不同的接口，默认使用web，具体区别见文档
+  titleKeywords?: string; // 禁止录制的标题关键字，英文逗号分开多个
+  debugLevel?: `verbose` | "basic"; // verbose参数时，录制器会输出更加详细的log
 }
 ```
 

package/lib/douyin_api.d.ts

@@ -6,6 +6,11 @@ import type { APIType, RealAPIType } from "./types.js";
  */
 export declare function resolveShortURL(shortURL: string): Promise<string>;
 export declare const getCookie: () => Promise<string>;
+/**
+ * 随机选择一个可用的 API 接口
+ * @returns 随机选择的 API 类型
+ */
+export declare function selectRandomAPI(exclude?: RealAPIType[]): RealAPIType;
 export declare function getRoomInfo(webRoomId: string, opts?: {
     auth?: string;
     doubleScreen?: boolean;

package/lib/douyin_api.js

@@ -107,8 +107,16 @@ function generateNonce() {
  * 随机选择一个可用的 API 接口
  * @returns 随机选择的 API 类型
  */
-function selectRandomAPI() {
+export function selectRandomAPI(exclude) {
     const availableAPIs = ["web", "webHTML", "mobile", "userHTML"];
+    if (exclude && exclude.length > 0) {
+        for (const api of exclude) {
+            const index = availableAPIs.indexOf(api);
+            if (index !== -1) {
+                availableAPIs.splice(index, 1);
+            }
+        }
+    }
     const randomIndex = Math.floor(Math.random() * availableAPIs.length);
     return availableAPIs[randomIndex];
 }
@@ -151,7 +159,7 @@ async function getRoomInfoByUserWeb(secUserId, opts = {}) {
             nickname: "",
             sec_uid: "",
             avatar: "",
-            api: "webHTML",
+            api: "userHTML",
             room: null,
         };
     }
@@ -244,7 +252,7 @@ async function getRoomInfoByHtml(webRoomId, opts = {}) {
             nickname: roomInfo?.anchor?.nickname ?? "",
             sec_uid: roomInfo?.anchor?.sec_uid ?? "",
             avatar: roomInfo?.anchor?.avatar_thumb?.url_list?.[0] ?? "",
-            api: "userHTML",
+            api: "webHTML",
             room: {
                 title: roomInfo?.room?.title ?? "",
                 cover: roomInfo?.room?.cover?.url_list?.[0] ?? "",

package/lib/index.js

@@ -1,6 +1,6 @@
 import path from "node:path";
 import mitt from "mitt";
-import { defaultFromJSON, defaultToJSON, genRecorderUUID, genRecordUUID, createBaseRecorder, } from "@bililive-tools/manager";
+import { defaultFromJSON, defaultToJSON, genRecorderUUID, genRecordUUID, utils, createBaseRecorder, } from "@bililive-tools/manager";
 import { getInfo, getStream } from "./stream.js";
 import { ensureFolderExist, singleton } from "./utils.js";
 import { resolveShortURL, parseUser } from "./douyin_api.js";
@@ -59,27 +59,48 @@ function createRecorder(opts) {
     });
     return recorderWithSupportUpdatedEvent;
 }
-const ffmpegOutputOptions = [
-    "-c",
-    "copy",
-    "-movflags",
-    "faststart+frag_keyframe+empty_moov",
-    "-min_frag_duration",
-    "10000000",
-];
-const ffmpegInputOptions = [
-    "-reconnect",
-    "1",
-    "-reconnect_streamed",
-    "1",
-    "-reconnect_delay_max",
-    "10",
-    "-rw_timeout",
-    "15000000",
-];
+const ffmpegOutputOptions = [];
+const ffmpegInputOptions = ["-rw_timeout", "10000000", "-timeout", "10000000"];
 const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isManualStart, }) {
-    if (this.recordHandle != null)
+    // 如果已经在录制中，只在需要检查标题关键词时才获取最新信息
+    if (this.recordHandle != null) {
+        // 只有当设置了标题关键词时，并且不是手动启动的录制，才获取最新的直播间信息
+        if (utils.shouldCheckTitleKeywords(isManualStart, this.titleKeywords)) {
+            const now = Date.now();
+            // 每5分钟检查一次标题变化
+            const titleCheckInterval = 5 * 60 * 1000; // 5分钟
+            // 获取上次检查时间
+            const lastCheckTime = typeof this.extra.lastTitleCheckTime === "number" ? this.extra.lastTitleCheckTime : 0;
+            // 如果距离上次检查时间不足指定间隔，则跳过检查
+            if (now - lastCheckTime < titleCheckInterval) {
+                return this.recordHandle;
+            }
+            // 更新检查时间
+            this.extra.lastTitleCheckTime = now;
+            // 获取直播间信息
+            const liveInfo = await getInfo(this.channelId, {
+                cookie: this.auth,
+                api: this.api,
+                uid: this.uid,
+            });
+            const { title } = liveInfo;
+            // 检查标题是否包含关键词
+            if (utils.hasBlockedTitleKeywords(title, this.titleKeywords)) {
+                this.state = "title-blocked";
+                this.emit("DebugLog", {
+                    type: "common",
+                    text: `检测到标题包含关键词，停止录制：直播间标题 "${title}" 包含关键词 "${this.titleKeywords}"`,
+                });
+                // 停止录制
+                await this.recordHandle.stop("直播间标题包含关键词");
+                // 返回 null，停止录制
+                return null;
+            }
+        }
+        // 已经在录制中，直接返回
         return this.recordHandle;
+    }
+    // 获取直播间信息
     try {
         const liveInfo = await getInfo(this.channelId, {
             cookie: this.auth,
@@ -103,6 +124,18 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         return null;
     if (!this.liveInfo.living)
         return null;
+    // 检查标题是否包含关键词，如果包含则不自动录制
+    // 手动开始录制时不检查标题关键词
+    if (utils.shouldCheckTitleKeywords(isManualStart, this.titleKeywords)) {
+        if (utils.hasBlockedTitleKeywords(this.liveInfo.title, this.titleKeywords)) {
+            this.state = "title-blocked";
+            this.emit("DebugLog", {
+                type: "common",
+                text: `跳过录制：直播间标题 "${this.liveInfo.title}" 包含关键词 "${this.titleKeywords}"`,
+            });
+            return null;
+        }
+    }
     let res;
     try {
         let strictQuality = false;
@@ -141,7 +174,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         this.state = "check-error";
         throw err;
     }
-    const { owner, title } = this.liveInfo;
+    const { owner, title, startTime } = this.liveInfo;
     this.state = "recording";
     const { currentStream: stream, sources: availableSources, streams: availableStreams } = res;
     this.availableStreams = availableStreams.map((s) => s.desc);
@@ -165,15 +198,23 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         const reason = args[0] instanceof Error ? args[0].message : String(args[0]);
         this.recordHandle?.stop(reason);
     };
-    let recorderType = this.recorderType === "mesio" ? "mesio" : "ffmpeg";
-    const recorder = createBaseRecorder(recorderType, {
+    const recordStartTime = new Date();
+    const recorder = createBaseRecorder(this.recorderType, {
         url: stream.url,
         outputOptions: ffmpegOutputOptions,
         inputOptions: ffmpegInputOptions,
         segment: this.segment ?? 0,
-        getSavePath: (opts) => getSavePath({ owner, title: opts.title ?? title, startTime: opts.startTime }),
+        getSavePath: (opts) => getSavePath({
+            owner,
+            title: opts.title ?? title,
+            startTime: opts.startTime,
+            liveStartTime: startTime,
+            recordStartTime,
+        }),
         disableDanma: this.disableProvideCommentsWhenRecording,
         videoFormat: this.videoFormat ?? "auto",
+        debugLevel: this.debugLevel ?? "none",
+        onlyAudio: stream.onlyAudio,
         headers: {
             Cookie: this.auth,
         },
@@ -184,6 +225,9 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
     const savePath = getSavePath({
         owner,
         title,
+        startTime: Date.now(),
+        liveStartTime: startTime,
+        recordStartTime,
     });
     try {
         ensureFolderExist(savePath);
@@ -192,8 +236,8 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         this.state = "idle";
         throw err;
     }
-    const handleVideoCreated = async ({ filename, title, cover }) => {
-        this.emit("videoFileCreated", { filename, cover });
+    const handleVideoCreated = async ({ filename, title, cover, rawFilename }) => {
+        this.emit("videoFileCreated", { filename, cover, rawFilename });
         if (title && this?.liveInfo) {
             this.liveInfo.title = title;
         }
@@ -223,6 +267,10 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         }
         this.emit("progress", progress);
     });
+    // 礼物消息缓存管理
+    const giftMessageCache = new Map();
+    // 礼物延迟处理时间(毫秒),可根据实际情况调整
+    const GIFT_DELAY = 5000;
     const client = new DouYinDanmaClient(this?.liveInfo?.liveId, {
         cookie: this.auth,
     });
@@ -272,8 +320,25 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
                 // },
             },
         };
-        this.emit("Message", gift);
-        extraDataController.addMessage(gift);
+        // 单独使用groupId并不可靠
+        const groupId = `${msg.groupId}_${msg.user.id}_${msg.giftId}`;
+        // 如果已存在相同 groupId 的礼物,清除旧的定时器
+        const existing = giftMessageCache.get(groupId);
+        if (existing) {
+            clearTimeout(existing.timer);
+        }
+        // 创建新的定时器
+        const timer = setTimeout(() => {
+            const cachedGift = giftMessageCache.get(groupId);
+            if (cachedGift) {
+                // 延迟时间到,添加最终的礼物消息
+                this.emit("Message", cachedGift.gift);
+                extraDataController.addMessage(cachedGift.gift);
+                giftMessageCache.delete(groupId);
+            }
+        }, GIFT_DELAY);
+        // 更新缓存
+        giftMessageCache.set(groupId, { gift, timer });
     });
     client.on("reconnect", (attempts) => {
         this.emit("DebugLog", {
@@ -337,6 +402,17 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
             return;
         this.state = "stopping-record";
         try {
+            // 清理所有礼物缓存定时器
+            for (const [_groupId, cached] of giftMessageCache.entries()) {
+                clearTimeout(cached.timer);
+                // 立即添加剩余的礼物消息
+                const extraDataController = recorder.getExtraDataController();
+                if (extraDataController) {
+                    this.emit("Message", cached.gift);
+                    extraDataController.addMessage(cached.gift);
+                }
+            }
+            giftMessageCache.clear();
             client.close();
             await recorder.stop();
         }
@@ -357,6 +433,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         id: genRecordUUID(),
         stream: stream.name,
         source: stream.source,
+        recorderType: recorder.type,
         url: stream.url,
         ffmpegArgs,
         savePath: savePath,

package/lib/stream.d.ts

@@ -29,6 +29,7 @@ export declare function getStream(opts: Pick<Recorder, "channelId" | "quality" |
         name: string;
         source: string;
         url: string;
+        onlyAudio: boolean;
     };
     living: boolean;
     roomId: string;

package/lib/stream.js

@@ -1,4 +1,4 @@
-import { getRoomInfo } from "./douyin_api.js";
+import { getRoomInfo, selectRandomAPI } from "./douyin_api.js";
 import { globalLoadBalancer } from "./loadBalancer/loadBalancer.js";
 export async function getInfo(channelId, opts) {
     let info;
@@ -31,6 +31,9 @@ export async function getStream(opts) {
         // userHTML 接口只能用于状态检测
         api = "web";
     }
+    else if (api === "random") {
+        api = selectRandomAPI(["userHTML"]);
+    }
     const info = await getRoomInfo(opts.channelId, {
         doubleScreen: opts.doubleScreen ?? true,
         auth: opts.cookie,
@@ -69,12 +72,23 @@ export async function getStream(opts) {
     if (!url) {
         throw new Error("未找到对应的流");
     }
+    let onlyAudio = false;
+    try {
+        const urlObj = new URL(url);
+        if (urlObj.searchParams.get("only_audio") == "1") {
+            onlyAudio = true;
+        }
+    }
+    catch (error) {
+        console.warn("解析流 URL 失败", error);
+    }
     return {
         ...info,
         currentStream: {
             name: qualityName,
             source: "自动",
             url: url,
+            onlyAudio,
         },
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bililive-tools/douyin-recorder",
-  "version": "1.7.1",
+  "version": "1.9.0",
   "description": "@bililive-tools douyin recorder implemention",
   "main": "./lib/index.js",
   "type": "module",
@@ -38,8 +38,8 @@
     "lodash-es": "^4.17.21",
     "mitt": "^3.0.1",
     "sm-crypto": "^0.3.13",
-    "@bililive-tools/manager": "^1.6.1",
-    "douyin-danma-listener": "0.2.0"
+    "douyin-danma-listener": "0.2.1",
+    "@bililive-tools/manager": "^1.9.0"
   },
   "devDependencies": {
     "@types/node": "*"