]>
Commit | Line | Data |
---|---|---|
1 | import { UploadFilesForCheck } from 'express' | |
2 | import { readFile } from 'fs-extra' | |
3 | import { getFileSize } from '@shared/extra-utils' | |
4 | import { CONSTRAINTS_FIELDS, MIMETYPES, VIDEO_LANGUAGES } from '../../initializers/constants' | |
5 | import { logger } from '../logger' | |
6 | import { exists, isFileValid } from './misc' | |
7 | ||
8 | function isVideoCaptionLanguageValid (value: any) { | |
9 | return exists(value) && VIDEO_LANGUAGES[value] !== undefined | |
10 | } | |
11 | ||
12 | // MacOS sends application/octet-stream | |
13 | const videoCaptionTypesRegex = [ ...Object.keys(MIMETYPES.VIDEO_CAPTIONS.MIMETYPE_EXT), 'application/octet-stream' ] | |
14 | .map(m => `(${m})`) | |
15 | .join('|') | |
16 | ||
17 | function isVideoCaptionFile (files: UploadFilesForCheck, field: string) { | |
18 | return isFileValid({ | |
19 | files, | |
20 | mimeTypeRegex: videoCaptionTypesRegex, | |
21 | field, | |
22 | maxSize: CONSTRAINTS_FIELDS.VIDEO_CAPTIONS.CAPTION_FILE.FILE_SIZE.max | |
23 | }) | |
24 | } | |
25 | ||
26 | async function isVTTFileValid (filePath: string) { | |
27 | const size = await getFileSize(filePath) | |
28 | const content = await readFile(filePath, 'utf8') | |
29 | ||
30 | logger.debug('Checking VTT file %s', filePath, { size, content }) | |
31 | ||
32 | if (size > CONSTRAINTS_FIELDS.VIDEO_CAPTIONS.CAPTION_FILE.FILE_SIZE.max) return false | |
33 | ||
34 | return content?.startsWith('WEBVTT') | |
35 | } | |
36 | ||
37 | // --------------------------------------------------------------------------- | |
38 | ||
39 | export { | |
40 | isVideoCaptionFile, | |
41 | isVTTFileValid, | |
42 | isVideoCaptionLanguageValid | |
43 | } |