PeerTube/server/controllers/api/v1/remote.js

'use strict'

const async = require('async')
const express = require('express')
const mongoose = require('mongoose')

const middlewares = require('../../../middlewares')
const secureMiddleware = middlewares.secure
const reqValidator = middlewares.reqValidators.remote
const logger = require('../../../helpers/logger')

const router = express.Router()
const Video = mongoose.model('Video')

router.post('/videos',
  reqValidator.signature,
  reqValidator.dataToDecrypt,
  secureMiddleware.decryptBody,
  reqValidator.remoteVideos,
  remoteVideos
)

// ---------------------------------------------------------------------------

module.exports = router

// ---------------------------------------------------------------------------

function remoteVideos (req, res, next) {
  const requests = req.body.data
  const fromUrl = req.body.signature.url

  // We need to process in the same order to keep consistency
  // TODO: optimization
  async.eachSeries(requests, function (request, callbackEach) {
    const videoData = request.data

    if (request.type === 'add') {
      addRemoteVideo(videoData, callbackEach)
    } else if (request.type === 'remove') {
      removeRemoteVideo(videoData, fromUrl, callbackEach)
    }
  }, function (err) {
    if (err) logger.error('Error managing remote videos.', { error: err })
  })

  // We don't need to keep the other pod waiting
  return res.type('json').status(204).end()
}

function addRemoteVideo (videoToCreateData, callback) {
  // Mongoose pre hook will automatically create the thumbnail on disk
  videoToCreateData.thumbnail = videoToCreateData.thumbnailBase64

  const video = new Video(videoToCreateData)
  video.save(callback)
}

function removeRemoteVideo (videoToRemoveData, fromUrl, callback) {
  // We need the list because we have to remove some other stuffs (thumbnail etc)
  Video.listByUrlAndMagnet(fromUrl, videoToRemoveData.magnetUri, function (err, videosList) {
    if (err) {
      logger.error('Cannot list videos from url and magnets.', { error: err })
      return callback(err)
    }

    async.each(videosList, function (video, callbackEach) {
      video.remove(callbackEach)
    }, callback)
  })
}