server/controllers/api/v1/videos.js

   1 'use strict'
   2
   3 const config = require('config')
   4 const crypto = require('crypto')
   5 const express = require('express')
   6 const multer = require('multer')
   7
   8 const logger = require('../../../helpers/logger')
   9 const friends = require('../../../lib/friends')
  10 const middleware = require('../../../middlewares')
  11 const cacheMiddleware = middleware.cache
  12 const reqValidator = middleware.reqValidators.videos
  13 const Videos = require('../../../models/videos') // model
  14 const videos = require('../../../lib/videos')
  15 const webtorrent = require('../../../lib/webtorrent')
  16
  17 const router = express.Router()
  18 const uploads = config.get('storage.uploads')
  19
  20 // multer configuration
  21 const storage = multer.diskStorage({
  22   destination: function (req, file, cb) {
  23     cb(null, uploads)
  24   },
  25
  26   filename: function (req, file, cb) {
  27     let extension = ''
  28     if (file.mimetype === 'video/webm') extension = 'webm'
  29     else if (file.mimetype === 'video/mp4') extension = 'mp4'
  30     else if (file.mimetype === 'video/ogg') extension = 'ogv'
  31     crypto.pseudoRandomBytes(16, function (err, raw) {
  32       const fieldname = err ? undefined : raw.toString('hex')
  33       cb(null, fieldname + '.' + extension)
  34     })
  35   }
  36 })
  37
  38 const reqFiles = multer({ storage: storage }).fields([{ name: 'videofile', maxCount: 1 }])
  39
  40 router.get('/', cacheMiddleware.cache(false), listVideos)
  41 router.post('/', reqFiles, reqValidator.videosAdd, cacheMiddleware.cache(false), addVideo)
  42 router.get('/:id', reqValidator.videosGet, cacheMiddleware.cache(false), getVideos)
  43 router.delete('/:id', reqValidator.videosRemove, cacheMiddleware.cache(false), removeVideo)
  44 router.get('/search/:name', reqValidator.videosSearch, cacheMiddleware.cache(false), searchVideos)
  45
  46 // ---------------------------------------------------------------------------
  47
  48 module.exports = router
  49
  50 // ---------------------------------------------------------------------------
  51
  52 function addVideo (req, res, next) {
  53   const video_file = req.files.videofile[0]
  54   const video_infos = req.body
  55
  56   videos.seed(video_file.path, function (err, torrent) {
  57     if (err) {
  58       logger.error('Cannot seed this video.')
  59       return next(err)
  60     }
  61
  62     const video_data = {
  63       name: video_infos.name,
  64       namePath: video_file.filename,
  65       description: video_infos.description,
  66       magnetUri: torrent.magnetURI
  67     }
  68
  69     Videos.add(video_data, function (err) {
  70       if (err) {
  71         // TODO unseed the video
  72         logger.error('Cannot insert this video in the database.')
  73         return next(err)
  74       }
  75
  76       // Now we'll add the video's meta data to our friends
  77       friends.addVideoToFriends(video_data)
  78
  79       // TODO : include Location of the new video -> 201
  80       res.type('json').status(204).end()
  81     })
  82   })
  83 }
  84
  85 function getVideos (req, res, next) {
  86   Videos.get(req.params.id, function (err, video_obj) {
  87     if (err) return next(err)
  88
  89     const state = videos.getVideoState(video_obj)
  90     if (state.exist === false) {
  91       return res.type('json').status(204).end()
  92     }
  93
  94     res.json(getFormatedVideo(video_obj))
  95   })
  96 }
  97
  98 function listVideos (req, res, next) {
  99   Videos.list(function (err, videos_list) {
 100     if (err) return next(err)
 101
 102     res.json(getFormatedVideos(videos_list))
 103   })
 104 }
 105
 106 function removeVideo (req, res, next) {
 107   const video_id = req.params.id
 108   Videos.get(video_id, function (err, video) {
 109     if (err) return next(err)
 110
 111     removeTorrent(video.magnetUri, function () {
 112       Videos.removeOwned(req.params.id, function (err) {
 113         if (err) return next(err)
 114
 115         const params = {
 116           name: video.name,
 117           magnetUri: video.magnetUri
 118         }
 119
 120         friends.removeVideoToFriends(params)
 121         res.type('json').status(204).end()
 122       })
 123     })
 124   })
 125 }
 126
 127 function searchVideos (req, res, next) {
 128   Videos.search(req.params.name, function (err, videos_list) {
 129     if (err) return next(err)
 130
 131     res.json(getFormatedVideos(videos_list))
 132   })
 133 }
 134
 135 // ---------------------------------------------------------------------------
 136
 137 function getFormatedVideo (video_obj) {
 138   const formated_video = {
 139     id: video_obj._id,
 140     name: video_obj.name,
 141     description: video_obj.description,
 142     podUrl: video_obj.podUrl,
 143     isLocal: videos.getVideoState(video_obj).owned,
 144     magnetUri: video_obj.magnetUri
 145   }
 146
 147   return formated_video
 148 }
 149
 150 function getFormatedVideos (videos_obj) {
 151   const formated_videos = []
 152
 153   videos_obj.forEach(function (video_obj) {
 154     formated_videos.push(getFormatedVideo(video_obj))
 155   })
 156
 157   return formated_videos
 158 }
 159
 160 // Maybe the torrent is not seeded, but we catch the error to don't stop the removing process
 161 function removeTorrent (magnetUri, callback) {
 162   try {
 163     webtorrent.remove(magnetUri, callback)
 164   } catch (err) {
 165     logger.warn('Cannot remove the torrent from WebTorrent', { err: err })
 166     return callback(null)
 167   }
 168 }