Server: add migration scripts to the new mongo schemes

1 files changed, 63 insertions, 0 deletions

diff --git a/server/initializers/migrations/0040-video-remote-id.js b/server/initializers/migrations/0040-video-remote-id.js
new file mode 100644
index 000000000..5cf856b2e
--- /dev/null
+++ b/server/initializers/migrations/0040-video-remote-id.js
@@ -0,0 +1,63 @@
+/*
+  Use remote id as identifier
+*/
+
+const each = require('async/each')
+const map = require('lodash/map')
+const mongoose = require('mongoose')
+const readline = require('readline')
+
+const rl = readline.createInterface({
+  input: process.stdin,
+  output: process.stdout
+})
+
+const logger = require('../../helpers/logger')
+const friends = require('../../lib/friends')
+
+const Pod = mongoose.model('Pod')
+const Video = mongoose.model('Video')
+
+exports.up = function (callback) {
+  Pod.find({}).lean().exec(function (err, pods) {
+    if (err) return callback(err)
+
+    // We need to quit friends first
+    if (pods.length === 0) {
+      return setVideosRemoteId(callback)
+    }
+
+    const timeout = setTimeout(function () {
+      throw new Error('You need to enter a value!')
+    }, 10000)
+
+    rl.question('I am sorry but I need to quit friends for upgrading. Do you want to continue? (yes/*)', function (answer) {
+      if (answer !== 'yes') throw new Error('I cannot continue.')
+
+      clearTimeout(timeout)
+      rl.close()
+
+      const urls = map(pods, 'url')
+      logger.info('Saying goodbye to: ' + urls.join(', '))
+
+      friends.quitFriends(function () {
+        setVideosRemoteId(callback)
+      })
+    })
+  })
+}
+
+exports.down = function (callback) {
+  throw new Error('Not implemented.')
+}
+
+function setVideosRemoteId (callback) {
+  Video.find({}, function (err, videos) {
+    if (err) return callback(err)
+
+    each(videos, function (video, callbackEach) {
+      video.remoteId = null
+      video.save(callbackEach)
+    }, callback)
+  })
+}