server/controllers/api/videos/editor.ts

   1 import express from 'express'
   2 import { createAnyReqFiles } from '@server/helpers/express-utils'
   3 import { CONFIG } from '@server/initializers/config'
   4 import { MIMETYPES } from '@server/initializers/constants'
   5 import { JobQueue } from '@server/lib/job-queue'
   6 import { buildTaskFileFieldname, getTaskFile } from '@server/lib/video-editor'
   7 import {
   8   HttpStatusCode,
   9   VideoEditionTaskPayload,
  10   VideoEditorCreateEdition,
  11   VideoEditorTask,
  12   VideoEditorTaskCut,
  13   VideoEditorTaskIntro,
  14   VideoEditorTaskOutro,
  15   VideoEditorTaskWatermark,
  16   VideoState
  17 } from '@shared/models'
  18 import { asyncMiddleware, authenticate, videosEditorAddEditionValidator } from '../../../middlewares'
  19
  20 const editorRouter = express.Router()
  21
  22 const tasksFiles = createAnyReqFiles(
  23   MIMETYPES.VIDEO.MIMETYPE_EXT,
  24   CONFIG.STORAGE.TMP_DIR,
  25   (req: express.Request, file: Express.Multer.File, cb: (err: Error, result?: boolean) => void) => {
  26     const body = req.body as VideoEditorCreateEdition
  27
  28     // Fetch array element
  29     const matches = file.fieldname.match(/tasks\[(\d+)\]/)
  30     if (!matches) return cb(new Error('Cannot find array element indice for ' + file.fieldname))
  31
  32     const indice = parseInt(matches[1])
  33     const task = body.tasks[indice]
  34
  35     if (!task) return cb(new Error('Cannot find array element of indice ' + indice + ' for ' + file.fieldname))
  36
  37     if (
  38       [ 'add-intro', 'add-outro', 'add-watermark' ].includes(task.name) &&
  39       file.fieldname === buildTaskFileFieldname(indice)
  40     ) {
  41       return cb(null, true)
  42     }
  43
  44     return cb(null, false)
  45   }
  46 )
  47
  48 editorRouter.post('/:videoId/editor/edit',
  49   authenticate,
  50   tasksFiles,
  51   asyncMiddleware(videosEditorAddEditionValidator),
  52   asyncMiddleware(createEditionTasks)
  53 )
  54
  55 // ---------------------------------------------------------------------------
  56
  57 export {
  58   editorRouter
  59 }
  60
  61 // ---------------------------------------------------------------------------
  62
  63 async function createEditionTasks (req: express.Request, res: express.Response) {
  64   const files = req.files as Express.Multer.File[]
  65   const body = req.body as VideoEditorCreateEdition
  66   const video = res.locals.videoAll
  67
  68   video.state = VideoState.TO_EDIT
  69   await video.save()
  70
  71   const payload = {
  72     videoUUID: video.uuid,
  73     tasks: body.tasks.map((t, i) => buildTaskPayload(t, i, files))
  74   }
  75
  76   JobQueue.Instance.createJob({ type: 'video-edition', payload })
  77
  78   return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
  79 }
  80
  81 const taskPayloadBuilders: {
  82   [id in VideoEditorTask['name']]: (task: VideoEditorTask, indice?: number, files?: Express.Multer.File[]) => VideoEditionTaskPayload
  83 } = {
  84   'add-intro': buildIntroOutroTask,
  85   'add-outro': buildIntroOutroTask,
  86   'cut': buildCutTask,
  87   'add-watermark': buildWatermarkTask
  88 }
  89
  90 function buildTaskPayload (task: VideoEditorTask, indice: number, files: Express.Multer.File[]): VideoEditionTaskPayload {
  91   return taskPayloadBuilders[task.name](task, indice, files)
  92 }
  93
  94 function buildIntroOutroTask (task: VideoEditorTaskIntro | VideoEditorTaskOutro, indice: number, files: Express.Multer.File[]) {
  95   return {
  96     name: task.name,
  97     options: {
  98       file: getTaskFile(files, indice).path
  99     }
 100   }
 101 }
 102
 103 function buildCutTask (task: VideoEditorTaskCut) {
 104   return {
 105     name: task.name,
 106     options: {
 107       start: task.options.start,
 108       end: task.options.end
 109     }
 110   }
 111 }
 112
 113 function buildWatermarkTask (task: VideoEditorTaskWatermark, indice: number, files: Express.Multer.File[]) {
 114   return {
 115     name: task.name,
 116     options: {
 117       file: getTaskFile(files, indice).path
 118     }
 119   }
 120 }