Add migration script for channels

1 files changed, 111 insertions, 0 deletions

diff --git a/server/initializers/migrations/0080-video-channels.ts b/server/initializers/migrations/0080-video-channels.ts
new file mode 100644
index 000000000..fc55ef3a2
--- /dev/null
+++ b/server/initializers/migrations/0080-video-channels.ts
@@ -0,0 +1,111 @@
+import * as Sequelize from 'sequelize'
+import * as uuidv4 from 'uuid/v4'
+
+async function up (utils: {
+  transaction: Sequelize.Transaction,
+  queryInterface: Sequelize.QueryInterface,
+  sequelize: Sequelize.Sequelize,
+  db: any
+}): Promise<void> {
+  const q = utils.queryInterface
+
+  // Assert not friends
+
+  // Create uuid column for author
+  const dataAuthorUUID = {
+    type: Sequelize.UUID,
+    defaultValue: Sequelize.UUIDV4,
+    allowNull: true
+  }
+  await q.addColumn('Authors', 'uuid', dataAuthorUUID)
+
+  // Set UUID to previous authors
+  {
+    const authors = await utils.db.Author.findAll()
+    for (const author of authors) {
+      author.uuid = uuidv4()
+      await author.save()
+    }
+  }
+
+  dataAuthorUUID.allowNull = false
+  await q.changeColumn('Authors', 'uuid', dataAuthorUUID)
+
+  // Create one author per user that does not already exist
+  const users = await utils.db.User.findAll()
+  for (const user of users) {
+    const author = await utils.db.Author.find({ where: { userId: user.id }})
+    if (!author) {
+      await utils.db.Author.create({
+        name: user.username,
+        podId: null, // It is our pod
+        userId: user.id
+      })
+    }
+  }
+
+  // Create video channels table
+  await utils.db.VideoChannel.sync()
+
+  // For each author, create its default video channel
+  const authors = await utils.db.Author.findAll()
+  for (const author of authors) {
+    await utils.db.VideoChannel.create({
+      name: `Default ${author.name} channel`,
+      remote: false,
+      authorId: author.id
+    })
+  }
+
+  // Create channelId column for videos
+  const dataChannelId = {
+    type: Sequelize.INTEGER,
+    defaultValue: null,
+    allowNull: true
+  }
+  await q.addColumn('Videos', 'channelId', dataChannelId)
+
+  const query = 'SELECT "id", "authorId" FROM "Videos"'
+  const options = {
+    type: Sequelize.QueryTypes.SELECT
+  }
+  const rawVideos = await utils.sequelize.query(query, options)
+
+  for (const rawVideo of rawVideos) {
+    const videoChannel = await utils.db.VideoChannel.findOne({ where: { authorId: rawVideo.authorId }})
+
+    const video = await utils.db.Video.findById(rawVideo.id)
+    video.channelId = videoChannel.id
+    await video.save()
+  }
+
+  dataChannelId.allowNull = false
+  await q.changeColumn('Videos', 'channelId', dataChannelId)
+
+  const constraintName = 'Videos_channelId_fkey'
+  const queryForeignKey = 'ALTER TABLE "Videos" ' +
+    ' ADD CONSTRAINT "' + constraintName + '"' +
+    ' FOREIGN KEY ("channelId") REFERENCES "VideoChannels" ON UPDATE CASCADE ON DELETE CASCADE'
+
+  await utils.sequelize.query(queryForeignKey)
+
+  await q.removeColumn('Videos', 'authorId')
+}
+
+function down (options) {
+  // update "Applications" SET "migrationVersion" = 75;
+  // delete from "Authors";
+  // alter table "Authors" drop column "uuid";
+  // ALTER SEQUENCE "Authors_id_seq" RESTART WITH 1
+  // INSERT INTO "Authors" ("name", "createdAt", "updatedAt", "userId") VALUES ('root', NOW(), NOW(), 1);
+  // alter table "Videos" drop column "channelId";
+  // drop table "VideoChannels";
+  // alter table "Videos" add column "authorId" INTEGER DEFAULT 1;
+  // alter table "Videos" ADD CONSTRAINT "coucou" FOREIGN KEY ("authorId") REFERENCES "Authors"
+  throw new Error('Not implemented.')
+}
+
+export {
+  up,
+  down
+}