@bililive-tools/douyin-recorder 1.7.1 → 1.8.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/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";
@@ -67,19 +67,47 @@ const ffmpegOutputOptions = [
     "-min_frag_duration",
     "10000000",
 ];
-const ffmpegInputOptions = [
-    "-reconnect",
-    "1",
-    "-reconnect_streamed",
-    "1",
-    "-reconnect_delay_max",
-    "10",
-    "-rw_timeout",
-    "15000000",
-];
+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 +131,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;
@@ -165,8 +205,7 @@ 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 recorder = createBaseRecorder(this.recorderType, {
         url: stream.url,
         outputOptions: ffmpegOutputOptions,
         inputOptions: ffmpegInputOptions,
@@ -174,6 +213,8 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         getSavePath: (opts) => getSavePath({ owner, title: opts.title ?? title, startTime: opts.startTime }),
         disableDanma: this.disableProvideCommentsWhenRecording,
         videoFormat: this.videoFormat ?? "auto",
+        debugLevel: this.debugLevel ?? "none",
+        onlyAudio: stream.onlyAudio,
         headers: {
             Cookie: this.auth,
         },
@@ -192,8 +233,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;
         }
@@ -253,6 +294,9 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
             return;
         if (this.saveGiftDanma === false)
             return;
+        // repeatEnd 表示礼物连击完毕，只记录这个礼物
+        if (!msg.repeatEnd)
+            return;
         const serverTimestamp = Number(msg.common.createTime) > 9999999999
             ? Number(msg.common.createTime)
             : Number(msg.common.createTime) * 1000;

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

@@ -69,12 +69,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.8.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"
+    "@bililive-tools/manager": "^1.8.0",
+    "douyin-danma-listener": "0.2.1"
   },
   "devDependencies": {
     "@types/node": "*"