]>
Commit | Line | Data |
---|---|---|
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(previewPath, thumbnailPath, size, 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 | } |