@bililive-tools/douyu-recorder 1.0.0

package/lib/dy_client/index.js

@@ -0,0 +1,106 @@
+/**
+ * 本来打算直接用 douyudm 这个库，但有比较多的地方不符合我的需求，
+ * 所以参考着这个库以及斗鱼的代码进行了重新实现。
+ * reference: https://github.com/flxxyz/douyudm
+ */
+import WebSocket from "ws";
+import mitt from "mitt";
+import { BufferCoder } from "./buffer_coder.js";
+import { STT } from "./stt.js";
+export function createDYClient(channelId, opts = {}) {
+    let ws = null;
+    let maxRetry = 5;
+    let coder = new BufferCoder();
+    let heartbeatTimer = null;
+    const send = (message) => ws?.send(coder.encode(STT.serialize(message)));
+    const sendLogin = () => send({ type: "loginreq", roomid: channelId });
+    const sendJoinGroup = () => send({ type: "joingroup", rid: channelId, gid: -9999 });
+    const sendHeartbeat = () => send({ type: "mrkl" });
+    const sendLogout = () => send({ type: "logout" });
+    const onOpen = () => {
+        sendLogin();
+        sendJoinGroup();
+        heartbeatTimer = setInterval(sendHeartbeat, 45e3);
+    };
+    const onClose = () => {
+        sendLogout();
+        if (heartbeatTimer) {
+            // @ts-ignore
+            clearInterval(heartbeatTimer);
+            heartbeatTimer = null;
+        }
+    };
+    const onError = (err) => {
+        if (maxRetry > 0) {
+            maxRetry -= 1;
+            stop();
+            setTimeout(() => {
+                start();
+            }, 3e3);
+        }
+        else {
+            client.emit("error", err);
+            client.emit("error", new Error("重连次数过多，停止重连"));
+        }
+    };
+    const onMessage = (message) => {
+        if (typeof message != "object" || message == null || !("type" in message)) {
+            console.warn("Unexpected message format", { message });
+            return;
+        }
+        client.emit("message", 
+        // TODO: 不太好验证 schema，先强制转了
+        message);
+    };
+    const start = () => {
+        if (ws != null)
+            return;
+        ws = new WebSocket(getRandomPortWSURL());
+        coder = new BufferCoder();
+        ws.binaryType = "arraybuffer";
+        ws.on("open", onOpen);
+        ws.on("error", onError);
+        ws.on("close", onClose);
+        ws.on("message", (data) => {
+            if (!(data instanceof ArrayBuffer)) {
+                throw new Error("Do not meet the types of ws.binaryType expected");
+            }
+            coder.decode(data, (messageText) => {
+                try {
+                    const message = STT.deserialize(messageText);
+                    // @ts-ignore
+                    if (message?.type === "comm_chatmsg" && message?.chatmsg) {
+                        // @ts-ignore
+                        message.chatmsg = STT.deserialize(message?.chatmsg);
+                    }
+                    onMessage(message);
+                }
+                catch (error) {
+                    client.emit("error", error);
+                }
+            });
+        });
+    };
+    const stop = () => {
+        if (ws == null)
+            return;
+        onClose();
+        ws = null;
+    };
+    if (!opts.notAutoStart) {
+        start();
+    }
+    const client = {
+        // @ts-ignore
+        ...mitt(),
+        start,
+        stop,
+        send,
+    };
+    return client;
+}
+function getRandomPortWSURL() {
+    const port = 8500 + ((min, max) => Math.floor(Math.random() * (max - min + 1) + min))(1, 6);
+    return `wss://danmuproxy.douyu.com:${port}/`;
+}
+//# sourceMappingURL=index.js.map

package/lib/dy_client/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/dy_client/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,IAAiB,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAoH/B,MAAM,UAAU,cAAc,CAC5B,SAAiB,EACjB,OAEI,EAAE;IAEN,IAAI,EAAE,GAAqB,IAAI,CAAC;IAChC,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;IAC9B,IAAI,cAAc,GAAwB,IAAI,CAAC;IAE/C,MAAM,IAAI,GAAG,CAAC,OAAgC,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAElG,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;IACtE,MAAM,aAAa,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;IACpF,MAAM,aAAa,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;IAElD,MAAM,MAAM,GAAG,GAAG,EAAE;QAClB,SAAS,EAAE,CAAC;QACZ,aAAa,EAAE,CAAC;QAChB,cAAc,GAAG,WAAW,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;IACpD,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,UAAU,EAAE,CAAC;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,aAAa;YACb,aAAa,CAAC,cAAc,CAAC,CAAC;YAC9B,cAAc,GAAG,IAAI,CAAC;QACxB,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,CAAC,GAAY,EAAE,EAAE;QAC/B,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACjB,QAAQ,IAAI,CAAC,CAAC;YACd,IAAI,EAAE,CAAC;YACP,UAAU,CAAC,GAAG,EAAE;gBACd,KAAK,EAAE,CAAC;YACV,CAAC,EAAE,GAAG,CAAC,CAAC;QACV,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,CAAC,OAAgB,EAAE,EAAE;QACrC,IAAI,OAAO,OAAO,IAAI,QAAQ,IAAI,OAAO,IAAI,IAAI,IAAI,CAAC,CAAC,MAAM,IAAI,OAAO,CAAC,EAAE,CAAC;YAC1E,OAAO,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,MAAM,CAAC,IAAI,CACT,SAAS;QACT,2BAA2B;QAC3B,OAAkB,CACnB,CAAC;IACJ,CAAC,CAAC;IAEF,MAAM,KAAK,GAAG,GAAG,EAAE;QACjB,IAAI,EAAE,IAAI,IAAI;YAAE,OAAO;QAEvB,EAAE,GAAG,IAAI,SAAS,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACzC,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;QAE1B,EAAE,CAAC,UAAU,GAAG,aAAa,CAAC;QAC9B,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtB,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACxB,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACxB,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,CAAC,CAAC,IAAI,YAAY,WAAW,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YAED,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;gBACjC,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,GAAG,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;oBAC7C,aAAa;oBACb,IAAI,OAAO,EAAE,IAAI,KAAK,cAAc,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;wBACzD,aAAa;wBACb,OAAO,CAAC,OAAO,GAAG,GAAG,CAAC,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACtD,CAAC;oBACD,SAAS,CAAC,OAAO,CAAC,CAAC;gBACrB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,IAAI,GAAG,GAAG,EAAE;QAChB,IAAI,EAAE,IAAI,IAAI;YAAE,OAAO;QAEvB,OAAO,EAAE,CAAC;QACV,EAAE,GAAG,IAAI,CAAC;IACZ,CAAC,CAAC;IAEF,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QACvB,KAAK,EAAE,CAAC;IACV,CAAC;IAED,MAAM,MAAM,GAAa;QACvB,aAAa;QACb,GAAG,IAAI,EAAE;QACT,KAAK;QACL,IAAI;QACJ,IAAI;KACL,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,kBAAkB;IACzB,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5F,OAAO,8BAA8B,IAAI,GAAG,CAAC;AAC/C,CAAC"}

package/lib/dy_client/stt.d.ts

@@ -0,0 +1,6 @@
+export declare namespace STT {
+    function escape(v: string): string;
+    function unescape(v: string): string;
+    function serialize(obj: unknown): string;
+    function deserialize(raw: string): unknown;
+}

package/lib/dy_client/stt.js

@@ -0,0 +1,46 @@
+export var STT;
+(function (STT) {
+    function escape(v) {
+        return v.toString().replace(/@/g, "@A").replace(/\//g, "@S");
+    }
+    STT.escape = escape;
+    function unescape(v) {
+        return v.toString().replace(/@S/g, "/").replace(/@A/g, "@");
+    }
+    STT.unescape = unescape;
+    function serialize(obj) {
+        if (obj == null)
+            throw new Error("Cant serialize null value");
+        if (Array.isArray(obj)) {
+            return obj.map((v) => STT.serialize(v)).join("");
+        }
+        if (typeof obj === "object") {
+            return Object.entries(obj)
+                .map(([k, v]) => `${k}@=${STT.serialize(v)}`)
+                .join("");
+        }
+        return STT.escape(obj.toString()) + "/";
+    }
+    STT.serialize = serialize;
+    function deserialize(raw) {
+        if (raw.includes("//")) {
+            return raw
+                .split("//")
+                .filter((e) => e !== "")
+                .map((item) => STT.deserialize(item));
+        }
+        if (raw.includes("@=")) {
+            return raw
+                .split("/")
+                .filter((part) => part !== "")
+                .reduce((obj, part) => {
+                const [key, val] = part.split("@=");
+                obj[key] = val ? STT.deserialize(val) : "";
+                return obj;
+            }, {});
+        }
+        return STT.unescape(raw);
+    }
+    STT.deserialize = deserialize;
+})(STT || (STT = {}));
+//# sourceMappingURL=stt.js.map

package/lib/dy_client/stt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stt.js","sourceRoot":"","sources":["../../src/dy_client/stt.ts"],"names":[],"mappings":"AAAA,MAAM,KAAW,GAAG,CAiDnB;AAjDD,WAAiB,GAAG;IAClB,SAAgB,MAAM,CAAC,CAAS;QAC9B,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC;IAFe,UAAM,SAErB,CAAA;IAED,SAAgB,QAAQ,CAAC,CAAS;QAChC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC9D,CAAC;IAFe,YAAQ,WAEvB,CAAA;IAED,SAAgB,SAAS,CAAC,GAAY;QACpC,IAAI,GAAG,IAAI,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAE9D,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;iBACvB,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5C,IAAI,CAAC,EAAE,CAAC,CAAC;QACd,CAAC;QAED,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,GAAG,GAAG,CAAC;IAC1C,CAAC;IAde,aAAS,YAcxB,CAAA;IAED,SAAgB,WAAW,CAAC,GAAW;QACrC,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG;iBACP,KAAK,CAAC,IAAI,CAAC;iBACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;iBACvB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1C,CAAC;QAED,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG;iBACP,KAAK,CAAC,GAAG,CAAC;iBACV,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC;iBAC7B,MAAM,CACL,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;gBACZ,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACpC,GAAG,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,OAAO,GAAG,CAAC;YACb,CAAC,EACD,EAA6B,CAC9B,CAAC;QACN,CAAC;QAED,OAAO,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAvBe,eAAW,cAuB1B,CAAA;AACH,CAAC,EAjDgB,GAAG,KAAH,GAAG,QAiDnB"}

package/lib/index.d.ts

@@ -0,0 +1,2 @@
+import { RecorderProvider } from "@bililive-tools/manager";
+export declare const provider: RecorderProvider<Record<string, unknown>>;

package/lib/index.js

@@ -0,0 +1,333 @@
+import mitt from "mitt";
+import { createFFMPEGBuilder, defaultFromJSON, defaultToJSON, genRecorderUUID, genRecordUUID, StreamManager, utils, } from "@bililive-tools/manager";
+import { getInfo, getStream } from "./stream.js";
+import { getRoomInfo } from "./dy_api.js";
+import { assert, ensureFolderExist } from "./utils.js";
+import { createDYClient } from "./dy_client/index.js";
+import { giftMap, colorTab } from "./danma.js";
+import { requester } from "./requester.js";
+function createRecorder(opts) {
+    // 内部实现时，应该只有 proxy 包裹的那一层会使用这个 recorder 标识符，不应该有直接通过
+    // 此标志来操作这个对象的地方，不然会跳过 proxy 的拦截。
+    const recorder = {
+        id: opts.id ?? genRecorderUUID(),
+        extra: opts.extra ?? {},
+        // @ts-ignore
+        ...mitt(),
+        ...opts,
+        availableStreams: [],
+        availableSources: [],
+        qualityMaxRetry: opts.qualityRetry ?? 0,
+        qualityRetry: opts.qualityRetry ?? 0,
+        state: "idle",
+        getChannelURL() {
+            return `https://www.douyu.com/${this.channelId}`;
+        },
+        checkLiveStatusAndRecord: utils.singleton(checkLiveStatusAndRecord),
+        toJSON() {
+            return defaultToJSON(provider, this);
+        },
+        async getLiveInfo() {
+            const channelId = this.channelId;
+            const info = await getInfo(channelId);
+            return {
+                channelId,
+                ...info,
+            };
+        },
+        async getStream() {
+            const res = await getStream({
+                channelId: this.channelId,
+                quality: this.quality,
+            });
+            return res.currentStream;
+        },
+    };
+    const recorderWithSupportUpdatedEvent = new Proxy(recorder, {
+        set(obj, prop, value) {
+            Reflect.set(obj, prop, value);
+            if (typeof prop === "string") {
+                obj.emit("Updated", [prop]);
+            }
+            return true;
+        },
+    });
+    return recorderWithSupportUpdatedEvent;
+}
+const ffmpegOutputOptions = [
+    "-c",
+    "copy",
+    "-movflags",
+    "faststart+frag_keyframe+empty_moov",
+    "-min_frag_duration",
+    "60000000",
+];
+const checkLiveStatusAndRecord = async function ({ getSavePath, qualityRetry, banLiveId, }) {
+    if (this.recordHandle != null)
+        return this.recordHandle;
+    const liveInfo = await getInfo(this.channelId);
+    this.liveInfo = liveInfo;
+    const { living, owner, title, cover } = liveInfo;
+    if (liveInfo.liveId === banLiveId) {
+        this.tempStopIntervalCheck = true;
+    }
+    else {
+        this.tempStopIntervalCheck = false;
+    }
+    if (this.tempStopIntervalCheck)
+        return null;
+    if (!living)
+        return null;
+    this.state = "recording";
+    let res;
+    // TODO: 先不做什么错误处理，就简单包一下预期上会有错误的地方
+    try {
+        let strictQuality = !!this.qualityRetry;
+        if (qualityRetry !== undefined) {
+            strictQuality = !!qualityRetry;
+        }
+        res = await getStream({
+            channelId: this.channelId,
+            quality: this.quality,
+            strictQuality,
+        });
+    }
+    catch (err) {
+        this.state = "idle";
+        this.qualityRetry -= 1;
+        throw err;
+    }
+    const { currentStream: stream, sources: availableSources, streams: availableStreams } = res;
+    this.availableStreams = availableStreams.map((s) => s.name);
+    this.availableSources = availableSources.map((s) => s.name);
+    this.usedStream = stream.name;
+    this.usedSource = stream.source;
+    // TODO: emit update event
+    const savePath = getSavePath({ owner, title });
+    const hasSegment = !!this.segment;
+    const streamManager = new StreamManager(this, getSavePath, owner, title, savePath, hasSegment);
+    try {
+        ensureFolderExist(savePath);
+    }
+    catch (err) {
+        this.state = "idle";
+        throw err;
+    }
+    const handleVideoCreated = async ({ filename }) => {
+        const extraDataController = streamManager?.getExtraDataController();
+        extraDataController?.setMeta({
+            room_id: this.channelId,
+            platform: provider?.id,
+            liveStartTimestamp: liveInfo.startTime?.getTime(),
+        });
+        if (this.saveCover) {
+            const coverPath = utils.replaceExtName(filename, ".jpg");
+            utils.downloadImage(cover, coverPath);
+        }
+    };
+    this.on("videoFileCreated", handleVideoCreated);
+    const client = createDYClient(Number(this.channelId), {
+        notAutoStart: true,
+    });
+    client.on("message", (msg) => {
+        const extraDataController = streamManager.getExtraDataController();
+        if (!extraDataController)
+            return;
+        switch (msg.type) {
+            case "chatmsg": {
+                const comment = {
+                    type: "comment",
+                    timestamp: Date.now(),
+                    text: msg.txt,
+                    color: colorTab[msg.col] ?? "#ffffff",
+                    sender: {
+                        uid: msg.uid,
+                        name: msg.nn,
+                        avatar: msg.ic,
+                        extra: {
+                            level: msg.level,
+                        },
+                    },
+                };
+                this.emit("Message", comment);
+                extraDataController.addMessage(comment);
+                break;
+            }
+            case "dgb": {
+                if (this.saveGiftDanma === false)
+                    return;
+                const gift = {
+                    type: "give_gift",
+                    timestamp: Date.now(),
+                    name: giftMap[msg.gfid]?.name ?? msg.gfn,
+                    price: (giftMap[msg.gfid]?.pc ?? 0) / 100,
+                    count: Number(msg.gfcnt),
+                    color: "#ffffff",
+                    sender: {
+                        uid: msg.uid,
+                        name: msg.nn,
+                        avatar: msg.ic,
+                        extra: {
+                            level: msg.level,
+                        },
+                    },
+                    extra: {
+                        hits: Number(msg.hits),
+                    },
+                };
+                this.emit("Message", gift);
+                extraDataController.addMessage(gift);
+                break;
+                // TODO: 还有一些其他礼物相关的 msg 要处理，目前先简单点只处理 dgb
+            }
+            case "comm_chatmsg": {
+                if (this.saveSCDanma === false)
+                    return;
+                switch (msg.btype) {
+                    case "voiceDanmu": {
+                        const comment = {
+                            type: "super_chat",
+                            timestamp: Date.now(),
+                            text: msg?.chatmsg?.txt,
+                            price: Number(msg.cprice) / 100,
+                            sender: {
+                                uid: msg.uid,
+                                name: msg?.chatmsg?.nn,
+                                avatar: msg?.chatmsg?.ic,
+                                extra: {
+                                    level: msg?.chatmsg?.level,
+                                },
+                            },
+                        };
+                        this.emit("Message", comment);
+                        extraDataController.addMessage(comment);
+                        break;
+                    }
+                }
+                break;
+            }
+        }
+    });
+    client.on("error", (err) => {
+        this.emit("DebugLog", { type: "common", text: String(err) });
+    });
+    // console.log("this.disableProvideCommentsWhenRecording", this.disableProvideCommentsWhenRecording);
+    if (!this.disableProvideCommentsWhenRecording) {
+        client.start();
+    }
+    let isEnded = false;
+    const onEnd = async (...args) => {
+        if (isEnded)
+            return;
+        isEnded = true;
+        this.emit("DebugLog", {
+            type: "common",
+            text: `ffmpeg end, reason: ${JSON.stringify(args, (_, v) => (v instanceof Error ? v.stack : v))}`,
+        });
+        const reason = args[0] instanceof Error ? args[0].message : String(args[0]);
+        this.recordHandle?.stop(reason);
+    };
+    const isInvalidStream = utils.createInvalidStreamChecker();
+    const timeoutChecker = utils.createTimeoutChecker(() => onEnd("ffmpeg timeout"), 10e3);
+    const command = createFFMPEGBuilder(stream.url)
+        .inputOptions("-user_agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.86 Safari/537.36")
+        .outputOptions(ffmpegOutputOptions)
+        .output(streamManager.videoFilePath)
+        .on("start", () => {
+        streamManager.handleVideoStarted();
+    })
+        .on("error", onEnd)
+        .on("end", () => onEnd("finished"))
+        .on("stderr", async (stderrLine) => {
+        assert(typeof stderrLine === "string");
+        if (utils.isFfmpegStartSegment(stderrLine)) {
+            await streamManager.handleVideoStarted(stderrLine);
+        }
+        // TODO:解析时间
+        this.emit("DebugLog", { type: "ffmpeg", text: stderrLine });
+        if (isInvalidStream(stderrLine)) {
+            onEnd("invalid stream");
+        }
+    })
+        .on("stderr", timeoutChecker.update);
+    if (hasSegment) {
+        command.outputOptions("-f", "segment", "-segment_time", String(this.segment * 60), "-reset_timestamps", "1");
+    }
+    const ffmpegArgs = command._getArguments();
+    command.run();
+    // TODO: 需要一个机制防止空录制，比如检查文件的大小变化、ffmpeg 的输出、直播状态等
+    const stop = utils.singleton(async (reason) => {
+        if (!this.recordHandle)
+            return;
+        this.state = "stopping-record";
+        // TODO: emit update event
+        timeoutChecker.stop();
+        try {
+            // @ts-ignore
+            command.ffmpegProc?.stdin?.write("q");
+            // TODO: 这里可能会有内存泄露，因为事件还没清，之后再检查下看看。
+            client.stop();
+        }
+        catch (err) {
+            // TODO: 这个 stop 经常报错，这里先把错误吞掉，以后再处理。
+            this.emit("DebugLog", { type: "common", text: String(err) });
+        }
+        this.usedStream = undefined;
+        this.usedSource = undefined;
+        // TODO: other codes
+        // TODO: emit update event
+        await streamManager.handleVideoCompleted();
+        this.emit("RecordStop", { recordHandle: this.recordHandle, reason });
+        this.off("videoFileCreated", handleVideoCreated);
+        this.recordHandle = undefined;
+        this.liveInfo = undefined;
+        this.state = "idle";
+        this.qualityRetry = this.qualityMaxRetry;
+    });
+    this.recordHandle = {
+        id: genRecordUUID(),
+        stream: stream.name,
+        source: stream.source,
+        url: stream.url,
+        ffmpegArgs,
+        savePath: savePath,
+        stop,
+    };
+    this.emit("RecordStart", this.recordHandle);
+    return this.recordHandle;
+};
+export const provider = {
+    id: "DouYu",
+    name: "斗鱼",
+    siteURL: "https://douyu.com/",
+    matchURL(channelURL) {
+        return /https?:\/\/(?:.*?\.)?douyu.com\//.test(channelURL);
+    },
+    async resolveChannelInfoFromURL(channelURL) {
+        if (!this.matchURL(channelURL))
+            return null;
+        channelURL = channelURL.trim();
+        const res = await requester.get(channelURL);
+        const html = res.data;
+        const matched = html.match(/\$ROOM\.room_id.?=(.*?);/);
+        if (!matched)
+            return null;
+        const room_id = matched[1].trim();
+        const roomInfo = await getRoomInfo(Number(room_id));
+        return {
+            id: matched[1].trim(),
+            title: roomInfo.room.room_name,
+            owner: roomInfo.room.nickname,
+        };
+    },
+    createRecorder(opts) {
+        return createRecorder({ providerId: provider.id, ...opts });
+    },
+    fromJSON(recorder) {
+        return defaultFromJSON(this, recorder);
+    },
+    setFFMPEGOutputArgs(args) {
+        ffmpegOutputOptions.splice(0, ffmpegOutputOptions.length, ...args);
+    },
+};
+//# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAIL,mBAAmB,EAEnB,eAAe,EACf,aAAa,EACb,eAAe,EACf,aAAa,EACb,aAAa,EACb,KAAK,GACN,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAE3C,SAAS,cAAc,CAAC,IAAwB;IAC9C,qDAAqD;IACrD,iCAAiC;IACjC,MAAM,QAAQ,GAAa;QACzB,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,eAAe,EAAE;QAChC,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;QACvB,aAAa;QACb,GAAG,IAAI,EAAE;QACT,GAAG,IAAI;QAEP,gBAAgB,EAAE,EAAE;QACpB,gBAAgB,EAAE,EAAE;QACpB,eAAe,EAAE,IAAI,CAAC,YAAY,IAAI,CAAC;QACvC,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,CAAC;QACpC,KAAK,EAAE,MAAM;QAEb,aAAa;YACX,OAAO,yBAAyB,IAAI,CAAC,SAAS,EAAE,CAAC;QACnD,CAAC;QACD,wBAAwB,EAAE,KAAK,CAAC,SAAS,CAAC,wBAAwB,CAAC;QAEnE,MAAM;YACJ,OAAO,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACvC,CAAC;QAED,KAAK,CAAC,WAAW;YACf,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;YACjC,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,CAAC;YACtC,OAAO;gBACL,SAAS;gBACT,GAAG,IAAI;aACR,CAAC;QACJ,CAAC;QACD,KAAK,CAAC,SAAS;YACb,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC;gBAC1B,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;YACH,OAAO,GAAG,CAAC,aAAa,CAAC;QAC3B,CAAC;KACF,CAAC;IAEF,MAAM,+BAA+B,GAAG,IAAI,KAAK,CAAC,QAAQ,EAAE;QAC1D,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK;YAClB,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YAE9B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC7B,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9B,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,+BAA+B,CAAC;AACzC,CAAC;AAED,MAAM,mBAAmB,GAAa;IACpC,IAAI;IACJ,MAAM;IACN,WAAW;IACX,oCAAoC;IACpC,oBAAoB;IACpB,UAAU;CACX,CAAC;AAEF,MAAM,wBAAwB,GAAyC,KAAK,WAAW,EACrF,WAAW,EACX,YAAY,EACZ,SAAS,GACV;IACC,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC,YAAY,CAAC;IAExD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC/C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IACzB,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC;IACjD,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAClC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;IACpC,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC;IACrC,CAAC;IACD,IAAI,IAAI,CAAC,qBAAqB;QAAE,OAAO,IAAI,CAAC;IAC5C,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IAEzB,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC;IACzB,IAAI,GAA0C,CAAC;IAC/C,mCAAmC;IACnC,IAAI,CAAC;QACH,IAAI,aAAa,GAAG,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;QACxC,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,aAAa,GAAG,CAAC,CAAC,YAAY,CAAC;QACjC,CAAC;QACD,GAAG,GAAG,MAAM,SAAS,CAAC;YACpB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QACpB,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;QACvB,MAAM,GAAG,CAAC;IACZ,CAAC;IACD,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,gBAAgB,EAAE,GAAG,GAAG,CAAC;IAC5F,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC5D,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC5D,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC;IAC9B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC;IAChC,0BAA0B;IAC1B,MAAM,QAAQ,GAAG,WAAW,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IAC/C,MAAM,UAAU,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;IAClC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;IAE/F,IAAI,CAAC;QACH,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QACpB,MAAM,GAAG,CAAC;IACZ,CAAC;IAED,MAAM,kBAAkB,GAAG,KAAK,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;QAChD,MAAM,mBAAmB,GAAG,aAAa,EAAE,sBAAsB,EAAE,CAAC;QACpE,mBAAmB,EAAE,OAAO,CAAC;YAC3B,OAAO,EAAE,IAAI,CAAC,SAAS;YACvB,QAAQ,EAAE,QAAQ,EAAE,EAAE;YACtB,kBAAkB,EAAE,QAAQ,CAAC,SAAS,EAAE,OAAO,EAAE;SAClD,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,SAAS,GAAG,KAAK,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YACzD,KAAK,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACxC,CAAC;IACH,CAAC,CAAC;IACF,IAAI,CAAC,EAAE,CAAC,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;IAEhD,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE;QACpD,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IACH,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;QAC3B,MAAM,mBAAmB,GAAG,aAAa,CAAC,sBAAsB,EAAE,CAAC;QACnE,IAAI,CAAC,mBAAmB;YAAE,OAAO;QACjC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,MAAM,OAAO,GAAY;oBACvB,IAAI,EAAE,SAAS;oBACf,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,IAAI,EAAE,GAAG,CAAC,GAAG;oBACb,KAAK,EAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,SAAS;oBACrC,MAAM,EAAE;wBACN,GAAG,EAAE,GAAG,CAAC,GAAG;wBACZ,IAAI,EAAE,GAAG,CAAC,EAAE;wBACZ,MAAM,EAAE,GAAG,CAAC,EAAE;wBACd,KAAK,EAAE;4BACL,KAAK,EAAE,GAAG,CAAC,KAAK;yBACjB;qBACF;iBACF,CAAC;gBACF,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBAC9B,mBAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;gBACxC,MAAM;YACR,CAAC;YACD,KAAK,KAAK,CAAC,CAAC,CAAC;gBACX,IAAI,IAAI,CAAC,aAAa,KAAK,KAAK;oBAAE,OAAO;gBACzC,MAAM,IAAI,GAAa;oBACrB,IAAI,EAAE,WAAW;oBACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,GAAG,CAAC,GAAG;oBACxC,KAAK,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,GAAG,GAAG;oBACzC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;oBACxB,KAAK,EAAE,SAAS;oBAChB,MAAM,EAAE;wBACN,GAAG,EAAE,GAAG,CAAC,GAAG;wBACZ,IAAI,EAAE,GAAG,CAAC,EAAE;wBACZ,MAAM,EAAE,GAAG,CAAC,EAAE;wBACd,KAAK,EAAE;4BACL,KAAK,EAAE,GAAG,CAAC,KAAK;yBACjB;qBACF;oBACD,KAAK,EAAE;wBACL,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;qBACvB;iBACF,CAAC;gBACF,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAC3B,mBAAmB,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACrC,MAAM;gBAEN,0CAA0C;YAC5C,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,IAAI,IAAI,CAAC,WAAW,KAAK,KAAK;oBAAE,OAAO;gBACvC,QAAQ,GAAG,CAAC,KAAK,EAAE,CAAC;oBAClB,KAAK,YAAY,CAAC,CAAC,CAAC;wBAClB,MAAM,OAAO,GAAc;4BACzB,IAAI,EAAE,YAAY;4BAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;4BACrB,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG;4BACvB,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,GAAG;4BAC/B,MAAM,EAAE;gCACN,GAAG,EAAE,GAAG,CAAC,GAAG;gCACZ,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE;gCACtB,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE;gCACxB,KAAK,EAAE;oCACL,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK;iCAC3B;6BACF;yBACF,CAAC;wBACF,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;wBAC9B,mBAAmB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;wBACxC,MAAM;oBACR,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;QACzB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IACH,qGAAqG;IACrG,IAAI,CAAC,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC9C,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,KAAK,GAAG,KAAK,EAAE,GAAG,IAAe,EAAE,EAAE;QACzC,IAAI,OAAO;YAAE,OAAO;QACpB,OAAO,GAAG,IAAI,CAAC;QAEf,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;YACpB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,uBAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;SAClG,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5E,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC,CAAC;IAEF,MAAM,eAAe,GAAG,KAAK,CAAC,0BAA0B,EAAE,CAAC;IAC3D,MAAM,cAAc,GAAG,KAAK,CAAC,oBAAoB,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,CAAC;IACvF,MAAM,OAAO,GAAG,mBAAmB,CAAC,MAAM,CAAC,GAAG,CAAC;SAC5C,YAAY,CACX,aAAa,EACb,oHAAoH,CACrH;SACA,aAAa,CAAC,mBAAmB,CAAC;SAClC,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC;SACnC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QAChB,aAAa,CAAC,kBAAkB,EAAE,CAAC;IACrC,CAAC,CAAC;SACD,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC;SAClB,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;SAClC,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;QACjC,MAAM,CAAC,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC;QACvC,IAAI,KAAK,CAAC,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC;YAC3C,MAAM,aAAa,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC;QACD,YAAY;QACZ,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAE5D,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;YAChC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CAAC;SACD,EAAE,CAAC,QAAQ,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;IACvC,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,CAAC,aAAa,CACnB,IAAI,EACJ,SAAS,EACT,eAAe,EACf,MAAM,CAAC,IAAI,CAAC,OAAQ,GAAG,EAAE,CAAC,EAC1B,mBAAmB,EACnB,GAAG,CACJ,CAAC;IACJ,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAC3C,OAAO,CAAC,GAAG,EAAE,CAAC;IAEd,iDAAiD;IAEjD,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAuB,KAAK,EAAE,MAAe,EAAE,EAAE;QAC3E,IAAI,CAAC,IAAI,CAAC,YAAY;YAAE,OAAO;QAC/B,IAAI,CAAC,KAAK,GAAG,iBAAiB,CAAC;QAC/B,0BAA0B;QAE1B,cAAc,CAAC,IAAI,EAAE,CAAC;QAEtB,IAAI,CAAC;YACH,aAAa;YACb,OAAO,CAAC,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;YACtC,qCAAqC;YACrC,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,qCAAqC;YACrC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,oBAAoB;QACpB,0BAA0B;QAC1B,MAAM,aAAa,CAAC,oBAAoB,EAAE,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,GAAG,CAAC,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;QACjD,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QACpB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,YAAY,GAAG;QAClB,EAAE,EAAE,aAAa,EAAE;QACnB,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,UAAU;QACV,QAAQ,EAAE,QAAQ;QAClB,IAAI;KACL,CAAC;IACF,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAE5C,OAAO,IAAI,CAAC,YAAY,CAAC;AAC3B,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,QAAQ,GAA8C;IACjE,EAAE,EAAE,OAAO;IACX,IAAI,EAAE,IAAI;IACV,OAAO,EAAE,oBAAoB;IAE7B,QAAQ,CAAC,UAAU;QACjB,OAAO,kCAAkC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,UAAU;QACxC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;YAAE,OAAO,IAAI,CAAC;QAE5C,UAAU,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAEtB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;QACvD,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAElC,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QAEpD,OAAO;YACL,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;YACrB,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,SAAS;YAC9B,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,QAAQ;SAC9B,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,IAAI;QACjB,OAAO,cAAc,CAAC,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,QAAQ,CAAC,QAAQ;QACf,OAAO,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,mBAAmB,CAAC,IAAI;QACtB,mBAAmB,CAAC,MAAM,CAAC,CAAC,EAAE,mBAAmB,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;IACrE,CAAC;CACF,CAAC"}

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,8 @@
+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,
+});
+//# sourceMappingURL=requester.js.map

package/lib/requester.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"requester.js","sourceRoot":"","sources":["../src/requester.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC;IACpC,OAAO,EAAE,IAAI;IACb,kDAAkD;IAClD,8FAA8F;IAC9F,KAAK,EAAE,KAAK;CACb,CAAC,CAAC"}

package/lib/stream.d.ts

@@ -0,0 +1,26 @@
+import { Recorder } from "@bililive-tools/manager";
+export declare function getInfo(channelId: string): Promise<{
+    living: boolean;
+    owner: string;
+    title: string;
+    startTime: Date;
+    avatar: string;
+    cover: string;
+    liveId: string;
+}>;
+export declare function getStream(opts: Pick<Recorder, "channelId" | "quality"> & {
+    rejectCache?: boolean;
+    strictQuality?: boolean;
+}): Promise<{
+    living: true;
+    sources: import("./dy_api.js").SourceProfile[];
+    streams: import("./dy_api.js").StreamProfile[];
+    isSupportRateSwitch: boolean;
+    isOriginalStream: boolean;
+    currentStream: {
+        source: string;
+        name: string;
+        rate: number;
+        url: string;
+    };
+}>;

package/lib/stream.js

@@ -0,0 +1,81 @@
+import { live } from "douyu-api";
+import { DouyuQualities, utils } from "@bililive-tools/manager";
+import { getLiveInfo } from "./dy_api.js";
+import { requester } from "./requester.js";
+export async function getInfo(channelId) {
+    const res = await requester.get(`http://open.douyucdn.cn/api/RoomApi/room/${channelId}`);
+    if (res.status !== 200) {
+        if (res.status === 404 && res.data === "Not Found") {
+            throw new Error("错误的地址 " + channelId);
+        }
+        throw new Error(`Unexpected status code, ${res.status}, ${res.data}`);
+    }
+    if (typeof res.data !== "object")
+        throw new Error(`Unexpected response, ${res.status}, ${res.data}`);
+    const json = res.data;
+    if (json.error === 101)
+        throw new Error("错误的地址 " + channelId);
+    if (json.error !== 0)
+        throw new Error("Unexpected error code, " + json.error);
+    let living = json.data.room_status === "1";
+    const data = await live.getRoomInfo(Number(channelId));
+    if (living) {
+        const isVideoLoop = data.room.videoLoop === 1;
+        if (isVideoLoop) {
+            living = false;
+        }
+    }
+    const startTime = new Date(data.room.show_time * 1000);
+    return {
+        living,
+        owner: data.room.nickname,
+        title: data.room.room_name,
+        avatar: data.room.avatar.big,
+        cover: data.room.room_pic,
+        startTime: startTime,
+        liveId: utils.md5(`${channelId}-${startTime?.getTime() ?? Date.now()}`),
+        // gifts: data.gift.map((g) => ({
+        //   id: g.id,
+        //   name: g.name,
+        //   img: g.himg,
+        //   cost: g.pc,
+        // })),
+    };
+}
+export async function getStream(opts) {
+    const qn = (DouyuQualities.includes(opts.quality) ? opts.quality : 0);
+    let liveInfo = await getLiveInfo({
+        channelId: opts.channelId,
+        rate: qn,
+    });
+    if (!liveInfo.living)
+        throw new Error("It must be called getStream when living");
+    // console.log(JSON.stringify(liveInfo, null, 2));
+    if (liveInfo.currentStream.rate !== qn && opts.strictQuality) {
+        throw new Error("Can not get expect quality because of strictQuality");
+    }
+    let expectSource = liveInfo.sources[0];
+    if (!expectSource) {
+        throw new Error("Source list is empty");
+    }
+    // 是否存在画质下没有source的情况，可能需要切换画质
+    if (liveInfo.currentStream.rate !== qn) {
+        if (liveInfo.streams.length === 0) {
+            throw new Error("Can not get expect quality because of no available stream");
+        }
+        else {
+            liveInfo = await getLiveInfo({
+                channelId: opts.channelId,
+                rate: liveInfo.streams[0].rate,
+            });
+            if (!liveInfo.living)
+                throw new Error("It must be called getStream when living");
+        }
+    }
+    // 流未准备好，防止刚开播时的无效录制。
+    // 该判断可能导致开播前 30 秒左右无法录制到，因为 streamStatus 在后端似乎有缓存，所以暂时不使用。
+    // TODO: 需要在 ffmpeg 那里加处理，防止无效录制
+    // if (!json.data.streamStatus) return
+    return liveInfo;
+}
+//# sourceMappingURL=stream.js.map

package/lib/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,cAAc,EAAY,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAC1E,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAE3C,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,SAAiB;IAe7C,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,GAAG,CAkB7B,4CAA4C,SAAS,EAAE,CAAC,CAAC;IAE3D,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ;QAC9B,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAErE,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;IACtB,IAAI,IAAI,CAAC,KAAK,KAAK,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;IAC9D,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9E,IAAI,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,KAAK,GAAG,CAAC;IAE3C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IACvD,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,CAAC;QAC9C,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,GAAG,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC;IACvD,OAAO;QACL,MAAM;QACN,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ;QACzB,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;QAC1B,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG;QAC5B,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ;QACzB,SAAS,EAAE,SAAS;QACpB,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS,IAAI,SAAS,EAAE,OAAO,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QAEvE,iCAAiC;QACjC,cAAc;QACd,kBAAkB;QAClB,iBAAiB;QACjB,gBAAgB;QAChB,OAAO;KACR,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,IAGC;IAED,MAAM,EAAE,GAAG,CACT,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAC7B,CAAC;IAErC,IAAI,QAAQ,GAAG,MAAM,WAAW,CAAC;QAC/B,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,IAAI,EAAE,EAAE;KACT,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAEjF,kDAAkD;IAElD,IAAI,QAAQ,CAAC,aAAa,CAAC,IAAI,KAAK,EAAE,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;QAC7D,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACvC,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;IAED,8BAA8B;IAC9B,IAAI,QAAQ,CAAC,aAAa,CAAC,IAAI,KAAK,EAAE,EAAE,CAAC;QACvC,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,MAAM,WAAW,CAAC;gBAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAED,qBAAqB;IACrB,2DAA2D;IAC3D,gCAAgC;IAChC,sCAAsC;IAEtC,OAAO,QAAQ,CAAC;AAClB,CAAC"}