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