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