Add script when the host/port of a pod change

pull/24/head
Chocobozzz 2016-11-22 22:12:13 +01:00
parent 04968ab477
commit 93d4a3b551
2 changed files with 52 additions and 0 deletions

View File

@ -32,6 +32,7 @@
"start": "node server",
"check": "scripty",
"upgrade": "scripty",
"update-host": "scripty",
"test": "scripty",
"help": "scripty",
"postinstall": "cd client && npm install"

51
scripts/update-host.js Executable file
View File

@ -0,0 +1,51 @@
#!/usr/bin/env node
'use strict'
// TODO: document this script
const fs = require('fs')
const mongoose = require('mongoose')
const parseTorrent = require('parse-torrent')
const constants = require('../server/initializers/constants')
const database = require('../server/initializers/database')
database.connect()
const friends = require('../server/lib/friends')
const Video = mongoose.model('Video')
friends.hasFriends(function (err, hasFriends) {
if (err) throw err
if (hasFriends === true) {
console.log('Cannot update host because you have friends!')
process.exit(-1)
}
console.log('Updating videos host in database.')
Video.update({ }, { podHost: constants.CONFIG.WEBSERVER.HOST }, { multi: true }, function (err) {
if (err) throw err
console.log('Updating torrent files.')
Video.find().lean().exec(function (err, videos) {
if (err) throw err
videos.forEach(function (video) {
const torrentName = video._id + '.torrent'
const torrentPath = constants.CONFIG.STORAGE.TORRENTS_DIR + torrentName
const filename = video._id + video.extname
const parsed = parseTorrent(fs.readFileSync(torrentPath))
parsed.announce = [ constants.CONFIG.WEBSERVER.WS + '://' + constants.CONFIG.WEBSERVER.HOST + '/tracker/socket' ]
parsed.urlList = [ constants.CONFIG.WEBSERVER.URL + constants.STATIC_PATHS.WEBSEED + filename ]
const buf = parseTorrent.toTorrentFile(parsed)
fs.writeFileSync(torrentPath, buf)
})
process.exit(0)
})
})
})