]>
Commit | Line | Data |
---|---|---|
1 | import * as Sequelize from 'sequelize' | |
2 | import { join } from 'path' | |
3 | import { CONFIG } from '../../initializers/constants' | |
4 | import { getVideoFileResolution } from '../../helpers/ffmpeg-utils' | |
5 | import { readdir, rename } from 'fs-extra' | |
6 | ||
7 | function up (utils: { | |
8 | transaction: Sequelize.Transaction, | |
9 | queryInterface: Sequelize.QueryInterface, | |
10 | sequelize: Sequelize.Sequelize, | |
11 | db: any | |
12 | }): Promise<void> { | |
13 | const torrentDir = CONFIG.STORAGE.TORRENTS_DIR | |
14 | const videoFileDir = CONFIG.STORAGE.VIDEOS_DIR | |
15 | ||
16 | return readdir(videoFileDir) | |
17 | .then(videoFiles => { | |
18 | const tasks: Promise<any>[] = [] | |
19 | for (const videoFile of videoFiles) { | |
20 | const matches = /^([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12})\.([a-z0-9]+)/.exec(videoFile) | |
21 | if (matches === null) { | |
22 | console.log('Invalid video file name %s.', videoFile) | |
23 | continue | |
24 | } | |
25 | ||
26 | const uuid = matches[1] | |
27 | const ext = matches[2] | |
28 | ||
29 | const p = getVideoFileResolution(join(videoFileDir, videoFile)) | |
30 | .then(height => { | |
31 | const oldTorrentName = uuid + '.torrent' | |
32 | const newTorrentName = uuid + '-' + height + '.torrent' | |
33 | return rename(join(torrentDir, oldTorrentName), join(torrentDir, newTorrentName)).then(() => height) | |
34 | }) | |
35 | .then(height => { | |
36 | const newVideoFileName = uuid + '-' + height + '.' + ext | |
37 | return rename(join(videoFileDir, videoFile), join(videoFileDir, newVideoFileName)).then(() => height) | |
38 | }) | |
39 | .then(height => { | |
40 | const query = 'UPDATE "VideoFiles" SET "resolution" = ' + height + | |
41 | ' WHERE "videoId" = (SELECT "id" FROM "Videos" WHERE "uuid" = \'' + uuid + '\')' | |
42 | return utils.sequelize.query(query) | |
43 | }) | |
44 | ||
45 | tasks.push(p) | |
46 | } | |
47 | ||
48 | return Promise.all(tasks).then(() => undefined) | |
49 | }) | |
50 | } | |
51 | ||
52 | function down (options) { | |
53 | throw new Error('Not implemented.') | |
54 | } | |
55 | ||
56 | export { | |
57 | up, | |
58 | down | |
59 | } |