@bililive-tools/manager 1.0.0

package/lib/record_extra_data_controller.d.ts

@@ -0,0 +1,28 @@
+import { Message } from "./common.js";
+export interface RecordExtraData {
+    meta: {
+        title?: string;
+        recordStartTimestamp: number;
+        recordStopTimestamp?: number;
+        liveStartTimestamp?: number;
+        ffmpegArgs?: string[];
+        platform?: string;
+        user_name?: string;
+        room_id?: string;
+    };
+    /** 这个数组预期上是一个根据 timestamp 排序的有序数组，方便做一些时间段查询 */
+    messages: Message[];
+}
+export interface RecordExtraDataController {
+    /** 设计上来说，外部程序不应该能直接修改 data 上的东西 */
+    readonly data: RecordExtraData;
+    addMessage: (message: Message) => void;
+    setMeta: (meta: Partial<RecordExtraData["meta"]>) => void;
+    flush: () => Promise<void>;
+}
+export declare function createRecordExtraDataController(savePath: string): RecordExtraDataController;
+/**
+ * 转换弹幕为b站格式xml
+ * @link: https://socialsisteryi.github.io/bilibili-API-collect/docs/danmaku/danmaku_xml.html#%E5%B1%9E%E6%80%A7-p
+ */
+export declare function convert2Xml(data: RecordExtraData): string;

package/lib/record_extra_data_controller.js

@@ -0,0 +1,157 @@
+/**
+ * 用一个视频之外的独立文件来存储元信息和弹幕、礼物等信息。
+ * 暂时使用 json 作为存储方案试试效果，猜测可能会有实时写入差、占用空间大的问题。
+ */
+import path from "node:path";
+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(),
+        },
+        messages: [],
+    };
+    const scheduleSave = asyncThrottle(() => save(), 30e3, {
+        immediateRunWhenEndOfDefer: true,
+    });
+    const save = () => {
+        return fs.promises.writeFile(savePath, JSON.stringify(data));
+    };
+    // TODO: 将所有数据存放在内存中可能存在问题
+    const addMessage = (comment) => {
+        data.messages.push(comment);
+        scheduleSave();
+    };
+    const setMeta = (meta) => {
+        data.meta = {
+            ...data.meta,
+            ...meta,
+        };
+        scheduleSave();
+    };
+    const flush = async () => {
+        scheduleSave.flush();
+        scheduleSave.cancel();
+        const xmlContent = convert2Xml(data);
+        const parsedPath = path.parse(savePath);
+        const xmlPath = path.join(parsedPath.dir, parsedPath.name + ".xml");
+        await fs.promises.writeFile(xmlPath, xmlContent);
+        await fs.promises.rm(savePath);
+    };
+    return {
+        data,
+        addMessage,
+        setMeta,
+        flush,
+    };
+}
+/**
+ * 转换弹幕为b站格式xml
+ * @link: https://socialsisteryi.github.io/bilibili-API-collect/docs/danmaku/danmaku_xml.html#%E5%B1%9E%E6%80%A7-p
+ */
+export function convert2Xml(data) {
+    const metadata = data.meta;
+    const builder = new XMLBuilder({
+        ignoreAttributes: false,
+        attributeNamePrefix: "@@",
+        format: true,
+    });
+    const comments = data.messages
+        .filter((item) => item.type === "comment")
+        .map((ele) => {
+        const progress = (ele.timestamp - metadata?.recordStartTimestamp) / 1000;
+        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),
+        };
+        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"]);
+    });
+    const gifts = data.messages
+        .filter((item) => item.type === "give_gift")
+        .map((ele) => {
+        const progress = (ele.timestamp - metadata?.recordStartTimestamp) / 1000;
+        const data = {
+            "@@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),
+            // "@@raw": JSON.stringify(ele),
+        };
+        return data;
+    });
+    const superChats = data.messages
+        .filter((item) => item.type === "super_chat")
+        .map((ele) => {
+        const progress = (ele.timestamp - metadata?.recordStartTimestamp) / 1000;
+        const data = {
+            "@@ts": progress,
+            "@@price": String(ele.price * 1000),
+            "#text": String(ele.text),
+            "@@user": String(ele.sender?.name),
+            "@@uid": String(ele?.sender?.uid),
+            // "@@raw": JSON.stringify(ele),
+        };
+        return data;
+    });
+    const guardGift = data.messages
+        .filter((item) => item.type === "guard")
+        .map((ele) => {
+        const progress = (ele.timestamp - metadata?.recordStartTimestamp) / 1000;
+        const data = {
+            "@@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),
+            // "@@raw": JSON.stringify(ele),
+        };
+        return data;
+    });
+    const xmlContent = builder.build({
+        i: {
+            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,
+            },
+            d: comments,
+            gift: gifts,
+            sc: superChats,
+            guard: guardGift,
+        },
+    });
+    return `<?xml version="1.0" encoding="utf-8"?>
+${xmlContent}`;
+}
+//# sourceMappingURL=record_extra_data_controller.js.map

package/lib/record_extra_data_controller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"record_extra_data_controller.js","sourceRoot":"","sources":["../src/record_extra_data_controller.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAGjC,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAyB3C,MAAM,UAAU,+BAA+B,CAAC,QAAgB;IAC9D,MAAM,IAAI,GAAoB;QAC5B,IAAI,EAAE;YACJ,oBAAoB,EAAE,IAAI,CAAC,GAAG,EAAE;SACjC;QACD,QAAQ,EAAE,EAAE;KACb,CAAC;IAEF,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE;QACrD,0BAA0B,EAAE,IAAI;KACjC,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,GAAG,EAAE;QAChB,OAAO,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC;IAEF,0BAA0B;IAC1B,MAAM,UAAU,GAA4C,CAAC,OAAO,EAAE,EAAE;QACtE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,YAAY,EAAE,CAAC;IACjB,CAAC,CAAC;IAEF,MAAM,OAAO,GAAyC,CAAC,IAAI,EAAE,EAAE;QAC7D,IAAI,CAAC,IAAI,GAAG;YACV,GAAG,IAAI,CAAC,IAAI;YACZ,GAAG,IAAI;SACR,CAAC;QACF,YAAY,EAAE,CAAC;IACjB,CAAC,CAAC;IAEF,MAAM,KAAK,GAAuC,KAAK,IAAI,EAAE;QAC3D,YAAY,CAAC,KAAK,EAAE,CAAC;QACrB,YAAY,CAAC,MAAM,EAAE,CAAC;QAEtB,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC;QACpE,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC,CAAC;IAEF,OAAO;QACL,IAAI;QACJ,UAAU;QACV,OAAO;QACP,KAAK;KACN,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,WAAW,CAAC,IAAqB;IAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IAE3B,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC;QAC7B,gBAAgB,EAAE,KAAK;QACvB,mBAAmB,EAAE,IAAI;QACzB,MAAM,EAAE,IAAI;KACb,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ;SAC3B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC;SACzC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,SAAS,GAAG,QAAQ,EAAE,oBAAoB,CAAC,GAAG,IAAI,CAAC;QACzE,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,QAAQ;YACtB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC;YAC/B,YAAY,EAAE,MAAM,CAAC,EAAE,CAAC;YACxB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,KAAK,IAAI,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1E,WAAW,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;YACrC,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC;YAChC,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC;YAChC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;YACnB,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;YACrB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC;YAClC,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;SAClC,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG;YACZ,IAAI,CAAC,YAAY,CAAC;YAClB,IAAI,CAAC,QAAQ,CAAC;YACd,IAAI,CAAC,YAAY,CAAC;YAClB,IAAI,CAAC,SAAS,CAAC;YACf,IAAI,CAAC,SAAS,CAAC;YACf,IAAI,CAAC,QAAQ,CAAC;YACd,IAAI,CAAC,WAAW,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC;YACb,IAAI,CAAC,UAAU,CAAC;SACjB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEL,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ;SACxB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC;SAC3C,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,SAAS,GAAG,QAAQ,EAAE,oBAAoB,CAAC,GAAG,IAAI,CAAC;QACzE,MAAM,IAAI,GAAG;YACX,MAAM,EAAE,QAAQ;YAChB,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;YAC9B,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAChC,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC;YACnC,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC;YAClC,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;YACjC,gCAAgC;SACjC,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEL,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ;SAC7B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC;SAC5C,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,SAAS,GAAG,QAAQ,EAAE,oBAAoB,CAAC,GAAG,IAAI,CAAC;QACzE,MAAM,IAAI,GAAG;YACX,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC;YACnC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;YACzB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC;YAClC,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;YACjC,gCAAgC;SACjC,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEL,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ;SAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC;SACvC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,SAAS,GAAG,QAAQ,EAAE,oBAAoB,CAAC,GAAG,IAAI,CAAC;QACzE,MAAM,IAAI,GAAG;YACX,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC;YACnC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;YAC9B,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAChC,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAC5B,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC;YAClC,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC;YACjC,gCAAgC;SACjC,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACL,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC;QAC/B,CAAC,EAAE;YACD,QAAQ,EAAE;gBACR,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,gBAAgB,EAAE,QAAQ,CAAC,oBAAoB;gBAC/C,eAAe,EAAE,QAAQ,CAAC,kBAAkB;gBAC5C,UAAU,EAAE,QAAQ,CAAC,KAAK;gBAC1B,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B;YACD,CAAC,EAAE,QAAQ;YACX,IAAI,EAAE,KAAK;YACX,EAAE,EAAE,UAAU;YACd,KAAK,EAAE,SAAS;SACjB;KACF,CAAC,CAAC;IACH,OAAO;EACP,UAAU,EAAE,CAAC;AACf,CAAC"}

package/lib/recorder.d.ts

@@ -0,0 +1,118 @@
+import { Emitter } from "mitt";
+import { ChannelId, Message, Quality } from "./common.js";
+import { RecorderProvider } from "./manager.js";
+import { AnyObject, PickRequired, UnknownObject } from "./utils.js";
+type FormatName = "auto" | "flv" | "hls" | "fmp4" | "flv_only" | "hls_only" | "fmp4_only";
+type CodecName = "auto" | "avc" | "hevc" | "avc_only" | "hevc_only";
+export interface RecorderCreateOpts<E extends AnyObject = UnknownObject> {
+    providerId: RecorderProvider<E>["id"];
+    channelId: ChannelId;
+    id?: string;
+    remarks?: string;
+    disableAutoCheck?: boolean;
+    disableProvideCommentsWhenRecording?: boolean;
+    quality: Quality;
+    streamPriorities: string[];
+    sourcePriorities: string[];
+    segment?: number;
+    saveGiftDanma?: boolean;
+    saveSCDanma?: boolean;
+    /** 保存封面 */
+    saveCover?: boolean;
+    /** 身份验证 */
+    auth?: string;
+    /** cookie所有者uid,B站弹幕录制 */
+    uid?: number;
+    /** 画质匹配重试次数 */
+    qualityRetry?: number;
+    /** B站是否使用m3u8代理 */
+    useM3U8Proxy?: boolean;
+    /**B站m3u8代理url */
+    m3u8ProxyUrl?: string;
+    /** 流格式 */
+    formatName?: FormatName;
+    /** 流编码 */
+    codecName?: CodecName;
+    extra?: Partial<E>;
+}
+export type SerializedRecorder<E extends AnyObject> = PickRequired<RecorderCreateOpts<E>, "id">;
+export type RecorderState = "idle" | "recording" | "stopping-record";
+export interface RecordHandle {
+    id: string;
+    stream: string;
+    source: string;
+    url: string;
+    ffmpegArgs?: string[];
+    savePath: string;
+    stop: (this: RecordHandle, reason?: string, tempStopIntervalCheck?: boolean) => Promise<void>;
+}
+export interface DebugLog {
+    type: string | "common" | "ffmpeg";
+    text: string;
+}
+export type GetSavePath = (data: {
+    owner: string;
+    title: string;
+    startTime?: number;
+}) => string;
+export interface Recorder<E extends AnyObject = UnknownObject> extends Emitter<{
+    RecordStart: RecordHandle;
+    RecordSegment?: RecordHandle;
+    videoFileCreated: {
+        filename: string;
+    };
+    videoFileCompleted: {
+        filename: string;
+    };
+    RecordStop: {
+        recordHandle: RecordHandle;
+        reason?: string;
+    };
+    Updated: (string | keyof Recorder)[];
+    Message: Message;
+    DebugLog: DebugLog;
+}>, RecorderCreateOpts<E> {
+    id: string;
+    extra: Partial<E>;
+    availableStreams: string[];
+    availableSources: string[];
+    usedStream?: string;
+    usedSource?: string;
+    state: RecorderState;
+    qualityMaxRetry: number;
+    qualityRetry: number;
+    uid?: number;
+    liveInfo?: {
+        living: boolean;
+        owner: string;
+        title: string;
+        startTime?: Date;
+        avatar: string;
+        cover: string;
+        liveId?: string;
+    };
+    tempStopIntervalCheck?: boolean;
+    getChannelURL: (this: Recorder<E>) => string;
+    checkLiveStatusAndRecord: (this: Recorder<E>, opts: {
+        getSavePath: GetSavePath;
+        qualityRetry?: number;
+        banLiveId?: string;
+    }) => Promise<RecordHandle | null>;
+    recordHandle?: RecordHandle;
+    toJSON: (this: Recorder<E>) => SerializedRecorder<E>;
+    getLiveInfo: (this: Recorder<E>) => Promise<{
+        owner: string;
+        title: string;
+        avatar: string;
+        cover: string;
+        channelId: ChannelId;
+        living: boolean;
+        startTime: Date;
+    }>;
+    getStream: (this: Recorder<E>) => Promise<{
+        source: string;
+        name: string;
+        url: string;
+    }>;
+}
+export {};

package/lib/recorder.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=recorder.js.map

package/lib/recorder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recorder.js","sourceRoot":"","sources":["../src/recorder.ts"],"names":[],"mappings":""}

package/lib/streamManager.d.ts

@@ -0,0 +1,30 @@
+import { createRecordExtraDataController } from "./record_extra_data_controller.js";
+import type { Recorder, GetSavePath } from "./recorder.js";
+export declare class Segment {
+    extraDataController: ReturnType<typeof createRecordExtraDataController> | null;
+    init: boolean;
+    getSavePath: GetSavePath;
+    owner: string;
+    title: string;
+    recorder: Recorder;
+    /** 原始的ffmpeg文件名，用于重命名 */
+    rawRecordingVideoPath: string;
+    /** 输出文件名名，不包含拓展名 */
+    outputVideoFilePath: string;
+    constructor(recorder: Recorder, getSavePath: GetSavePath, owner: string, title: string);
+    handleSegmentEnd(): Promise<void>;
+    onSegmentStart(stderrLine: string): Promise<void>;
+}
+export declare class StreamManager {
+    private segmentManager;
+    private extraDataController;
+    recorder: Recorder;
+    owner: string;
+    title: string;
+    recordSavePath: string;
+    constructor(recorder: Recorder, getSavePath: GetSavePath, owner: string, title: string, recordSavePath: string, hasSegment: boolean);
+    handleVideoStarted(stderrLine?: string): Promise<void>;
+    handleVideoCompleted(): Promise<void>;
+    getExtraDataController(): import("./record_extra_data_controller.js").RecordExtraDataController | null;
+    get videoFilePath(): string;
+}

package/lib/streamManager.js

@@ -0,0 +1,119 @@
+import fs from "fs-extra";
+import { createRecordExtraDataController } from "./record_extra_data_controller.js";
+import { replaceExtName, ensureFolderExist } from "./utils.js";
+export class Segment {
+    extraDataController = null;
+    init = true;
+    getSavePath;
+    owner;
+    title;
+    recorder;
+    /** 原始的ffmpeg文件名，用于重命名 */
+    rawRecordingVideoPath;
+    /** 输出文件名名，不包含拓展名 */
+    outputVideoFilePath;
+    constructor(recorder, getSavePath, owner, title) {
+        this.getSavePath = getSavePath;
+        this.owner = owner;
+        this.title = title;
+        this.recorder = recorder;
+    }
+    async handleSegmentEnd() {
+        if (!this.outputVideoFilePath) {
+            this.recorder.emit("DebugLog", {
+                type: "common",
+                text: "Should call onSegmentStart first",
+            });
+            return;
+        }
+        this.extraDataController?.setMeta({ recordStopTimestamp: Date.now() });
+        try {
+            await Promise.all([
+                fs.rename(this.rawRecordingVideoPath, `${this.outputVideoFilePath}.ts`),
+                this.extraDataController?.flush(),
+            ]);
+            this.recorder.emit("videoFileCompleted", { filename: `${this.outputVideoFilePath}.ts` });
+        }
+        catch (err) {
+            this.recorder.emit("DebugLog", {
+                type: "common",
+                text: "videoFileCompleted error " + String(err),
+            });
+        }
+    }
+    async onSegmentStart(stderrLine) {
+        if (!this.init) {
+            await this.handleSegmentEnd();
+        }
+        this.init = false;
+        const startTime = Date.now();
+        this.outputVideoFilePath = this.getSavePath({
+            owner: this.owner,
+            title: this.title,
+            startTime: startTime,
+        });
+        ensureFolderExist(this.outputVideoFilePath);
+        this.extraDataController = createRecordExtraDataController(`${this.outputVideoFilePath}.json`);
+        this.extraDataController.setMeta({ title: this.title, user_name: this.owner });
+        const regex = /'([^']+)'/;
+        const match = stderrLine.match(regex);
+        if (match) {
+            const filename = match[1];
+            this.rawRecordingVideoPath = filename;
+            this.recorder.emit("videoFileCreated", { filename: `${this.outputVideoFilePath}.ts` });
+        }
+        else {
+            this.recorder.emit("DebugLog", { type: "ffmpeg", text: "No match found" });
+        }
+    }
+}
+export class StreamManager {
+    segmentManager = null;
+    extraDataController = null;
+    recorder;
+    owner;
+    title;
+    recordSavePath;
+    constructor(recorder, getSavePath, owner, title, recordSavePath, hasSegment) {
+        this.recordSavePath = recordSavePath;
+        this.recorder = recorder;
+        this.owner = owner;
+        this.title = title;
+        if (hasSegment) {
+            this.segmentManager = new Segment(recorder, getSavePath, owner, title);
+        }
+        else {
+            const extraDataSavePath = replaceExtName(recordSavePath, ".json");
+            this.extraDataController = createRecordExtraDataController(extraDataSavePath);
+            this.extraDataController.setMeta({ title, user_name: owner });
+            // TODO: 增加platform参数，直播开始时间
+        }
+    }
+    async handleVideoStarted(stderrLine) {
+        if (this.segmentManager) {
+            if (stderrLine) {
+                await this.segmentManager.onSegmentStart(stderrLine);
+            }
+        }
+        else {
+            this.recorder.emit("videoFileCreated", { filename: this.videoFilePath });
+        }
+    }
+    async handleVideoCompleted() {
+        if (this.segmentManager) {
+            await this.segmentManager.handleSegmentEnd();
+        }
+        else {
+            this.getExtraDataController()?.setMeta({ recordStopTimestamp: Date.now() });
+            await this.getExtraDataController()?.flush();
+            this.recorder.emit("videoFileCompleted", { filename: this.videoFilePath });
+        }
+    }
+    getExtraDataController() {
+        return this.segmentManager?.extraDataController || this.extraDataController;
+    }
+    get videoFilePath() {
+        return this.segmentManager ? `${this.recordSavePath}-PART%03d.ts` : `${this.recordSavePath}.ts`;
+    }
+}
+//# sourceMappingURL=streamManager.js.map

package/lib/streamManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamManager.js","sourceRoot":"","sources":["../src/streamManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,+BAA+B,EAAE,MAAM,mCAAmC,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAI/D,MAAM,OAAO,OAAO;IAClB,mBAAmB,GAA8D,IAAI,CAAC;IACtF,IAAI,GAAG,IAAI,CAAC;IACZ,WAAW,CAAc;IACzB,KAAK,CAAS;IACd,KAAK,CAAS;IACd,QAAQ,CAAW;IACnB,yBAAyB;IACzB,qBAAqB,CAAU;IAC/B,oBAAoB;IACpB,mBAAmB,CAAU;IAE7B,YAAY,QAAkB,EAAE,WAAwB,EAAE,KAAa,EAAE,KAAa;QACpF,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE;gBAC7B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,kCAAkC;aACzC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QACD,IAAI,CAAC,mBAAmB,EAAE,OAAO,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAEvE,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,mBAAmB,KAAK,CAAC;gBACvE,IAAI,CAAC,mBAAmB,EAAE,KAAK,EAAE;aAClC,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC;QAC3F,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE;gBAC7B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,2BAA2B,GAAG,MAAM,CAAC,GAAG,CAAC;aAChD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,UAAkB;QACrC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC;QAClB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,WAAW,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,SAAS,EAAE,SAAS;SACrB,CAAC,CAAC;QAEH,iBAAiB,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAE5C,IAAI,CAAC,mBAAmB,GAAG,+BAA+B,CAAC,GAAG,IAAI,CAAC,mBAAmB,OAAO,CAAC,CAAC;QAC/F,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QAE/E,MAAM,KAAK,GAAG,WAAW,CAAC;QAC1B,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;YACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC;QACzF,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IAChB,cAAc,GAAmB,IAAI,CAAC;IACtC,mBAAmB,GAA8D,IAAI,CAAC;IAC9F,QAAQ,CAAW;IACnB,KAAK,CAAS;IACd,KAAK,CAAS;IACd,cAAc,CAAS;IAEvB,YACE,QAAkB,EAClB,WAAwB,EACxB,KAAa,EACb,KAAa,EACb,cAAsB,EACtB,UAAmB;QAEnB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QAEnB,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,cAAc,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACzE,CAAC;aAAM,CAAC;YACN,MAAM,iBAAiB,GAAG,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;YAClE,IAAI,CAAC,mBAAmB,GAAG,+BAA+B,CAAC,iBAAiB,CAAC,CAAC;YAC9E,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9D,4BAA4B;QAC9B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAmB;QAC1C,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,oBAAoB;QACxB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,sBAAsB,EAAE,EAAE,OAAO,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC5E,MAAM,IAAI,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,CAAC;YAC7C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAED,sBAAsB;QACpB,OAAO,IAAI,CAAC,cAAc,EAAE,mBAAmB,IAAI,IAAI,CAAC,mBAAmB,CAAC;IAC9E,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,cAAc,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,KAAK,CAAC;IAClG,CAAC;CACF"}

package/lib/utils.d.ts

@@ -0,0 +1,62 @@
+import { DebouncedFunc } from "lodash-es";
+export type AnyObject = Record<string, any>;
+export type UnknownObject = Record<string, unknown>;
+export type PickRequired<T, K extends keyof T> = T & Pick<Required<T>, K>;
+export declare function asyncThrottle(fn: () => Promise<void>, time: number, opts?: {
+    immediateRunWhenEndOfDefer?: boolean;
+}): DebouncedFunc<() => void>;
+export declare function replaceExtName(filePath: string, newExtName: string): string;
+/**
+ * 接收 fn ，返回一个和 fn 签名一致的函数 fn'。当已经有一个 fn' 在运行时，再调用
+ * fn' 会直接返回运行中 fn' 的 Promise，直到 Promise 结束 pending 状态
+ */
+export declare function singleton<Fn extends (...args: any) => Promise<any>>(fn: Fn): Fn;
+/**
+ * 从数组中按照特定算法提取一些值（允许同个索引重复提取）。
+ * 算法的行为类似 flex 的 space-between。
+ *
+ * examples:
+ * ```
+ * console.log(getValuesFromArrayLikeFlexSpaceBetween([1, 2, 3, 4, 5, 6, 7], 1))
+ * // [1]
+ * console.log(getValuesFromArrayLikeFlexSpaceBetween([1, 2, 3, 4, 5, 6, 7], 3))
+ * // [1, 4, 7]
+ * console.log(getValuesFromArrayLikeFlexSpaceBetween([1, 2, 3, 4, 5, 6, 7], 4))
+ * // [1, 3, 5, 7]
+ * console.log(getValuesFromArrayLikeFlexSpaceBetween([1, 2, 3, 4, 5, 6, 7], 11))
+ * // [1, 1, 2, 3, 3, 4, 5, 5, 6, 7, 7]
+ * ```
+ */
+export declare function getValuesFromArrayLikeFlexSpaceBetween<T>(array: T[], columnCount: number): T[];
+export declare function ensureFolderExist(fileOrFolderPath: string): void;
+export declare function assert(assertion: unknown, msg?: string): asserts assertion;
+export declare function assertStringType(data: unknown, msg?: string): asserts data is string;
+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 formatDate(date: Date, format: string): string;
+export declare function removeSystemReservedChars(filename: string): string;
+export declare function isFfmpegStartSegment(line: string): boolean;
+export declare const formatTemplate: (string: string, ...args: any[]) => string;
+export declare function createInvalidStreamChecker(): (ffmpegLogLine: string) => boolean;
+export declare function createTimeoutChecker(onTimeout: () => void, time: number): {
+    update: () => void;
+    stop: () => void;
+};
+declare function downloadImage(imageUrl: string, savePath: string): Promise<void>;
+declare const _default: {
+    replaceExtName: typeof replaceExtName;
+    singleton: typeof singleton;
+    getValuesFromArrayLikeFlexSpaceBetween: typeof getValuesFromArrayLikeFlexSpaceBetween;
+    ensureFolderExist: typeof ensureFolderExist;
+    assert: typeof assert;
+    assertStringType: typeof assertStringType;
+    assertNumberType: typeof assertNumberType;
+    assertObjectType: typeof assertObjectType;
+    asyncThrottle: typeof asyncThrottle;
+    isFfmpegStartSegment: typeof isFfmpegStartSegment;
+    createInvalidStreamChecker: typeof createInvalidStreamChecker;
+    createTimeoutChecker: typeof createTimeoutChecker;
+    downloadImage: typeof downloadImage;
+    md5: (str: string) => string;
+};
+export default _default;