mirror of https://github.com/Chocobozzz/PeerTube
61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import express from 'express'
|
|
import { param } from 'express-validator'
|
|
import { basename } from 'path'
|
|
import { isSafeFilename } from '@server/helpers/custom-validators/misc'
|
|
import { hasVideoStudioTaskFile, HttpStatusCode, RunnerJobStudioTranscodingPayload } from '@shared/models'
|
|
import { areValidationErrors, doesVideoExist, isValidVideoIdParam } from '../shared'
|
|
|
|
const tags = [ 'runner' ]
|
|
|
|
export const runnerJobGetVideoTranscodingFileValidator = [
|
|
isValidVideoIdParam('videoId'),
|
|
|
|
async (req: express.Request, res: express.Response, next: express.NextFunction) => {
|
|
if (areValidationErrors(req, res)) return
|
|
|
|
if (!await doesVideoExist(req.params.videoId, res, 'all')) return
|
|
|
|
const runnerJob = res.locals.runnerJob
|
|
|
|
if (runnerJob.privatePayload.videoUUID !== res.locals.videoAll.uuid) {
|
|
return res.fail({
|
|
status: HttpStatusCode.FORBIDDEN_403,
|
|
message: 'Job is not associated to this video',
|
|
tags: [ ...tags, res.locals.videoAll.uuid ]
|
|
})
|
|
}
|
|
|
|
return next()
|
|
}
|
|
]
|
|
|
|
export const runnerJobGetVideoStudioTaskFileValidator = [
|
|
param('filename').custom(v => isSafeFilename(v)),
|
|
|
|
(req: express.Request, res: express.Response, next: express.NextFunction) => {
|
|
if (areValidationErrors(req, res)) return
|
|
|
|
const filename = req.params.filename
|
|
|
|
const payload = res.locals.runnerJob.payload as RunnerJobStudioTranscodingPayload
|
|
|
|
const found = Array.isArray(payload?.tasks) && payload.tasks.some(t => {
|
|
if (hasVideoStudioTaskFile(t)) {
|
|
return basename(t.options.file) === filename
|
|
}
|
|
|
|
return false
|
|
})
|
|
|
|
if (!found) {
|
|
return res.fail({
|
|
status: HttpStatusCode.BAD_REQUEST_400,
|
|
message: 'File is not associated to this edition task',
|
|
tags: [ ...tags, res.locals.videoAll.uuid ]
|
|
})
|
|
}
|
|
|
|
return next()
|
|
}
|
|
]
|