import { registerTSPaths } from '../server/helpers/register-ts-paths' registerTSPaths() import { VIDEO_TRANSCODING_FPS } from '../server/initializers/constants' import { getDurationFromVideoFile, getVideoFileBitrate, getVideoFileFPS, getVideoFileResolution } from '../server/helpers/ffmpeg-utils' import { getMaxBitrate } from '../shared/models/videos' import { VideoModel } from '../server/models/video/video' import { optimizeOriginalVideofile } from '../server/lib/video-transcoding' import { initDatabaseModels } from '../server/initializers' import { basename, dirname } from 'path' import { copy, move, remove } from 'fs-extra' import { createTorrentAndSetInfoHash } from '@server/helpers/webtorrent' import { getVideoFilePath } from '@server/lib/video-paths' run() .then(() => process.exit(0)) .catch(err => { console.error(err) process.exit(-1) }) let currentVideoId = null let currentFile = null process.on('SIGINT', async function () { console.log('Cleaning up temp files') await remove(`${currentFile}_backup`) await remove(`${dirname(currentFile)}/${currentVideoId}-transcoded.mp4`) process.exit(0) }) async function run () { await initDatabaseModels(true) const localVideos = await VideoModel.listLocal() for (const video of localVideos) { currentVideoId = video.id for (const file of video.VideoFiles) { currentFile = getVideoFilePath(video, file) const [ videoBitrate, fps, resolution ] = await Promise.all([ getVideoFileBitrate(currentFile), getVideoFileFPS(currentFile), getVideoFileResolution(currentFile) ]) const maxBitrate = getMaxBitrate(resolution.videoFileResolution, fps, VIDEO_TRANSCODING_FPS) const isMaxBitrateExceeded = videoBitrate > maxBitrate if (isMaxBitrateExceeded) { console.log( 'Optimizing video file %s with bitrate %s kbps (max: %s kbps)', basename(currentFile), videoBitrate / 1000, maxBitrate / 1000 ) const backupFile = `${currentFile}_backup` await copy(currentFile, backupFile) await optimizeOriginalVideofile(video, file) const originalDuration = await getDurationFromVideoFile(backupFile) const newDuration = await getDurationFromVideoFile(currentFile) if (originalDuration === newDuration) { console.log('Finished optimizing %s', basename(currentFile)) await remove(backupFile) continue } console.log('Failed to optimize %s, restoring original', basename(currentFile)) await move(backupFile, currentFile, { overwrite: true }) await createTorrentAndSetInfoHash(video, file) await file.save() } } } console.log('Finished optimizing videos') }