@floating-live/bilibili 0.15.3 → 0.16.1

package/room.js

@@ -1,230 +1,294 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.RoomBilibili = void 0;
-const floating_live_1 = require("floating-live");
-const bilibili_live_ws_1 = require("bilibili-live-ws");
-const parser_1 = require("./parser");
-const utils_1 = require("./utils");
-const floating_live_2 = require("floating-live");
-class RoomBilibili extends floating_live_1.LiveRoom {
-    constructor(
-    /** 房间id */
-    id, options) {
-        super();
-        /** 平台id */
-        this.platform = "bilibili";
-        /** 直播间room_id */
-        this.roomId = 0;
-        /** 直播间短房号 */
-        this.shortId = 0;
-        /** 房间信息 */
-        this.detail = {
-            /** 直播标题 */
-            title: "",
-            /** 分区 */
-            area: [],
-            /** 封面url */
-            cover: "",
-        };
-        this.stats = {
-            /** 观看数 */
-            view: 0,
-            /** 点赞数 */
-            like: 0,
-        };
-        /** 房间是否可用 */
-        this.available = false;
-        /** 是否连接上房间 */
-        this.connected = false;
-        /** 主播信息 */
-        this.anchor = { name: "", id: 0 };
-        /** 直播间弹幕api模块 */
-        this.client = null;
-        /** 连接凭证 */
-        this.tokens = {
-            userId: 0,
-            buvid: "",
-            token: "",
-        };
-        /** 是否处于正在打开的状态 */
-        this.opening = false;
-        /** 是否处于重连状态 */
-        this.reconnecting = false;
-        this.id = id;
-        this.connectInterval = options?.connectInterval || 100;
-        this.init(options);
-    }
-    /** 初始化 */
-    async init(options = {}) {
-        const { info, tokens, credentials, open } = options;
-        if (open)
-            this.opening = true;
-        if (info) {
-            this.updateInfo(info);
-        }
-        else {
-            await this.getInfo();
-        }
-        if (tokens?.userId && tokens?.buvid && tokens?.token) {
-            this.setTokens(tokens);
-        }
-        else {
-            await this.setCredentials(credentials || "", tokens);
-        }
-        if (open) {
-            this.opening = false;
-            this.openWS();
-        }
-        this.emit("init");
-    }
-    /** 获取房间信息 */
-    async getInfo() {
-        await this.fetchInfo()
-            .then((info) => {
-            this.roomId = info.id;
-            this.updateInfo(info);
-        })
-            .catch((error) => {
-            this.emit("info_error", error);
-        });
-    }
-    /** 请求房间信息 */
-    async fetchInfo() {
-        const rawInfo = await (0, utils_1.getInfoByRoom)(this.id);
-        return (0, parser_1.parseInfo)(rawInfo);
-    }
-    /** 更新直播间信息 */
-    updateInfo({ roomId, shortId, status, timestamp, detail, stats, anchor, liveId, available, }) {
-        this.roomId = roomId;
-        this.shortId = shortId;
-        this.status = status;
-        this.timestamp = timestamp;
-        this.detail = detail;
-        this.stats = stats;
-        this.anchor = anchor;
-        this.liveId = liveId;
-        this.available = available;
-        this.emit("info", this.info);
-    }
-    /** 开启直播间监听 */
-    async open() {
-        // 如果直播间监听已打开或处于正在打开状态，则返回
-        if (this.opened || this.opening)
-            return;
-        this.opening = true;
-        await this.getInfo();
-        this.opening = false;
-        this.openWS();
-    }
-    openWS() {
-        // 如果直播间不可用，则返回
-        if (!this.available)
-            return;
-        this.opened = true;
-        this.initClient();
-        this.emit("open");
-    }
-    /** 设置登录凭据 */
-    async setCredentials(credentials, userToken) {
-        this.setTokens(await this.fetchTokens(credentials, userToken));
-    }
-    /** 请求连接tokens */
-    async fetchTokens(credentials = "", userToken) {
-        const cookie = (0, utils_1.parseCookieString)(credentials);
-        const buvid = userToken?.buvid ||
-            cookie["buvid3"] ||
-            (await (0, utils_1.getBuvid)(credentials)) ||
-            "";
-        const userId = userToken?.userId ||
-            parseInt(cookie["DedeUserID"]) ||
-            (await (0, utils_1.getLoginUid)(credentials));
-        const token = await (0, utils_1.getToken)(this.roomId, credentials);
-        return {
-            userId,
-            buvid,
-            token,
-        };
-    }
-    /** 设置直播连接tokens */
-    setTokens(tokens) {
-        this.tokens = tokens;
-        if (this.opened) {
-            // 如果已打开，则重连
-            this.reconnect();
-        }
-    }
-    /** 重连 */
-    reconnect() {
-        if (this.reconnecting)
-            return;
-        this.reconnecting = true;
-        const lastClient = this.client;
-        if (!lastClient) {
-            this.reconnecting = false;
-            return;
-        }
-        this.initClient();
-        this.client?.once("open", () => {
-            lastClient.removeAllListeners();
-            lastClient.close();
-            this.reconnecting = false;
-        });
-    }
-    emitConnention(status) {
-        this.connection = status;
-        switch (status) {
-            case floating_live_2.ConnectStatus.connecting:
-                this.emit("connecting");
-                break;
-            case floating_live_2.ConnectStatus.connected:
-                this.emit("connected");
-                break;
-            case floating_live_2.ConnectStatus.entered:
-                this.emit("enter");
-                break;
-            case floating_live_2.ConnectStatus.disconnected:
-                this.emit("disconnect");
-                break;
-        }
-    }
-    /** 创建直播弹幕客户端 */
-    initClient() {
-        this.emitConnention(floating_live_2.ConnectStatus.connecting);
-        // 与Websocket服务器连接
-        const client = new bilibili_live_ws_1.KeepLiveWS(this.roomId || this.id, {
-            uid: this.tokens.userId,
-            buvid: this.tokens.buvid,
-            key: this.tokens.token,
-        });
-        client.interval = this.connectInterval;
-        client.on("open", () => {
-            // 连接到服务器
-            this.opened && this.emitConnention(floating_live_2.ConnectStatus.connected);
-        });
-        client.on("close", () => {
-            // 断开连接
-            this.opened && this.emitConnention(floating_live_2.ConnectStatus.disconnected);
-        });
-        client.on("live", () => {
-            // 连接到直播间
-            this.emitConnention(floating_live_2.ConnectStatus.entered);
-        });
-        client.on("msg", (data) => {
-            const m = (0, parser_1.parseMessage)(data, this);
-            m && this.emitMessage(m);
-            this.emitRaw(data);
-        });
-        this.client = client;
-    }
-    /** 关闭直播间监听 */
-    close() {
-        if (!this.opened)
-            return;
-        this.opened = false;
-        this.client?.close();
-        this.emitConnention(floating_live_2.ConnectStatus.off);
-        this.emit("close");
-    }
-}
-exports.RoomBilibili = RoomBilibili;
-exports.default = RoomBilibili;
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var _RoomBilibili_tokens, _RoomBilibili_apiClient;
+import { LiveWS, parseLiveConfig } from "bilibili-live-danmaku";
+import { parseGetInfoByRoom, parseMessage, parseRoomBaseInfo } from "./parser";
+import { LiveConnectionStatus, LiveRoom, LiveRoomOpenStatus, } from "floating-live";
+import { BilibiliApiClientEx } from "./utils/request";
+const symbolAbortController = Symbol("abortController");
+export class RoomBilibili extends LiveRoom {
+    get userAgent() {
+        return __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").userAgent;
+    }
+    set userAgent(ua) {
+        __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").userAgent = ua;
+    }
+    constructor(
+    /** 房间id */
+    id, options = {}) {
+        super();
+        /** 平台id */
+        this.platform = "bilibili";
+        /** 直播间room_id */
+        this.roomId = 0;
+        /** 直播间短房号 */
+        this.shortId = 0;
+        /** 房间信息 */
+        this.detail = {
+            /** 直播标题 */
+            title: "",
+            /** 分区 */
+            area: [],
+            /** 封面url */
+            cover: "",
+        };
+        this.stats = {
+            /** 观看数 */
+            view: 0,
+            /** 点赞数 */
+            like: 0,
+        };
+        /** 是否连接上房间 */
+        this.connected = false;
+        /** 主播信息 */
+        this.anchor = { name: "", id: 0 };
+        /** 直播间弹幕api客户端 */
+        this.client = null;
+        /** 连接凭证 */
+        _RoomBilibili_tokens.set(this, void 0);
+        /** 用户cookie */
+        _RoomBilibili_apiClient.set(this, void 0);
+        /** 是否处于重连状态 */
+        this.reconnecting = false;
+        /** 最后一次连接时间 */
+        this.lastConnectionTime = 0;
+        this.id = id;
+        this.connectInterval = options.connectInterval ?? 500;
+        this.connectTimeout = options.connectTimeout ?? 45000;
+        this.autoReconnect = options.autoReconnect ?? false;
+        __classPrivateFieldSet(this, _RoomBilibili_tokens, options.tokens, "f");
+        __classPrivateFieldSet(this, _RoomBilibili_apiClient, new BilibiliApiClientEx({
+            userAgent: options.userAgent,
+            cookie: options.credentials,
+            fetch: options.fetch,
+        }), "f");
+        options.data && this.setData(options.data);
+        const init = options.init ?? true;
+        if (init) {
+            this.whenInit = this.init(options).then(() => this);
+        }
+        else {
+            this.whenInit = new Promise((res) => res(this));
+        }
+    }
+    /** 初始化
+     * 自动获取并设置 `credentials` ，自动获取房间信息和连接 `token`
+     */
+    async init(options = {}) {
+        const { open, data, tokens } = options;
+        if (open)
+            this.openStatus = 1;
+        // 设置必要 cookie
+        // 如果cookie中没有buvid3字段，会自动获取一个
+        await this.setCredentials(__classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").cookie, false);
+        // 自动更新房间信息
+        if (!data)
+            await this.update();
+        // 自动更新房间 tokens
+        if (!tokens)
+            await this.updateTokens();
+        // 自动开启
+        if (open) {
+            this.openWebsocketClient();
+        }
+        this.emit("init");
+    }
+    /** 更新房间信息 */
+    async update() {
+        const data = await this.fetchData();
+        this.roomId = data.id;
+        this.setData(data);
+        return data;
+    }
+    /** 获取房间信息 */
+    async fetchData() {
+        if (this.customFetchData)
+            return this.customFetchData(this.id);
+        try {
+            const res = await __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").xliveGetInfoByRoom({
+                room_id: this.id,
+            });
+            return parseGetInfoByRoom(res.data);
+        }
+        catch (e) {
+            // fallback
+            const res = await __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").xliveGetRoomBaseInfo({
+                room_ids: [this.id],
+            });
+            for (const r in res.data.by_room_ids) {
+                return parseRoomBaseInfo(res.data.by_room_ids[r]);
+            }
+            throw new Error("房间不存在");
+        }
+    }
+    /** 设置直播间数据 */
+    setData({ roomId, shortId, status, timestamp, detail, stats, anchor, liveId, available, }) {
+        this.roomId = roomId;
+        this.shortId = shortId;
+        this.status = status;
+        this.timestamp = timestamp;
+        this.detail = detail;
+        this.stats = stats;
+        this.anchor = anchor;
+        this.liveId = liveId;
+        this.available = available;
+        const data = this.toData();
+        this.emit("update", { room: data });
+        return data;
+    }
+    /** 开启直播间监听 */
+    async open() {
+        // 如果直播间监听已打开或处于正在打开状态，则返回
+        if (this.openStatus)
+            return;
+        this.openStatus = LiveRoomOpenStatus.opening;
+        await this.update();
+        this.openWebsocketClient();
+    }
+    /** 设置登录凭据 */
+    async setCredentials(credentials, updateTokens = true) {
+        __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").setCookie(credentials);
+        // 在没有buvid3的情况下，访问bilibili主页，获得buvid3
+        if (!__classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").cookies.has("buvid3")) {
+            __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").wwwBilibili();
+        }
+        if (updateTokens)
+            this.updateTokens();
+    }
+    /** 更新连接tokens */
+    async updateTokens() {
+        this.setTokens(await this.fetchTokens());
+    }
+    /** 请求连接tokens */
+    async fetchTokens() {
+        if (this.customFetchTokens) {
+            return this.customFetchTokens(this.roomId, __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").cookie);
+        }
+        const buvid = __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").cookies.get("buvid3");
+        const uid = parseInt(__classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").cookies.get("DedeUserID"));
+        const { key } = await parseLiveConfig((await __classPrivateFieldGet(this, _RoomBilibili_apiClient, "f").xliveGetDanmuInfo({ id: this.roomId })).data);
+        return {
+            uid,
+            buvid,
+            key,
+        };
+    }
+    /** 设置直播连接tokens */
+    setTokens(tokens) {
+        __classPrivateFieldSet(this, _RoomBilibili_tokens, tokens, "f");
+        if (this.opened) {
+            // 如果已打开，则重连
+            this.reconnect();
+        }
+    }
+    /** 无缝重连 */
+    reconnect() {
+        // 如果正在重连，则返回
+        if (this.reconnecting)
+            return;
+        this.reconnecting = true;
+        const lastClient = this.client;
+        this.initClient();
+        // 重连成功后，关闭上一个连接
+        this.client?.addEventListener("open", () => {
+            lastClient?.[symbolAbortController].abort();
+            lastClient?.close();
+            this.reconnecting = false;
+        }, { once: true });
+    }
+    emitConnention(status) {
+        this.connectionStatus = status;
+        switch (status) {
+            case LiveConnectionStatus.connecting:
+                this.emit("connecting");
+                break;
+            case LiveConnectionStatus.connected:
+                this.emit("connected");
+                break;
+            case LiveConnectionStatus.entered:
+                this.emit("enter");
+                break;
+            case LiveConnectionStatus.disconnected:
+                this.emit("disconnect");
+                break;
+        }
+    }
+    /** 打开直播弹幕客户端 */
+    openWebsocketClient() {
+        // 如果直播间不可用，则返回
+        if (!this.available) {
+            this.openStatus = LiveRoomOpenStatus.closed;
+            return;
+        }
+        this.initClient();
+        this.openStatus = LiveRoomOpenStatus.opened;
+        this.emit("open");
+    }
+    /** 创建直播弹幕客户端 */
+    initClient() {
+        this.emitConnention(LiveConnectionStatus.connecting);
+        // 与Websocket服务器连接
+        const client = new LiveWS(this.roomId || this.id, {
+            uid: __classPrivateFieldGet(this, _RoomBilibili_tokens, "f")?.uid,
+            buvid: __classPrivateFieldGet(this, _RoomBilibili_tokens, "f")?.buvid,
+            key: __classPrivateFieldGet(this, _RoomBilibili_tokens, "f")?.key,
+        });
+        console.log(__classPrivateFieldGet(this, _RoomBilibili_tokens, "f"));
+        const controller = new AbortController();
+        const signal = controller.signal;
+        client[symbolAbortController] = controller;
+        this.lastConnectionTime = Date.now();
+        console.log(client.ws.url);
+        client.addEventListener("open", () => {
+            // 连接到服务器
+            this.opened && this.emitConnention(LiveConnectionStatus.connected);
+        }, { signal });
+        client.addEventListener("close", () => {
+            // 断开连接
+            this.opened && this.emitConnention(LiveConnectionStatus.disconnected);
+            // 自动重连
+            if (this.opened && this.autoReconnect) {
+                const timeout = this.lastConnectionTime + this.connectInterval - Date.now();
+                if (timeout > 0) {
+                    const t = setTimeout(() => {
+                        this.reconnect();
+                        clearTimeout(t);
+                    }, timeout);
+                }
+                else {
+                    this.reconnect();
+                }
+            }
+        }, { signal });
+        client.addEventListener("CONNECT_SUCCESS", () => {
+            // 连接到直播间
+            this.emitConnention(LiveConnectionStatus.entered);
+        }, { signal });
+        client.addEventListener("MESSAGE", ({ data }) => {
+            const m = parseMessage(data, this);
+            m && this.emitMessage(m);
+            this.emitRaw(data);
+        }, { signal });
+        this.client = client;
+    }
+    /** 关闭直播间监听 */
+    close() {
+        if (!this.opened)
+            return;
+        this.openStatus = LiveRoomOpenStatus.closed;
+        this.client?.close();
+        this.emitConnention(LiveConnectionStatus.off);
+        this.emit("close");
+    }
+}
+_RoomBilibili_tokens = new WeakMap(), _RoomBilibili_apiClient = new WeakMap();
+export default RoomBilibili;