diff options
Diffstat (limited to 'server/middlewares')
-rw-r--r-- | server/middlewares/validators/redundancy.ts | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/server/middlewares/validators/redundancy.ts b/server/middlewares/validators/redundancy.ts new file mode 100644 index 000000000..d91b47574 --- /dev/null +++ b/server/middlewares/validators/redundancy.ts | |||
@@ -0,0 +1,80 @@ | |||
1 | import * as express from 'express' | ||
2 | import 'express-validator' | ||
3 | import { param, body } from 'express-validator/check' | ||
4 | import { exists, isBooleanValid, isIdOrUUIDValid, toIntOrNull } from '../../helpers/custom-validators/misc' | ||
5 | import { isVideoExist } from '../../helpers/custom-validators/videos' | ||
6 | import { logger } from '../../helpers/logger' | ||
7 | import { areValidationErrors } from './utils' | ||
8 | import { VideoModel } from '../../models/video/video' | ||
9 | import { VideoRedundancyModel } from '../../models/redundancy/video-redundancy' | ||
10 | import { isHostValid } from '../../helpers/custom-validators/servers' | ||
11 | import { getServerActor } from '../../helpers/utils' | ||
12 | import { ActorFollowModel } from '../../models/activitypub/actor-follow' | ||
13 | import { SERVER_ACTOR_NAME } from '../../initializers' | ||
14 | import { ServerModel } from '../../models/server/server' | ||
15 | |||
16 | const videoRedundancyGetValidator = [ | ||
17 | param('videoId').custom(isIdOrUUIDValid).not().isEmpty().withMessage('Should have a valid video id'), | ||
18 | param('resolution') | ||
19 | .customSanitizer(toIntOrNull) | ||
20 | .custom(exists).withMessage('Should have a valid resolution'), | ||
21 | param('fps') | ||
22 | .optional() | ||
23 | .customSanitizer(toIntOrNull) | ||
24 | .custom(exists).withMessage('Should have a valid fps'), | ||
25 | |||
26 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
27 | logger.debug('Checking videoRedundancyGetValidator parameters', { parameters: req.params }) | ||
28 | |||
29 | if (areValidationErrors(req, res)) return | ||
30 | if (!await isVideoExist(req.params.videoId, res)) return | ||
31 | |||
32 | const video: VideoModel = res.locals.video | ||
33 | const videoFile = video.VideoFiles.find(f => { | ||
34 | return f.resolution === req.params.resolution && (!req.params.fps || f.fps === req.params.fps) | ||
35 | }) | ||
36 | |||
37 | if (!videoFile) return res.status(404).json({ error: 'Video file not found.' }) | ||
38 | res.locals.videoFile = videoFile | ||
39 | |||
40 | const videoRedundancy = await VideoRedundancyModel.loadByFileId(videoFile.id) | ||
41 | if (!videoRedundancy)return res.status(404).json({ error: 'Video redundancy not found.' }) | ||
42 | res.locals.videoRedundancy = videoRedundancy | ||
43 | |||
44 | return next() | ||
45 | } | ||
46 | ] | ||
47 | |||
48 | const updateServerRedundancyValidator = [ | ||
49 | param('host').custom(isHostValid).withMessage('Should have a valid host'), | ||
50 | body('redundancyAllowed') | ||
51 | .toBoolean() | ||
52 | .custom(isBooleanValid).withMessage('Should have a valid redundancyAllowed attribute'), | ||
53 | |||
54 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
55 | logger.debug('Checking updateServerRedundancy parameters', { parameters: req.params }) | ||
56 | |||
57 | if (areValidationErrors(req, res)) return | ||
58 | |||
59 | const server = await ServerModel.loadByHost(req.params.host) | ||
60 | |||
61 | if (!server) { | ||
62 | return res | ||
63 | .status(404) | ||
64 | .json({ | ||
65 | error: `Server ${req.params.host} not found.` | ||
66 | }) | ||
67 | .end() | ||
68 | } | ||
69 | |||
70 | res.locals.server = server | ||
71 | return next() | ||
72 | } | ||
73 | ] | ||
74 | |||
75 | // --------------------------------------------------------------------------- | ||
76 | |||
77 | export { | ||
78 | videoRedundancyGetValidator, | ||
79 | updateServerRedundancyValidator | ||
80 | } | ||