From 0a6658fdcbd779ada8f3758048c326e997902d5a Mon Sep 17 00:00:00 2001 From: Chocobozzz Date: Tue, 11 Jul 2017 16:01:56 +0200 Subject: Use global uuid instead of remoteId for videos --- server/middlewares/validators/users.ts | 15 +++++++++++++-- server/middlewares/validators/videos.ts | 24 +++++++++++++++++------- 2 files changed, 30 insertions(+), 9 deletions(-) (limited to 'server/middlewares/validators') diff --git a/server/middlewares/validators/users.ts b/server/middlewares/validators/users.ts index 9db4fff77..90a46752c 100644 --- a/server/middlewares/validators/users.ts +++ b/server/middlewares/validators/users.ts @@ -1,9 +1,12 @@ import 'express-validator' import * as express from 'express' +import * as Promise from 'bluebird' +import * as validator from 'validator' import { database as db } from '../../initializers/database' import { checkErrors } from './utils' import { logger } from '../../helpers' +import { VideoInstance } from '../../models' function usersAddValidator (req: express.Request, res: express.Response, next: express.NextFunction) { req.checkBody('username', 'Should have a valid username').isUserUsernameValid() @@ -59,12 +62,20 @@ function usersUpdateValidator (req: express.Request, res: express.Response, next } function usersVideoRatingValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('videoId', 'Should have a valid video id').notEmpty().isUUID(4) + req.checkParams('videoId', 'Should have a valid video id').notEmpty().isVideoIdOrUUIDValid() logger.debug('Checking usersVideoRating parameters', { parameters: req.params }) checkErrors(req, res, function () { - db.Video.load(req.params.videoId) + let videoPromise: Promise + + if (validator.isUUID(req.params.videoId)) { + videoPromise = db.Video.loadByUUID(req.params.videoId) + } else { + videoPromise = db.Video.load(req.params.videoId) + } + + videoPromise .then(video => { if (!video) return res.status(404).send('Video not found') diff --git a/server/middlewares/validators/videos.ts b/server/middlewares/validators/videos.ts index 013466487..0a88e064e 100644 --- a/server/middlewares/validators/videos.ts +++ b/server/middlewares/validators/videos.ts @@ -1,10 +1,13 @@ import 'express-validator' import * as express from 'express' +import * as Promise from 'bluebird' +import * as validator from 'validator' import { database as db } from '../../initializers/database' import { checkErrors } from './utils' import { CONSTRAINTS_FIELDS, SEARCHABLE_COLUMNS } from '../../initializers' import { logger, isVideoDurationValid } from '../../helpers' +import { VideoInstance } from '../../models' function videosAddValidator (req: express.Request, res: express.Response, next: express.NextFunction) { // FIXME: Don't write an error message, it seems there is a bug with express-validator @@ -40,7 +43,7 @@ function videosAddValidator (req: express.Request, res: express.Response, next: } function videosUpdateValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('id', 'Should have a valid id').notEmpty().isUUID(4) + req.checkParams('id', 'Should have a valid id').notEmpty().isVideoIdOrUUIDValid() req.checkBody('name', 'Should have a valid name').optional().isVideoNameValid() req.checkBody('category', 'Should have a valid category').optional().isVideoCategoryValid() req.checkBody('licence', 'Should have a valid licence').optional().isVideoLicenceValid() @@ -68,7 +71,7 @@ function videosUpdateValidator (req: express.Request, res: express.Response, nex } function videosGetValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('id', 'Should have a valid id').notEmpty().isUUID(4) + req.checkParams('id', 'Should have a valid id').notEmpty().isVideoIdOrUUIDValid() logger.debug('Checking videosGet parameters', { parameters: req.params }) @@ -78,7 +81,7 @@ function videosGetValidator (req: express.Request, res: express.Response, next: } function videosRemoveValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('id', 'Should have a valid id').notEmpty().isUUID(4) + req.checkParams('id', 'Should have a valid id').notEmpty().isVideoIdOrUUIDValid() logger.debug('Checking videosRemove parameters', { parameters: req.params }) @@ -105,7 +108,7 @@ function videosSearchValidator (req: express.Request, res: express.Response, nex } function videoAbuseReportValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('id', 'Should have a valid id').notEmpty().isUUID(4) + req.checkParams('id', 'Should have a valid id').notEmpty().isVideoIdOrUUIDValid() req.checkBody('reason', 'Should have a valid reason').isVideoAbuseReasonValid() logger.debug('Checking videoAbuseReport parameters', { parameters: req.body }) @@ -116,7 +119,7 @@ function videoAbuseReportValidator (req: express.Request, res: express.Response, } function videoRateValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('id', 'Should have a valid id').notEmpty().isUUID(4) + req.checkParams('id', 'Should have a valid id').notEmpty().isVideoIdOrUUIDValid() req.checkBody('rating', 'Should have a valid rate type').isVideoRatingTypeValid() logger.debug('Checking videoRate parameters', { parameters: req.body }) @@ -127,7 +130,7 @@ function videoRateValidator (req: express.Request, res: express.Response, next: } function videosBlacklistValidator (req: express.Request, res: express.Response, next: express.NextFunction) { - req.checkParams('id', 'Should have a valid id').notEmpty().isUUID(4) + req.checkParams('id', 'Should have a valid id').notEmpty().isVideoIdOrUUIDValid() logger.debug('Checking videosBlacklist parameters', { parameters: req.params }) @@ -157,7 +160,14 @@ export { // --------------------------------------------------------------------------- function checkVideoExists (id: string, res: express.Response, callback: () => void) { - db.Video.loadAndPopulateAuthorAndPodAndTags(id).then(video => { + let promise: Promise + if (validator.isInt(id)) { + promise = db.Video.loadAndPopulateAuthorAndPodAndTags(+id) + } else { // UUID + promise = db.Video.loadByUUIDAndPopulateAuthorAndPodAndTags(id) + } + + promise.then(video => { if (!video) return res.status(404).send('Video not found') res.locals.video = video -- cgit v1.2.3