Add video recomandation by tags (#1001)

[github/Chocobozzz/PeerTube.git] / client / src / app / videos / recommendations / recent-videos-recommendation.service.ts

import { Inject, Injectable } from '@angular/core'
import { RecommendationService } from '@app/videos/recommendations/recommendations.service'
import { Video } from '@app/shared/video/video.model'
import { RecommendationInfo } from '@app/shared/video/recommendation-info.model'
import { VideoService } from '@app/shared/video/video.service'
import { map } from 'rxjs/operators'
import { Observable } from 'rxjs'
import { SearchService } from '@app/search/search.service'
import { AdvancedSearch } from '@app/search/advanced-search.model'

/**
 * Provides "recommendations" by providing the most recently uploaded videos.
 */
@Injectable()
export class RecentVideosRecommendationService implements RecommendationService {

  readonly pageSize = 5

  constructor (
    private videos: VideoService,
    private searchService: SearchService
  ) {
  }

  getRecommendations (recommendation: RecommendationInfo): Observable<Video[]> {
    return this.fetchPage(1, recommendation)
      .pipe(
        map(vids => {
          const otherVideos = vids.filter(v => v.uuid !== recommendation.uuid)
          return otherVideos.slice(0, this.pageSize)
        })
      )
  }

  private fetchPage (page: number, recommendation: RecommendationInfo): Observable<Video[]> {
    let pagination = { currentPage: page, itemsPerPage: this.pageSize + 1 }
    if (!recommendation.tags) {
      return this.videos.getVideos(pagination, '-createdAt')
        .pipe(
          map(v => v.videos)
        )
    }
    if (recommendation.tags.length === 0) {
      return this.videos.getVideos(pagination, '-createdAt')
        .pipe(
          map(v => v.videos)
        )
    }
    return this.searchService.searchVideos('',
      pagination,
      new AdvancedSearch({ tagsOneOf: recommendation.tags.join(','), sort: '-createdAt' })
    ).pipe(
      map(v => v.videos)
    )
  }
}