@bililive-tools/huya-recorder/lib/utils.js

bililive-tools huya recorder implemention

import fs from "node:fs";
import path from "node:path";
import { range } from "lodash-es";
/**
 * 从数组中按照特定算法提取一些值（允许同个索引重复提取）。
 * 算法的行为类似 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 function getValuesFromArrayLikeFlexSpaceBetween(array, columnCount) {
    if (columnCount < 1)
        return [];
    if (columnCount === 1)
        return [array[0]];
    const spacingCount = columnCount - 1;
    const spacingLength = array.length / spacingCount;
    const columns = range(1, columnCount + 1);
    const columnValues = columns.map((column, idx, columns) => {
        // 首个和最后的列是特殊的，因为它们不在范围内，而是在两端
        if (idx === 0) {
            return array[0];
        }
        else if (idx === columns.length - 1) {
            return array[array.length - 1];
        }
        const beforeSpacingCount = column - 1;
        const colPos = beforeSpacingCount * spacingLength;
        return array[Math.floor(colPos)];
    });
    return columnValues;
}
export function ensureFolderExist(fileOrFolderPath) {
    const folder = path.dirname(fileOrFolderPath);
    if (!fs.existsSync(folder)) {
        fs.mkdirSync(folder, { recursive: true });
    }
}
export function assert(assertion, msg) {
    if (!assertion) {
        throw new Error(msg);
    }
}
export function assertStringType(data, msg) {
    assert(typeof data === "string", msg);
}
export function assertNumberType(data, msg) {
    assert(typeof data === "number", msg);
}
export function assertObjectType(data, msg) {
    assert(typeof data === "object", msg);
}
// 根据formatPriorities获取最终的sources
// 如果formatPriorities为空或者undefined，则使用['flv','hls']
// 如果有参数，按照顺序进行匹配，如果匹配的值不存在或者为空，则使用下一个参数，最后返回的是流数组
export function getFormatSources(sources, formatPriorities = ["flv", "hls"]) {
    for (const format of formatPriorities) {
        if (sources[format] && sources[format].length > 0) {
            return {
                sources: sources[format],
                formatName: format,
            };
        }
    }
    // 如果没有匹配到任何格式,使用默认的flv格式
    return null;
}