Add migration script

1 files changed, 71 insertions, 0 deletions

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()
+  }
+}