@bililive-tools/bilibili-recorder 1.2.0 → 1.4.1

package/README.md

@@ -51,6 +51,7 @@ interface Options {
   useM3U8Proxy?: boolean; // 是否使用m3u8代理，由于hls及fmp4存在一个小时超时时间，需自行实现代理避免
   m3u8ProxyUrl?: string; // 代理链接，文档待补充
   videoFormat?: "auto"; // 视频格式： "auto", "ts", "mkv" ，auto模式下, 分段使用 "ts"，不分段使用 "mp4"
+  onlyAudio?: boolean; // 只录制音频，默认为否
 }
 ```
 

package/lib/bilibili_api.d.ts

@@ -1,4 +1,4 @@
-type LiveStatus = 0 | 1 | 2;
+export type LiveStatus = 0 | 1 | 2;
 export declare function getRoomInit(roomIdOrShortId: number): Promise<{
     room_id: number;
     short_id: number;
@@ -71,6 +71,7 @@ export declare function getPlayURL(roomId: number, opts?: {
 export declare function getRoomPlayInfo(roomIdOrShortId: number, opts?: {
     qn?: number;
     cookie?: string;
+    onlyAudio?: boolean;
 }): Promise<{
     uid: number;
     room_id: number;
@@ -85,6 +86,7 @@ export declare function getRoomPlayInfo(roomIdOrShortId: number, opts?: {
         };
     };
 }>;
+export declare function getBuvidConf(): Promise<any>;
 export interface ProtocolInfo {
     protocol_name: "http_stream" | "http_hls";
     format: FormatInfo[];
@@ -110,4 +112,3 @@ export interface SourceProfile {
     extra: string;
     stream_ttl: number;
 }
-export {};

package/lib/bilibili_api.js

@@ -61,6 +61,7 @@ export async function getRoomPlayInfo(roomIdOrShortId, opts = {}) {
             codec: "0,1",
             // 0 flv, 1 ts, 2 fmp4
             format: "0,1,2",
+            only_audio: opts.onlyAudio ? "1" : "0",
         },
         headers: {
             Cookie: opts.cookie,
@@ -69,3 +70,14 @@ export async function getRoomPlayInfo(roomIdOrShortId, opts = {}) {
     assert(res.data.code === 0, `Unexpected resp, code ${res.data.code}, msg ${res.data.message}`);
     return res.data.data;
 }
+export async function getBuvidConf() {
+    const res = await fetch("https://api.bilibili.com/x/frontend/finger/spi", {
+        headers: {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+        },
+    });
+    if (!res.ok)
+        throw new Error(`Failed to get buvid conf: ${res.statusText}`);
+    const data = await res.json();
+    return data;
+}

package/lib/danma.d.ts

@@ -6,7 +6,7 @@ declare class DanmaClient extends EventEmitter {
     private uid;
     private retryCount;
     constructor(roomId: number, auth: string | undefined, uid: number | undefined);
-    start(): void;
+    start(): Promise<void>;
     stop(): void;
 }
 export default DanmaClient;

package/lib/danma.js

@@ -1,5 +1,46 @@
 import { EventEmitter } from "node:events";
 import { startListen } from "./blive-message-listener/index.js";
+import { getBuvidConf } from "./bilibili_api.js";
+// 全局缓存，一天过期时间 (24 * 60 * 60 * 1000 ms)
+const CACHE_DURATION = 24 * 60 * 60 * 1000;
+let buvidCache = null;
+// 获取带缓存的 buvid 配置
+async function getCachedBuvidConf() {
+    const now = Date.now();
+    // 检查缓存是否有效
+    if (buvidCache && now - buvidCache.timestamp < CACHE_DURATION) {
+        return buvidCache.data;
+    }
+    // 缓存失效或不存在，重新获取（带重试）
+    const info = await getBuvidConfWithRetry();
+    buvidCache = {
+        data: info,
+        timestamp: now,
+    };
+    return info;
+}
+// 带重试功能的 getBuvidConf
+async function getBuvidConfWithRetry(maxRetries = 3, retryDelay = 1000) {
+    let lastError;
+    for (let attempt = 1; attempt <= maxRetries; attempt++) {
+        try {
+            const result = await getBuvidConf();
+            return result;
+        }
+        catch (error) {
+            lastError = error;
+            // 如果是最后一次尝试，直接抛出错误
+            if (attempt === maxRetries) {
+                throw error;
+            }
+            // 等待指定时间后重试，使用指数退避策略
+            const delay = retryDelay * Math.pow(2, attempt - 1);
+            await new Promise((resolve) => setTimeout(resolve, delay));
+        }
+    }
+    // 这里不应该到达，但为了类型安全
+    throw lastError;
+}
 class DanmaClient extends EventEmitter {
     client = null;
     roomId;
@@ -12,7 +53,9 @@ class DanmaClient extends EventEmitter {
         this.auth = auth;
         this.uid = uid;
     }
-    start() {
+    async start() {
+        const info = await getCachedBuvidConf();
+        const buvid3 = info.data.b_3;
         const handler = {
             onIncomeDanmu: (msg) => {
                 let content = msg.body.content;
@@ -21,7 +64,7 @@ class DanmaClient extends EventEmitter {
                     return;
                 const comment = {
                     type: "comment",
-                    timestamp: msg.timestamp,
+                    timestamp: msg.body.timestamp,
                     text: content,
                     color: msg.body.content_color,
                     mode: msg.body.type,
@@ -41,7 +84,7 @@ class DanmaClient extends EventEmitter {
                 const content = msg.body.content.replaceAll(/[\r\n]/g, "");
                 const comment = {
                     type: "super_chat",
-                    timestamp: msg.timestamp,
+                    timestamp: msg.raw.send_time,
                     text: content,
                     price: msg.body.price,
                     sender: {
@@ -79,7 +122,7 @@ class DanmaClient extends EventEmitter {
             onGift: (msg) => {
                 const gift = {
                     type: "give_gift",
-                    timestamp: msg.timestamp,
+                    timestamp: msg.raw.send_time,
                     name: msg.body.gift_name,
                     count: msg.body.amount,
                     price: msg.body.coin_type === "silver" ? 0 : msg.body.price / 1000,
@@ -98,11 +141,26 @@ class DanmaClient extends EventEmitter {
                 };
                 this.emit("Message", gift);
             },
+            onRoomInfoChange: (msg) => {
+                this.emit("RoomInfoChange", msg);
+            },
         };
+        let lastAuth = "";
+        if (this.auth?.includes("buvid3")) {
+            lastAuth = this.auth;
+        }
+        else {
+            if (this.auth) {
+                lastAuth = `${this.auth}; buvid3=${buvid3}`;
+            }
+            else {
+                lastAuth = `buvid3=${buvid3}`;
+            }
+        }
         this.client = startListen(this.roomId, handler, {
             ws: {
                 headers: {
-                    Cookie: this.auth ?? "",
+                    Cookie: lastAuth,
                 },
                 uid: this.uid ?? 0,
             },

package/lib/index.js

@@ -2,7 +2,7 @@ import path from "node:path";
 import mitt from "mitt";
 import { defaultFromJSON, defaultToJSON, genRecorderUUID, genRecordUUID, utils, FFMPEGRecorder, } from "@bililive-tools/manager";
 import { getInfo, getStream, getLiveStatus, getStrictStream } from "./stream.js";
-import { ensureFolderExist } from "./utils.js";
+import { ensureFolderExist, hasKeyword } from "./utils.js";
 import DanmaClient from "./danma.js";
 function createRecorder(opts) {
     // 内部实现时，应该只有 proxy 包裹的那一层会使用这个 recorder 标识符，不应该有直接通过
@@ -67,7 +67,7 @@ const ffmpegOutputOptions = [
     "-movflags",
     "faststart+frag_keyframe+empty_moov",
     "-min_frag_duration",
-    "60000000",
+    "10000000",
 ];
 const ffmpegInputOptions = [
     "-reconnect",
@@ -103,7 +103,21 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
         return null;
     if (!living)
         return null;
-    this.emit("LiveStart", { liveId });
+    // 检查标题是否包含关键词，如果包含则不自动录制
+    // 手动开始录制时不检查标题关键词
+    if (!isManualStart &&
+        this.titleKeywords &&
+        typeof this.titleKeywords === "string" &&
+        this.titleKeywords.trim()) {
+        const hasTitleKeyword = hasKeyword(_title, this.titleKeywords);
+        if (hasTitleKeyword) {
+            this.emit("DebugLog", {
+                type: "common",
+                text: `跳过录制：直播间标题 "${_title}" 包含关键词 "${this.titleKeywords}"`,
+            });
+            return null;
+        }
+    }
     const liveInfo = await getInfo(this.channelId);
     const { owner, title, roomId } = liveInfo;
     this.liveInfo = liveInfo;
@@ -127,6 +141,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
             strictQuality: strictQuality,
             formatName: this.formatName,
             codecName: this.codecName,
+            onlyAudio: this.onlyAudio,
         });
     }
     catch (err) {
@@ -165,8 +180,13 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
             });
         }, 50 * 60 * 1000);
     }
+    let isCutting = false;
     let isEnded = false;
     const onEnd = (...args) => {
+        if (isCutting) {
+            isCutting = false;
+            return;
+        }
         if (isEnded)
             return;
         isEnded = true;
@@ -182,11 +202,14 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
         outputOptions: ffmpegOutputOptions,
         inputOptions: ffmpegInputOptions,
         segment: this.segment ?? 0,
-        getSavePath: (opts) => getSavePath({ owner, title, startTime: opts.startTime }),
+        getSavePath: (opts) => getSavePath({ owner, title: opts.title ?? title, startTime: opts.startTime }),
         isHls: streamOptions.protocol_name === "http_hls",
         disableDanma: this.disableProvideCommentsWhenRecording,
         videoFormat: this.videoFormat,
-    }, onEnd);
+    }, onEnd, async () => {
+        const info = await getInfo(this.channelId);
+        return info;
+    });
     const savePath = getSavePath({
         owner,
         title,
@@ -198,14 +221,20 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
         this.state = "idle";
         throw err;
     }
-    const handleVideoCreated = async ({ filename }) => {
-        this.emit("videoFileCreated", { filename });
+    const handleVideoCreated = async ({ filename, title, cover }) => {
+        this.emit("videoFileCreated", { filename, cover });
+        if (title && this?.liveInfo) {
+            this.liveInfo.title = title;
+        }
+        if (cover && this?.liveInfo) {
+            this.liveInfo.cover = cover;
+        }
         const extraDataController = recorder.getExtraDataController();
         extraDataController?.setMeta({
             room_id: String(roomId),
             platform: provider?.id,
             liveStartTimestamp: liveInfo.startTime?.getTime(),
-            recordStopTimestamp: Date.now(),
+            // recordStopTimestamp: Date.now(),
             title: title,
             user_name: owner,
         });
@@ -225,7 +254,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
     });
     let danmaClient = new DanmaClient(roomId, this.auth, this.uid);
     if (!this.disableProvideCommentsWhenRecording) {
-        danmaClient = danmaClient.on("Message", (msg) => {
+        danmaClient.on("Message", (msg) => {
             const extraDataController = recorder.getExtraDataController();
             if (!extraDataController)
                 return;
@@ -236,10 +265,37 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
             this.emit("Message", msg);
             extraDataController.addMessage(msg);
         });
+        danmaClient.on("onRoomInfoChange", (msg) => {
+            if (!isManualStart &&
+                this.titleKeywords &&
+                typeof this.titleKeywords === "string" &&
+                this.titleKeywords.trim()) {
+                const title = msg?.body?.title ?? "";
+                const hasTitleKeyword = hasKeyword(title, this.titleKeywords);
+                if (hasTitleKeyword) {
+                    this.emit("DebugLog", {
+                        type: "common",
+                        text: `检测到标题包含关键词，停止录制：直播间标题 "${title}" 包含关键词 "${this.titleKeywords}"`,
+                    });
+                    // 停止录制
+                    this.recordHandle && this.recordHandle.stop("直播间标题包含关键词");
+                }
+            }
+        });
         danmaClient.start();
     }
     const ffmpegArgs = recorder.getArguments();
     recorder.run();
+    const cut = utils.singleton(async () => {
+        if (!this.recordHandle)
+            return;
+        if (isCutting)
+            return;
+        isCutting = true;
+        await recorder.stop();
+        recorder.createCommand();
+        recorder.run();
+    });
     const stop = utils.singleton(async (reason) => {
         if (!this.recordHandle)
             return;
@@ -271,6 +327,7 @@ const checkLiveStatusAndRecord = async function ({ getSavePath, isManualStart, b
         ffmpegArgs,
         savePath: savePath,
         stop,
+        cut,
     };
     this.emit("RecordStart", this.recordHandle);
     return this.recordHandle;

package/lib/stream.d.ts

@@ -30,6 +30,7 @@ export declare function getStream(opts: Pick<Recorder, "channelId" | "quality">
     strictQuality?: boolean;
     formatName: RecorderCreateOpts["formatName"];
     codecName: RecorderCreateOpts["codecName"];
+    onlyAudio?: boolean;
 }): Promise<{
     currentStream: {
         name: string;

package/lib/stream.js

@@ -155,10 +155,17 @@ async function getLiveInfo(roomIdOrShortId, opts) {
     let streamInfo;
     let streamOptions;
     for (const condition of conditons) {
-        streamInfo = res.playurl_info.playurl.stream
+        const streamList = res.playurl_info.playurl.stream
             .find(({ protocol_name }) => protocol_name === condition.protocol_name)
             ?.format.find(({ format_name }) => format_name === condition.format_name)
-            ?.codec.find(({ codec_name }) => codec_name === condition.codec_name);
+            ?.codec.filter(({ codec_name }) => codec_name === condition.codec_name);
+        if (streamList && streamList.length > 1) {
+            // 由于录播姬直推hevc时，指定qn，服务端仍会返回其他画质的流，这里需要指定找一下流
+            streamInfo = streamList.find((item) => item.current_qn === opts.qn);
+        }
+        if (!streamInfo) {
+            streamInfo = streamList?.[0];
+        }
         if (streamInfo) {
             streamOptions = {
                 ...condition,
@@ -167,7 +174,11 @@ async function getLiveInfo(roomIdOrShortId, opts) {
             break;
         }
     }
-    console.log("streamOptions", streamOptions, res.playurl_info.playurl.stream);
+    // console.log(
+    //   "streamOptions",
+    //   streamOptions,
+    //   JSON.stringify(res.playurl_info.playurl.stream, null, 2),
+    // );
     assert(streamInfo, "没有找到支持的流");
     const streams = streamInfo.accept_qn.map((qn) => {
         const qnDesc = res.playurl_info.playurl.g_qn_desc.find((item) => item.qn === qn);
@@ -200,6 +211,7 @@ export async function getStream(opts) {
         cookie: opts.cookie,
         formatName: opts.formatName,
         codecName: opts.codecName,
+        onlyAudio: opts.onlyAudio,
     });
     // console.log(JSON.stringify(liveInfo, null, 2));
     if (liveInfo.current_qn !== qn && opts.strictQuality) {
@@ -213,6 +225,7 @@ export async function getStream(opts) {
             cookie: opts.cookie,
             formatName: opts.formatName,
             codecName: opts.codecName,
+            onlyAudio: opts.onlyAudio,
         });
     }
     let expectSource = liveInfo.sources[0];

package/lib/utils.d.ts

@@ -21,3 +21,4 @@ export declare function assertStringType(data: unknown, msg?: string): asserts d
 export declare function assertNumberType(data: unknown, msg?: string): asserts data is number;
 export declare function assertObjectType(data: unknown, msg?: string): asserts data is object;
 export declare function createInvalidStreamChecker(count?: number): (ffmpegLogLine: string) => boolean;
+export declare function hasKeyword(title: string, titleKeywords: string): boolean;

package/lib/utils.js

@@ -81,3 +81,11 @@ export function createInvalidStreamChecker(count = 10) {
         return false;
     };
 }
+export function hasKeyword(title, titleKeywords) {
+    const keywords = titleKeywords
+        .split(",")
+        .map((k) => k.trim())
+        .filter((k) => k);
+    const hasTitleKeyword = keywords.some((keyword) => title.toLowerCase().includes(keyword.toLowerCase()));
+    return hasTitleKeyword;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bililive-tools/bilibili-recorder",
-  "version": "1.2.0",
+  "version": "1.4.1",
   "description": "bililive-tools bilibili recorder implemention",
   "main": "./lib/index.js",
   "type": "module",
@@ -36,10 +36,10 @@
   "dependencies": {
     "blive-message-listener": "^0.5.0",
     "mitt": "^3.0.1",
-    "tiny-bilibili-ws": "^1.0.1",
+    "tiny-bilibili-ws": "^1.0.2",
     "lodash-es": "^4.17.21",
     "axios": "^1.7.8",
-    "@bililive-tools/manager": "^1.2.0"
+    "@bililive-tools/manager": "^1.3.0"
   },
   "scripts": {
     "build": "tsc",