diff options
author | Chocobozzz <me@florianbigard.com> | 2021-10-27 14:37:04 +0200 |
---|---|---|
committer | Chocobozzz <chocobozzz@cpy.re> | 2021-10-29 11:48:21 +0200 |
commit | 2760b454a761f6af3138b2fb5f34340772ab0d1e (patch) | |
tree | 2b3a2d81478f8b432eb54cce4caa5a760c494627 /server/models/video/sql/shared | |
parent | e4611b54910d8e7f2b4f8a97ee2d9cc8e1054127 (diff) | |
download | PeerTube-2760b454a761f6af3138b2fb5f34340772ab0d1e.tar.gz PeerTube-2760b454a761f6af3138b2fb5f34340772ab0d1e.tar.zst PeerTube-2760b454a761f6af3138b2fb5f34340772ab0d1e.zip |
Deprecate filter video query
Introduce include and isLocal instead
Diffstat (limited to 'server/models/video/sql/shared')
3 files changed, 95 insertions, 9 deletions
diff --git a/server/models/video/sql/shared/abstract-videos-model-query-builder.ts b/server/models/video/sql/shared/abstract-videos-model-query-builder.ts index 0d7e64574..29827db2a 100644 --- a/server/models/video/sql/shared/abstract-videos-model-query-builder.ts +++ b/server/models/video/sql/shared/abstract-videos-model-query-builder.ts | |||
@@ -1,3 +1,5 @@ | |||
1 | import { createSafeIn } from '@server/models/utils' | ||
2 | import { MUserAccountId } from '@server/types/models' | ||
1 | import validator from 'validator' | 3 | import validator from 'validator' |
2 | import { AbstractVideosQueryBuilder } from './abstract-videos-query-builder' | 4 | import { AbstractVideosQueryBuilder } from './abstract-videos-query-builder' |
3 | import { VideoTables } from './video-tables' | 5 | import { VideoTables } from './video-tables' |
@@ -188,6 +190,32 @@ export class AbstractVideosModelQueryBuilder extends AbstractVideosQueryBuilder | |||
188 | } | 190 | } |
189 | } | 191 | } |
190 | 192 | ||
193 | protected includeBlockedOwnerAndServer (serverAccountId: number, user?: MUserAccountId) { | ||
194 | const blockerIds = [ serverAccountId ] | ||
195 | if (user) blockerIds.push(user.Account.id) | ||
196 | |||
197 | const inClause = createSafeIn(this.sequelize, blockerIds) | ||
198 | |||
199 | this.addJoin( | ||
200 | 'LEFT JOIN "accountBlocklist" AS "VideoChannel->Account->AccountBlocklist" ' + | ||
201 | 'ON "VideoChannel->Account"."id" = "VideoChannel->Account->AccountBlocklist"."targetAccountId" ' + | ||
202 | 'AND "VideoChannel->Account->AccountBlocklist"."accountId" IN (' + inClause + ')' | ||
203 | ) | ||
204 | |||
205 | this.addJoin( | ||
206 | 'LEFT JOIN "serverBlocklist" AS "VideoChannel->Account->Actor->Server->ServerBlocklist" ' + | ||
207 | 'ON "VideoChannel->Account->Actor->Server->ServerBlocklist"."targetServerId" = "VideoChannel->Account->Actor"."serverId" ' + | ||
208 | 'AND "VideoChannel->Account->Actor->Server->ServerBlocklist"."accountId" IN (' + inClause + ') ' | ||
209 | ) | ||
210 | |||
211 | this.attributes = { | ||
212 | ...this.attributes, | ||
213 | |||
214 | ...this.buildAttributesObject('VideoChannel->Account->AccountBlocklist', this.tables.getBlocklistAttributes()), | ||
215 | ...this.buildAttributesObject('VideoChannel->Account->Actor->Server->ServerBlocklist', this.tables.getBlocklistAttributes()) | ||
216 | } | ||
217 | } | ||
218 | |||
191 | protected includeScheduleUpdate () { | 219 | protected includeScheduleUpdate () { |
192 | this.addJoin( | 220 | this.addJoin( |
193 | 'LEFT OUTER JOIN "scheduleVideoUpdate" AS "ScheduleVideoUpdate" ON "video"."id" = "ScheduleVideoUpdate"."videoId"' | 221 | 'LEFT OUTER JOIN "scheduleVideoUpdate" AS "ScheduleVideoUpdate" ON "video"."id" = "ScheduleVideoUpdate"."videoId"' |
diff --git a/server/models/video/sql/shared/video-model-builder.ts b/server/models/video/sql/shared/video-model-builder.ts index 33a0181e9..0eac95661 100644 --- a/server/models/video/sql/shared/video-model-builder.ts +++ b/server/models/video/sql/shared/video-model-builder.ts | |||
@@ -1,11 +1,14 @@ | |||
1 | 1 | ||
2 | import { AccountModel } from '@server/models/account/account' | 2 | import { AccountModel } from '@server/models/account/account' |
3 | import { AccountBlocklistModel } from '@server/models/account/account-blocklist' | ||
3 | import { ActorModel } from '@server/models/actor/actor' | 4 | import { ActorModel } from '@server/models/actor/actor' |
4 | import { ActorImageModel } from '@server/models/actor/actor-image' | 5 | import { ActorImageModel } from '@server/models/actor/actor-image' |
5 | import { VideoRedundancyModel } from '@server/models/redundancy/video-redundancy' | 6 | import { VideoRedundancyModel } from '@server/models/redundancy/video-redundancy' |
6 | import { ServerModel } from '@server/models/server/server' | 7 | import { ServerModel } from '@server/models/server/server' |
8 | import { ServerBlocklistModel } from '@server/models/server/server-blocklist' | ||
7 | import { TrackerModel } from '@server/models/server/tracker' | 9 | import { TrackerModel } from '@server/models/server/tracker' |
8 | import { UserVideoHistoryModel } from '@server/models/user/user-video-history' | 10 | import { UserVideoHistoryModel } from '@server/models/user/user-video-history' |
11 | import { VideoInclude } from '@shared/models' | ||
9 | import { ScheduleVideoUpdateModel } from '../../schedule-video-update' | 12 | import { ScheduleVideoUpdateModel } from '../../schedule-video-update' |
10 | import { TagModel } from '../../tag' | 13 | import { TagModel } from '../../tag' |
11 | import { ThumbnailModel } from '../../thumbnail' | 14 | import { ThumbnailModel } from '../../thumbnail' |
@@ -33,6 +36,8 @@ export class VideoModelBuilder { | |||
33 | private thumbnailsDone: Set<any> | 36 | private thumbnailsDone: Set<any> |
34 | private historyDone: Set<any> | 37 | private historyDone: Set<any> |
35 | private blacklistDone: Set<any> | 38 | private blacklistDone: Set<any> |
39 | private accountBlocklistDone: Set<any> | ||
40 | private serverBlocklistDone: Set<any> | ||
36 | private liveDone: Set<any> | 41 | private liveDone: Set<any> |
37 | private redundancyDone: Set<any> | 42 | private redundancyDone: Set<any> |
38 | private scheduleVideoUpdateDone: Set<any> | 43 | private scheduleVideoUpdateDone: Set<any> |
@@ -51,7 +56,14 @@ export class VideoModelBuilder { | |||
51 | 56 | ||
52 | } | 57 | } |
53 | 58 | ||
54 | buildVideosFromRows (rows: SQLRow[], rowsWebTorrentFiles?: SQLRow[], rowsStreamingPlaylist?: SQLRow[]) { | 59 | buildVideosFromRows (options: { |
60 | rows: SQLRow[] | ||
61 | include?: VideoInclude | ||
62 | rowsWebTorrentFiles?: SQLRow[] | ||
63 | rowsStreamingPlaylist?: SQLRow[] | ||
64 | }) { | ||
65 | const { rows, rowsWebTorrentFiles, rowsStreamingPlaylist, include } = options | ||
66 | |||
55 | this.reinit() | 67 | this.reinit() |
56 | 68 | ||
57 | for (const row of rows) { | 69 | for (const row of rows) { |
@@ -77,6 +89,15 @@ export class VideoModelBuilder { | |||
77 | this.setBlacklisted(row, videoModel) | 89 | this.setBlacklisted(row, videoModel) |
78 | this.setScheduleVideoUpdate(row, videoModel) | 90 | this.setScheduleVideoUpdate(row, videoModel) |
79 | this.setLive(row, videoModel) | 91 | this.setLive(row, videoModel) |
92 | } else { | ||
93 | if (include & VideoInclude.BLACKLISTED) { | ||
94 | this.setBlacklisted(row, videoModel) | ||
95 | } | ||
96 | |||
97 | if (include & VideoInclude.BLOCKED_OWNER) { | ||
98 | this.setBlockedOwner(row, videoModel) | ||
99 | this.setBlockedServer(row, videoModel) | ||
100 | } | ||
80 | } | 101 | } |
81 | } | 102 | } |
82 | 103 | ||
@@ -91,15 +112,18 @@ export class VideoModelBuilder { | |||
91 | this.videoStreamingPlaylistMemo = {} | 112 | this.videoStreamingPlaylistMemo = {} |
92 | this.videoFileMemo = {} | 113 | this.videoFileMemo = {} |
93 | 114 | ||
94 | this.thumbnailsDone = new Set<number>() | 115 | this.thumbnailsDone = new Set() |
95 | this.historyDone = new Set<number>() | 116 | this.historyDone = new Set() |
96 | this.blacklistDone = new Set<number>() | 117 | this.blacklistDone = new Set() |
97 | this.liveDone = new Set<number>() | 118 | this.liveDone = new Set() |
98 | this.redundancyDone = new Set<number>() | 119 | this.redundancyDone = new Set() |
99 | this.scheduleVideoUpdateDone = new Set<number>() | 120 | this.scheduleVideoUpdateDone = new Set() |
121 | |||
122 | this.accountBlocklistDone = new Set() | ||
123 | this.serverBlocklistDone = new Set() | ||
100 | 124 | ||
101 | this.trackersDone = new Set<string>() | 125 | this.trackersDone = new Set() |
102 | this.tagsDone = new Set<string>() | 126 | this.tagsDone = new Set() |
103 | 127 | ||
104 | this.videos = [] | 128 | this.videos = [] |
105 | } | 129 | } |
@@ -162,6 +186,8 @@ export class VideoModelBuilder { | |||
162 | const accountModel = new AccountModel(this.grab(row, this.tables.getAccountAttributes(), 'VideoChannel.Account'), this.buildOpts) | 186 | const accountModel = new AccountModel(this.grab(row, this.tables.getAccountAttributes(), 'VideoChannel.Account'), this.buildOpts) |
163 | accountModel.Actor = this.buildActor(row, 'VideoChannel.Account') | 187 | accountModel.Actor = this.buildActor(row, 'VideoChannel.Account') |
164 | 188 | ||
189 | accountModel.BlockedBy = [] | ||
190 | |||
165 | channelModel.Account = accountModel | 191 | channelModel.Account = accountModel |
166 | 192 | ||
167 | videoModel.VideoChannel = channelModel | 193 | videoModel.VideoChannel = channelModel |
@@ -180,6 +206,8 @@ export class VideoModelBuilder { | |||
180 | ? new ServerModel(this.grab(row, this.tables.getServerAttributes(), serverPrefix), this.buildOpts) | 206 | ? new ServerModel(this.grab(row, this.tables.getServerAttributes(), serverPrefix), this.buildOpts) |
181 | : null | 207 | : null |
182 | 208 | ||
209 | if (serverModel) serverModel.BlockedBy = [] | ||
210 | |||
183 | const actorModel = new ActorModel(this.grab(row, this.tables.getActorAttributes(), actorPrefix), this.buildOpts) | 211 | const actorModel = new ActorModel(this.grab(row, this.tables.getActorAttributes(), actorPrefix), this.buildOpts) |
184 | actorModel.Avatar = avatarModel | 212 | actorModel.Avatar = avatarModel |
185 | actorModel.Server = serverModel | 213 | actorModel.Server = serverModel |
@@ -297,6 +325,32 @@ export class VideoModelBuilder { | |||
297 | this.blacklistDone.add(id) | 325 | this.blacklistDone.add(id) |
298 | } | 326 | } |
299 | 327 | ||
328 | private setBlockedOwner (row: SQLRow, videoModel: VideoModel) { | ||
329 | const id = row['VideoChannel.Account.AccountBlocklist.id'] | ||
330 | if (!id) return | ||
331 | |||
332 | const key = `${videoModel.id}-${id}` | ||
333 | if (this.accountBlocklistDone.has(key)) return | ||
334 | |||
335 | const attributes = this.grab(row, this.tables.getBlocklistAttributes(), 'VideoChannel.Account.AccountBlocklist') | ||
336 | videoModel.VideoChannel.Account.BlockedBy.push(new AccountBlocklistModel(attributes, this.buildOpts)) | ||
337 | |||
338 | this.accountBlocklistDone.add(key) | ||
339 | } | ||
340 | |||
341 | private setBlockedServer (row: SQLRow, videoModel: VideoModel) { | ||
342 | const id = row['VideoChannel.Account.Actor.Server.ServerBlocklist.id'] | ||
343 | if (!id || this.serverBlocklistDone.has(id)) return | ||
344 | |||
345 | const key = `${videoModel.id}-${id}` | ||
346 | if (this.serverBlocklistDone.has(key)) return | ||
347 | |||
348 | const attributes = this.grab(row, this.tables.getBlocklistAttributes(), 'VideoChannel.Account.Actor.Server.ServerBlocklist') | ||
349 | videoModel.VideoChannel.Account.Actor.Server.BlockedBy.push(new ServerBlocklistModel(attributes, this.buildOpts)) | ||
350 | |||
351 | this.serverBlocklistDone.add(key) | ||
352 | } | ||
353 | |||
300 | private setScheduleVideoUpdate (row: SQLRow, videoModel: VideoModel) { | 354 | private setScheduleVideoUpdate (row: SQLRow, videoModel: VideoModel) { |
301 | const id = row['ScheduleVideoUpdate.id'] | 355 | const id = row['ScheduleVideoUpdate.id'] |
302 | if (!id || this.scheduleVideoUpdateDone.has(id)) return | 356 | if (!id || this.scheduleVideoUpdateDone.has(id)) return |
diff --git a/server/models/video/sql/shared/video-tables.ts b/server/models/video/sql/shared/video-tables.ts index 75823864d..042b9d5da 100644 --- a/server/models/video/sql/shared/video-tables.ts +++ b/server/models/video/sql/shared/video-tables.ts | |||
@@ -139,6 +139,10 @@ export class VideoTables { | |||
139 | return [ 'id', 'reason', 'unfederated' ] | 139 | return [ 'id', 'reason', 'unfederated' ] |
140 | } | 140 | } |
141 | 141 | ||
142 | getBlocklistAttributes () { | ||
143 | return [ 'id' ] | ||
144 | } | ||
145 | |||
142 | getScheduleUpdateAttributes () { | 146 | getScheduleUpdateAttributes () { |
143 | return [ | 147 | return [ |
144 | 'id', | 148 | 'id', |