server/tests/shared/generate.ts

   1 import { expect } from 'chai'
   2 import ffmpeg from 'fluent-ffmpeg'
   3 import { ensureDir, pathExists } from 'fs-extra'
   4 import { dirname } from 'path'
   5 import { buildAbsoluteFixturePath, getMaxBitrate } from '@shared/core-utils'
   6 import { getVideoStreamBitrate, getVideoStreamFPS, getVideoStreamDimensionsInfo } from '@shared/extra-utils'
   7
   8 async function ensureHasTooBigBitrate (fixturePath: string) {
   9   const bitrate = await getVideoStreamBitrate(fixturePath)
  10   const dataResolution = await getVideoStreamDimensionsInfo(fixturePath)
  11   const fps = await getVideoStreamFPS(fixturePath)
  12
  13   const maxBitrate = getMaxBitrate({ ...dataResolution, fps })
  14   expect(bitrate).to.be.above(maxBitrate)
  15 }
  16
  17 async function generateHighBitrateVideo () {
  18   const tempFixturePath = buildAbsoluteFixturePath('video_high_bitrate_1080p.mp4', true)
  19
  20   await ensureDir(dirname(tempFixturePath))
  21
  22   const exists = await pathExists(tempFixturePath)
  23   if (!exists) {
  24     console.log('Generating high bitrate video.')
  25
  26     // Generate a random, high bitrate video on the fly, so we don't have to include
  27     // a large file in the repo. The video needs to have a certain minimum length so
  28     // that FFmpeg properly applies bitrate limits.
  29     // https://stackoverflow.com/a/15795112
  30     return new Promise<string>((res, rej) => {
  31       ffmpeg()
  32         .outputOptions([ '-f rawvideo', '-video_size 1920x1080', '-i /dev/urandom' ])
  33         .outputOptions([ '-ac 2', '-f s16le', '-i /dev/urandom', '-t 10' ])
  34         .outputOptions([ '-maxrate 10M', '-bufsize 10M' ])
  35         .output(tempFixturePath)
  36         .on('error', rej)
  37         .on('end', () => res(tempFixturePath))
  38         .run()
  39     })
  40   }
  41
  42   await ensureHasTooBigBitrate(tempFixturePath)
  43
  44   return tempFixturePath
  45 }
  46
  47 async function generateVideoWithFramerate (fps = 60) {
  48   const tempFixturePath = buildAbsoluteFixturePath(`video_${fps}fps.mp4`, true)
  49
  50   await ensureDir(dirname(tempFixturePath))
  51
  52   const exists = await pathExists(tempFixturePath)
  53   if (!exists) {
  54     console.log('Generating video with framerate %d.', fps)
  55
  56     return new Promise<string>((res, rej) => {
  57       ffmpeg()
  58         .outputOptions([ '-f rawvideo', '-video_size 1280x720', '-i /dev/urandom' ])
  59         .outputOptions([ '-ac 2', '-f s16le', '-i /dev/urandom', '-t 10' ])
  60         .outputOptions([ `-r ${fps}` ])
  61         .output(tempFixturePath)
  62         .on('error', rej)
  63         .on('end', () => res(tempFixturePath))
  64         .run()
  65     })
  66   }
  67
  68   return tempFixturePath
  69 }
  70
  71 export {
  72   generateHighBitrateVideo,
  73   generateVideoWithFramerate
  74 }