scripts/optimize-old-videos.ts

   1 import { registerTSPaths } from '../server/helpers/register-ts-paths'
   2 registerTSPaths()
   3
   4 import { VIDEO_TRANSCODING_FPS } from '../server/initializers/constants'
   5 import { getDurationFromVideoFile, getVideoFileBitrate, getVideoFileFPS, getVideoFileResolution } from '../server/helpers/ffmpeg-utils'
   6 import { getMaxBitrate } from '../shared/models/videos'
   7 import { VideoModel } from '../server/models/video/video'
   8 import { optimizeOriginalVideofile } from '../server/lib/video-transcoding'
   9 import { initDatabaseModels } from '../server/initializers'
  10 import { basename, dirname } from 'path'
  11 import { copy, move, remove } from 'fs-extra'
  12 import { createTorrentAndSetInfoHash } from '@server/helpers/webtorrent'
  13 import { getVideoFilePath } from '@server/lib/video-paths'
  14
  15 run()
  16   .then(() => process.exit(0))
  17   .catch(err => {
  18     console.error(err)
  19     process.exit(-1)
  20   })
  21
  22 let currentVideoId = null
  23 let currentFile = null
  24
  25 process.on('SIGINT', async function () {
  26   console.log('Cleaning up temp files')
  27   await remove(`${currentFile}_backup`)
  28   await remove(`${dirname(currentFile)}/${currentVideoId}-transcoded.mp4`)
  29   process.exit(0)
  30 })
  31
  32 async function run () {
  33   await initDatabaseModels(true)
  34
  35   const localVideos = await VideoModel.listLocal()
  36
  37   for (const video of localVideos) {
  38     currentVideoId = video.id
  39
  40     for (const file of video.VideoFiles) {
  41       currentFile = getVideoFilePath(video, file)
  42
  43       const [ videoBitrate, fps, resolution ] = await Promise.all([
  44         getVideoFileBitrate(currentFile),
  45         getVideoFileFPS(currentFile),
  46         getVideoFileResolution(currentFile)
  47       ])
  48
  49       const maxBitrate = getMaxBitrate(resolution.videoFileResolution, fps, VIDEO_TRANSCODING_FPS)
  50       const isMaxBitrateExceeded = videoBitrate > maxBitrate
  51       if (isMaxBitrateExceeded) {
  52         console.log(
  53           'Optimizing video file %s with bitrate %s kbps (max: %s kbps)',
  54           basename(currentFile), videoBitrate / 1000, maxBitrate / 1000
  55         )
  56
  57         const backupFile = `${currentFile}_backup`
  58         await copy(currentFile, backupFile)
  59
  60         await optimizeOriginalVideofile(video, file)
  61
  62         const originalDuration = await getDurationFromVideoFile(backupFile)
  63         const newDuration = await getDurationFromVideoFile(currentFile)
  64
  65         if (originalDuration === newDuration) {
  66           console.log('Finished optimizing %s', basename(currentFile))
  67           await remove(backupFile)
  68           continue
  69         }
  70
  71         console.log('Failed to optimize %s, restoring original', basename(currentFile))
  72         await move(backupFile, currentFile, { overwrite: true })
  73         await createTorrentAndSetInfoHash(video, file)
  74         await file.save()
  75       }
  76     }
  77   }
  78
  79   console.log('Finished optimizing videos')
  80 }