server/helpers/custom-validators/video-captions.ts

   1 import { getFileSize } from '@shared/extra-utils'
   2 import { readFile } from 'fs-extra'
   3 import { CONSTRAINTS_FIELDS, MIMETYPES, VIDEO_LANGUAGES } from '../../initializers/constants'
   4 import { exists, isFileValid } from './misc'
   5
   6 function isVideoCaptionLanguageValid (value: any) {
   7   return exists(value) && VIDEO_LANGUAGES[value] !== undefined
   8 }
   9
  10 const videoCaptionTypesRegex = Object.keys(MIMETYPES.VIDEO_CAPTIONS.MIMETYPE_EXT)
  11                                 .concat([ 'application/octet-stream' ]) // MacOS sends application/octet-stream
  12                                 .map(m => `(${m})`)
  13                                 .join('|')
  14 function isVideoCaptionFile (files: { [ fieldname: string ]: Express.Multer.File[] } | Express.Multer.File[], field: string) {
  15   return isFileValid(files, videoCaptionTypesRegex, field, CONSTRAINTS_FIELDS.VIDEO_CAPTIONS.CAPTION_FILE.FILE_SIZE.max)
  16 }
  17
  18 async function isVTTFileValid (filePath: string) {
  19   const size = await getFileSize(filePath)
  20
  21   if (size > CONSTRAINTS_FIELDS.VIDEO_CAPTIONS.CAPTION_FILE.FILE_SIZE.max) return false
  22
  23   const content = await readFile(filePath, 'utf8')
  24
  25   return content?.startsWith('WEBVTT\n')
  26 }
  27
  28 // ---------------------------------------------------------------------------
  29
  30 export {
  31   isVideoCaptionFile,
  32   isVTTFileValid,
  33   isVideoCaptionLanguageValid
  34 }