scripts/regenerate-thumbnails.ts

   1 import { registerTSPaths } from '../server/helpers/register-ts-paths'
   2 registerTSPaths()
   3
   4 import * as Bluebird from 'bluebird'
   5 import { program } from 'commander'
   6 import { pathExists, remove } from 'fs-extra'
   7 import { generateImageFilename, processImage } from '@server/helpers/image-utils'
   8 import { THUMBNAILS_SIZE } from '@server/initializers/constants'
   9 import { VideoModel } from '@server/models/video/video'
  10 import { MVideo } from '@server/types/models'
  11 import { initDatabaseModels } from '@server/initializers/database'
  12
  13 program
  14   .description('Regenerate local thumbnails using preview files')
  15   .parse(process.argv)
  16
  17 run()
  18   .then(() => process.exit(0))
  19   .catch(err => console.error(err))
  20
  21 async function run () {
  22   await initDatabaseModels(true)
  23
  24   const videos = await VideoModel.listLocal()
  25
  26   await Bluebird.map(videos, v => {
  27     return processVideo(v)
  28       .catch(err => console.error('Cannot process video %s.', v.url, err))
  29   }, { concurrency: 20 })
  30 }
  31
  32 async function processVideo (videoArg: MVideo) {
  33   const video = await VideoModel.loadWithFiles(videoArg.id)
  34
  35   console.log('Processing video %s.', video.name)
  36
  37   const thumbnail = video.getMiniature()
  38   const preview = video.getPreview()
  39
  40   const previewPath = preview.getPath()
  41
  42   if (!await pathExists(previewPath)) {
  43     throw new Error(`Preview ${previewPath} does not exist on disk`)
  44   }
  45
  46   const size = {
  47     width: THUMBNAILS_SIZE.width,
  48     height: THUMBNAILS_SIZE.height
  49   }
  50
  51   const oldPath = thumbnail.getPath()
  52
  53   // Update thumbnail
  54   thumbnail.filename = generateImageFilename()
  55   thumbnail.width = size.width
  56   thumbnail.height = size.height
  57
  58   const thumbnailPath = thumbnail.getPath()
  59   await processImage(previewPath, thumbnailPath, size, true)
  60
  61   // Save new attributes
  62   await thumbnail.save()
  63
  64   // Remove old thumbnail
  65   await remove(oldPath)
  66
  67   // Don't federate, remote instances will refresh the thumbnails after a while
  68 }