1 import * as debug from 'debug'
2 import { Observable, Subject } from 'rxjs'
3 import { filter, first, map } from 'rxjs/operators'
4 import { Injectable } from '@angular/core'
5 import { buildBulkObservable } from '@app/helpers'
6 import { ResultList } from '@shared/models/common'
7 import { Video, VideoChannel } from '../shared-main'
8 import { VideoPlaylist } from '../shared-video-playlist'
9 import { SearchService } from './search.service'
10 import { AdvancedSearch } from './advanced-search.model'
12 const logger = debug('peertube:search:FindInBulkService')
14 type BulkObservables <P extends number | string, R> = {
16 result: Observable<{ params: P[], response: R }>
20 export class FindInBulkService {
22 private advancedSearchForBulk: AdvancedSearch
24 private getVideoInBulk: BulkObservables<string, ResultList<Video>>
25 private getChannelInBulk: BulkObservables<string, ResultList<VideoChannel>>
26 private getPlaylistInBulk: BulkObservables<string, ResultList<VideoPlaylist>>
29 private searchService: SearchService
31 this.getVideoInBulk = this.buildBulkObservableObject(this.getVideosInBulk.bind(this))
32 this.getChannelInBulk = this.buildBulkObservableObject(this.getChannelsInBulk.bind(this))
33 this.getPlaylistInBulk = this.buildBulkObservableObject(this.getPlaylistsInBulk.bind(this))
35 this.advancedSearchForBulk = new AdvancedSearch({ searchTarget: 'local' })
38 getVideo (uuid: string): Observable<Video> {
39 logger('Schedule video fetch for uuid %s.', uuid)
42 observableObject: this.getVideoInBulk,
43 finder: v => v.uuid === uuid,
48 getChannel (handle: string): Observable<VideoChannel> {
49 logger('Schedule channel fetch for handle %s.', handle)
52 observableObject: this.getChannelInBulk,
53 finder: c => c.nameWithHost === handle || c.nameWithHostForced === handle,
58 getPlaylist (uuid: string): Observable<VideoPlaylist> {
59 logger('Schedule playlist fetch for uuid %s.', uuid)
62 observableObject: this.getPlaylistInBulk,
63 finder: p => p.uuid === uuid,
68 private getData <P extends number | string, R> (options: {
69 observableObject: BulkObservables<P, ResultList<R>>
71 finder: (d: R) => boolean
73 const { observableObject, param, finder } = options
75 return new Observable<R>(obs => {
76 observableObject.result
78 filter(result => result.params.includes(param)),
80 map(result => result.response.data),
81 map(data => data.find(finder))
83 .subscribe(result => {
85 obs.error(new Error($localize`Element ${param} not found`))
92 observableObject.notifier.next(param)
96 private getVideosInBulk (uuids: string[]) {
97 logger('Fetching videos %s.', uuids.join(', '))
99 return this.searchService.searchVideos({
101 componentPagination: { itemsPerPage: uuids.length, currentPage: 1 },
102 advancedSearch: this.advancedSearchForBulk
106 private getChannelsInBulk (handles: string[]) {
107 logger('Fetching channels %s.', handles.join(', '))
109 return this.searchService.searchVideoChannels({
111 componentPagination: { itemsPerPage: handles.length, currentPage: 1 },
112 advancedSearch: this.advancedSearchForBulk
116 private getPlaylistsInBulk (uuids: string[]) {
117 logger('Fetching playlists %s.', uuids.join(', '))
119 return this.searchService.searchVideoPlaylists({
121 componentPagination: { itemsPerPage: uuids.length, currentPage: 1 },
122 advancedSearch: this.advancedSearchForBulk
126 private buildBulkObservableObject <P extends number | string, R> (bulkGet: (params: P[]) => Observable<R>) {
127 const notifier = new Subject<P>()
132 result: buildBulkObservable({
135 notifierObservable: notifier.asObservable()