PeerTube/server/initializers/migrations/0075-video-resolutions.ts

60 lines
1.9 KiB
TypeScript
Raw Normal View History

2017-10-03 16:04:14 +02:00
import * as Sequelize from 'sequelize'
import { join } from 'path'
2019-04-11 11:33:44 +02:00
import { CONFIG } from '../../initializers/config'
2020-11-20 17:16:55 +01:00
import { getVideoFileResolution } from '../../helpers/ffprobe-utils'
2018-08-27 16:23:34 +02:00
import { readdir, rename } from 'fs-extra'
2017-10-03 16:04:14 +02:00
function up (utils: {
2020-01-31 16:56:52 +01:00
transaction: Sequelize.Transaction
queryInterface: Sequelize.QueryInterface
sequelize: Sequelize.Sequelize
2017-10-03 16:04:14 +02:00
db: any
}): Promise<void> {
const torrentDir = CONFIG.STORAGE.TORRENTS_DIR
const videoFileDir = CONFIG.STORAGE.VIDEOS_DIR
2018-08-27 16:23:34 +02:00
return readdir(videoFileDir)
2017-10-03 16:04:14 +02:00
.then(videoFiles => {
const tasks: Promise<any>[] = []
for (const videoFile of videoFiles) {
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)
if (matches === null) {
console.log('Invalid video file name %s.', videoFile)
continue
}
const uuid = matches[1]
const ext = matches[2]
2018-02-27 15:57:28 +01:00
const p = getVideoFileResolution(join(videoFileDir, videoFile))
.then(height => {
const oldTorrentName = uuid + '.torrent'
const newTorrentName = uuid + '-' + height + '.torrent'
2018-08-27 16:23:34 +02:00
return rename(join(torrentDir, oldTorrentName), join(torrentDir, newTorrentName)).then(() => height)
})
.then(height => {
const newVideoFileName = uuid + '-' + height + '.' + ext
2018-08-27 16:23:34 +02:00
return rename(join(videoFileDir, videoFile), join(videoFileDir, newVideoFileName)).then(() => height)
})
.then(height => {
const query = 'UPDATE "VideoFiles" SET "resolution" = ' + height +
' WHERE "videoId" = (SELECT "id" FROM "Videos" WHERE "uuid" = \'' + uuid + '\')'
return utils.sequelize.query(query)
})
2017-10-03 16:04:14 +02:00
tasks.push(p)
}
return Promise.all(tasks).then(() => undefined)
})
}
function down (options) {
throw new Error('Not implemented.')
}
export {
up,
down
}