From f285faa04e84b45e62bd05e7050dc500113b0356 Mon Sep 17 00:00:00 2001 From: Chocobozzz Date: Fri, 11 Nov 2016 15:20:03 +0100 Subject: Server: generate magnet uri on the fly --- server/models/video.js | 102 ++++++++++++++++++++++++++++++++++--------------- 1 file changed, 72 insertions(+), 30 deletions(-) (limited to 'server/models/video.js') diff --git a/server/models/video.js b/server/models/video.js index 6cffa87af..19136ba25 100644 --- a/server/models/video.js +++ b/server/models/video.js @@ -3,10 +3,10 @@ const createTorrent = require('create-torrent') const ffmpeg = require('fluent-ffmpeg') const fs = require('fs') +const magnetUtil = require('magnet-uri') const parallel = require('async/parallel') const parseTorrent = require('parse-torrent') const pathUtils = require('path') -const magnet = require('magnet-uri') const mongoose = require('mongoose') const constants = require('../initializers/constants') @@ -25,7 +25,9 @@ const VideoSchema = mongoose.Schema({ }, remoteId: mongoose.Schema.Types.ObjectId, description: String, - magnetUri: String, + magnet: { + infoHash: String + }, podUrl: String, author: String, duration: Number, @@ -39,7 +41,6 @@ const VideoSchema = mongoose.Schema({ VideoSchema.path('name').validate(customVideosValidators.isVideoNameValid) VideoSchema.path('description').validate(customVideosValidators.isVideoDescriptionValid) -VideoSchema.path('magnetUri').validate(customVideosValidators.isVideoMagnetUriValid) VideoSchema.path('podUrl').validate(customVideosValidators.isVideoPodUrlValid) VideoSchema.path('author').validate(customVideosValidators.isVideoAuthorValid) VideoSchema.path('duration').validate(customVideosValidators.isVideoDurationValid) @@ -51,8 +52,10 @@ VideoSchema.path('thumbnail').validate(function (value) { VideoSchema.path('tags').validate(customVideosValidators.isVideoTagsValid) VideoSchema.methods = { - getFilename, - getJPEGName, + generateMagnetUri, + getVideoFilename, + getThumbnailName, + getPreviewName, getTorrentName, isOwned, toFormatedJSON, @@ -103,8 +106,8 @@ VideoSchema.pre('save', function (next) { const tasks = [] if (video.isOwned()) { - const videoPath = pathUtils.join(constants.CONFIG.STORAGE.VIDEOS_DIR, video.getFilename()) - this.podUrl = constants.CONFIG.WEBSERVER.URL + const videoPath = pathUtils.join(constants.CONFIG.STORAGE.VIDEOS_DIR, video.getVideoFilename()) + this.podUrl = constants.CONFIG.WEBSERVER.HOSTNAME + ':' + constants.CONFIG.WEBSERVER.PORT tasks.push( // TODO: refractoring @@ -114,7 +117,7 @@ VideoSchema.pre('save', function (next) { [ constants.CONFIG.WEBSERVER.WS + '://' + constants.CONFIG.WEBSERVER.HOSTNAME + ':' + constants.CONFIG.WEBSERVER.PORT + '/tracker/socket' ] ], urlList: [ - constants.CONFIG.WEBSERVER.URL + constants.STATIC_PATHS.WEBSEED + video.getFilename() + constants.CONFIG.WEBSERVER.URL + constants.STATIC_PATHS.WEBSEED + video.getVideoFilename() ] } @@ -125,9 +128,9 @@ VideoSchema.pre('save', function (next) { if (err) return callback(err) const parsedTorrent = parseTorrent(torrent) - parsedTorrent.xs = video.podUrl + constants.STATIC_PATHS.TORRENTS + video.getTorrentName() - video.magnetUri = magnet.encode(parsedTorrent) + video.magnet.infoHash = parsedTorrent.infoHash + console.log(parsedTorrent) callback(null) }) }) @@ -150,16 +153,57 @@ mongoose.model('Video', VideoSchema) // ------------------------------ METHODS ------------------------------ -function getFilename () { - return this._id + this.extname +function generateMagnetUri () { + let baseUrlHttp, baseUrlWs + + if (this.isOwned()) { + baseUrlHttp = constants.CONFIG.WEBSERVER.URL + baseUrlWs = constants.CONFIG.WEBSERVER.WS + '://' + constants.CONFIG.WEBSERVER.HOSTNAME + ':' + constants.CONFIG.WEBSERVER.PORT + } else { + baseUrlHttp = constants.REMOTE_SCHEME.HTTP + this.podUrl + baseUrlWs = constants.REMOTE_SCHEME.WS + this.podUrl + } + + const xs = baseUrlHttp + constants.STATIC_PATHS.TORRENTS + this.getTorrentName() + const announce = baseUrlWs + '/tracker/socket' + const urlList = [ baseUrlHttp + constants.STATIC_PATHS.WEBSEED + this.getVideoFilename() ] + + const magnetHash = { + xs, + announce, + urlList, + infoHash: this.magnet.infoHash, + name: this.name + } + + return magnetUtil.encode(magnetHash) } -function getJPEGName () { +function getVideoFilename () { + if (this.isOwned()) return this._id + this.extname + + return this.remoteId + this.extname +} + +function getThumbnailName () { + // We always have a copy of the thumbnail return this._id + '.jpg' } +function getPreviewName () { + const extension = '.jpg' + + if (this.isOwned()) return this._id + extension + + return this.remoteId + extension +} + function getTorrentName () { - return this._id + '.torrent' + const extension = '.torrent' + + if (this.isOwned()) return this._id + extension + + return this.remoteId + extension } function isOwned () { @@ -171,13 +215,13 @@ function toFormatedJSON () { id: this._id, name: this.name, description: this.description, - podUrl: this.podUrl.replace(/^https?:\/\//, ''), + podUrl: this.podUrl, isLocal: this.isOwned(), - magnetUri: this.magnetUri, + magnetUri: this.generateMagnetUri(), author: this.author, duration: this.duration, tags: this.tags, - thumbnailPath: constants.STATIC_PATHS.THUMBNAILS + '/' + this.getJPEGName(), + thumbnailPath: constants.STATIC_PATHS.THUMBNAILS + '/' + this.getThumbnailName(), createdDate: this.createdDate } @@ -188,7 +232,7 @@ function toRemoteJSON (callback) { const self = this // Convert thumbnail to base64 - const thumbnailPath = pathUtils.join(constants.CONFIG.STORAGE.THUMBNAILS_DIR, this.getJPEGName()) + const thumbnailPath = pathUtils.join(constants.CONFIG.STORAGE.THUMBNAILS_DIR, this.getThumbnailName()) fs.readFile(thumbnailPath, function (err, thumbnailData) { if (err) { logger.error('Cannot read the thumbnail of the video') @@ -198,7 +242,7 @@ function toRemoteJSON (callback) { const remoteVideo = { name: self.name, description: self.description, - magnetUri: self.magnetUri, + magnet: self.magnet, remoteId: self._id, author: self.author, duration: self.duration, @@ -267,11 +311,11 @@ function search (value, field, start, count, sort, callback) { // --------------------------------------------------------------------------- function removeThumbnail (video, callback) { - fs.unlink(constants.CONFIG.STORAGE.THUMBNAILS_DIR + video.getJPEGName(), callback) + fs.unlink(constants.CONFIG.STORAGE.THUMBNAILS_DIR + video.getThumbnailName(), callback) } function removeFile (video, callback) { - fs.unlink(constants.CONFIG.STORAGE.VIDEOS_DIR + video.getFilename(), callback) + fs.unlink(constants.CONFIG.STORAGE.VIDEOS_DIR + video.getVideoFilename(), callback) } function removeTorrent (video, callback) { @@ -280,22 +324,21 @@ function removeTorrent (video, callback) { function removePreview (video, callback) { // Same name than video thumnail - // TODO: refractoring - fs.unlink(constants.CONFIG.STORAGE.PREVIEWS_DIR + video.getJPEGName(), callback) + fs.unlink(constants.CONFIG.STORAGE.PREVIEWS_DIR + video.getPreviewName(), callback) } function createPreview (video, videoPath, callback) { - generateImage(video, videoPath, constants.CONFIG.STORAGE.PREVIEWS_DIR, callback) + generateImage(video, videoPath, constants.CONFIG.STORAGE.PREVIEWS_DIR, video.getPreviewName(), callback) } function createThumbnail (video, videoPath, callback) { - generateImage(video, videoPath, constants.CONFIG.STORAGE.THUMBNAILS_DIR, constants.THUMBNAILS_SIZE, callback) + generateImage(video, videoPath, constants.CONFIG.STORAGE.THUMBNAILS_DIR, video.getThumbnailName(), constants.THUMBNAILS_SIZE, callback) } function generateThumbnailFromBase64 (video, thumbnailData, callback) { // Creating the thumbnail for this remote video) - const thumbnailName = video.getJPEGName() + const thumbnailName = video.getThumbnailName() const thumbnailPath = constants.CONFIG.STORAGE.THUMBNAILS_DIR + thumbnailName fs.writeFile(thumbnailPath, thumbnailData, { encoding: 'base64' }, function (err) { if (err) return callback(err) @@ -304,10 +347,9 @@ function generateThumbnailFromBase64 (video, thumbnailData, callback) { }) } -function generateImage (video, videoPath, folder, size, callback) { - const filename = video.getJPEGName() +function generateImage (video, videoPath, folder, imageName, size, callback) { const options = { - filename, + filename: imageName, count: 1, folder } @@ -321,7 +363,7 @@ function generateImage (video, videoPath, folder, size, callback) { ffmpeg(videoPath) .on('error', callback) .on('end', function () { - callback(null, filename) + callback(null, imageName) }) .thumbnail(options) } -- cgit v1.2.3