]>
Commit | Line | Data |
---|---|---|
1 | import * as debug from 'debug' | |
2 | import { Observable, Subject } from 'rxjs' | |
3 | import { first, map } from 'rxjs/operators' | |
4 | import { Injectable, NgZone } 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 | ||
11 | const logger = debug('peertube:search:FindInBulkService') | |
12 | ||
13 | type BulkObservables <P extends number | string, R> = { | |
14 | notifier: Subject<P> | |
15 | result: Observable<R> | |
16 | } | |
17 | ||
18 | @Injectable() | |
19 | export class FindInBulkService { | |
20 | ||
21 | private getVideoInBulk: BulkObservables<string, ResultList<Video>> | |
22 | private getChannelInBulk: BulkObservables<string, ResultList<VideoChannel>> | |
23 | private getPlaylistInBulk: BulkObservables<string, ResultList<VideoPlaylist>> | |
24 | ||
25 | constructor ( | |
26 | private searchService: SearchService, | |
27 | private ngZone: NgZone | |
28 | ) { | |
29 | this.getVideoInBulk = this.buildBulkObservableObject(this.getVideosInBulk.bind(this)) | |
30 | this.getChannelInBulk = this.buildBulkObservableObject(this.getChannelsInBulk.bind(this)) | |
31 | this.getPlaylistInBulk = this.buildBulkObservableObject(this.getPlaylistsInBulk.bind(this)) | |
32 | } | |
33 | ||
34 | getVideo (uuid: string): Observable<Video> { | |
35 | logger('Schedule video fetch for uuid %s.', uuid) | |
36 | ||
37 | return this.getData({ | |
38 | observableObject: this.getVideoInBulk, | |
39 | finder: v => v.uuid === uuid, | |
40 | param: uuid | |
41 | }) | |
42 | } | |
43 | ||
44 | getChannel (handle: string): Observable<VideoChannel> { | |
45 | logger('Schedule channel fetch for handle %s.', handle) | |
46 | ||
47 | return this.getData({ | |
48 | observableObject: this.getChannelInBulk, | |
49 | finder: c => c.nameWithHost === handle || c.nameWithHostForced === handle, | |
50 | param: handle | |
51 | }) | |
52 | } | |
53 | ||
54 | getPlaylist (uuid: string): Observable<VideoPlaylist> { | |
55 | logger('Schedule playlist fetch for uuid %s.', uuid) | |
56 | ||
57 | return this.getData({ | |
58 | observableObject: this.getPlaylistInBulk, | |
59 | finder: p => p.uuid === uuid, | |
60 | param: uuid | |
61 | }) | |
62 | } | |
63 | ||
64 | private getData <P extends number | string, R> (options: { | |
65 | observableObject: BulkObservables<P, ResultList<R>> | |
66 | param: P | |
67 | finder: (d: R) => boolean | |
68 | }) { | |
69 | const { observableObject, param, finder } = options | |
70 | ||
71 | return new Observable<R>(obs => { | |
72 | observableObject.result | |
73 | .pipe( | |
74 | first(), | |
75 | map(({ data }) => data), | |
76 | map(data => data.find(finder)) | |
77 | ) | |
78 | .subscribe(result => { | |
79 | if (!result) { | |
80 | obs.error(new Error($localize`Element ${param} not found`)) | |
81 | } else { | |
82 | obs.next(result) | |
83 | obs.complete() | |
84 | } | |
85 | }) | |
86 | ||
87 | observableObject.notifier.next(param) | |
88 | }) | |
89 | } | |
90 | ||
91 | private getVideosInBulk (uuids: string[]) { | |
92 | logger('Fetching videos %s.', uuids.join(', ')) | |
93 | ||
94 | return this.searchService.searchVideos({ uuids }) | |
95 | } | |
96 | ||
97 | private getChannelsInBulk (handles: string[]) { | |
98 | logger('Fetching channels %s.', handles.join(', ')) | |
99 | ||
100 | return this.searchService.searchVideoChannels({ handles }) | |
101 | } | |
102 | ||
103 | private getPlaylistsInBulk (uuids: string[]) { | |
104 | logger('Fetching playlists %s.', uuids.join(', ')) | |
105 | ||
106 | return this.searchService.searchVideoPlaylists({ uuids }) | |
107 | } | |
108 | ||
109 | private buildBulkObservableObject <T extends number | string, R> (bulkGet: (params: T[]) => Observable<R>) { | |
110 | const notifier = new Subject<T>() | |
111 | ||
112 | return { | |
113 | notifier, | |
114 | ||
115 | result: buildBulkObservable({ | |
116 | time: 500, | |
117 | bulkGet, | |
118 | ngZone: this.ngZone, | |
119 | notifierObservable: notifier.asObservable() | |
120 | }) | |
121 | } | |
122 | } | |
123 | } |