PeerTube/server/controllers/lazy-static.ts

138 lines
4.3 KiB
TypeScript
Raw Normal View History

2021-08-27 07:32:44 -05:00
import cors from 'cors'
import express from 'express'
import { VideosTorrentCache } from '@server/lib/files-cache/videos-torrent-cache'
import { MActorImage } from '@server/types/models'
2021-07-16 03:42:24 -05:00
import { HttpStatusCode } from '../../shared/models/http/http-error-codes'
import { logger } from '../helpers/logger'
import { ACTOR_IMAGES_SIZE, LAZY_STATIC_PATHS, STATIC_MAX_AGE } from '../initializers/constants'
2019-08-09 04:32:40 -05:00
import { VideosCaptionCache, VideosPreviewCache } from '../lib/files-cache'
2021-06-03 09:02:29 -05:00
import { actorImagePathUnsafeCache, pushActorImageProcessInQueue } from '../lib/local-actor'
2019-08-09 04:32:40 -05:00
import { asyncMiddleware } from '../middlewares'
2021-05-11 04:15:29 -05:00
import { ActorImageModel } from '../models/actor/actor-image'
2019-08-09 04:32:40 -05:00
const lazyStaticRouter = express.Router()
lazyStaticRouter.use(cors())
lazyStaticRouter.use(
LAZY_STATIC_PATHS.AVATARS + ':filename',
2021-04-06 04:35:56 -05:00
asyncMiddleware(getActorImage)
)
lazyStaticRouter.use(
LAZY_STATIC_PATHS.BANNERS + ':filename',
asyncMiddleware(getActorImage)
2019-08-09 04:32:40 -05:00
)
lazyStaticRouter.use(
LAZY_STATIC_PATHS.PREVIEWS + ':filename',
2019-08-09 04:32:40 -05:00
asyncMiddleware(getPreview)
)
lazyStaticRouter.use(
2021-02-15 07:08:16 -06:00
LAZY_STATIC_PATHS.VIDEO_CAPTIONS + ':filename',
2019-08-09 04:32:40 -05:00
asyncMiddleware(getVideoCaption)
)
lazyStaticRouter.use(
LAZY_STATIC_PATHS.TORRENTS + ':filename',
asyncMiddleware(getTorrent)
)
2019-08-09 04:32:40 -05:00
// ---------------------------------------------------------------------------
export {
lazyStaticRouter,
getPreview,
getVideoCaption
}
// ---------------------------------------------------------------------------
2021-06-14 09:14:45 -05:00
async function getActorImage (req: express.Request, res: express.Response, next: express.NextFunction) {
2019-08-09 04:32:40 -05:00
const filename = req.params.filename
2021-04-06 04:35:56 -05:00
if (actorImagePathUnsafeCache.has(filename)) {
return res.sendFile(actorImagePathUnsafeCache.get(filename), { maxAge: STATIC_MAX_AGE.SERVER })
2019-08-09 04:32:40 -05:00
}
2021-04-06 04:35:56 -05:00
const image = await ActorImageModel.loadByName(filename)
if (!image) return res.status(HttpStatusCode.NOT_FOUND_404).end()
2021-04-06 04:35:56 -05:00
if (image.onDisk === false) {
if (!image.fileUrl) return res.status(HttpStatusCode.NOT_FOUND_404).end()
2019-08-09 04:32:40 -05:00
2021-04-06 04:35:56 -05:00
logger.info('Lazy serve remote actor image %s.', image.fileUrl)
2019-08-09 04:32:40 -05:00
2019-08-09 08:04:36 -05:00
try {
await pushActorImageProcessInQueue({
filename: image.filename,
fileUrl: image.fileUrl,
size: getActorImageSize(image),
type: image.type
})
2019-08-09 08:04:36 -05:00
} catch (err) {
2021-04-06 04:35:56 -05:00
logger.warn('Cannot process remote actor image %s.', image.fileUrl, { err })
return res.status(HttpStatusCode.NOT_FOUND_404).end()
2019-08-09 08:04:36 -05:00
}
2019-08-09 04:32:40 -05:00
2021-04-06 04:35:56 -05:00
image.onDisk = true
image.save()
.catch(err => logger.error('Cannot save new actor image disk state.', { err }))
2019-08-09 04:32:40 -05:00
}
2021-04-06 04:35:56 -05:00
const path = image.getPath()
2019-08-09 04:32:40 -05:00
2021-04-06 04:35:56 -05:00
actorImagePathUnsafeCache.set(filename, path)
2021-06-14 09:14:45 -05:00
return res.sendFile(path, { maxAge: STATIC_MAX_AGE.LAZY_SERVER }, (err: any) => {
if (!err) return
// It seems this actor image is not on the disk anymore
if (err.status === HttpStatusCode.NOT_FOUND_404 && !image.isOwned()) {
logger.error('Cannot lazy serve actor image %s.', filename, { err })
actorImagePathUnsafeCache.delete(filename)
2021-06-14 09:14:45 -05:00
image.onDisk = false
image.save()
.catch(err => logger.error('Cannot save new actor image disk state.', { err }))
}
return next(err)
})
2019-08-09 04:32:40 -05:00
}
function getActorImageSize (image: MActorImage): { width: number, height: number } {
if (image.width && image.height) {
return {
height: image.height,
width: image.width
}
}
return ACTOR_IMAGES_SIZE[image.type][0]
}
2019-08-09 04:32:40 -05:00
async function getPreview (req: express.Request, res: express.Response) {
const result = await VideosPreviewCache.Instance.getFilePath(req.params.filename)
if (!result) return res.status(HttpStatusCode.NOT_FOUND_404).end()
2019-08-09 04:32:40 -05:00
return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.LAZY_SERVER })
2019-08-09 04:32:40 -05:00
}
async function getVideoCaption (req: express.Request, res: express.Response) {
2021-02-15 07:08:16 -06:00
const result = await VideosCaptionCache.Instance.getFilePath(req.params.filename)
if (!result) return res.status(HttpStatusCode.NOT_FOUND_404).end()
2019-08-09 04:32:40 -05:00
return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.LAZY_SERVER })
}
async function getTorrent (req: express.Request, res: express.Response) {
const result = await VideosTorrentCache.Instance.getFilePath(req.params.filename)
if (!result) return res.status(HttpStatusCode.NOT_FOUND_404).end()
2021-02-18 04:28:00 -06:00
// Torrents still use the old naming convention (video uuid + .torrent)
2019-08-09 04:32:40 -05:00
return res.sendFile(result.path, { maxAge: STATIC_MAX_AGE.SERVER })
}