PeerTube/scripts/migrations/peertube-2.1.ts

import { registerTSPaths } from '../../server/helpers/register-ts-paths'
registerTSPaths()

import { initDatabaseModels, sequelizeTypescript } from '../../server/initializers/database'
import * as Sequelize from 'sequelize'
import { join } from 'path'
import { HLS_STREAMING_PLAYLIST_DIRECTORY, STATIC_PATHS, WEBSERVER } from '@server/initializers/constants'
import { pathExists, stat, writeFile } from 'fs-extra'
import { createTorrentPromise } from '@server/helpers/webtorrent'
import { CONFIG } from '@server/initializers/config'
import parseTorrent from 'parse-torrent'
import { logger } from '@server/helpers/logger'

run()
  .then(() => process.exit(0))
  .catch(err => {
    console.error(err)
    process.exit(-1)
  })

async function run () {
  logger.info('Creating torrents and updating database for HSL files.')

  await initDatabaseModels(true)

  const query = 'select "videoFile".id as id, "videoFile".resolution as resolution, "video".uuid as uuid from "videoFile" ' +
    'inner join "videoStreamingPlaylist" ON "videoStreamingPlaylist".id = "videoFile"."videoStreamingPlaylistId" ' +
    'inner join video ON video.id = "videoStreamingPlaylist"."videoId" ' +
    'WHERE video.remote IS FALSE'
  const options = {
    type: Sequelize.QueryTypes.SELECT
  }
  const res = await sequelizeTypescript.query(query, options)

  for (const row of res) {
    const videoFilename = `${row['uuid']}-${row['resolution']}-fragmented.mp4`
    const videoFilePath = join(HLS_STREAMING_PLAYLIST_DIRECTORY, row['uuid'], videoFilename)

    logger.info('Processing %s.', videoFilePath)

    if (!await pathExists(videoFilePath)) {
      console.warn('Cannot generate torrent of %s: file does not exist.', videoFilePath)
      continue
    }

    const createTorrentOptions = {
      // Keep the extname, it's used by the client to stream the file inside a web browser
      name: `video ${row['uuid']}`,
      createdBy: 'PeerTube',
      announceList: [
        [ WEBSERVER.WS + '://' + WEBSERVER.HOSTNAME + ':' + WEBSERVER.PORT + '/tracker/socket' ],
        [ WEBSERVER.URL + '/tracker/announce' ]
      ],
      urlList: [ WEBSERVER.URL + join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, row['uuid'], videoFilename) ]
    }
    const torrent = await createTorrentPromise(videoFilePath, createTorrentOptions)

    const torrentName = `${row['uuid']}-${row['resolution']}-hls.torrent`
    const filePath = join(CONFIG.STORAGE.TORRENTS_DIR, torrentName)

    await writeFile(filePath, torrent)

    const parsedTorrent = parseTorrent(torrent)
    const infoHash = parsedTorrent.infoHash

    const stats = await stat(videoFilePath)
    const size = stats.size

    const queryUpdate = 'UPDATE "videoFile" SET "infoHash" = ?, "size" = ? WHERE id = ?'

    const options = {
      type: Sequelize.QueryTypes.UPDATE,
      replacements: [ infoHash, size, row['id'] ]
    }
    await sequelizeTypescript.query(queryUpdate, options)
  }
}