rc-plugin/utils/yt-dlp-util.js
2024-10-17 19:33:01 +08:00

78 lines
2.2 KiB
JavaScript

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 截取时间段
*/
export async function ytDlpHelper(path, url, isOversea, proxy, merge = false, graphics, timeRange) {
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]"` : "";
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);
}
});
});
}