server/initializers/migrations/0065-video-file-size.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

import * as Sequelize from 'sequelize'
import * as Promise from 'bluebird'
import { stat } from 'fs-extra'
import { VideoModel } from '../../models/video/video'
import { getVideoFilePath } from '@server/lib/video-paths'

function up (utils: {
  transaction: Sequelize.Transaction
  queryInterface: Sequelize.QueryInterface
  sequelize: Sequelize.Sequelize
  db: any
}): Promise<void> {
  return utils.db.Video.listOwnedAndPopulateAuthorAndTags()
    .then((videos: VideoModel[]) => {
      const tasks: Promise<any>[] = []

      videos.forEach(video => {
        video.VideoFiles.forEach(videoFile => {
          const p = new Promise((res, rej) => {
            stat(getVideoFilePath(video, videoFile), (err, stats) => {
              if (err) return rej(err)

              videoFile.size = stats.size
              videoFile.save().then(res).catch(rej)
            })
          })

          tasks.push(p)
        })
      })

      return tasks
    })
    .then((tasks: Promise<any>[]) => {
      return Promise.all(tasks)
    })
}

function down (options) {
  throw new Error('Not implemented.')
}

export {
  up,
  down
}