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/ffprobe-utils'
   6 import { getMaxBitrate } from '../shared/models/videos'
   7 import { VideoModel } from '../server/models/video/video'
   8 import { optimizeOriginalVideofile } from '../server/lib/transcoding/video-transcoding'
   9 import { initDatabaseModels } from '../server/initializers/database'
  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 localVideo of localVideos) {
  38     const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(localVideo.id)
  39
  40     currentVideoId = video.id
  41
  42     for (const file of video.VideoFiles) {
  43       currentFile = getVideoFilePath(video, file)
  44
  45       const [ videoBitrate, fps, resolution ] = await Promise.all([
  46         getVideoFileBitrate(currentFile),
  47         getVideoFileFPS(currentFile),
  48         getVideoFileResolution(currentFile)
  49       ])
  50
  51       const maxBitrate = getMaxBitrate(resolution.videoFileResolution, fps, VIDEO_TRANSCODING_FPS)
  52       const isMaxBitrateExceeded = videoBitrate > maxBitrate
  53       if (isMaxBitrateExceeded) {
  54         console.log(
  55           'Optimizing video file %s with bitrate %s kbps (max: %s kbps)',
  56           basename(currentFile), videoBitrate / 1000, maxBitrate / 1000
  57         )
  58
  59         const backupFile = `${currentFile}_backup`
  60         await copy(currentFile, backupFile)
  61
  62         await optimizeOriginalVideofile(video, file)
  63
  64         const originalDuration = await getDurationFromVideoFile(backupFile)
  65         const newDuration = await getDurationFromVideoFile(currentFile)
  66
  67         if (originalDuration === newDuration) {
  68           console.log('Finished optimizing %s', basename(currentFile))
  69           await remove(backupFile)
  70           continue
  71         }
  72
  73         console.log('Failed to optimize %s, restoring original', basename(currentFile))
  74         await move(backupFile, currentFile, { overwrite: true })
  75         await createTorrentAndSetInfoHash(video, file)
  76         await file.save()
  77       }
  78     }
  79   }
  80
  81   console.log('Finished optimizing videos')
  82 }