-import { readFileSync, writeFileSync } from 'fs'
-import * as parseTorrent from 'parse-torrent'
+import * as Promise from 'bluebird'
-import { CONFIG, STATIC_PATHS } from '../server/initializers/constants'
import { database as db } from '../server/initializers/database'
import { hasFriends } from '../server/lib/friends'
-db.init(true, function () {
- hasFriends(function (err, itHasFriends) {
- if (err) throw err
-
+db.init(true)
+ .then(() => {
+ return hasFriends()
+ })
+ .then(itHasFriends => {
if (itHasFriends === true) {
console.log('Cannot update host because you have friends!')
process.exit(-1)
}
console.log('Updating torrent files.')
- db.Video.list(function (err, videos) {
- if (err) throw err
-
- videos.forEach(function (video) {
- const torrentName = video.id + '.torrent'
- const torrentPath = CONFIG.STORAGE.TORRENTS_DIR + torrentName
- const filename = video.id + video.extname
+ return db.Video.list()
+ })
+ .then(videos => {
+ const tasks: Promise<any>[] = []
- const parsed = parseTorrent(readFileSync(torrentPath))
- parsed.announce = [ CONFIG.WEBSERVER.WS + '://' + CONFIG.WEBSERVER.HOST + '/tracker/socket' ]
- parsed.urlList = [ CONFIG.WEBSERVER.URL + STATIC_PATHS.WEBSEED + filename ]
+ videos.forEach(video => {
+ console.log('Updating video ' + video.uuid)
- const buf = parseTorrent.toTorrentFile(parsed)
- writeFileSync(torrentPath, buf)
+ video.VideoFiles.forEach(file => {
+ tasks.push(video.createTorrentAndSetInfoHash(file))
})
-
- process.exit(0)
})
+
+ return Promise.all(tasks)
+ })
+ .then(() => {
+ process.exit(0)
})
-})