import { exec, execSync } from "child_process";

/**
 * 构建梯子参数
 * @param isOversea
 * @param proxy
 * @returns {string|string}
 */
function constructProxyParam(isOversea, proxy) {
    return isOversea ? "" : `--proxy ${proxy}`;
}


/**
 * 获取时长
 * @param url
 * @param isOversea
 * @param proxy
 * @returns string
 */
export function ytDlpGetDuration(url, isOversea, proxy) {
    return execSync(`yt-dlp --get-duration --skip-download ${constructProxyParam(isOversea, proxy)} ${url}`);
}

/**
 * 获取标题
 * @param url
 * @param isOversea
 * @param proxy
 * @returns string
 */
export function ytDlpGetTilt(url, isOversea, proxy) {
    return execSync(`yt-dlp --get-title --skip-download ${constructProxyParam(isOversea, proxy)} ${url} --encoding utf8`);
}

/**
 * 获取封面
 * @param path
 * @param url
 * @param isOversea
 * @param proxy
 */
export function ytDlpGetThumbnail(path, url, isOversea, proxy) {
    return execSync(`yt-dlp --write-thumbnail --convert-thumbnails png --skip-download ${constructProxyParam(isOversea, proxy)} ${url} -P ${path} -o "thumbnail.%(ext)s"`);
}

/**
 * yt-dlp 工具类
 * @returns {Promise<void>}
 * @param path       下载路径
 * @param url        下载链接
 * @param isOversea  是否是海外用户
 * @param proxy      代理地址
 * @param merge      是否合并输出为 mp4 格式 (仅适用于视频合并需求)
 * @param graphics   YouTube画质参数
 * @param timeRange  截取时间段
 * @param maxThreads  最大线程数
 */
export async function ytDlpHelper(path, url, isOversea, proxy, merge = false, graphics, timeRange, maxThreads) {
    return new Promise((resolve, reject) => {
        const mergeOption = merge ? '--merge-output-format "mp4"' : '';

        const fParam = url.includes("youtu") ? `--download-sections "*${timeRange}" -f "bv${graphics}[ext=mp4]+ba[ext=m4a]" --concurrent-fragments ${maxThreads} ` : "";

        const command = `yt-dlp ${fParam} ${constructProxyParam(isOversea, proxy)} -P ${path} -o "temp.%(ext)s" ${url}`;

        logger.info(`[R插件][yt-dlp审计] ${command}`)

        exec(command, (error, stdout) => {
            if (error) {
                logger.error(`[R插件][yt-dlp审计] 执行命令时出错: ${error}`);
                reject(error);
            } else {
                resolve(stdout);
            }
        });
    });
}