@bililive-tools/douyu-recorder/lib/stream.js

bililive-tools douyu recorder implemention

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);
    const recordStartTime = new Date();
    return {
        living,
        owner: data.room.nickname,
        title: data.room.room_name,
        avatar: data.room.avatar.big,
        cover: data.room.room_pic,
        liveStartTime: startTime,
        liveId: utils.md5(`${channelId}-${startTime?.getTime() ?? Date.now()}`),
        recordStartTime: recordStartTime,
        // 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 cdn = opts.source === "auto" ? undefined : opts.source;
    if (opts.source === "auto" && opts.avoidEdgeCDN) {
        cdn = "hw-h5";
    }
    let liveInfo = await getLiveInfo({
        channelId: opts.channelId,
        rate: qn,
        cdn,
        onlyAudio: opts.onlyAudio,
    });
    if (!liveInfo.living)
        throw new Error("It must be called getStream when living");
    //如果是scdn，那么找到第一个非scdn的源，重新请求一次
    if (liveInfo.currentStream.source === "scdn") {
        const nonScdnSource = liveInfo.sources.find((source) => source.cdn !== "scdnctshh");
        if (nonScdnSource) {
            liveInfo = await getLiveInfo({
                channelId: opts.channelId,
                rate: qn,
                cdn: nonScdnSource?.cdn,
                onlyAudio: opts.onlyAudio,
            });
        }
    }
    if (!liveInfo.living)
        throw new Error("It must be called getStream when living");
    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,
                onlyAudio: opts.onlyAudio,
            });
            if (!liveInfo.living)
                throw new Error("It must be called getStream when living");
        }
    }
    return liveInfo;
}