rc-plugin/utils/yt-dlp-util.js

import { exec } from 'child_process';

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

/**
 * 构造cookie参数
 * 目前只支持YouTube构造cookie，否则就必须修改`url.includes("youtu")`
 * @param url
 * @param cookiePath
 * @returns {string}
 */
function constructCookiePath(url, cookiePath) {
  return cookiePath !== '' && url.includes('youtu') ? `--cookies ${cookiePath}` : '';
}

/**
 * yt-dlp获取标题的时候可能需要的一个编码参数，也在一定程度上解决部分window系统乱码问题
 * @param url
 * @returns {string}
 */
function constructEncodingParam(url) {
  return '--encoding UTF-8'; // 始终为标题获取使用 UTF-8 编码
}

/**
 * 获取时长
 * @param url
 * @param isOversea
 * @param proxy
 * @param cookiePath
 * @returns string
 */
export function ytDlpGetDuration(url, isOversea, proxy, cookiePath = '') {
  return new Promise((resolve, reject) => {
    // 构造 cookie 参数
    const cookieParam = constructCookiePath(url, cookiePath);
    const command = `yt-dlp --get-duration --skip-download ${cookieParam} ${constructProxyParam(isOversea, proxy)} "${url}"`;
    exec(command, (error, stdout, stderr) => {
      if (error) {
        logger.error(
          `[R插件][yt-dlp审计] Error executing ytDlpGetDuration: ${error}. Stderr: ${stderr}`
        );
        reject(error);
      } else {
        resolve(stdout.trim());
      }
    });
  });
}

/**
 * 获取标题
 * @param url
 * @param isOversea
 * @param proxy
 * @param cookiePath
 * @returns string
 */
export async function ytDlpGetTilt(url, isOversea, proxy, cookiePath = '') {
  return new Promise((resolve, reject) => {
    // 构造 cookie 参数
    const cookieParam = constructCookiePath(url, cookiePath);
    // 构造 编码 参数
    const encodingParam = constructEncodingParam(url);
    const command = `yt-dlp --get-title --skip-download ${cookieParam} ${constructProxyParam(isOversea, proxy)} "${url}" ${encodingParam}`;
    exec(command, (error, stdout, stderr) => {
      if (error) {
        logger.error(
          `[R插件][yt-dlp审计] Error executing ytDlpGetTilt: ${error}. Stderr: ${stderr}`
        );
        reject(error);
      } else {
        resolve(stdout.trim());
      }
    });
  });
}

/**
 * 获取封面
 * @param path
 * @param url
 * @param isOversea
 * @param proxy
 * @param cookiePath
 * @param thumbnailFilenamePrefix 缩略图文件名前缀 (不含扩展名)
 */
export function ytDlpGetThumbnail(
  path,
  url,
  isOversea,
  proxy,
  cookiePath = '',
  thumbnailFilenamePrefix = 'thumbnail'
) {
  return new Promise((resolve, reject) => {
    const cookieParam = constructCookiePath(url, cookiePath);
    const finalThumbnailName = thumbnailFilenamePrefix || 'thumbnail';
    const command = `yt-dlp --write-thumbnail --convert-thumbnails png --skip-download ${cookieParam} ${constructProxyParam(isOversea, proxy)} "${url}" -P "${path}" -o "${finalThumbnailName}.%(ext)s"`;

    exec(command, (error, stdout, stderr) => {
      if (error) {
        logger.error(
          `[R插件][yt-dlp审计] Error executing ytDlpGetThumbnail: ${error}. Stderr: ${stderr}`
        );
        return reject(error);
      }
      // 从yt-dlp的输出中提取文件名
      const match = stdout.match(/Writing thumbnail to: (.*)/);
      if (match && match[1]) {
        const thumbnailPath = match[1].trim();
        // 只返回文件名部分
        const thumbnailFilename = thumbnailPath.split(/[\\/]/).pop();
        logger.info(`[R插件][yt-dlp审计] Thumbnail downloaded: ${thumbnailFilename}`);
        resolve(thumbnailFilename);
      } else {
        // 兜底方案：如果无法从输出中解析，则按原逻辑拼接
        logger.warn(
          '[R插件][yt-dlp审计] Could not parse thumbnail filename from stdout. Falling back to default.'
        );
        // 尝试查找文件，因为yt-dlp可能没有输出我们期望的格式
        const expectedPngPath = `${finalThumbnailName}.png`;
        resolve(expectedPngPath);
      }
    });
  });
}

/**
 * yt-dlp 工具类
 * @returns {Promise<void>}
 * @param path       下载路径
 * @param url        下载链接
 * @param isOversea  是否是海外用户
 * @param proxy      代理地址
 * @param merge      是否合并输出为 mp4 格式 (仅适用于视频合并需求)
 * @param graphics   YouTube画质参数
 * @param timeRange  截取时间段
 * @param maxThreads 最大并发
 * @param outputFilename 输出文件名 (不含扩展名)
 * @param cookiePath Cookie所在位置
 */
export async function ytDlpHelper(
  path,
  url,
  isOversea,
  proxy,
  maxThreads,
  outputFilename,
  merge = false,
  graphics,
  timeRange,
  cookiePath = ''
) {
  return new Promise((resolve, reject) => {
    let command = '';
    // 构造 cookie 参数
    const cookieParam = constructCookiePath(url, cookiePath);
    // 确保 outputFilename 不为空，提供一个默认值以防万一
    const finalOutputFilename = outputFilename || 'temp_download';

    if (url.includes('music')) {
      // 这里是 YouTube Music的处理逻辑
      // e.g yt-dlp -x --audio-format mp3 https://youtu.be/5wEtefq9VzM -o test.mp3
      command = `yt-dlp -x --audio-format flac -f ba ${cookieParam} ${constructProxyParam(isOversea, proxy)} -P "${path}" -o "${finalOutputFilename}.flac" "${url}"`;
    } else {
      // 正常情况下的处理逻辑
      const fParam = url.includes('youtu')
        ? `--download-sections "*${timeRange}" -f "bv${graphics}[ext=mp4]+ba[ext=m4a]" `
        : '';

      command = `yt-dlp -N ${maxThreads} ${fParam} --concurrent-fragments ${maxThreads} ${cookieParam} ${constructProxyParam(isOversea, proxy)} -P "${path}" -o "${finalOutputFilename}.%(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);
      }
    });
  });
}