diff options
author | Chocobozzz <me@florianbigard.com> | 2022-10-12 16:09:02 +0200 |
---|---|---|
committer | Chocobozzz <chocobozzz@cpy.re> | 2022-10-24 14:48:24 +0200 |
commit | 3545e72c686ff1725bbdfd8d16d693e2f4aa75a3 (patch) | |
tree | e7f1d12ef5dae1e1142c3a8d0b681c1dbbb0de10 /scripts/migrations | |
parent | 38a3ccc7f8ad0ea94362b58c732af7c387ab46be (diff) | |
download | PeerTube-3545e72c686ff1725bbdfd8d16d693e2f4aa75a3.tar.gz PeerTube-3545e72c686ff1725bbdfd8d16d693e2f4aa75a3.tar.zst PeerTube-3545e72c686ff1725bbdfd8d16d693e2f4aa75a3.zip |
Put private videos under a specific subdirectory
Diffstat (limited to 'scripts/migrations')
-rw-r--r-- | scripts/migrations/peertube-2.1.ts | 74 |
1 files changed, 0 insertions, 74 deletions
diff --git a/scripts/migrations/peertube-2.1.ts b/scripts/migrations/peertube-2.1.ts deleted file mode 100644 index 2e316d996..000000000 --- a/scripts/migrations/peertube-2.1.ts +++ /dev/null | |||
@@ -1,74 +0,0 @@ | |||
1 | import { pathExists, stat, writeFile } from 'fs-extra' | ||
2 | import parseTorrent from 'parse-torrent' | ||
3 | import { join } from 'path' | ||
4 | import * as Sequelize from 'sequelize' | ||
5 | import { logger } from '@server/helpers/logger' | ||
6 | import { createTorrentPromise } from '@server/helpers/webtorrent' | ||
7 | import { CONFIG } from '@server/initializers/config' | ||
8 | import { HLS_STREAMING_PLAYLIST_DIRECTORY, STATIC_PATHS, WEBSERVER } from '@server/initializers/constants' | ||
9 | import { initDatabaseModels, sequelizeTypescript } from '../../server/initializers/database' | ||
10 | |||
11 | run() | ||
12 | .then(() => process.exit(0)) | ||
13 | .catch(err => { | ||
14 | console.error(err) | ||
15 | process.exit(-1) | ||
16 | }) | ||
17 | |||
18 | async function run () { | ||
19 | logger.info('Creating torrents and updating database for HSL files.') | ||
20 | |||
21 | await initDatabaseModels(true) | ||
22 | |||
23 | const query = 'select "videoFile".id as id, "videoFile".resolution as resolution, "video".uuid as uuid from "videoFile" ' + | ||
24 | 'inner join "videoStreamingPlaylist" ON "videoStreamingPlaylist".id = "videoFile"."videoStreamingPlaylistId" ' + | ||
25 | 'inner join video ON video.id = "videoStreamingPlaylist"."videoId" ' + | ||
26 | 'WHERE video.remote IS FALSE' | ||
27 | const options = { | ||
28 | type: Sequelize.QueryTypes.SELECT | ||
29 | } | ||
30 | const res = await sequelizeTypescript.query(query, options) | ||
31 | |||
32 | for (const row of res) { | ||
33 | const videoFilename = `${row['uuid']}-${row['resolution']}-fragmented.mp4` | ||
34 | const videoFilePath = join(HLS_STREAMING_PLAYLIST_DIRECTORY, row['uuid'], videoFilename) | ||
35 | |||
36 | logger.info('Processing %s.', videoFilePath) | ||
37 | |||
38 | if (!await pathExists(videoFilePath)) { | ||
39 | console.warn('Cannot generate torrent of %s: file does not exist.', videoFilePath) | ||
40 | continue | ||
41 | } | ||
42 | |||
43 | const createTorrentOptions = { | ||
44 | // Keep the extname, it's used by the client to stream the file inside a web browser | ||
45 | name: `video ${row['uuid']}`, | ||
46 | createdBy: 'PeerTube', | ||
47 | announceList: [ | ||
48 | [ WEBSERVER.WS + '://' + WEBSERVER.HOSTNAME + ':' + WEBSERVER.PORT + '/tracker/socket' ], | ||
49 | [ WEBSERVER.URL + '/tracker/announce' ] | ||
50 | ], | ||
51 | urlList: [ WEBSERVER.URL + join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, row['uuid'], videoFilename) ] | ||
52 | } | ||
53 | const torrent = await createTorrentPromise(videoFilePath, createTorrentOptions) | ||
54 | |||
55 | const torrentName = `${row['uuid']}-${row['resolution']}-hls.torrent` | ||
56 | const filePath = join(CONFIG.STORAGE.TORRENTS_DIR, torrentName) | ||
57 | |||
58 | await writeFile(filePath, torrent) | ||
59 | |||
60 | const parsedTorrent = parseTorrent(torrent) | ||
61 | const infoHash = parsedTorrent.infoHash | ||
62 | |||
63 | const stats = await stat(videoFilePath) | ||
64 | const size = stats.size | ||
65 | |||
66 | const queryUpdate = 'UPDATE "videoFile" SET "infoHash" = ?, "size" = ? WHERE id = ?' | ||
67 | |||
68 | const options = { | ||
69 | type: Sequelize.QueryTypes.UPDATE, | ||
70 | replacements: [ infoHash, size, row['id'] ] | ||
71 | } | ||
72 | await sequelizeTypescript.query(queryUpdate, options) | ||
73 | } | ||
74 | } | ||