]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/models/account/user.ts
Fix user video quota with webtorrent disabled
[github/Chocobozzz/PeerTube.git] / server / models / account / user.ts
1 import { col, FindOptions, fn, literal, Op, QueryTypes, where, WhereOptions } from 'sequelize'
2 import {
3 AfterDestroy,
4 AfterUpdate,
5 AllowNull,
6 BeforeCreate,
7 BeforeUpdate,
8 Column,
9 CreatedAt,
10 DataType,
11 Default,
12 DefaultScope,
13 HasMany,
14 HasOne,
15 Is,
16 IsEmail,
17 Model,
18 Scopes,
19 Table,
20 UpdatedAt
21 } from 'sequelize-typescript'
22 import { hasUserRight, MyUser, USER_ROLE_LABELS, UserRight, VideoAbuseState, VideoPlaylistType, VideoPrivacy } from '../../../shared'
23 import { User, UserRole } from '../../../shared/models/users'
24 import {
25 isNoInstanceConfigWarningModal,
26 isNoWelcomeModal,
27 isUserAdminFlagsValid,
28 isUserAutoPlayNextVideoPlaylistValid,
29 isUserAutoPlayNextVideoValid,
30 isUserAutoPlayVideoValid,
31 isUserBlockedReasonValid,
32 isUserBlockedValid,
33 isUserEmailVerifiedValid,
34 isUserNSFWPolicyValid,
35 isUserPasswordValid,
36 isUserRoleValid,
37 isUserUsernameValid,
38 isUserVideoLanguages,
39 isUserVideoQuotaDailyValid,
40 isUserVideoQuotaValid,
41 isUserVideosHistoryEnabledValid,
42 isUserWebTorrentEnabledValid
43 } from '../../helpers/custom-validators/users'
44 import { comparePassword, cryptPassword } from '../../helpers/peertube-crypto'
45 import { OAuthTokenModel } from '../oauth/oauth-token'
46 import { getSort, throwIfNotValid } from '../utils'
47 import { VideoChannelModel } from '../video/video-channel'
48 import { VideoPlaylistModel } from '../video/video-playlist'
49 import { AccountModel } from './account'
50 import { NSFWPolicyType } from '../../../shared/models/videos/nsfw-policy.type'
51 import { values } from 'lodash'
52 import { DEFAULT_USER_THEME_NAME, NSFW_POLICY_TYPES } from '../../initializers/constants'
53 import { clearCacheByUserId } from '../../lib/oauth-model'
54 import { UserNotificationSettingModel } from './user-notification-setting'
55 import { VideoModel } from '../video/video'
56 import { ActorModel } from '../activitypub/actor'
57 import { ActorFollowModel } from '../activitypub/actor-follow'
58 import { VideoImportModel } from '../video/video-import'
59 import { UserAdminFlag } from '../../../shared/models/users/user-flag.model'
60 import { isThemeNameValid } from '../../helpers/custom-validators/plugins'
61 import { getThemeOrDefault } from '../../lib/plugins/theme-utils'
62 import * as Bluebird from 'bluebird'
63 import {
64 MMyUserFormattable,
65 MUserDefault,
66 MUserFormattable,
67 MUserId,
68 MUserNotifSettingChannelDefault,
69 MUserWithNotificationSetting,
70 MVideoFullLight
71 } from '@server/typings/models'
72
73 enum ScopeNames {
74 FOR_ME_API = 'FOR_ME_API',
75 WITH_VIDEOCHANNELS = 'WITH_VIDEOCHANNELS',
76 WITH_STATS = 'WITH_STATS'
77 }
78
79 @DefaultScope(() => ({
80 include: [
81 {
82 model: AccountModel,
83 required: true
84 },
85 {
86 model: UserNotificationSettingModel,
87 required: true
88 }
89 ]
90 }))
91 @Scopes(() => ({
92 [ScopeNames.FOR_ME_API]: {
93 include: [
94 {
95 model: AccountModel,
96 include: [
97 {
98 model: VideoChannelModel
99 },
100 {
101 attributes: [ 'id', 'name', 'type' ],
102 model: VideoPlaylistModel.unscoped(),
103 required: true,
104 where: {
105 type: {
106 [Op.ne]: VideoPlaylistType.REGULAR
107 }
108 }
109 }
110 ]
111 },
112 {
113 model: UserNotificationSettingModel,
114 required: true
115 }
116 ]
117 },
118 [ScopeNames.WITH_VIDEOCHANNELS]: {
119 include: [
120 {
121 model: AccountModel,
122 include: [
123 {
124 model: VideoChannelModel
125 },
126 {
127 attributes: [ 'id', 'name', 'type' ],
128 model: VideoPlaylistModel.unscoped(),
129 required: true,
130 where: {
131 type: {
132 [Op.ne]: VideoPlaylistType.REGULAR
133 }
134 }
135 }
136 ]
137 }
138 ]
139 },
140 [ScopeNames.WITH_STATS]: {
141 attributes: {
142 include: [
143 [
144 literal(
145 '(' +
146 UserModel.generateUserQuotaBaseSQL({
147 withSelect: false,
148 whereUserId: '"UserModel"."id"'
149 }) +
150 ')'
151 ),
152 'videoQuotaUsed'
153 ],
154 [
155 literal(
156 '(' +
157 'SELECT COUNT("video"."id") ' +
158 'FROM "video" ' +
159 'INNER JOIN "videoChannel" ON "videoChannel"."id" = "video"."channelId" ' +
160 'INNER JOIN "account" ON "account"."id" = "videoChannel"."accountId" ' +
161 'WHERE "account"."userId" = "UserModel"."id"' +
162 ')'
163 ),
164 'videosCount'
165 ],
166 [
167 literal(
168 '(' +
169 `SELECT concat_ws(':', "abuses", "acceptedAbuses") ` +
170 'FROM (' +
171 'SELECT COUNT("videoAbuse"."id") AS "abuses", ' +
172 `COUNT("videoAbuse"."id") FILTER (WHERE "videoAbuse"."state" = ${VideoAbuseState.ACCEPTED}) AS "acceptedAbuses" ` +
173 'FROM "videoAbuse" ' +
174 'INNER JOIN "video" ON "videoAbuse"."videoId" = "video"."id" ' +
175 'INNER JOIN "videoChannel" ON "videoChannel"."id" = "video"."channelId" ' +
176 'INNER JOIN "account" ON "account"."id" = "videoChannel"."accountId" ' +
177 'WHERE "account"."userId" = "UserModel"."id"' +
178 ') t' +
179 ')'
180 ),
181 'videoAbusesCount'
182 ],
183 [
184 literal(
185 '(' +
186 'SELECT COUNT("videoAbuse"."id") ' +
187 'FROM "videoAbuse" ' +
188 'INNER JOIN "account" ON "account"."id" = "videoAbuse"."reporterAccountId" ' +
189 'WHERE "account"."userId" = "UserModel"."id"' +
190 ')'
191 ),
192 'videoAbusesCreatedCount'
193 ],
194 [
195 literal(
196 '(' +
197 'SELECT COUNT("videoComment"."id") ' +
198 'FROM "videoComment" ' +
199 'INNER JOIN "account" ON "account"."id" = "videoComment"."accountId" ' +
200 'WHERE "account"."userId" = "UserModel"."id"' +
201 ')'
202 ),
203 'videoCommentsCount'
204 ]
205 ]
206 }
207 }
208 }))
209 @Table({
210 tableName: 'user',
211 indexes: [
212 {
213 fields: [ 'username' ],
214 unique: true
215 },
216 {
217 fields: [ 'email' ],
218 unique: true
219 }
220 ]
221 })
222 export class UserModel extends Model<UserModel> {
223
224 @AllowNull(false)
225 @Is('UserPassword', value => throwIfNotValid(value, isUserPasswordValid, 'user password'))
226 @Column
227 password: string
228
229 @AllowNull(false)
230 @Is('UserUsername', value => throwIfNotValid(value, isUserUsernameValid, 'user name'))
231 @Column
232 username: string
233
234 @AllowNull(false)
235 @IsEmail
236 @Column(DataType.STRING(400))
237 email: string
238
239 @AllowNull(true)
240 @IsEmail
241 @Column(DataType.STRING(400))
242 pendingEmail: string
243
244 @AllowNull(true)
245 @Default(null)
246 @Is('UserEmailVerified', value => throwIfNotValid(value, isUserEmailVerifiedValid, 'email verified boolean', true))
247 @Column
248 emailVerified: boolean
249
250 @AllowNull(false)
251 @Is('UserNSFWPolicy', value => throwIfNotValid(value, isUserNSFWPolicyValid, 'NSFW policy'))
252 @Column(DataType.ENUM(...values(NSFW_POLICY_TYPES)))
253 nsfwPolicy: NSFWPolicyType
254
255 @AllowNull(false)
256 @Default(true)
257 @Is('UserWebTorrentEnabled', value => throwIfNotValid(value, isUserWebTorrentEnabledValid, 'WebTorrent enabled'))
258 @Column
259 webTorrentEnabled: boolean
260
261 @AllowNull(false)
262 @Default(true)
263 @Is('UserVideosHistoryEnabled', value => throwIfNotValid(value, isUserVideosHistoryEnabledValid, 'Videos history enabled'))
264 @Column
265 videosHistoryEnabled: boolean
266
267 @AllowNull(false)
268 @Default(true)
269 @Is('UserAutoPlayVideo', value => throwIfNotValid(value, isUserAutoPlayVideoValid, 'auto play video boolean'))
270 @Column
271 autoPlayVideo: boolean
272
273 @AllowNull(false)
274 @Default(false)
275 @Is('UserAutoPlayNextVideo', value => throwIfNotValid(value, isUserAutoPlayNextVideoValid, 'auto play next video boolean'))
276 @Column
277 autoPlayNextVideo: boolean
278
279 @AllowNull(false)
280 @Default(true)
281 @Is(
282 'UserAutoPlayNextVideoPlaylist',
283 value => throwIfNotValid(value, isUserAutoPlayNextVideoPlaylistValid, 'auto play next video for playlists boolean')
284 )
285 @Column
286 autoPlayNextVideoPlaylist: boolean
287
288 @AllowNull(true)
289 @Default(null)
290 @Is('UserVideoLanguages', value => throwIfNotValid(value, isUserVideoLanguages, 'video languages'))
291 @Column(DataType.ARRAY(DataType.STRING))
292 videoLanguages: string[]
293
294 @AllowNull(false)
295 @Default(UserAdminFlag.NONE)
296 @Is('UserAdminFlags', value => throwIfNotValid(value, isUserAdminFlagsValid, 'user admin flags'))
297 @Column
298 adminFlags?: UserAdminFlag
299
300 @AllowNull(false)
301 @Default(false)
302 @Is('UserBlocked', value => throwIfNotValid(value, isUserBlockedValid, 'blocked boolean'))
303 @Column
304 blocked: boolean
305
306 @AllowNull(true)
307 @Default(null)
308 @Is('UserBlockedReason', value => throwIfNotValid(value, isUserBlockedReasonValid, 'blocked reason', true))
309 @Column
310 blockedReason: string
311
312 @AllowNull(false)
313 @Is('UserRole', value => throwIfNotValid(value, isUserRoleValid, 'role'))
314 @Column
315 role: number
316
317 @AllowNull(false)
318 @Is('UserVideoQuota', value => throwIfNotValid(value, isUserVideoQuotaValid, 'video quota'))
319 @Column(DataType.BIGINT)
320 videoQuota: number
321
322 @AllowNull(false)
323 @Is('UserVideoQuotaDaily', value => throwIfNotValid(value, isUserVideoQuotaDailyValid, 'video quota daily'))
324 @Column(DataType.BIGINT)
325 videoQuotaDaily: number
326
327 @AllowNull(false)
328 @Default(DEFAULT_USER_THEME_NAME)
329 @Is('UserTheme', value => throwIfNotValid(value, isThemeNameValid, 'theme'))
330 @Column
331 theme: string
332
333 @AllowNull(false)
334 @Default(false)
335 @Is(
336 'UserNoInstanceConfigWarningModal',
337 value => throwIfNotValid(value, isNoInstanceConfigWarningModal, 'no instance config warning modal')
338 )
339 @Column
340 noInstanceConfigWarningModal: boolean
341
342 @AllowNull(false)
343 @Default(false)
344 @Is(
345 'UserNoInstanceConfigWarningModal',
346 value => throwIfNotValid(value, isNoWelcomeModal, 'no welcome modal')
347 )
348 @Column
349 noWelcomeModal: boolean
350
351 @CreatedAt
352 createdAt: Date
353
354 @UpdatedAt
355 updatedAt: Date
356
357 @HasOne(() => AccountModel, {
358 foreignKey: 'userId',
359 onDelete: 'cascade',
360 hooks: true
361 })
362 Account: AccountModel
363
364 @HasOne(() => UserNotificationSettingModel, {
365 foreignKey: 'userId',
366 onDelete: 'cascade',
367 hooks: true
368 })
369 NotificationSetting: UserNotificationSettingModel
370
371 @HasMany(() => VideoImportModel, {
372 foreignKey: 'userId',
373 onDelete: 'cascade'
374 })
375 VideoImports: VideoImportModel[]
376
377 @HasMany(() => OAuthTokenModel, {
378 foreignKey: 'userId',
379 onDelete: 'cascade'
380 })
381 OAuthTokens: OAuthTokenModel[]
382
383 @BeforeCreate
384 @BeforeUpdate
385 static cryptPasswordIfNeeded (instance: UserModel) {
386 if (instance.changed('password')) {
387 return cryptPassword(instance.password)
388 .then(hash => {
389 instance.password = hash
390 return undefined
391 })
392 }
393 }
394
395 @AfterUpdate
396 @AfterDestroy
397 static removeTokenCache (instance: UserModel) {
398 return clearCacheByUserId(instance.id)
399 }
400
401 static countTotal () {
402 return this.count()
403 }
404
405 static listForApi (start: number, count: number, sort: string, search?: string) {
406 let where: WhereOptions
407
408 if (search) {
409 where = {
410 [Op.or]: [
411 {
412 email: {
413 [Op.iLike]: '%' + search + '%'
414 }
415 },
416 {
417 username: {
418 [Op.iLike]: '%' + search + '%'
419 }
420 }
421 ]
422 }
423 }
424
425 const query: FindOptions = {
426 attributes: {
427 include: [
428 [
429 literal(
430 '(' +
431 UserModel.generateUserQuotaBaseSQL({
432 withSelect: false,
433 whereUserId: '"UserModel"."id"'
434 }) +
435 ')'
436 ),
437 'videoQuotaUsed'
438 ] as any // FIXME: typings
439 ]
440 },
441 offset: start,
442 limit: count,
443 order: getSort(sort),
444 where
445 }
446
447 return UserModel.findAndCountAll(query)
448 .then(({ rows, count }) => {
449 return {
450 data: rows,
451 total: count
452 }
453 })
454 }
455
456 static listWithRight (right: UserRight): Bluebird<MUserDefault[]> {
457 const roles = Object.keys(USER_ROLE_LABELS)
458 .map(k => parseInt(k, 10) as UserRole)
459 .filter(role => hasUserRight(role, right))
460
461 const query = {
462 where: {
463 role: {
464 [Op.in]: roles
465 }
466 }
467 }
468
469 return UserModel.findAll(query)
470 }
471
472 static listUserSubscribersOf (actorId: number): Bluebird<MUserWithNotificationSetting[]> {
473 const query = {
474 include: [
475 {
476 model: UserNotificationSettingModel.unscoped(),
477 required: true
478 },
479 {
480 attributes: [ 'userId' ],
481 model: AccountModel.unscoped(),
482 required: true,
483 include: [
484 {
485 attributes: [],
486 model: ActorModel.unscoped(),
487 required: true,
488 where: {
489 serverId: null
490 },
491 include: [
492 {
493 attributes: [],
494 as: 'ActorFollowings',
495 model: ActorFollowModel.unscoped(),
496 required: true,
497 where: {
498 targetActorId: actorId
499 }
500 }
501 ]
502 }
503 ]
504 }
505 ]
506 }
507
508 return UserModel.unscoped().findAll(query)
509 }
510
511 static listByUsernames (usernames: string[]): Bluebird<MUserDefault[]> {
512 const query = {
513 where: {
514 username: usernames
515 }
516 }
517
518 return UserModel.findAll(query)
519 }
520
521 static loadById (id: number, withStats = false): Bluebird<MUserDefault> {
522 const scopes = [
523 ScopeNames.WITH_VIDEOCHANNELS
524 ]
525
526 if (withStats) scopes.push(ScopeNames.WITH_STATS)
527
528 return UserModel.scope(scopes).findByPk(id)
529 }
530
531 static loadByUsername (username: string): Bluebird<MUserDefault> {
532 const query = {
533 where: {
534 username: { [Op.iLike]: username }
535 }
536 }
537
538 return UserModel.findOne(query)
539 }
540
541 static loadForMeAPI (username: string): Bluebird<MUserNotifSettingChannelDefault> {
542 const query = {
543 where: {
544 username: { [Op.iLike]: username }
545 }
546 }
547
548 return UserModel.scope(ScopeNames.FOR_ME_API).findOne(query)
549 }
550
551 static loadByEmail (email: string): Bluebird<MUserDefault> {
552 const query = {
553 where: {
554 email
555 }
556 }
557
558 return UserModel.findOne(query)
559 }
560
561 static loadByUsernameOrEmail (username: string, email?: string): Bluebird<MUserDefault> {
562 if (!email) email = username
563
564 const query = {
565 where: {
566 [Op.or]: [
567 where(fn('lower', col('username')), fn('lower', username)),
568
569 { email }
570 ]
571 }
572 }
573
574 return UserModel.findOne(query)
575 }
576
577 static loadByVideoId (videoId: number): Bluebird<MUserDefault> {
578 const query = {
579 include: [
580 {
581 required: true,
582 attributes: [ 'id' ],
583 model: AccountModel.unscoped(),
584 include: [
585 {
586 required: true,
587 attributes: [ 'id' ],
588 model: VideoChannelModel.unscoped(),
589 include: [
590 {
591 required: true,
592 attributes: [ 'id' ],
593 model: VideoModel.unscoped(),
594 where: {
595 id: videoId
596 }
597 }
598 ]
599 }
600 ]
601 }
602 ]
603 }
604
605 return UserModel.findOne(query)
606 }
607
608 static loadByVideoImportId (videoImportId: number): Bluebird<MUserDefault> {
609 const query = {
610 include: [
611 {
612 required: true,
613 attributes: [ 'id' ],
614 model: VideoImportModel.unscoped(),
615 where: {
616 id: videoImportId
617 }
618 }
619 ]
620 }
621
622 return UserModel.findOne(query)
623 }
624
625 static loadByChannelActorId (videoChannelActorId: number): Bluebird<MUserDefault> {
626 const query = {
627 include: [
628 {
629 required: true,
630 attributes: [ 'id' ],
631 model: AccountModel.unscoped(),
632 include: [
633 {
634 required: true,
635 attributes: [ 'id' ],
636 model: VideoChannelModel.unscoped(),
637 where: {
638 actorId: videoChannelActorId
639 }
640 }
641 ]
642 }
643 ]
644 }
645
646 return UserModel.findOne(query)
647 }
648
649 static loadByAccountActorId (accountActorId: number): Bluebird<MUserDefault> {
650 const query = {
651 include: [
652 {
653 required: true,
654 attributes: [ 'id' ],
655 model: AccountModel.unscoped(),
656 where: {
657 actorId: accountActorId
658 }
659 }
660 ]
661 }
662
663 return UserModel.findOne(query)
664 }
665
666 static getOriginalVideoFileTotalFromUser (user: MUserId) {
667 // Don't use sequelize because we need to use a sub query
668 const query = UserModel.generateUserQuotaBaseSQL({
669 withSelect: true,
670 whereUserId: '$userId'
671 })
672
673 return UserModel.getTotalRawQuery(query, user.id)
674 }
675
676 // Returns cumulative size of all video files uploaded in the last 24 hours.
677 static getOriginalVideoFileTotalDailyFromUser (user: MUserId) {
678 // Don't use sequelize because we need to use a sub query
679 const query = UserModel.generateUserQuotaBaseSQL({
680 withSelect: true,
681 whereUserId: '$userId',
682 where: '"video"."createdAt" > now() - interval \'24 hours\''
683 })
684
685 return UserModel.getTotalRawQuery(query, user.id)
686 }
687
688 static async getStats () {
689 const totalUsers = await UserModel.count()
690
691 return {
692 totalUsers
693 }
694 }
695
696 static autoComplete (search: string) {
697 const query = {
698 where: {
699 username: {
700 [Op.like]: `%${search}%`
701 }
702 },
703 limit: 10
704 }
705
706 return UserModel.findAll(query)
707 .then(u => u.map(u => u.username))
708 }
709
710 canGetVideo (video: MVideoFullLight) {
711 const videoUserId = video.VideoChannel.Account.userId
712
713 if (video.isBlacklisted()) {
714 return videoUserId === this.id || this.hasRight(UserRight.MANAGE_VIDEO_BLACKLIST)
715 }
716
717 if (video.privacy === VideoPrivacy.PRIVATE) {
718 return video.VideoChannel && videoUserId === this.id || this.hasRight(UserRight.MANAGE_VIDEO_BLACKLIST)
719 }
720
721 if (video.privacy === VideoPrivacy.INTERNAL) return true
722
723 return false
724 }
725
726 hasRight (right: UserRight) {
727 return hasUserRight(this.role, right)
728 }
729
730 hasAdminFlag (flag: UserAdminFlag) {
731 return this.adminFlags & flag
732 }
733
734 isPasswordMatch (password: string) {
735 return comparePassword(password, this.password)
736 }
737
738 toFormattedJSON (this: MUserFormattable, parameters: { withAdminFlags?: boolean } = {}): User {
739 const videoQuotaUsed = this.get('videoQuotaUsed')
740 const videoQuotaUsedDaily = this.get('videoQuotaUsedDaily')
741 const videosCount = this.get('videosCount')
742 const [ videoAbusesCount, videoAbusesAcceptedCount ] = (this.get('videoAbusesCount') as string || ':').split(':')
743 const videoAbusesCreatedCount = this.get('videoAbusesCreatedCount')
744 const videoCommentsCount = this.get('videoCommentsCount')
745
746 const json: User = {
747 id: this.id,
748 username: this.username,
749 email: this.email,
750 theme: getThemeOrDefault(this.theme, DEFAULT_USER_THEME_NAME),
751
752 pendingEmail: this.pendingEmail,
753 emailVerified: this.emailVerified,
754
755 nsfwPolicy: this.nsfwPolicy,
756 webTorrentEnabled: this.webTorrentEnabled,
757 videosHistoryEnabled: this.videosHistoryEnabled,
758 autoPlayVideo: this.autoPlayVideo,
759 autoPlayNextVideo: this.autoPlayNextVideo,
760 autoPlayNextVideoPlaylist: this.autoPlayNextVideoPlaylist,
761 videoLanguages: this.videoLanguages,
762
763 role: this.role,
764 roleLabel: USER_ROLE_LABELS[this.role],
765
766 videoQuota: this.videoQuota,
767 videoQuotaDaily: this.videoQuotaDaily,
768 videoQuotaUsed: videoQuotaUsed !== undefined
769 ? parseInt(videoQuotaUsed + '', 10)
770 : undefined,
771 videoQuotaUsedDaily: videoQuotaUsedDaily !== undefined
772 ? parseInt(videoQuotaUsedDaily + '', 10)
773 : undefined,
774 videosCount: videosCount !== undefined
775 ? parseInt(videosCount + '', 10)
776 : undefined,
777 videoAbusesCount: videoAbusesCount
778 ? parseInt(videoAbusesCount, 10)
779 : undefined,
780 videoAbusesAcceptedCount: videoAbusesAcceptedCount
781 ? parseInt(videoAbusesAcceptedCount, 10)
782 : undefined,
783 videoAbusesCreatedCount: videoAbusesCreatedCount !== undefined
784 ? parseInt(videoAbusesCreatedCount + '', 10)
785 : undefined,
786 videoCommentsCount: videoCommentsCount !== undefined
787 ? parseInt(videoCommentsCount + '', 10)
788 : undefined,
789
790 noInstanceConfigWarningModal: this.noInstanceConfigWarningModal,
791 noWelcomeModal: this.noWelcomeModal,
792
793 blocked: this.blocked,
794 blockedReason: this.blockedReason,
795
796 account: this.Account.toFormattedJSON(),
797
798 notificationSettings: this.NotificationSetting
799 ? this.NotificationSetting.toFormattedJSON()
800 : undefined,
801
802 videoChannels: [],
803
804 createdAt: this.createdAt
805 }
806
807 if (parameters.withAdminFlags) {
808 Object.assign(json, { adminFlags: this.adminFlags })
809 }
810
811 if (Array.isArray(this.Account.VideoChannels) === true) {
812 json.videoChannels = this.Account.VideoChannels
813 .map(c => c.toFormattedJSON())
814 .sort((v1, v2) => {
815 if (v1.createdAt < v2.createdAt) return -1
816 if (v1.createdAt === v2.createdAt) return 0
817
818 return 1
819 })
820 }
821
822 return json
823 }
824
825 toMeFormattedJSON (this: MMyUserFormattable): MyUser {
826 const formatted = this.toFormattedJSON()
827
828 const specialPlaylists = this.Account.VideoPlaylists
829 .map(p => ({ id: p.id, name: p.name, type: p.type }))
830
831 return Object.assign(formatted, { specialPlaylists })
832 }
833
834 async isAbleToUploadVideo (videoFile: { size: number }) {
835 if (this.videoQuota === -1 && this.videoQuotaDaily === -1) return Promise.resolve(true)
836
837 const [ totalBytes, totalBytesDaily ] = await Promise.all([
838 UserModel.getOriginalVideoFileTotalFromUser(this),
839 UserModel.getOriginalVideoFileTotalDailyFromUser(this)
840 ])
841
842 const uploadedTotal = videoFile.size + totalBytes
843 const uploadedDaily = videoFile.size + totalBytesDaily
844
845 if (this.videoQuotaDaily === -1) return uploadedTotal < this.videoQuota
846 if (this.videoQuota === -1) return uploadedDaily < this.videoQuotaDaily
847
848 return uploadedTotal < this.videoQuota && uploadedDaily < this.videoQuotaDaily
849 }
850
851 private static generateUserQuotaBaseSQL (options: {
852 whereUserId: '$userId' | '"UserModel"."id"'
853 withSelect: boolean
854 where?: string
855 }) {
856 const andWhere = options.where
857 ? 'AND ' + options.where
858 : ''
859
860 const videoChannelJoin = 'INNER JOIN "videoChannel" ON "videoChannel"."id" = "video"."channelId" ' +
861 'INNER JOIN "account" ON "videoChannel"."accountId" = "account"."id" ' +
862 `WHERE "account"."userId" = ${options.whereUserId} ${andWhere}`
863
864 const webtorrentFiles = 'SELECT "videoFile"."size" AS "size", "video"."id" AS "videoId" FROM "videoFile" ' +
865 'INNER JOIN "video" ON "videoFile"."videoId" = "video"."id" ' +
866 videoChannelJoin
867
868 const hlsFiles = 'SELECT "videoFile"."size" AS "size", "video"."id" AS "videoId" FROM "videoFile" ' +
869 'INNER JOIN "videoStreamingPlaylist" ON "videoFile"."videoStreamingPlaylistId" = "videoStreamingPlaylist".id ' +
870 'INNER JOIN "video" ON "videoStreamingPlaylist"."videoId" = "video"."id" ' +
871 videoChannelJoin
872
873 return 'SELECT COALESCE(SUM("size"), 0) AS "total" ' +
874 'FROM (' +
875 `SELECT MAX("t1"."size") AS "size" FROM (${webtorrentFiles} UNION ${hlsFiles}) t1 ` +
876 'GROUP BY "t1"."videoId"' +
877 ') t2'
878 }
879
880 private static getTotalRawQuery (query: string, userId: number) {
881 const options = {
882 bind: { userId },
883 type: QueryTypes.SELECT as QueryTypes.SELECT
884 }
885
886 return UserModel.sequelize.query<{ total: string }>(query, options)
887 .then(([ { total } ]) => {
888 if (total === null) return 0
889
890 return parseInt(total, 10)
891 })
892 }
893 }