]>
Commit | Line | Data |
---|---|---|
d7a25329 | 1 | import * as Sequelize from 'sequelize' |
d441f2ed C |
2 | import { join } from 'path' |
3 | import { HLS_STREAMING_PLAYLIST_DIRECTORY, WEBSERVER } from '@server/initializers/constants' | |
4 | import { CONFIG } from '@server/initializers/config' | |
5 | import { pathExists, stat, writeFile } from 'fs-extra' | |
6 | import * as parseTorrent from 'parse-torrent' | |
7 | import { createTorrentPromise } from '@server/helpers/webtorrent' | |
d7a25329 C |
8 | |
9 | async function up (utils: { | |
10 | transaction: Sequelize.Transaction, | |
11 | queryInterface: Sequelize.QueryInterface, | |
12 | sequelize: Sequelize.Sequelize, | |
13 | db: any | |
14 | }): Promise<void> { | |
15 | { | |
16 | const data = { | |
17 | type: Sequelize.INTEGER, | |
18 | allowNull: true, | |
19 | references: { | |
20 | model: 'videoStreamingPlaylist', | |
21 | key: 'id' | |
22 | }, | |
23 | onDelete: 'CASCADE' | |
24 | } | |
25 | ||
26 | await utils.queryInterface.addColumn('videoFile', 'videoStreamingPlaylistId', data) | |
27 | } | |
28 | ||
29 | { | |
30 | const data = { | |
31 | type: Sequelize.INTEGER, | |
32 | allowNull: true | |
33 | } | |
34 | ||
35 | await utils.queryInterface.changeColumn('videoFile', 'videoId', data) | |
36 | } | |
d441f2ed C |
37 | |
38 | { | |
39 | await utils.queryInterface.removeIndex('videoFile', 'video_file_video_id_resolution_fps') | |
40 | } | |
41 | ||
42 | { | |
43 | const query = 'insert into "videoFile" ' + | |
44 | '(resolution, size, "infoHash", "videoId", "createdAt", "updatedAt", fps, extname, "videoStreamingPlaylistId")' + | |
45 | '(SELECT "videoFile".resolution, "videoFile".size, \'fake\', NULL, "videoFile"."createdAt", "videoFile"."updatedAt", "videoFile"."fps", ' + | |
46 | '"videoFile".extname, "videoStreamingPlaylist".id FROM "videoStreamingPlaylist" ' + | |
47 | 'inner join video ON video.id = "videoStreamingPlaylist"."videoId" inner join "videoFile" ON "videoFile"."videoId" = video.id)' | |
48 | ||
49 | await utils.sequelize.query(query, { transaction: utils.transaction }) | |
50 | } | |
51 | ||
52 | { | |
53 | const query = 'select "videoFile".id as id, "videoFile".resolution as resolution, "video".uuid as uuid from "videoFile" ' + | |
54 | 'inner join "videoStreamingPlaylist" ON "videoStreamingPlaylist".id = "videoFile"."videoStreamingPlaylistId" ' + | |
55 | 'inner join video ON video.id = "videoStreamingPlaylist"."videoId" ' + | |
56 | 'WHERE video.remote IS FALSE' | |
57 | const options = { | |
58 | type: Sequelize.QueryTypes.SELECT, | |
59 | transaction: utils.transaction | |
60 | } | |
61 | const res = await utils.sequelize.query(query, options) | |
62 | ||
63 | for (const row of res) { | |
64 | const videoFilename = `${row['uuid']}-${row['resolution']}-fragmented.mp4` | |
65 | const videoFilePath = join(HLS_STREAMING_PLAYLIST_DIRECTORY, row['uuid'], videoFilename) | |
66 | ||
67 | if (!await pathExists(videoFilePath)) { | |
68 | console.warn('Cannot generate torrent of %s: file does not exist.', videoFilePath) | |
69 | continue | |
70 | } | |
71 | ||
72 | const createTorrentOptions = { | |
73 | // Keep the extname, it's used by the client to stream the file inside a web browser | |
74 | name: `video ${row['uuid']}`, | |
75 | createdBy: 'PeerTube', | |
76 | announceList: [ | |
77 | [ WEBSERVER.WS + '://' + WEBSERVER.HOSTNAME + ':' + WEBSERVER.PORT + '/tracker/socket' ], | |
78 | [ WEBSERVER.URL + '/tracker/announce' ] | |
79 | ], | |
80 | urlList: [ WEBSERVER.URL + join(HLS_STREAMING_PLAYLIST_DIRECTORY, row['uuid'], videoFilename) ] | |
81 | } | |
82 | const torrent = await createTorrentPromise(videoFilePath, createTorrentOptions) | |
83 | ||
84 | const torrentName = `${row['uuid']}-${row['resolution']}-hls.torrent` | |
85 | const filePath = join(CONFIG.STORAGE.TORRENTS_DIR, torrentName) | |
86 | ||
87 | await writeFile(filePath, torrent) | |
88 | ||
89 | const parsedTorrent = parseTorrent(torrent) | |
90 | const infoHash = parsedTorrent.infoHash | |
91 | ||
92 | const stats = await stat(videoFilePath) | |
93 | const size = stats.size | |
94 | ||
95 | const queryUpdate = 'UPDATE "videoFile" SET "infoHash" = ?, "size" = ? WHERE id = ?' | |
96 | ||
97 | const options = { | |
98 | type: Sequelize.QueryTypes.UPDATE, | |
99 | replacements: [ infoHash, size, row['id'] ], | |
100 | transaction: utils.transaction | |
101 | } | |
102 | await utils.sequelize.query(queryUpdate, options) | |
103 | ||
104 | } | |
105 | } | |
d7a25329 C |
106 | } |
107 | ||
108 | function down (options) { | |
109 | throw new Error('Not implemented.') | |
110 | } | |
111 | ||
112 | export { | |
113 | up, | |
114 | down | |
115 | } |