rc-plugin/utils/tdl-util.js

import { exec } from 'child_process';
import path from 'path'

/**
 * 执行 TDL 进行下载
 * @param url
 * @param curPath
 * @param isOversea
 * @param proxyAddr
 * @param videoDownloadConcurrency
 * @returns {Promise<string>}
 */
export async function startTDL(url, curPath, isOversea, proxyAddr, videoDownloadConcurrency = 1) {
    return new Promise((resolve, reject) => {
        curPath = path.resolve(curPath);
        const proxyStr = isOversea ? `` : `--proxy ${ proxyAddr }`;
        const concurrencyStr = videoDownloadConcurrency > 1 ? `-t ${ videoDownloadConcurrency } -s 524288 -l ${ videoDownloadConcurrency }` : '';
        const command = `tdl dl -u ${ url } -d ${ curPath } ${ concurrencyStr } ${ proxyStr }`
        logger.mark(`[R插件][TDL] ${ command }`);
        exec(command, (error, stdout, stderr) => {
            if (error) {
                reject(`[R插件][TDL]执行出错: ${ error.message }`);
                return;
            }
            if (stderr) {
                reject(`[R插件][TDL]错误信息: ${ stderr }`);
                return;
            }
            resolve(stdout);
        })
    })
}

/**
 * 保存小飞机内容到小飞机的收藏
 * @param url
 * @param isOversea
 * @param proxyAddr
 * @returns {Promise<unknown>}
 */
export async function saveTDL(url, isOversea, proxyAddr) {
    return new Promise((resolve, reject) => {
        const proxyStr = isOversea ? `` : `--proxy ${ proxyAddr }`;
        const command = `tdl forward --from ${ url } ${ proxyStr }`
        exec(command, (error, stdout, stderr) => {
            if (error) {
                reject(`[R插件][TDL保存]执行出错: ${ error.message }`);
                return;
            }
            if (stderr) {
                reject(`[R插件][TDL保存]错误信息: ${ stderr }`);
                return;
            }
            resolve(stdout);
        })
    })
}

/**
 * 上传文件到飞机收藏夹
 * @param filePath
 * @param isOversea
 * @param proxyAddr
 * @returns {Promise<void>}
 */
export async function uploadTDL(filePath, isOversea, proxyAddr) {
    return new Promise((resolve, reject) => {
        const proxyStr = isOversea ? `` : `--proxy ${ proxyAddr }`;
        const command = `tdl up -p ${ filePath } ${ proxyStr }`;
        exec(command, (error, stdout, stderr) => {
            if (error) {
                reject(`[R插件][TDL上传]执行出错: ${ error.message }`);
                return;
            }
            if (stderr) {
                reject(`[R插件][TDL上传]错误信息: ${ stderr }`);
                return;
            }
            resolve(stdout);
        })
    })
}