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