Function downloadStream

src/engine/message.ts:26–139

(stream: HuyaStreamInfo)

const logger = log4js.getLogger("message");
export const downloadStream = (stream: HuyaStreamInfo) => {
  // 每段视频持续时间，单位s
  const partDuration = "1800"
  // let huyaRoomId = getRoomArrInfo(infoJson.streamerInfo)[0].roomLink;
  // let huyaRoomTitle = getRoomArrInfo(infoJson.streamerInfo)[0].roomTitle;
  // console.log("开始下载 ", stream.streamName);
  logger.info("开始下载 ", stream.streamName)
  // const cmd = `ffmpeg -i "${streamUrl}" -f mp4 res.MP4`;
  // 命名加上时间戳
  // console.log("streamUrl", stream.streamUrl);
  const timeV = dayjs().format("YYYY-MM-DD");
  const cmd = `ffmpeg`;
  // console.log("11", huyaRoomId + `${timeV}res.MP4`);
  let savePath = join(rootPath, "/download")
  let startNumber = 0
  if (!fs.existsSync(savePath)) {
    fs.mkdirSync(savePath)
  }
  let dirName = join(savePath, stream.streamName)
  if (!fs.existsSync(dirName)) {
    fs.mkdirSync(dirName)
  }
  dirName = join(dirName, timeV)
  if (!fs.existsSync(dirName)) {
    fs.mkdirSync(dirName)
  } else { //文件夹存在，说明有视频未上传，接着前面的序号下载
    let ps = fs.readdirSync(dirName);
    startNumber = ps.length
  }
  const fileName: string = join(dirName, `${stream.streamName}-${timeV}-part-%03d.mp4`);
  //伪装了请求头，避免服务器返回403
  const fakeX: any = {
    'Accept': '*/*',
    'Accept-Encoding': 'gzip, deflate, br',
    'Accept-Language': 'zh,zh-TW;q=0.9,en-US;q=0.8,en;q=0.7,zh-CN;q=0.6,ru;q=0.5',
    'Origin': 'https://www.huya.com',
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.106 Safari/537.36'
  }
  let fakeHeaders = ""
  for (let key of Object.keys(fakeX)) {
    fakeHeaders = `${fakeHeaders}${key}: ${fakeX[key]}\r\n`
  }
  const huyaApp = spawn(cmd, [
    "-headers",
    fakeHeaders,
    "-i",
    stream.streamUrl,
    "-c:v",
    "copy",
    "-c:a",
    "copy",
    "-f",
    "segment",
    "-segment_time",
    partDuration,
    "-segment_start_number",
    startNumber.toString(),
    fileName,