@bililive-tools/manager 1.5.0 → 1.6.1

package/lib/xml_stream_controller.d.ts

@@ -0,0 +1,23 @@
+import { Message } from "./common.js";
+export interface XmlStreamData {
+    meta: {
+        title?: string;
+        recordStartTimestamp: number;
+        recordStopTimestamp?: number;
+        liveStartTimestamp?: number;
+        ffmpegArgs?: string[];
+        platform?: string;
+        user_name?: string;
+        room_id?: string;
+    };
+    /** 缓存的消息，待写入到文件 */
+    pendingMessages: Message[];
+}
+export interface XmlStreamController {
+    /** 设计上来说，外部程序不应该能直接修改 data 上的东西 */
+    readonly data: XmlStreamData;
+    addMessage: (message: Message) => void;
+    setMeta: (meta: Partial<XmlStreamData["meta"]>) => void;
+    flush: () => Promise<void>;
+}
+export declare function createRecordExtraDataController(savePath: string): XmlStreamController;

package/lib/xml_stream_controller.js

@@ -0,0 +1,240 @@
+/**
+ * XML流式写入控制器，用于实时写入弹幕、礼物等信息到XML文件
+ * 相比原有的json方案，这个实现每隔5秒就会写入数据，减少内存占用和数据丢失风险
+ */
+import fs from "node:fs";
+import { XMLBuilder } from "fast-xml-parser";
+import { pick } from "lodash-es";
+import { asyncThrottle } from "./utils.js";
+export function createRecordExtraDataController(savePath) {
+    const data = {
+        meta: {
+            recordStartTimestamp: Date.now(),
+        },
+        pendingMessages: [],
+    };
+    let hasCompleted = false;
+    let isWriting = false;
+    let isInitialized = false;
+    // 初始化文件
+    const initializeFile = async () => {
+        if (isInitialized)
+            return;
+        isInitialized = true;
+        try {
+            // 创建XML文件头，使用占位符预留metadata位置
+            const header = `<?xml version="1.0" encoding="utf-8"?>\n<i>\n<!--METADATA_PLACEHOLDER-->\n`;
+            await fs.promises.writeFile(savePath, header);
+        }
+        catch (error) {
+            console.error("初始化XML文件失败:", error);
+            isInitialized = false;
+            throw error;
+        }
+    };
+    // 每10秒写入一次数据
+    const scheduleWrite = asyncThrottle(() => writeToFile(), 10e3, {
+        immediateRunWhenEndOfDefer: true,
+    });
+    const writeToFile = async () => {
+        if (isWriting || hasCompleted || data.pendingMessages.length === 0) {
+            return;
+        }
+        // 确保文件已初始化
+        await initializeFile();
+        isWriting = true;
+        try {
+            // 获取待写入的消息
+            const messagesToWrite = [...data.pendingMessages];
+            data.pendingMessages = [];
+            // 生成XML内容
+            const xmlContent = generateXmlContent(data.meta, messagesToWrite);
+            // 追加写入文件
+            await appendToXmlFile(savePath, xmlContent);
+        }
+        catch (error) {
+            console.error("写入XML文件失败:", error);
+            // 如果写入失败，将消息重新加入队列
+            data.pendingMessages = [...data.pendingMessages];
+        }
+        finally {
+            isWriting = false;
+        }
+    };
+    const addMessage = (message) => {
+        if (hasCompleted)
+            return;
+        // if (!isInitialized) return;
+        data.pendingMessages.push(message);
+        // 确保文件已初始化
+        initializeFile().catch(console.error);
+        scheduleWrite();
+    };
+    const setMeta = (meta) => {
+        if (hasCompleted)
+            return;
+        data.meta = {
+            ...data.meta,
+            ...meta,
+        };
+    };
+    const flush = async () => {
+        if (hasCompleted)
+            return;
+        hasCompleted = true;
+        scheduleWrite.cancel();
+        await initializeFile().catch(console.error);
+        // 写入剩余的数据
+        if (data.pendingMessages.length > 0) {
+            await writeToFile();
+        }
+        // 完成XML文件（添加结束标签等）
+        await finalizeXmlFile(savePath, data.meta);
+        // 清理内存
+        data.pendingMessages = [];
+    };
+    return {
+        data,
+        addMessage,
+        setMeta,
+        flush,
+    };
+}
+/**
+ * 生成XML内容片段
+ */
+function generateXmlContent(metadata, messages) {
+    const builder = new XMLBuilder({
+        ignoreAttributes: false,
+        attributeNamePrefix: "@@",
+        format: true,
+    });
+    const comments = messages
+        .filter((item) => item.type === "comment")
+        .map((ele) => {
+        const progress = Math.max((ele.timestamp - metadata.recordStartTimestamp) / 1000, 0);
+        const data = {
+            "@@p": "",
+            "@@progress": progress,
+            "@@mode": String(ele.mode ?? 1),
+            "@@fontsize": String(25),
+            "@@color": String(parseInt((ele.color || "#ffffff").replace("#", ""), 16)),
+            "@@midHash": String(ele?.sender?.uid),
+            "#text": String(ele?.text || ""),
+            "@@ctime": String(ele.timestamp),
+            "@@pool": String(0),
+            "@@weight": String(0),
+            "@@user": String(ele.sender?.name),
+            "@@uid": String(ele?.sender?.uid),
+            "@@timestamp": String(ele.timestamp),
+        };
+        data["@@p"] = [
+            data["@@progress"],
+            data["@@mode"],
+            data["@@fontsize"],
+            data["@@color"],
+            data["@@ctime"],
+            data["@@pool"],
+            data["@@midHash"],
+            data["@@uid"],
+            data["@@weight"],
+        ].join(",");
+        return pick(data, ["@@p", "#text", "@@user", "@@uid", "@@timestamp"]);
+    });
+    const gifts = messages
+        .filter((item) => item.type === "give_gift")
+        .map((ele) => {
+        const progress = Math.max((ele.timestamp - metadata.recordStartTimestamp) / 1000, 0);
+        return {
+            "@@ts": progress,
+            "@@giftname": String(ele.name),
+            "@@giftcount": String(ele.count),
+            "@@price": String(ele.price * 1000),
+            "@@user": String(ele.sender?.name),
+            "@@uid": String(ele?.sender?.uid),
+            "@@timestamp": String(ele.timestamp),
+        };
+    });
+    const superChats = messages
+        .filter((item) => item.type === "super_chat")
+        .map((ele) => {
+        const progress = Math.max((ele.timestamp - metadata.recordStartTimestamp) / 1000, 0);
+        return {
+            "@@ts": progress,
+            "@@price": String(ele.price * 1000),
+            "#text": String(ele.text),
+            "@@user": String(ele.sender?.name),
+            "@@uid": String(ele?.sender?.uid),
+            "@@timestamp": String(ele.timestamp),
+        };
+    });
+    const guardGift = messages
+        .filter((item) => item.type === "guard")
+        .map((ele) => {
+        const progress = Math.max((ele.timestamp - metadata.recordStartTimestamp) / 1000, 0);
+        return {
+            "@@ts": progress,
+            "@@price": String(ele.price * 1000),
+            "@@giftname": String(ele.name),
+            "@@giftcount": String(ele.count),
+            "@@level": String(ele.level),
+            "@@user": String(ele.sender?.name),
+            "@@uid": String(ele?.sender?.uid),
+            "@@timestamp": String(ele.timestamp),
+        };
+    });
+    // 构建这一批消息的XML片段
+    const fragment = {
+        d: comments,
+        gift: gifts,
+        sc: superChats,
+        guard: guardGift,
+    };
+    return builder.build(fragment);
+}
+/**
+ * 追加内容到XML文件
+ */
+async function appendToXmlFile(filePath, content) {
+    try {
+        // 直接追加内容
+        await fs.promises.appendFile(filePath, content);
+    }
+    catch (error) {
+        console.error(`写入XML文件失败: ${filePath}`, error);
+        throw error;
+    }
+}
+/**
+ * 完成XML文件写入
+ */
+async function finalizeXmlFile(filePath, metadata) {
+    try {
+        const builder = new XMLBuilder({
+            ignoreAttributes: false,
+            attributeNamePrefix: "@@",
+            format: true,
+        });
+        // 生成metadata XML
+        const metadataXml = builder.build({
+            metadata: {
+                platform: metadata.platform,
+                video_start_time: metadata.recordStartTimestamp,
+                live_start_time: metadata.liveStartTimestamp,
+                room_title: metadata.title,
+                user_name: metadata.user_name,
+                room_id: metadata.room_id,
+            },
+        });
+        // 读取文件内容
+        const content = await fs.promises.readFile(filePath, "utf-8");
+        // 替换占位符为实际的metadata，并添加结束标签
+        const finalContent = content.replace("<!--METADATA_PLACEHOLDER-->", metadataXml) + "</i>";
+        // 写回文件
+        await fs.promises.writeFile(filePath, finalContent);
+    }
+    catch (error) {
+        console.error(`完成XML文件写入失败: ${filePath}`, error);
+        throw error;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bililive-tools/manager",
-  "version": "1.5.0",
+  "version": "1.6.1",
   "description": "Batch scheduling recorders",
   "main": "./lib/index.js",
   "type": "module",
@@ -32,9 +32,9 @@
   "author": "renmu123",
   "license": "LGPL",
   "dependencies": {
-    "@renmu/fluent-ffmpeg": "2.3.2",
+    "@renmu/fluent-ffmpeg": "2.3.3",
     "fast-xml-parser": "^4.5.0",
-    "filenamify": "^6.0.0",
+    "filenamify": "^7.0.0",
     "mitt": "^3.0.1",
     "string-argv": "^0.3.2",
     "lodash-es": "^4.17.21",

package/lib/FFMPEGRecorder.d.ts

@@ -1,49 +0,0 @@
-import EventEmitter from "node:events";
-export declare class FFMPEGRecorder extends EventEmitter {
-    private onEnd;
-    private onUpdateLiveInfo;
-    private command;
-    private streamManager;
-    private timeoutChecker;
-    hasSegment: boolean;
-    getSavePath: (data: {
-        startTime: number;
-        title?: string;
-    }) => string;
-    segment: number;
-    ffmpegOutputOptions: string[];
-    inputOptions: string[];
-    isHls: boolean;
-    disableDanma: boolean;
-    url: string;
-    headers: {
-        [key: string]: string | undefined;
-    } | undefined;
-    constructor(opts: {
-        url: string;
-        getSavePath: (data: {
-            startTime: number;
-            title?: string;
-        }) => string;
-        segment: number;
-        outputOptions: string[];
-        inputOptions?: string[];
-        isHls?: boolean;
-        disableDanma?: boolean;
-        videoFormat?: "auto" | "ts" | "mkv";
-        headers?: {
-            [key: string]: string | undefined;
-        };
-    }, onEnd: (...args: unknown[]) => void, onUpdateLiveInfo: () => Promise<{
-        title?: string;
-        cover?: string;
-    }>);
-    createCommand(): import("@renmu/fluent-ffmpeg").FfmpegCommand;
-    formatLine(line: string): {
-        time: string | null;
-    } | null;
-    run(): void;
-    getArguments(): string[];
-    stop(): Promise<void>;
-    getExtraDataController(): import("./record_extra_data_controller.js").RecordExtraDataController | null;
-}