PeerTube/server/middlewares/validators/redundancy.ts

179 lines
6.4 KiB
TypeScript
Raw Normal View History

2018-09-11 16:27:07 +02:00
import * as express from 'express'
2020-01-10 10:11:28 +01:00
import { body, param, query } from 'express-validator'
import { exists, isBooleanValid, isIdOrUUIDValid, isIdValid, toBooleanOrNull, toIntOrNull } from '../../helpers/custom-validators/misc'
2018-09-11 16:27:07 +02:00
import { logger } from '../../helpers/logger'
import { areValidationErrors } from './utils'
import { VideoRedundancyModel } from '../../models/redundancy/video-redundancy'
import { isHostValid } from '../../helpers/custom-validators/servers'
import { ServerModel } from '../../models/server/server'
2019-07-23 10:40:39 +02:00
import { doesVideoExist } from '../../helpers/middlewares'
2020-01-10 10:11:28 +01:00
import { isVideoRedundancyTarget } from '@server/helpers/custom-validators/video-redundancies'
2018-09-11 16:27:07 +02:00
2019-01-29 08:37:25 +01:00
const videoFileRedundancyGetValidator = [
2018-09-11 16:27:07 +02:00
param('videoId').custom(isIdOrUUIDValid).not().isEmpty().withMessage('Should have a valid video id'),
param('resolution')
.customSanitizer(toIntOrNull)
.custom(exists).withMessage('Should have a valid resolution'),
param('fps')
.optional()
.customSanitizer(toIntOrNull)
.custom(exists).withMessage('Should have a valid fps'),
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
2019-01-29 08:37:25 +01:00
logger.debug('Checking videoFileRedundancyGetValidator parameters', { parameters: req.params })
2018-09-11 16:27:07 +02:00
if (areValidationErrors(req, res)) return
2019-03-19 09:26:50 +01:00
if (!await doesVideoExist(req.params.videoId, res)) return
2018-09-11 16:27:07 +02:00
2019-08-15 11:53:26 +02:00
const video = res.locals.videoAll
2019-10-21 14:50:55 +02:00
const paramResolution = req.params.resolution as unknown as number // We casted to int above
const paramFPS = req.params.fps as unknown as number // We casted to int above
2018-09-11 16:27:07 +02:00
const videoFile = video.VideoFiles.find(f => {
2019-10-21 14:50:55 +02:00
return f.resolution === paramResolution && (!req.params.fps || paramFPS)
2018-09-11 16:27:07 +02:00
})
if (!videoFile) return res.status(404).json({ error: 'Video file not found.' })
res.locals.videoFile = videoFile
const videoRedundancy = await VideoRedundancyModel.loadLocalByFileId(videoFile.id)
2019-01-29 08:37:25 +01:00
if (!videoRedundancy) return res.status(404).json({ error: 'Video redundancy not found.' })
res.locals.videoRedundancy = videoRedundancy
return next()
}
]
const videoPlaylistRedundancyGetValidator = [
2019-10-21 14:50:55 +02:00
param('videoId')
.custom(isIdOrUUIDValid)
.not().isEmpty().withMessage('Should have a valid video id'),
param('streamingPlaylistType')
.customSanitizer(toIntOrNull)
.custom(exists).withMessage('Should have a valid streaming playlist type'),
2019-01-29 08:37:25 +01:00
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
logger.debug('Checking videoPlaylistRedundancyGetValidator parameters', { parameters: req.params })
if (areValidationErrors(req, res)) return
2019-03-19 09:26:50 +01:00
if (!await doesVideoExist(req.params.videoId, res)) return
2019-01-29 08:37:25 +01:00
2019-08-15 11:53:26 +02:00
const video = res.locals.videoAll
2019-10-21 14:50:55 +02:00
const paramPlaylistType = req.params.streamingPlaylistType as unknown as number // We casted to int above
const videoStreamingPlaylist = video.VideoStreamingPlaylists.find(p => p.type === paramPlaylistType)
2019-01-29 08:37:25 +01:00
if (!videoStreamingPlaylist) return res.status(404).json({ error: 'Video playlist not found.' })
res.locals.videoStreamingPlaylist = videoStreamingPlaylist
const videoRedundancy = await VideoRedundancyModel.loadLocalByStreamingPlaylistId(videoStreamingPlaylist.id)
if (!videoRedundancy) return res.status(404).json({ error: 'Video redundancy not found.' })
2018-09-11 16:27:07 +02:00
res.locals.videoRedundancy = videoRedundancy
return next()
}
]
const updateServerRedundancyValidator = [
param('host').custom(isHostValid).withMessage('Should have a valid host'),
body('redundancyAllowed')
2019-07-25 16:23:44 +02:00
.customSanitizer(toBooleanOrNull)
2018-09-11 16:27:07 +02:00
.custom(isBooleanValid).withMessage('Should have a valid redundancyAllowed attribute'),
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
logger.debug('Checking updateServerRedundancy parameters', { parameters: req.params })
if (areValidationErrors(req, res)) return
const server = await ServerModel.loadByHost(req.params.host)
if (!server) {
return res
.status(404)
.json({
error: `Server ${req.params.host} not found.`
})
.end()
}
res.locals.server = server
return next()
}
]
2020-01-10 10:11:28 +01:00
const listVideoRedundanciesValidator = [
query('target')
.custom(isVideoRedundancyTarget).withMessage('Should have a valid video redundancies target'),
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
logger.debug('Checking listVideoRedundanciesValidator parameters', { parameters: req.query })
if (areValidationErrors(req, res)) return
return next()
}
]
const addVideoRedundancyValidator = [
body('videoId')
.custom(isIdValid)
.withMessage('Should have a valid video id'),
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
logger.debug('Checking addVideoRedundancyValidator parameters', { parameters: req.query })
if (areValidationErrors(req, res)) return
if (!await doesVideoExist(req.body.videoId, res, 'only-video')) return
if (res.locals.onlyVideo.remote === false) {
return res.status(400)
.json({ error: 'Cannot create a redundancy on a local video' })
.end()
}
const alreadyExists = await VideoRedundancyModel.isLocalByVideoUUIDExists(res.locals.onlyVideo.uuid)
if (alreadyExists) {
return res.status(409)
.json({ error: 'This video is already duplicated by your instance.' })
}
return next()
}
]
const removeVideoRedundancyValidator = [
param('redundancyId')
.custom(isIdValid)
.withMessage('Should have a valid redundancy id'),
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
logger.debug('Checking removeVideoRedundancyValidator parameters', { parameters: req.query })
if (areValidationErrors(req, res)) return
const redundancy = await VideoRedundancyModel.loadByIdWithVideo(parseInt(req.params.redundancyId, 10))
if (!redundancy) {
return res.status(404)
.json({ error: 'Video redundancy not found' })
.end()
}
res.locals.videoRedundancy = redundancy
return next()
}
]
2018-09-11 16:27:07 +02:00
// ---------------------------------------------------------------------------
export {
2019-01-29 08:37:25 +01:00
videoFileRedundancyGetValidator,
videoPlaylistRedundancyGetValidator,
2020-01-10 10:11:28 +01:00
updateServerRedundancyValidator,
listVideoRedundanciesValidator,
addVideoRedundancyValidator,
removeVideoRedundancyValidator
2018-09-11 16:27:07 +02:00
}