]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/helpers/custom-validators/video-imports.ts
emit more specific status codes on video upload (#3423)
[github/Chocobozzz/PeerTube.git] / server / helpers / custom-validators / video-imports.ts
1 import 'multer'
2 import validator from 'validator'
3 import { CONSTRAINTS_FIELDS, MIMETYPES, VIDEO_IMPORT_STATES } from '../../initializers/constants'
4 import { exists, isFileValid } from './misc'
5 import * as express from 'express'
6 import { VideoImportModel } from '../../models/video/video-import'
7 import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes'
8
9 function isVideoImportTargetUrlValid (url: string) {
10 const isURLOptions = {
11 require_host: true,
12 require_tld: true,
13 require_protocol: true,
14 require_valid_protocol: true,
15 protocols: [ 'http', 'https' ]
16 }
17
18 return exists(url) &&
19 validator.isURL('' + url, isURLOptions) &&
20 validator.isLength('' + url, CONSTRAINTS_FIELDS.VIDEO_IMPORTS.URL)
21 }
22
23 function isVideoImportStateValid (value: any) {
24 return exists(value) && VIDEO_IMPORT_STATES[value] !== undefined
25 }
26
27 const videoTorrentImportRegex = Object.keys(MIMETYPES.TORRENT.MIMETYPE_EXT)
28 .concat([ 'application/octet-stream' ]) // MacOS sends application/octet-stream
29 .map(m => `(${m})`)
30 .join('|')
31 function isVideoImportTorrentFile (files: { [ fieldname: string ]: Express.Multer.File[] } | Express.Multer.File[]) {
32 return isFileValid(files, videoTorrentImportRegex, 'torrentfile', CONSTRAINTS_FIELDS.VIDEO_IMPORTS.TORRENT_FILE.FILE_SIZE.max, true)
33 }
34
35 async function doesVideoImportExist (id: number, res: express.Response) {
36 const videoImport = await VideoImportModel.loadAndPopulateVideo(id)
37
38 if (!videoImport) {
39 res.status(HttpStatusCode.NOT_FOUND_404)
40 .json({ error: 'Video import not found' })
41 .end()
42
43 return false
44 }
45
46 res.locals.videoImport = videoImport
47 return true
48 }
49
50 // ---------------------------------------------------------------------------
51
52 export {
53 isVideoImportStateValid,
54 isVideoImportTargetUrlValid,
55 doesVideoImportExist,
56 isVideoImportTorrentFile
57 }