1 files changed, 157 insertions, 0 deletions

diff --git a/server/initializers/migrations/0055-video-uuid.ts b/server/initializers/migrations/0055-video-uuid.ts
new file mode 100644
index 000000000..9bc65917c
--- /dev/null
+++ b/server/initializers/migrations/0055-video-uuid.ts
@@ -0,0 +1,157 @@
+import * as Sequelize from 'sequelize'
+import * as Promise from 'bluebird'
+
+function up (utils: {
+  transaction: Sequelize.Transaction,
+  queryInterface: Sequelize.QueryInterface,
+  sequelize: Sequelize.Sequelize
+}): Promise<void> {
+  const q = utils.queryInterface
+
+  const dataUUID = {
+    type: Sequelize.UUID,
+    defaultValue: Sequelize.UUIDV4,
+    allowNull: true
+  }
+
+  return q.addColumn('Videos', 'uuid', dataUUID)
+    .then(() => {
+      const query = 'UPDATE "Videos" SET "uuid" = "id" WHERE "remoteId" IS NULL'
+      return utils.sequelize.query(query)
+    })
+    .then(() => {
+      const query = 'UPDATE "Videos" SET "uuid" = "remoteId" WHERE "remoteId" IS NOT NULL'
+      return utils.sequelize.query(query)
+    })
+    .then(() => {
+      dataUUID.defaultValue = null
+
+      return q.changeColumn('Videos', 'uuid', dataUUID)
+    })
+    .then(() => {
+      return removeForeignKey(utils.sequelize, 'RequestVideoQadus')
+    })
+    .then(() => {
+      return removeForeignKey(utils.sequelize, 'RequestVideoEvents')
+    })
+    .then(() => {
+      return removeForeignKey(utils.sequelize, 'BlacklistedVideos')
+    })
+    .then(() => {
+      return removeForeignKey(utils.sequelize, 'UserVideoRates')
+    })
+    .then(() => {
+      return removeForeignKey(utils.sequelize, 'VideoAbuses')
+    })
+    .then(() => {
+      return removeForeignKey(utils.sequelize, 'VideoTags')
+    })
+    .then(() => {
+      const query = 'ALTER TABLE "Videos" DROP CONSTRAINT "Videos_pkey"'
+      return utils.sequelize.query(query)
+    })
+    .then(() => {
+      const query = 'ALTER TABLE "Videos" ADD COLUMN "id2" SERIAL PRIMARY KEY'
+      return utils.sequelize.query(query)
+    })
+    .then(() => {
+      return q.renameColumn('Videos', 'id', 'oldId')
+    })
+    .then(() => {
+      return q.renameColumn('Videos', 'id2', 'id')
+    })
+    .then(() => {
+      return changeForeignKey(q, utils.sequelize, 'RequestVideoQadus', false)
+    })
+    .then(() => {
+      return changeForeignKey(q, utils.sequelize, 'RequestVideoEvents', false)
+    })
+    .then(() => {
+      return changeForeignKey(q, utils.sequelize, 'BlacklistedVideos', false)
+    })
+    .then(() => {
+      return changeForeignKey(q, utils.sequelize, 'UserVideoRates', false)
+    })
+    .then(() => {
+      return changeForeignKey(q, utils.sequelize, 'VideoAbuses', false)
+    })
+    .then(() => {
+      return changeForeignKey(q, utils.sequelize, 'VideoTags', true)
+    })
+    .then(() => {
+      return q.removeColumn('Videos', 'oldId')
+    })
+    .then(() => {
+      const dataRemote = {
+        type: Sequelize.BOOLEAN,
+        defaultValue: false,
+        allowNull: false
+      }
+      return q.addColumn('Videos', 'remote', dataRemote)
+    })
+    .then(() => {
+      const query = 'UPDATE "Videos" SET "remote" = false WHERE "remoteId" IS NULL'
+      return utils.sequelize.query(query)
+    })
+    .then(() => {
+      const query = 'UPDATE "Videos" SET "remote" = true WHERE "remoteId" IS NOT NULL'
+      return utils.sequelize.query(query)
+    })
+    .then(() => {
+      return q.removeColumn('Videos', 'remoteId')
+    })
+}
+
+function down (options) {
+  throw new Error('Not implemented.')
+}
+
+function removeForeignKey (sequelize: Sequelize.Sequelize, tableName: string) {
+  const query = 'ALTER TABLE "' + tableName + '" DROP CONSTRAINT "' + tableName + '_videoId_fkey' + '"'
+  return sequelize.query(query)
+}
+
+function changeForeignKey (q: Sequelize.QueryInterface, sequelize: Sequelize.Sequelize, tableName: string, allowNull: boolean) {
+  const data = {
+    type: Sequelize.INTEGER,
+    allowNull: true
+  }
+
+  return q.addColumn(tableName, 'videoId2', data)
+    .then(() => {
+      const query = 'UPDATE "' + tableName + '" SET "videoId2" = ' +
+                    '(SELECT "id" FROM "Videos" WHERE "' + tableName + '"."videoId" = "Videos"."oldId")'
+      return sequelize.query(query)
+    })
+    .then(() => {
+      if (allowNull === false) {
+        data.allowNull = false
+
+        return q.changeColumn(tableName, 'videoId2', data)
+      }
+
+      return Promise.resolve()
+    })
+    .then(() => {
+      return q.removeColumn(tableName, 'videoId')
+    })
+    .then(() => {
+      return q.renameColumn(tableName, 'videoId2', 'videoId')
+    })
+    .then(() => {
+      return q.addIndex(tableName, [ 'videoId' ])
+    })
+    .then(() => {
+      const constraintName = tableName + '_videoId_fkey'
+      const query = 'ALTER TABLE "' + tableName + '" ' +
+                    ' ADD CONSTRAINT "' + constraintName + '"' +
+                    ' FOREIGN KEY ("videoId") REFERENCES "Videos" ON DELETE CASCADE'
+
+      return sequelize.query(query)
+    })
+}
+
+export {
+  up,
+  down
+}