import { AbstractVideosQueryBuilder } from './abstract-videos-query-builder' import { VideoAttributes } from './video-attributes' import { VideoModelBuilder } from './video-model-builder' export class AbstractVideosModelQueryBuilder extends AbstractVideosQueryBuilder { protected attributes: { [key: string]: string } = {} protected joins: string[] = [] protected videoAttributes: VideoAttributes protected videoModelBuilder: VideoModelBuilder constructor (private readonly mode: 'list' | 'get') { super() this.videoAttributes = new VideoAttributes(this.mode) this.videoModelBuilder = new VideoModelBuilder(this.mode, this.videoAttributes) } protected buildSelect () { return 'SELECT ' + Object.keys(this.attributes).map(key => { const value = this.attributes[key] if (value) return `${key} AS ${value}` return key }).join(', ') } protected includeChannels () { this.joins.push( 'INNER JOIN "videoChannel" AS "VideoChannel" ON "video"."channelId" = "VideoChannel"."id"', 'INNER JOIN "actor" AS "VideoChannel->Actor" ON "VideoChannel"."actorId" = "VideoChannel->Actor"."id"', 'LEFT OUTER JOIN "server" AS "VideoChannel->Actor->Server" ON "VideoChannel->Actor"."serverId" = "VideoChannel->Actor->Server"."id"', 'LEFT OUTER JOIN "actorImage" AS "VideoChannel->Actor->Avatar" ' + 'ON "VideoChannel->Actor"."avatarId" = "VideoChannel->Actor->Avatar"."id"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoChannel', this.videoAttributes.getChannelAttributes()), ...this.buildActorInclude('VideoChannel->Actor'), ...this.buildAvatarInclude('VideoChannel->Actor->Avatar'), ...this.buildServerInclude('VideoChannel->Actor->Server') } } protected includeAccounts () { this.joins.push( 'INNER JOIN "account" AS "VideoChannel->Account" ON "VideoChannel"."accountId" = "VideoChannel->Account"."id"', 'INNER JOIN "actor" AS "VideoChannel->Account->Actor" ON "VideoChannel->Account"."actorId" = "VideoChannel->Account->Actor"."id"', 'LEFT OUTER JOIN "server" AS "VideoChannel->Account->Actor->Server" ' + 'ON "VideoChannel->Account->Actor"."serverId" = "VideoChannel->Account->Actor->Server"."id"', 'LEFT OUTER JOIN "actorImage" AS "VideoChannel->Account->Actor->Avatar" ' + 'ON "VideoChannel->Account->Actor"."avatarId" = "VideoChannel->Account->Actor->Avatar"."id"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoChannel->Account', this.videoAttributes.getAccountAttributes()), ...this.buildActorInclude('VideoChannel->Account->Actor'), ...this.buildAvatarInclude('VideoChannel->Account->Actor->Avatar'), ...this.buildServerInclude('VideoChannel->Account->Actor->Server') } } protected includeThumbnails () { this.joins.push('LEFT OUTER JOIN "thumbnail" AS "Thumbnails" ON "video"."id" = "Thumbnails"."videoId"') this.attributes = { ...this.attributes, ...this.buildAttributesObject('Thumbnails', this.videoAttributes.getThumbnailAttributes()) } } protected includeFiles () { this.joins.push( 'LEFT JOIN "videoFile" AS "VideoFiles" ON "VideoFiles"."videoId" = "video"."id"', 'LEFT JOIN "videoStreamingPlaylist" AS "VideoStreamingPlaylists" ON "VideoStreamingPlaylists"."videoId" = "video"."id"', 'LEFT JOIN "videoFile" AS "VideoStreamingPlaylists->VideoFiles" ' + 'ON "VideoStreamingPlaylists->VideoFiles"."videoStreamingPlaylistId" = "VideoStreamingPlaylists"."id"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoFiles', this.videoAttributes.getFileAttributes()), ...this.buildAttributesObject('VideoStreamingPlaylists', this.videoAttributes.getStreamingPlaylistAttributes()), ...this.buildAttributesObject('VideoStreamingPlaylists->VideoFiles', this.videoAttributes.getFileAttributes()) } } protected includeUserHistory (userId: number) { this.joins.push( 'LEFT OUTER JOIN "userVideoHistory" ' + 'ON "video"."id" = "userVideoHistory"."videoId" AND "userVideoHistory"."userId" = :userVideoHistoryId' ) this.replacements.userVideoHistoryId = userId this.attributes = { ...this.attributes, ...this.buildAttributesObject('userVideoHistory', this.videoAttributes.getUserHistoryAttributes()) } } protected includePlaylist (playlistId: number) { this.joins.push( 'INNER JOIN "videoPlaylistElement" as "VideoPlaylistElement" ON "videoPlaylistElement"."videoId" = "video"."id" ' + 'AND "VideoPlaylistElement"."videoPlaylistId" = :videoPlaylistId' ) this.replacements.videoPlaylistId = playlistId this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoPlaylistElement', this.videoAttributes.getPlaylistAttributes()) } } protected includeTags () { this.joins.push( 'LEFT OUTER JOIN (' + '"videoTag" AS "Tags->VideoTagModel" INNER JOIN "tag" AS "Tags" ON "Tags"."id" = "Tags->VideoTagModel"."tagId"' + ') ' + 'ON "video"."id" = "Tags->VideoTagModel"."videoId"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('Tags', this.videoAttributes.getTagAttributes()), ...this.buildAttributesObject('Tags->VideoTagModel', this.videoAttributes.getVideoTagAttributes()) } } protected includeBlacklisted () { this.joins.push( 'LEFT OUTER JOIN "videoBlacklist" AS "VideoBlacklist" ON "video"."id" = "VideoBlacklist"."videoId"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoBlacklist', this.videoAttributes.getBlacklistedAttributes()) } } protected includeScheduleUpdate () { this.joins.push( 'LEFT OUTER JOIN "scheduleVideoUpdate" AS "ScheduleVideoUpdate" ON "video"."id" = "ScheduleVideoUpdate"."videoId"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('ScheduleVideoUpdate', this.videoAttributes.getScheduleUpdateAttributes()) } } protected includeLive () { this.joins.push( 'LEFT OUTER JOIN "videoLive" AS "VideoLive" ON "video"."id" = "VideoLive"."videoId"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoLive', this.videoAttributes.getLiveAttributes()) } } protected includeTrackers () { this.joins.push( 'LEFT OUTER JOIN (' + '"videoTracker" AS "Trackers->VideoTrackerModel" ' + 'INNER JOIN "tracker" AS "Trackers" ON "Trackers"."id" = "Trackers->VideoTrackerModel"."trackerId"' + ') ON "video"."id" = "Trackers->VideoTrackerModel"."videoId"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('Trackers', this.videoAttributes.getTrackerAttributes()), ...this.buildAttributesObject('Trackers->VideoTrackerModel', this.videoAttributes.getVideoTrackerAttributes()) } } protected includeRedundancies () { this.joins.push( 'LEFT OUTER JOIN "videoRedundancy" AS "VideoStreamingPlaylists->RedundancyVideos" ' + 'ON "VideoStreamingPlaylists"."id" = "VideoStreamingPlaylists->RedundancyVideos"."videoStreamingPlaylistId"', 'LEFT OUTER JOIN "videoRedundancy" AS "VideoFiles->RedundancyVideos" ON ' + '"VideoFiles"."id" = "VideoFiles->RedundancyVideos"."videoFileId"' ) this.attributes = { ...this.attributes, ...this.buildAttributesObject('VideoFiles->RedundancyVideos', this.videoAttributes.getRedundancyAttributes()), ...this.buildAttributesObject('VideoStreamingPlaylists->RedundancyVideos', this.videoAttributes.getRedundancyAttributes()) } } protected buildActorInclude (prefixKey: string) { return this.buildAttributesObject(prefixKey, this.videoAttributes.getActorAttributes()) } protected buildAvatarInclude (prefixKey: string) { return this.buildAttributesObject(prefixKey, this.videoAttributes.getAvatarAttributes()) } protected buildServerInclude (prefixKey: string) { return this.buildAttributesObject(prefixKey, this.videoAttributes.getServerAttributes()) } protected buildAttributesObject (prefixKey: string, attributeKeys: string[]) { const result: { [id: string]: string} = {} const prefixValue = prefixKey.replace(/->/g, '.') for (const attribute of attributeKeys) { result[`"${prefixKey}"."${attribute}"`] = `"${prefixValue}.${attribute}"` } return result } }