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

   1 import * as Sequelize from 'sequelize'
   2 import { join } from 'path'
   3 import { CONFIG } from '../../initializers/config'
   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 }