server/models/video/sql/video-model-get-query-builder.ts

   1 import { Sequelize, Transaction } from 'sequelize'
   2 import { AbstractVideosModelQueryBuilder } from './shared/abstract-videos-model-query-builder'
   3 import { VideoFileQueryBuilder } from './shared/video-file-query-builder'
   4 import { VideoModelBuilder } from './shared/video-model-builder'
   5 import { VideoTables } from './shared/video-tables'
   6
   7 /**
   8  *
   9  * Build a GET SQL query, fetch rows and create the video model
  10  *
  11  */
  12
  13 export type GetType =
  14   'api' |
  15   'full-light' |
  16   'account-blacklist-files' |
  17   'all-files' |
  18   'thumbnails' |
  19   'thumbnails-blacklist' |
  20   'id' |
  21   'blacklist-rights'
  22
  23 export type BuildVideoGetQueryOptions = {
  24   id?: number | string
  25   url?: string
  26
  27   type: GetType
  28
  29   userId?: number
  30   transaction?: Transaction
  31
  32   logging?: boolean
  33 }
  34
  35 export class VideosModelGetQueryBuilder {
  36   videoQueryBuilder: VideosModelGetQuerySubBuilder
  37   webtorrentFilesQueryBuilder: VideoFileQueryBuilder
  38   streamingPlaylistFilesQueryBuilder: VideoFileQueryBuilder
  39
  40   private readonly videoModelBuilder: VideoModelBuilder
  41
  42   private static readonly videoFilesInclude = new Set<GetType>([ 'api', 'full-light', 'account-blacklist-files', 'all-files' ])
  43
  44   constructor (protected readonly sequelize: Sequelize) {
  45     this.videoQueryBuilder = new VideosModelGetQuerySubBuilder(sequelize)
  46     this.webtorrentFilesQueryBuilder = new VideoFileQueryBuilder(sequelize)
  47     this.streamingPlaylistFilesQueryBuilder = new VideoFileQueryBuilder(sequelize)
  48
  49     this.videoModelBuilder = new VideoModelBuilder('get', new VideoTables('get'))
  50   }
  51
  52   async queryVideo (options: BuildVideoGetQueryOptions) {
  53     const [ videoRows, webtorrentFilesRows, streamingPlaylistFilesRows ] = await Promise.all([
  54       this.videoQueryBuilder.queryVideos(options),
  55
  56       VideosModelGetQueryBuilder.videoFilesInclude.has(options.type)
  57         ? this.webtorrentFilesQueryBuilder.queryWebTorrentVideos(options)
  58         : Promise.resolve(undefined),
  59
  60       VideosModelGetQueryBuilder.videoFilesInclude.has(options.type)
  61         ? this.streamingPlaylistFilesQueryBuilder.queryStreamingPlaylistVideos(options)
  62         : Promise.resolve(undefined)
  63     ])
  64
  65     const videos = this.videoModelBuilder.buildVideosFromRows(videoRows, webtorrentFilesRows, streamingPlaylistFilesRows)
  66
  67     if (videos.length > 1) {
  68       throw new Error('Video results is more than ')
  69     }
  70
  71     if (videos.length === 0) return null
  72     return videos[0]
  73   }
  74 }
  75
  76 export class VideosModelGetQuerySubBuilder extends AbstractVideosModelQueryBuilder {
  77   protected attributes: { [key: string]: string }
  78
  79   protected webtorrentFilesQuery: string
  80   protected streamingPlaylistFilesQuery: string
  81
  82   private static readonly trackersInclude = new Set<GetType>([ 'api' ])
  83   private static readonly liveInclude = new Set<GetType>([ 'api', 'full-light' ])
  84   private static readonly scheduleUpdateInclude = new Set<GetType>([ 'api', 'full-light' ])
  85   private static readonly tagsInclude = new Set<GetType>([ 'api', 'full-light' ])
  86   private static readonly userHistoryInclude = new Set<GetType>([ 'api', 'full-light' ])
  87   private static readonly accountInclude = new Set<GetType>([ 'api', 'full-light', 'account-blacklist-files' ])
  88   private static readonly ownerUserInclude = new Set<GetType>([ 'blacklist-rights' ])
  89
  90   private static readonly blacklistedInclude = new Set<GetType>([
  91     'api',
  92     'full-light',
  93     'account-blacklist-files',
  94     'thumbnails-blacklist',
  95     'blacklist-rights'
  96   ])
  97
  98   private static readonly thumbnailsInclude = new Set<GetType>([
  99     'api',
 100     'full-light',
 101     'account-blacklist-files',
 102     'all-files',
 103     'thumbnails',
 104     'thumbnails-blacklist'
 105   ])
 106
 107   constructor (protected readonly sequelize: Sequelize) {
 108     super('get')
 109   }
 110
 111   queryVideos (options: BuildVideoGetQueryOptions) {
 112     this.buildMainGetQuery(options)
 113
 114     return this.runQuery(options)
 115   }
 116
 117   private buildMainGetQuery (options: BuildVideoGetQueryOptions) {
 118     this.attributes = {
 119       '"video".*': ''
 120     }
 121
 122     if (VideosModelGetQuerySubBuilder.thumbnailsInclude.has(options.type)) {
 123       this.includeThumbnails()
 124     }
 125
 126     if (VideosModelGetQuerySubBuilder.blacklistedInclude.has(options.type)) {
 127       this.includeBlacklisted()
 128     }
 129
 130     if (VideosModelGetQuerySubBuilder.accountInclude.has(options.type)) {
 131       this.includeChannels()
 132       this.includeAccounts()
 133     }
 134
 135     if (VideosModelGetQuerySubBuilder.tagsInclude.has(options.type)) {
 136       this.includeTags()
 137     }
 138
 139     if (VideosModelGetQuerySubBuilder.scheduleUpdateInclude.has(options.type)) {
 140       this.includeScheduleUpdate()
 141     }
 142
 143     if (VideosModelGetQuerySubBuilder.liveInclude.has(options.type)) {
 144       this.includeLive()
 145     }
 146
 147     if (options.userId && VideosModelGetQuerySubBuilder.userHistoryInclude.has(options.type)) {
 148       this.includeUserHistory(options.userId)
 149     }
 150
 151     if (VideosModelGetQuerySubBuilder.ownerUserInclude.has(options.type)) {
 152       this.includeOwnerUser()
 153     }
 154
 155     if (VideosModelGetQuerySubBuilder.trackersInclude.has(options.type)) {
 156       this.includeTrackers()
 157     }
 158
 159     this.whereId(options)
 160
 161     this.query = this.buildQuery(options)
 162   }
 163
 164   private buildQuery (options: BuildVideoGetQueryOptions) {
 165     const order = VideosModelGetQuerySubBuilder.tagsInclude.has(options.type)
 166       ? 'ORDER BY "Tags"."name" ASC'
 167       : ''
 168
 169     const from = `SELECT * FROM "video" ${this.where} LIMIT 1`
 170
 171     return `${this.buildSelect()} FROM (${from}) AS "video" ${this.joins} ${order}`
 172   }
 173 }