From fc097052abfbd646950a26f02b31b9925a5f3961 Mon Sep 17 00:00:00 2001 From: Chocobozzz Date: Fri, 21 Oct 2022 11:56:19 +0200 Subject: Add migration script --- scripts/migrations/peertube-5.0.ts | 71 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 71 insertions(+) create mode 100644 scripts/migrations/peertube-5.0.ts (limited to 'scripts/migrations') diff --git a/scripts/migrations/peertube-5.0.ts b/scripts/migrations/peertube-5.0.ts new file mode 100644 index 000000000..a0f51a64c --- /dev/null +++ b/scripts/migrations/peertube-5.0.ts @@ -0,0 +1,71 @@ +import { ensureDir } from 'fs-extra' +import { Op } from 'sequelize' +import { updateTorrentMetadata } from '@server/helpers/webtorrent' +import { DIRECTORIES } from '@server/initializers/constants' +import { moveFilesIfPrivacyChanged } from '@server/lib/video-privacy' +import { VideoModel } from '@server/models/video/video' +import { MVideoFullLight } from '@server/types/models' +import { VideoPrivacy } from '@shared/models' +import { initDatabaseModels } from '../../server/initializers/database' + +run() + .then(() => process.exit(0)) + .catch(err => { + console.error(err) + process.exit(-1) + }) + +async function run () { + console.log('Moving private video files in dedicated folders.') + + await ensureDir(DIRECTORIES.HLS_STREAMING_PLAYLIST.PRIVATE) + await ensureDir(DIRECTORIES.VIDEOS.PRIVATE) + + await initDatabaseModels(true) + + const videos = await VideoModel.unscoped().findAll({ + attributes: [ 'uuid' ], + where: { + privacy: { + [Op.in]: [ VideoPrivacy.PRIVATE, VideoPrivacy.INTERNAL ] + } + } + }) + + for (const { uuid } of videos) { + try { + console.log('Moving files of video %s.', uuid) + + const video = await VideoModel.loadFull(uuid) + + try { + await moveFilesIfPrivacyChanged(video, VideoPrivacy.PUBLIC) + } catch (err) { + console.error('Cannot move files of video %s.', uuid, err) + } + + try { + await updateTorrents(video) + } catch (err) { + console.error('Cannot regenerate torrents of video %s.', uuid, err) + } + } catch (err) { + console.error('Cannot process video %s.', uuid, err) + } + } +} + +async function updateTorrents (video: MVideoFullLight) { + for (const file of video.VideoFiles) { + await updateTorrentMetadata(video, file) + + await file.save() + } + + const playlist = video.getHLSPlaylist() + for (const file of (playlist?.VideoFiles || [])) { + await updateTorrentMetadata(playlist, file) + + await file.save() + } +} -- cgit v1.2.3