|
@@ -0,0 +1,77 @@
|
|
|
+import { registerTSPaths } from '../../server/helpers/register-ts-paths'
|
|
|
+registerTSPaths()
|
|
|
+
|
|
|
+import { initDatabaseModels, sequelizeTypescript } from '../../server/initializers'
|
|
|
+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 * as 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)
|
|
|
+ }
|
|
|
+}
|