123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166 |
- import express from 'express'
- import { sanitizeUrl } from '@server/helpers/core-utils.js'
- import { pickSearchVideoQuery } from '@server/helpers/query.js'
- import { doJSONRequest } from '@server/helpers/requests.js'
- import { CONFIG } from '@server/initializers/config.js'
- import { WEBSERVER } from '@server/initializers/constants.js'
- import { findLatestAPRedirection } from '@server/lib/activitypub/activity.js'
- import { getOrCreateAPVideo } from '@server/lib/activitypub/videos/index.js'
- import { Hooks } from '@server/lib/plugins/hooks.js'
- import { buildMutedForSearchIndex, isSearchIndexSearch, isURISearch } from '@server/lib/search.js'
- import { getServerActor } from '@server/models/application/application.js'
- import { HttpStatusCode, ResultList, Video, VideosSearchQueryAfterSanitize } from '@peertube/peertube-models'
- import { buildNSFWFilter, isUserAbleToSearchRemoteURI } from '../../../helpers/express-utils.js'
- import { logger } from '../../../helpers/logger.js'
- import { getFormattedObjects } from '../../../helpers/utils.js'
- import {
- asyncMiddleware,
- commonVideosFiltersValidator,
- openapiOperationDoc,
- optionalAuthenticate,
- paginationValidator,
- setDefaultPagination,
- setDefaultSearchSort,
- videosSearchSortValidator,
- videosSearchValidator
- } from '../../../middlewares/index.js'
- import { guessAdditionalAttributesFromQuery } from '../../../models/video/formatter/index.js'
- import { VideoModel } from '../../../models/video/video.js'
- import { MVideoAccountLightBlacklistAllFiles } from '../../../types/models/index.js'
- import { searchLocalUrl } from './shared/index.js'
- const searchVideosRouter = express.Router()
- searchVideosRouter.get('/videos',
- openapiOperationDoc({ operationId: 'searchVideos' }),
- paginationValidator,
- setDefaultPagination,
- videosSearchSortValidator,
- setDefaultSearchSort,
- optionalAuthenticate,
- commonVideosFiltersValidator,
- videosSearchValidator,
- asyncMiddleware(searchVideos)
- )
- // ---------------------------------------------------------------------------
- export { searchVideosRouter }
- // ---------------------------------------------------------------------------
- function searchVideos (req: express.Request, res: express.Response) {
- const query = pickSearchVideoQuery(req.query)
- const search = query.search
- if (isURISearch(search)) {
- return searchVideoURI(search, res)
- }
- if (isSearchIndexSearch(query)) {
- return searchVideosIndex(query, res)
- }
- return searchVideosDB(query, res)
- }
- async function searchVideosIndex (query: VideosSearchQueryAfterSanitize, res: express.Response) {
- const result = await buildMutedForSearchIndex(res)
- let body = { ...query, ...result }
- // Use the default instance NSFW policy if not specified
- if (!body.nsfw) {
- const nsfwPolicy = res.locals.oauth
- ? res.locals.oauth.token.User.nsfwPolicy
- : CONFIG.INSTANCE.DEFAULT_NSFW_POLICY
- body.nsfw = nsfwPolicy === 'do_not_list'
- ? 'false'
- : 'both'
- }
- body = await Hooks.wrapObject(body, 'filter:api.search.videos.index.list.params')
- const url = sanitizeUrl(CONFIG.SEARCH.SEARCH_INDEX.URL) + '/api/v1/search/videos'
- try {
- logger.debug('Doing videos search index request on %s.', url, { body })
- const { body: searchIndexResult } = await doJSONRequest<ResultList<Video>>(url, { method: 'POST', json: body })
- const jsonResult = await Hooks.wrapObject(searchIndexResult, 'filter:api.search.videos.index.list.result')
- return res.json(jsonResult)
- } catch (err) {
- logger.warn('Cannot use search index to make video search.', { err })
- return res.fail({
- status: HttpStatusCode.INTERNAL_SERVER_ERROR_500,
- message: 'Cannot use search index to make video search'
- })
- }
- }
- async function searchVideosDB (query: VideosSearchQueryAfterSanitize, res: express.Response) {
- const serverActor = await getServerActor()
- const apiOptions = await Hooks.wrapObject({
- ...query,
- displayOnlyForFollower: {
- actorId: serverActor.id,
- orLocalVideos: true
- },
- nsfw: buildNSFWFilter(res, query.nsfw),
- user: res.locals.oauth
- ? res.locals.oauth.token.User
- : undefined
- }, 'filter:api.search.videos.local.list.params')
- const resultList = await Hooks.wrapPromiseFun(
- VideoModel.searchAndPopulateAccountAndServer,
- apiOptions,
- 'filter:api.search.videos.local.list.result'
- )
- return res.json(getFormattedObjects(resultList.data, resultList.total, guessAdditionalAttributesFromQuery(query)))
- }
- async function searchVideoURI (url: string, res: express.Response) {
- let video: MVideoAccountLightBlacklistAllFiles
- // Check if we can fetch a remote video with the URL
- if (isUserAbleToSearchRemoteURI(res)) {
- try {
- const syncParam = {
- rates: false,
- shares: false,
- comments: false,
- refreshVideo: false
- }
- const result = await getOrCreateAPVideo({
- videoObject: await findLatestAPRedirection(url),
- syncParam
- })
- video = result ? result.video : undefined
- } catch (err) {
- logger.info('Cannot search remote video %s.', url, { err })
- }
- } else {
- video = await searchLocalUrl(sanitizeLocalUrl(url), url => VideoModel.loadByUrlAndPopulateAccount(url))
- }
- return res.json({
- total: video ? 1 : 0,
- data: video ? [ video.toFormattedJSON() ] : []
- })
- }
- function sanitizeLocalUrl (url: string) {
- if (!url) return ''
- // Handle alternative video URLs
- return url.replace(new RegExp('^' + WEBSERVER.URL + '/w/'), WEBSERVER.URL + '/videos/watch/')
- }
|