Add video files migration

2 files changed, 56 insertions, 1 deletions

diff --git a/server/initializers/constants.ts b/server/initializers/constants.ts
index 6b0984186..c4c7ffdac 100644
--- a/server/initializers/constants.ts
+++ b/server/initializers/constants.ts
@@ -24,7 +24,7 @@ import { CONFIG, registerConfigChangedHandler } from './config'
 
 // ---------------------------------------------------------------------------
 
-const LAST_MIGRATION_VERSION = 580
+const LAST_MIGRATION_VERSION = 585
 
 // ---------------------------------------------------------------------------
 
diff --git a/server/initializers/migrations/0585-video-file-names.ts b/server/initializers/migrations/0585-video-file-names.ts
new file mode 100644
index 000000000..dd5fec3a1
--- /dev/null
+++ b/server/initializers/migrations/0585-video-file-names.ts
@@ -0,0 +1,55 @@
+import * as Sequelize from 'sequelize'
+
+async function up (utils: {
+  transaction: Sequelize.Transaction
+  queryInterface: Sequelize.QueryInterface
+  sequelize: Sequelize.Sequelize
+  db: any
+}): Promise<void> {
+  for (const column of [ 'filename', 'fileUrl', 'torrentFilename', 'torrentUrl' ]) {
+    const data = {
+      type: Sequelize.STRING,
+      allowNull: true,
+      defaultValue: null
+    }
+
+    await utils.queryInterface.addColumn('videoFile', column, data)
+  }
+
+  // Generate filenames for webtorrent files
+  {
+    const webtorrentQuery = `SELECT "videoFile".id, "video".uuid, "videoFile".resolution, "videoFile".extname ` +
+    `FROM video INNER JOIN "videoFile" ON "videoFile"."videoId" = video.id`
+
+    const query = `UPDATE "videoFile" ` +
+                  `SET filename = t.uuid || '-' || t.resolution || t.extname, ` +
+                  `"torrentFilename" = t.uuid || '-' || t.resolution || '.torrent' ` +
+                  `FROM (${webtorrentQuery}) AS t WHERE t.id = "videoFile"."id"`
+
+    await utils.sequelize.query(query)
+  }
+
+  // Generate filenames for HLS files
+  {
+    const hlsQuery = `SELECT "videoFile".id, "video".uuid, "videoFile".resolution, "videoFile".extname ` +
+                     `FROM video ` +
+                     `INNER JOIN "videoStreamingPlaylist" ON "videoStreamingPlaylist"."videoId" = video.id ` +
+                     `INNER JOIN "videoFile" ON "videoFile"."videoStreamingPlaylistId" = "videoStreamingPlaylist".id`
+
+    const query = `UPDATE "videoFile" ` +
+                  `SET filename = t.uuid || '-' || t.resolution || '-fragmented' || t.extname, ` +
+                  `"torrentFilename" = t.uuid || '-' || t.resolution || '-hls.torrent' ` +
+                  `FROM (${hlsQuery}) AS t WHERE t.id = "videoFile"."id"`
+
+    await utils.sequelize.query(query)
+  }
+}
+
+function down (options) {
+  throw new Error('Not implemented.')
+}
+
+export {
+  up,
+  down
+}