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 import { HttpStatusCode } from '../../shared/core-utils/miscs/http-error-codes'
  10
  11 const lazyStaticRouter = express.Router()
  12
  13 lazyStaticRouter.use(cors())
  14
  15 lazyStaticRouter.use(
  16   LAZY_STATIC_PATHS.AVATARS + ':filename',
  17   asyncMiddleware(getAvatar)
  18 )
  19
  20 lazyStaticRouter.use(
  21   LAZY_STATIC_PATHS.PREVIEWS + ':filename',
  22   asyncMiddleware(getPreview)
  23 )
  24
  25 lazyStaticRouter.use(
  26   LAZY_STATIC_PATHS.VIDEO_CAPTIONS + ':filename',
  27   asyncMiddleware(getVideoCaption)
  28 )
  29
  30 // ---------------------------------------------------------------------------
  31
  32 export {
  33   lazyStaticRouter,
  34   getPreview,
  35   getVideoCaption
  36 }
  37
  38 // ---------------------------------------------------------------------------
  39
  40 async function getAvatar (req: express.Request, res: express.Response) {
  41   const filename = req.params.filename
  42
  43   if (avatarPathUnsafeCache.has(filename)) {
  44     return res.sendFile(avatarPathUnsafeCache.get(filename), { maxAge: STATIC_MAX_AGE.SERVER })
  45   }
  46
  47   const avatar = await AvatarModel.loadByName(filename)
  48   if (!avatar) return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
  49
  50   if (avatar.onDisk === false) {
  51     if (!avatar.fileUrl) return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
  52
  53     logger.info('Lazy serve remote avatar image %s.', avatar.fileUrl)
  54
  55     try {
  56       await pushAvatarProcessInQueue({ filename: avatar.filename, fileUrl: avatar.fileUrl })
  57     } catch (err) {
  58       logger.warn('Cannot process remote avatar %s.', avatar.fileUrl, { err })
  59       return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
  60     }
  61
  62     avatar.onDisk = true
  63     avatar.save()
  64       .catch(err => logger.error('Cannot save new avatar disk state.', { err }))
  65   }
  66
  67   const path = avatar.getPath()
  68
  69   avatarPathUnsafeCache.set(filename, path)
  70   return res.sendFile(path, { maxAge: STATIC_MAX_AGE.SERVER })
  71 }
  72
  73 async function getPreview (req: express.Request, res: express.Response) {
  74   const result = await VideosPreviewCache.Instance.getFilePath(req.params.filename)
  75   if (!result) return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
  76
  77   return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
  78 }
  79
  80 async function getVideoCaption (req: express.Request, res: express.Response) {
  81   const result = await VideosCaptionCache.Instance.getFilePath(req.params.filename)
  82   if (!result) return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
  83
  84   return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
  85 }