server/controllers/api/videos/live.ts

   1 import * as express from 'express'
   2 import { v4 as uuidv4 } from 'uuid'
   3 import { createReqFiles } from '@server/helpers/express-utils'
   4 import { CONFIG } from '@server/initializers/config'
   5 import { ASSETS_PATH, MIMETYPES } from '@server/initializers/constants'
   6 import { getVideoActivityPubUrl } from '@server/lib/activitypub/url'
   7 import { buildLocalVideoFromReq, buildVideoThumbnailsFromReq, setVideoTags } from '@server/lib/video'
   8 import { videoLiveAddValidator, videoLiveGetValidator, videoLiveUpdateValidator } from '@server/middlewares/validators/videos/video-live'
   9 import { VideoLiveModel } from '@server/models/video/video-live'
  10 import { MVideoDetails, MVideoFullLight } from '@server/types/models'
  11 import { LiveVideoCreate, LiveVideoUpdate, VideoState } from '../../../../shared'
  12 import { logger } from '../../../helpers/logger'
  13 import { sequelizeTypescript } from '../../../initializers/database'
  14 import { createVideoMiniatureFromExisting } from '../../../lib/thumbnail'
  15 import { asyncMiddleware, asyncRetryTransactionMiddleware, authenticate } from '../../../middlewares'
  16 import { VideoModel } from '../../../models/video/video'
  17
  18 const liveRouter = express.Router()
  19
  20 const reqVideoFileLive = createReqFiles(
  21   [ 'thumbnailfile', 'previewfile' ],
  22   MIMETYPES.IMAGE.MIMETYPE_EXT,
  23   {
  24     thumbnailfile: CONFIG.STORAGE.TMP_DIR,
  25     previewfile: CONFIG.STORAGE.TMP_DIR
  26   }
  27 )
  28
  29 liveRouter.post('/live',
  30   authenticate,
  31   reqVideoFileLive,
  32   asyncMiddleware(videoLiveAddValidator),
  33   asyncRetryTransactionMiddleware(addLiveVideo)
  34 )
  35
  36 liveRouter.get('/live/:videoId',
  37   authenticate,
  38   asyncMiddleware(videoLiveGetValidator),
  39   asyncRetryTransactionMiddleware(getLiveVideo)
  40 )
  41
  42 liveRouter.put('/live/:videoId',
  43   authenticate,
  44   asyncMiddleware(videoLiveGetValidator),
  45   videoLiveUpdateValidator,
  46   asyncRetryTransactionMiddleware(updateLiveVideo)
  47 )
  48
  49 // ---------------------------------------------------------------------------
  50
  51 export {
  52   liveRouter
  53 }
  54
  55 // ---------------------------------------------------------------------------
  56
  57 async function getLiveVideo (req: express.Request, res: express.Response) {
  58   const videoLive = res.locals.videoLive
  59
  60   return res.json(videoLive.toFormattedJSON())
  61 }
  62
  63 async function updateLiveVideo (req: express.Request, res: express.Response) {
  64   const body: LiveVideoUpdate = req.body
  65
  66   const videoLive = res.locals.videoLive
  67   videoLive.saveReplay = body.saveReplay || false
  68
  69   await videoLive.save()
  70
  71   return res.sendStatus(204)
  72 }
  73
  74 async function addLiveVideo (req: express.Request, res: express.Response) {
  75   const videoInfo: LiveVideoCreate = req.body
  76
  77   // Prepare data so we don't block the transaction
  78   const videoData = buildLocalVideoFromReq(videoInfo, res.locals.videoChannel.id)
  79   videoData.isLive = true
  80   videoData.state = VideoState.WAITING_FOR_LIVE
  81   videoData.duration = 0
  82
  83   const video = new VideoModel(videoData) as MVideoDetails
  84   video.url = getVideoActivityPubUrl(video) // We use the UUID, so set the URL after building the object
  85
  86   const videoLive = new VideoLiveModel()
  87   videoLive.saveReplay = videoInfo.saveReplay || false
  88   videoLive.streamKey = uuidv4()
  89
  90   const [ thumbnailModel, previewModel ] = await buildVideoThumbnailsFromReq({
  91     video,
  92     files: req.files,
  93     fallback: type => {
  94       return createVideoMiniatureFromExisting({
  95         inputPath: ASSETS_PATH.DEFAULT_LIVE_BACKGROUND,
  96         video,
  97         type,
  98         automaticallyGenerated: true,
  99         keepOriginal: true
 100       })
 101     }
 102   })
 103
 104   const { videoCreated } = await sequelizeTypescript.transaction(async t => {
 105     const sequelizeOptions = { transaction: t }
 106
 107     const videoCreated = await video.save(sequelizeOptions) as MVideoFullLight
 108
 109     if (thumbnailModel) await videoCreated.addAndSaveThumbnail(thumbnailModel, t)
 110     if (previewModel) await videoCreated.addAndSaveThumbnail(previewModel, t)
 111
 112     // Do not forget to add video channel information to the created video
 113     videoCreated.VideoChannel = res.locals.videoChannel
 114
 115     videoLive.videoId = videoCreated.id
 116     await videoLive.save(sequelizeOptions)
 117
 118     await setVideoTags({ video, tags: videoInfo.tags, transaction: t })
 119
 120     logger.info('Video live %s with uuid %s created.', videoInfo.name, videoCreated.uuid)
 121
 122     return { videoCreated }
 123   })
 124
 125   return res.json({
 126     video: {
 127       id: videoCreated.id,
 128       uuid: videoCreated.uuid
 129     }
 130   })
 131 }