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

   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 { exists, isFileValid } from './misc'
   6
   7 function isVideoCaptionLanguageValid (value: any) {
   8   return exists(value) && VIDEO_LANGUAGES[value] !== undefined
   9 }
  10
  11 const videoCaptionTypesRegex = Object.keys(MIMETYPES.VIDEO_CAPTIONS.MIMETYPE_EXT)
  12                                 .concat([ 'application/octet-stream' ]) // MacOS sends application/octet-stream
  13                                 .map(m => `(${m})`)
  14                                 .join('|')
  15 function isVideoCaptionFile (files: UploadFilesForCheck, field: string) {
  16   return isFileValid({
  17     files,
  18     mimeTypeRegex: videoCaptionTypesRegex,
  19     field,
  20     maxSize: CONSTRAINTS_FIELDS.VIDEO_CAPTIONS.CAPTION_FILE.FILE_SIZE.max
  21   })
  22 }
  23
  24 async function isVTTFileValid (filePath: string) {
  25   const size = await getFileSize(filePath)
  26
  27   if (size > CONSTRAINTS_FIELDS.VIDEO_CAPTIONS.CAPTION_FILE.FILE_SIZE.max) return false
  28
  29   const content = await readFile(filePath, 'utf8')
  30
  31   return content?.startsWith('WEBVTT\n')
  32 }
  33
  34 // ---------------------------------------------------------------------------
  35
  36 export {
  37   isVideoCaptionFile,
  38   isVTTFileValid,
  39   isVideoCaptionLanguageValid
  40 }