scripts/regenerate-thumbnails.ts

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