client/src/app/shared/shared-search/find-in-bulk.service.ts

   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'
  11
  12 const debugLogger = debug('peertube:search:FindInBulkService')
  13
  14 type BulkObservables <P extends number | string, R> = {
  15   notifier: Subject<P>
  16   result: Observable<{ params: P[], response: R }>
  17 }
  18
  19 @Injectable()
  20 export class FindInBulkService {
  21
  22   private advancedSearchForBulk: AdvancedSearch
  23
  24   private getVideoInBulk: BulkObservables<string, ResultList<Video>>
  25   private getChannelInBulk: BulkObservables<string, ResultList<VideoChannel>>
  26   private getPlaylistInBulk: BulkObservables<string, ResultList<VideoPlaylist>>
  27
  28   constructor (
  29     private searchService: SearchService
  30   ) {
  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))
  34
  35     this.advancedSearchForBulk = new AdvancedSearch({ searchTarget: 'local' })
  36   }
  37
  38   getVideo (uuid: string): Observable<Video> {
  39     debugLogger('Schedule video fetch for uuid %s.', uuid)
  40
  41     return this.getData({
  42       observableObject: this.getVideoInBulk,
  43       finder: v => v.uuid === uuid || v.shortUUID === uuid,
  44       param: uuid
  45     })
  46   }
  47
  48   getChannel (handle: string): Observable<VideoChannel> {
  49     debugLogger('Schedule channel fetch for handle %s.', handle)
  50
  51     return this.getData({
  52       observableObject: this.getChannelInBulk,
  53       finder: c => c.nameWithHost === handle || c.nameWithHostForced === handle,
  54       param: handle
  55     })
  56   }
  57
  58   getPlaylist (uuid: string): Observable<VideoPlaylist> {
  59     debugLogger('Schedule playlist fetch for uuid %s.', uuid)
  60
  61     return this.getData({
  62       observableObject: this.getPlaylistInBulk,
  63       finder: p => p.uuid === uuid,
  64       param: uuid
  65     })
  66   }
  67
  68   private getData <P extends number | string, R> (options: {
  69     observableObject: BulkObservables<P, ResultList<R>>
  70     param: P
  71     finder: (d: R) => boolean
  72   }) {
  73     const { observableObject, param, finder } = options
  74
  75     return new Observable<R>(obs => {
  76       observableObject.result
  77         .pipe(
  78           filter(result => result.params.includes(param)),
  79           first(),
  80           map(result => result.response.data),
  81           map(data => data.find(finder))
  82         )
  83         .subscribe({
  84           next: result => {
  85             if (!result) {
  86               obs.error(new Error($localize`Element ${param} not found`))
  87               return
  88             }
  89
  90             obs.next(result)
  91             obs.complete()
  92           },
  93
  94           error: err => obs.error(err)
  95         })
  96
  97       observableObject.notifier.next(param)
  98     })
  99   }
 100
 101   private getVideosInBulk (uuids: string[]) {
 102     debugLogger('Fetching videos %s.', uuids.join(', '))
 103
 104     return this.searchService.searchVideos({
 105       uuids,
 106       componentPagination: { itemsPerPage: uuids.length, currentPage: 1 },
 107       advancedSearch: this.advancedSearchForBulk
 108     })
 109   }
 110
 111   private getChannelsInBulk (handles: string[]) {
 112     debugLogger('Fetching channels %s.', handles.join(', '))
 113
 114     return this.searchService.searchVideoChannels({
 115       handles,
 116       componentPagination: { itemsPerPage: handles.length, currentPage: 1 },
 117       advancedSearch: this.advancedSearchForBulk
 118     })
 119   }
 120
 121   private getPlaylistsInBulk (uuids: string[]) {
 122     debugLogger('Fetching playlists %s.', uuids.join(', '))
 123
 124     return this.searchService.searchVideoPlaylists({
 125       uuids,
 126       componentPagination: { itemsPerPage: uuids.length, currentPage: 1 },
 127       advancedSearch: this.advancedSearchForBulk
 128     })
 129   }
 130
 131   private buildBulkObservableObject <P extends number | string, R> (bulkGet: (params: P[]) => Observable<R>) {
 132     const notifier = new Subject<P>()
 133
 134     return {
 135       notifier,
 136
 137       result: buildBulkObservable({
 138         time: 500,
 139         bulkGet,
 140         notifierObservable: notifier.asObservable()
 141       })
 142     }
 143   }
 144 }