server/controllers/lazy-static.ts

   1 import * as cors from 'cors'
   2 import * as express from 'express'
   3 import { LAZY_STATIC_PATHS, STATIC_MAX_AGE } from '../initializers/constants'
   4 import { VideosCaptionCache, VideosPreviewCache } from '../lib/files-cache'
   5 import { asyncMiddleware } from '../middlewares'
   6 import { AvatarModel } from '../models/avatar/avatar'
   7 import { logger } from '../helpers/logger'
   8 import { avatarPathUnsafeCache, pushAvatarProcessInQueue } from '../lib/avatar'
   9
  10 const lazyStaticRouter = express.Router()
  11
  12 lazyStaticRouter.use(cors())
  13
  14 lazyStaticRouter.use(
  15   LAZY_STATIC_PATHS.AVATARS + ':filename',
  16   asyncMiddleware(getAvatar)
  17 )
  18
  19 lazyStaticRouter.use(
  20   LAZY_STATIC_PATHS.PREVIEWS + ':uuid.jpg',
  21   asyncMiddleware(getPreview)
  22 )
  23
  24 lazyStaticRouter.use(
  25   LAZY_STATIC_PATHS.VIDEO_CAPTIONS + ':videoId-:captionLanguage([a-z]+).vtt',
  26   asyncMiddleware(getVideoCaption)
  27 )
  28
  29 // ---------------------------------------------------------------------------
  30
  31 export {
  32   lazyStaticRouter,
  33   getPreview,
  34   getVideoCaption
  35 }
  36
  37 // ---------------------------------------------------------------------------
  38
  39 async function getAvatar (req: express.Request, res: express.Response) {
  40   const filename = req.params.filename
  41
  42   if (avatarPathUnsafeCache.has(filename)) {
  43     return res.sendFile(avatarPathUnsafeCache.get(filename), { maxAge: STATIC_MAX_AGE.SERVER })
  44   }
  45
  46   const avatar = await AvatarModel.loadByName(filename)
  47   if (!avatar) return res.sendStatus(404)
  48
  49   if (avatar.onDisk === false) {
  50     if (!avatar.fileUrl) return res.sendStatus(404)
  51
  52     logger.info('Lazy serve remote avatar image %s.', avatar.fileUrl)
  53
  54     try {
  55       await pushAvatarProcessInQueue({ filename: avatar.filename, fileUrl: avatar.fileUrl })
  56     } catch (err) {
  57       logger.warn('Cannot process remote avatar %s.', avatar.fileUrl, { err })
  58       return res.sendStatus(404)
  59     }
  60
  61     avatar.onDisk = true
  62     avatar.save()
  63       .catch(err => logger.error('Cannot save new avatar disk state.', { err }))
  64   }
  65
  66   const path = avatar.getPath()
  67
  68   avatarPathUnsafeCache.set(filename, path)
  69   return res.sendFile(path, { maxAge: STATIC_MAX_AGE.SERVER })
  70 }
  71
  72 async function getPreview (req: express.Request, res: express.Response) {
  73   const result = await VideosPreviewCache.Instance.getFilePath(req.params.uuid)
  74   if (!result) return res.sendStatus(404)
  75
  76   return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
  77 }
  78
  79 async function getVideoCaption (req: express.Request, res: express.Response) {
  80   const result = await VideosCaptionCache.Instance.getFilePath({
  81     videoId: req.params.videoId,
  82     language: req.params.captionLanguage
  83   })
  84   if (!result) return res.sendStatus(404)
  85
  86   return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
  87 }