123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- import * as express from 'express'
- import { buildNSFWFilter } from '../../helpers/express-utils'
- import { VideoModel } from '../../models/video/video'
- import { asyncMiddleware, optionalAuthenticate, videosOverviewValidator } from '../../middlewares'
- import { TagModel } from '../../models/video/tag'
- import { CategoryOverview, ChannelOverview, TagOverview, VideosOverview } from '../../../shared/models/overviews'
- import { MEMOIZE_TTL, OVERVIEWS } from '../../initializers/constants'
- import * as memoizee from 'memoizee'
- import { logger } from '@server/helpers/logger'
- const overviewsRouter = express.Router()
- overviewsRouter.get('/videos',
- videosOverviewValidator,
- optionalAuthenticate,
- asyncMiddleware(getVideosOverview)
- )
- // ---------------------------------------------------------------------------
- export { overviewsRouter }
- // ---------------------------------------------------------------------------
- const buildSamples = memoizee(async function () {
- const [ categories, channels, tags ] = await Promise.all([
- VideoModel.getRandomFieldSamples('category', OVERVIEWS.VIDEOS.SAMPLE_THRESHOLD, OVERVIEWS.VIDEOS.SAMPLES_COUNT),
- VideoModel.getRandomFieldSamples('channelId', OVERVIEWS.VIDEOS.SAMPLE_THRESHOLD, OVERVIEWS.VIDEOS.SAMPLES_COUNT),
- TagModel.getRandomSamples(OVERVIEWS.VIDEOS.SAMPLE_THRESHOLD, OVERVIEWS.VIDEOS.SAMPLES_COUNT)
- ])
- const result = { categories, channels, tags }
- logger.debug('Building samples for overview endpoint.', { result })
- return result
- }, { maxAge: MEMOIZE_TTL.OVERVIEWS_SAMPLE })
- // This endpoint could be quite long, but we cache it
- async function getVideosOverview (req: express.Request, res: express.Response) {
- const attributes = await buildSamples()
- const page = req.query.page || 1
- const index = page - 1
- const categories: CategoryOverview[] = []
- const channels: ChannelOverview[] = []
- const tags: TagOverview[] = []
- await Promise.all([
- getVideosByCategory(attributes.categories, index, res, categories),
- getVideosByChannel(attributes.channels, index, res, channels),
- getVideosByTag(attributes.tags, index, res, tags)
- ])
- const result: VideosOverview = {
- categories,
- channels,
- tags
- }
- return res.json(result)
- }
- async function getVideosByTag (tagsSample: string[], index: number, res: express.Response, acc: TagOverview[]) {
- if (tagsSample.length <= index) return
- const tag = tagsSample[index]
- const videos = await getVideos(res, { tagsOneOf: [ tag ] })
- if (videos.length === 0) return
- acc.push({
- tag,
- videos
- })
- }
- async function getVideosByCategory (categoriesSample: number[], index: number, res: express.Response, acc: CategoryOverview[]) {
- if (categoriesSample.length <= index) return
- const category = categoriesSample[index]
- const videos = await getVideos(res, { categoryOneOf: [ category ] })
- if (videos.length === 0) return
- acc.push({
- category: videos[0].category,
- videos
- })
- }
- async function getVideosByChannel (channelsSample: number[], index: number, res: express.Response, acc: ChannelOverview[]) {
- if (channelsSample.length <= index) return
- const channelId = channelsSample[index]
- const videos = await getVideos(res, { videoChannelId: channelId })
- if (videos.length === 0) return
- acc.push({
- channel: videos[0].channel,
- videos
- })
- }
- async function getVideos (
- res: express.Response,
- where: { videoChannelId?: number, tagsOneOf?: string[], categoryOneOf?: number[] }
- ) {
- const query = Object.assign({
- start: 0,
- count: 12,
- sort: '-createdAt',
- includeLocalVideos: true,
- nsfw: buildNSFWFilter(res),
- user: res.locals.oauth ? res.locals.oauth.token.User : undefined,
- withFiles: false,
- countVideos: false
- }, where)
- const { data } = await VideoModel.listForApi(query)
- return data.map(d => d.toFormattedJSON())
- }
|