PeerTube/scripts/regenerate-thumbnails.ts

68 lines
1.8 KiB
TypeScript
Raw Normal View History

2021-03-12 17:04:49 +01:00
import { registerTSPaths } from '../server/helpers/register-ts-paths'
registerTSPaths()
2021-08-27 14:32:44 +02:00
import { map } from 'bluebird'
2021-06-25 17:39:27 +02:00
import { program } from 'commander'
import { pathExists, remove } from 'fs-extra'
2021-04-08 11:23:45 +02:00
import { generateImageFilename, processImage } from '@server/helpers/image-utils'
2021-03-12 17:04:49 +01:00
import { THUMBNAILS_SIZE } from '@server/initializers/constants'
import { VideoModel } from '@server/models/video/video'
import { initDatabaseModels } from '@server/initializers/database'
program
.description('Regenerate local thumbnails using preview files')
.parse(process.argv)
run()
.then(() => process.exit(0))
.catch(err => console.error(err))
async function run () {
await initDatabaseModels(true)
const ids = await VideoModel.listLocalIds()
2021-03-12 17:04:49 +01:00
await map(ids, id => {
return processVideo(id)
.catch(err => console.error('Cannot process video %d.', id, err))
2021-03-12 17:04:49 +01:00
}, { concurrency: 20 })
}
async function processVideo (id: number) {
const video = await VideoModel.loadWithFiles(id)
2021-03-12 17:04:49 +01:00
2021-03-29 17:23:48 +02:00
console.log('Processing video %s.', video.name)
2021-03-12 17:04:49 +01:00
const thumbnail = video.getMiniature()
const preview = video.getPreview()
const previewPath = preview.getPath()
if (!await pathExists(previewPath)) {
throw new Error(`Preview ${previewPath} does not exist on disk`)
}
const size = {
width: THUMBNAILS_SIZE.width,
height: THUMBNAILS_SIZE.height
}
const oldPath = thumbnail.getPath()
// Update thumbnail
2021-04-08 11:23:45 +02:00
thumbnail.filename = generateImageFilename()
thumbnail.width = size.width
thumbnail.height = size.height
const thumbnailPath = thumbnail.getPath()
2021-03-12 17:04:49 +01:00
await processImage(previewPath, thumbnailPath, size, true)
// Save new attributes
await thumbnail.save()
// Remove old thumbnail
await remove(oldPath)
// Don't federate, remote instances will refresh the thumbnails after a while
2021-03-12 17:04:49 +01:00
}