@bililive-tools/huya-recorder 1.7.1 → 1.9.0

package/README.md

@@ -37,6 +37,7 @@ interface Options {
   quality: number; // 见画质参数
   qualityRetry?: number; // 画质匹配重试次数, -1为强制匹配画质，0为自动配置，正整数为最大匹配次数
   streamPriorities: []; // 废弃
+  titleKeywords?: string; // 禁止录制的标题关键字，英文逗号分开多个
   sourcePriorities: []; // 按提供的源优先级去给CDN列表排序，并过滤掉不在优先级配置中的源，在未匹配到的情况下会优先使用TX的CDN，具体参数见 CDN 参数
   formatPriorities?: string[]; // 支持，`flv`和`hls` 参数，默认为['flv','hls']
   disableAutoCheck?: boolean; // 为 true 时 manager 将跳过自动检查
@@ -47,6 +48,7 @@ interface Options {
   api?: "auto" | "mp" | "web"; // 默认为auto，在星秀区使用mp接口，其他使用web接口，你也可以强制指定
   videoFormat?: "auto"; // 视频格式： "auto", "ts", "mkv" ，auto模式下, 分段使用 "ts"，不分段使用 "mp4"
   recorderType?: "auto" | "ffmpeg" | "mesio"; // 底层录制器，使用mesio时videoFormat参数无效
+  debugLevel?: `verbose` | "basic"; // verbose参数时，录制器会输出更加详细的log
 }
 ```
 

package/lib/huya_api.js

@@ -1,10 +1,7 @@
-import axios from "axios";
+import { requester } from "./requester.js";
 import { utils } from "@bililive-tools/manager";
 import { assert, getFormatSources } from "./utils.js";
 import { initInfo } from "./anticode.js";
-const requester = axios.create({
-    timeout: 10e3,
-});
 export async function getRoomInfo(roomIdOrShortId, opts = {}) {
     const res = await requester.get(`https://www.huya.com/${roomIdOrShortId}`);
     const html = res.data;

package/lib/huya_mobile_api.js

@@ -1,16 +1,14 @@
 // import { createHash, randomInt } from "node:crypto";
 // import { URLSearchParams } from "node:url";
-import axios from "axios";
+import { requester } from "./requester.js";
 import { utils } from "@bililive-tools/manager";
 import { assert, getFormatSources } from "./utils.js";
-const requester = axios.create({
-    timeout: 10e3,
-    headers: {
-        "User-Agent": "Mozilla/5.0 (Linux; Android 5.0; SM-G900P Build/LRX21T) AppleWebKit/537.36 (KHTML, like Gecko)",
-    },
-});
 export async function getRoomInfo(roomIdOrShortId, formatPriorities = ["flv", "hls"]) {
-    const res = await requester.get(`https://mp.huya.com/cache.php?m=Live&do=profileRoom&roomid=${roomIdOrShortId}`);
+    const res = await requester.get(`https://mp.huya.com/cache.php?m=Live&do=profileRoom&roomid=${roomIdOrShortId}`, {
+        headers: {
+            "User-Agent": "Mozilla/5.0 (Linux; Android 5.0; SM-G900P Build/LRX21T) AppleWebKit/537.36 (KHTML, like Gecko)",
+        },
+    });
     const html = res.data;
     assert(html, `Unexpected resp, hyPlayerConfig is null`);
     if (res.status !== 200) {

package/lib/index.js

@@ -56,27 +56,44 @@ 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);
+            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);
         this.liveInfo = liveInfo;
@@ -86,7 +103,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         this.state = "check-error";
         throw error;
     }
-    const { living, owner, title } = this.liveInfo;
+    const { living, owner, title, startTime } = this.liveInfo;
     if (this.liveInfo.liveId === banLiveId) {
         this.tempStopIntervalCheck = true;
     }
@@ -97,6 +114,18 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         return null;
     if (!living)
         return null;
+    // 检查标题是否包含关键词，如果包含则不自动录制
+    // 手动开始录制时不检查标题关键词
+    if (utils.shouldCheckTitleKeywords(isManualStart, this.titleKeywords)) {
+        if (utils.hasBlockedTitleKeywords(title, this.titleKeywords)) {
+            this.state = "title-blocked";
+            this.emit("DebugLog", {
+                type: "common",
+                text: `跳过录制：直播间标题 "${title}" 包含关键词 "${this.titleKeywords}"`,
+            });
+            return null;
+        }
+    }
     let res;
     // TODO: 先不做什么错误处理，就简单包一下预期上会有错误的地方
     try {
@@ -149,15 +178,22 @@ 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",
     }, onEnd, async () => {
         const info = await getInfo(this.channelId);
         return info;
@@ -165,6 +201,9 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
     const savePath = getSavePath({
         owner,
         title,
+        startTime: Date.now(),
+        liveStartTime: startTime,
+        recordStartTime,
     });
     try {
         ensureFolderExist(savePath);
@@ -173,8 +212,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;
         }
@@ -255,7 +294,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, banLiveId, isMan
         client.on("retry", (e) => {
             this.emit("DebugLog", {
                 type: "common",
-                text: `huya danmu retry: ${e.count}/${e.max}`,
+                text: `${this?.liveInfo?.owner}:${this.channelId} huya danmu retry: ${e.count}/${e.max}`,
             });
         });
         client.start();
@@ -297,6 +336,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/requester.d.ts

@@ -0,0 +1,2 @@
+import axios from "axios";
+export declare const requester: axios.AxiosInstance;

package/lib/requester.js

@@ -0,0 +1,7 @@
+import axios from "axios";
+export const requester = axios.create({
+    timeout: 10e3,
+    // axios 会自动读取环境变量中的 http_proxy 和 https_proxy 并应用，
+    // 但会导致请求报错 "Client network socket disconnected before secure TLS connection was established"。
+    proxy: false,
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bililive-tools/huya-recorder",
-  "version": "1.7.1",
+  "version": "1.9.0",
   "description": "bililive-tools huya recorder implemention",
   "main": "./lib/index.js",
   "type": "module",
@@ -37,8 +37,8 @@
     "mitt": "^3.0.1",
     "lodash-es": "^4.17.21",
     "axios": "^1.7.8",
-    "@bililive-tools/manager": "^1.6.1",
-    "huya-danma-listener": "0.1.2"
+    "huya-danma-listener": "0.1.3",
+    "@bililive-tools/manager": "^1.9.0"
   },
   "devDependencies": {},
   "scripts": {