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.onDisk === false) {
  48     if (!avatar.fileUrl) return res.sendStatus(404)
  49
  50     logger.info('Lazy serve remote avatar image %s.', avatar.fileUrl)
  51
  52     try {
  53       await pushAvatarProcessInQueue({ filename: avatar.filename, fileUrl: avatar.fileUrl })
  54     } catch (err) {
  55       logger.warn('Cannot process remote avatar %s.', avatar.fileUrl, { err })
  56       return res.sendStatus(404)
  57     }
  58
  59     avatar.onDisk = true
  60     avatar.save()
  61       .catch(err => logger.error('Cannot save new avatar disk state.', { err }))
  62   }
  63
  64   const path = avatar.getPath()
  65
  66   avatarPathUnsafeCache.set(filename, path)
  67   return res.sendFile(path, { maxAge: STATIC_MAX_AGE.SERVER })
  68 }
  69
  70 async function getPreview (req: express.Request, res: express.Response) {
  71   const result = await VideosPreviewCache.Instance.getFilePath(req.params.uuid)
  72   if (!result) return res.sendStatus(404)
  73
  74   return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
  75 }
  76
  77 async function getVideoCaption (req: express.Request, res: express.Response) {
  78   const result = await VideosCaptionCache.Instance.getFilePath({
  79     videoId: req.params.videoId,
  80     language: req.params.captionLanguage
  81   })
  82   if (!result) return res.sendStatus(404)
  83
  84   return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
  85 }