mirror of https://github.com/Chocobozzz/PeerTube
117 lines
4.1 KiB
TypeScript
117 lines
4.1 KiB
TypeScript
import { HttpStatusCode, VideoCaptionGenerate } from '@peertube/peertube-models'
|
|
import { Hooks } from '@server/lib/plugins/hooks.js'
|
|
import { createLocalCaption, createTranscriptionTaskIfNeeded } from '@server/lib/video-captions.js'
|
|
import { VideoJobInfoModel } from '@server/models/video/video-job-info.js'
|
|
import express from 'express'
|
|
import { createReqFiles } from '../../../helpers/express-utils.js'
|
|
import { logger, loggerTagsFactory } from '../../../helpers/logger.js'
|
|
import { getFormattedObjects } from '../../../helpers/utils.js'
|
|
import { MIMETYPES } from '../../../initializers/constants.js'
|
|
import { sequelizeTypescript } from '../../../initializers/database.js'
|
|
import { federateVideoIfNeeded } from '../../../lib/activitypub/videos/index.js'
|
|
import { asyncMiddleware, asyncRetryTransactionMiddleware, authenticate } from '../../../middlewares/index.js'
|
|
import {
|
|
addVideoCaptionValidator,
|
|
deleteVideoCaptionValidator,
|
|
generateVideoCaptionValidator,
|
|
listVideoCaptionsValidator
|
|
} from '../../../middlewares/validators/index.js'
|
|
import { VideoCaptionModel } from '../../../models/video/video-caption.js'
|
|
|
|
const lTags = loggerTagsFactory('api', 'video-caption')
|
|
|
|
const reqVideoCaptionAdd = createReqFiles([ 'captionfile' ], MIMETYPES.VIDEO_CAPTIONS.MIMETYPE_EXT)
|
|
|
|
const videoCaptionsRouter = express.Router()
|
|
|
|
videoCaptionsRouter.post('/:videoId/captions/generate',
|
|
authenticate,
|
|
asyncMiddleware(generateVideoCaptionValidator),
|
|
asyncMiddleware(createGenerateVideoCaption)
|
|
)
|
|
|
|
videoCaptionsRouter.get('/:videoId/captions',
|
|
asyncMiddleware(listVideoCaptionsValidator),
|
|
asyncMiddleware(listVideoCaptions)
|
|
)
|
|
|
|
videoCaptionsRouter.put('/:videoId/captions/:captionLanguage',
|
|
authenticate,
|
|
reqVideoCaptionAdd,
|
|
asyncMiddleware(addVideoCaptionValidator),
|
|
asyncRetryTransactionMiddleware(createVideoCaption)
|
|
)
|
|
|
|
videoCaptionsRouter.delete('/:videoId/captions/:captionLanguage',
|
|
authenticate,
|
|
asyncMiddleware(deleteVideoCaptionValidator),
|
|
asyncRetryTransactionMiddleware(deleteVideoCaption)
|
|
)
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
videoCaptionsRouter
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
async function createGenerateVideoCaption (req: express.Request, res: express.Response) {
|
|
const video = res.locals.videoAll
|
|
|
|
const body = req.body as VideoCaptionGenerate
|
|
if (body.forceTranscription === true) {
|
|
await VideoJobInfoModel.abortAllTasks(video.uuid, 'pendingTranscription')
|
|
}
|
|
|
|
await createTranscriptionTaskIfNeeded(video)
|
|
|
|
return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
|
|
}
|
|
|
|
async function listVideoCaptions (req: express.Request, res: express.Response) {
|
|
const data = await VideoCaptionModel.listVideoCaptions(res.locals.onlyVideo.id)
|
|
|
|
return res.json(getFormattedObjects(data, data.length))
|
|
}
|
|
|
|
async function createVideoCaption (req: express.Request, res: express.Response) {
|
|
const videoCaptionPhysicalFile: Express.Multer.File = req.files['captionfile'][0]
|
|
const video = res.locals.videoAll
|
|
|
|
const captionLanguage = req.params.captionLanguage
|
|
|
|
const videoCaption = await createLocalCaption({
|
|
video,
|
|
language: captionLanguage,
|
|
path: videoCaptionPhysicalFile.path,
|
|
automaticallyGenerated: false
|
|
})
|
|
|
|
await sequelizeTypescript.transaction(async t => {
|
|
await federateVideoIfNeeded(video, false, t)
|
|
})
|
|
|
|
Hooks.runAction('action:api.video-caption.created', { caption: videoCaption, req, res })
|
|
|
|
return res.status(HttpStatusCode.NO_CONTENT_204).end()
|
|
}
|
|
|
|
async function deleteVideoCaption (req: express.Request, res: express.Response) {
|
|
const video = res.locals.videoAll
|
|
const videoCaption = res.locals.videoCaption
|
|
|
|
await sequelizeTypescript.transaction(async t => {
|
|
await videoCaption.destroy({ transaction: t })
|
|
|
|
// Send video update
|
|
await federateVideoIfNeeded(video, false, t)
|
|
})
|
|
|
|
logger.info('Video caption %s of video %s deleted.', videoCaption.language, video.uuid, lTags(video.uuid))
|
|
|
|
Hooks.runAction('action:api.video-caption.deleted', { caption: videoCaption, req, res })
|
|
|
|
return res.type('json').status(HttpStatusCode.NO_CONTENT_204).end()
|
|
}
|