client/src/app/shared/users/user.model.ts

   1 import {
   2   Account as AccountServerModel,
   3   hasUserRight,
   4   User as UserServerModel,
   5   UserRight,
   6   UserRole,
   7   VideoChannel
   8 } from '../../../../../shared'
   9 import { NSFWPolicyType } from '../../../../../shared/models/videos/nsfw-policy.type'
  10 import { Account } from '@app/shared/account/account.model'
  11 import { Avatar } from '../../../../../shared/models/avatars/avatar.model'
  12
  13 export type UserConstructorHash = {
  14   id: number,
  15   username: string,
  16   email: string,
  17   role: UserRole,
  18   videoQuota?: number,
  19   videoQuotaDaily?: number,
  20   nsfwPolicy?: NSFWPolicyType,
  21   webTorrentEnabled?: boolean,
  22   autoPlayVideo?: boolean,
  23   createdAt?: Date,
  24   account?: AccountServerModel,
  25   videoChannels?: VideoChannel[]
  26
  27   blocked?: boolean
  28   blockedReason?: string
  29 }
  30 export class User implements UserServerModel {
  31   id: number
  32   username: string
  33   email: string
  34   role: UserRole
  35   nsfwPolicy: NSFWPolicyType
  36   webTorrentEnabled: boolean
  37   autoPlayVideo: boolean
  38   videoQuota: number
  39   videoQuotaDaily: number
  40   account: Account
  41   videoChannels: VideoChannel[]
  42   createdAt: Date
  43
  44   blocked: boolean
  45   blockedReason?: string
  46   [key: string]: any
  47
  48   constructor (hash: UserConstructorHash) {
  49     this.id = hash.id
  50     this.username = hash.username
  51     this.email = hash.email
  52     this.role = hash.role
  53
  54     this.videoChannels = hash.videoChannels
  55     this.videoQuota = hash.videoQuota
  56     this.videoQuotaDaily = hash.videoQuotaDaily
  57     this.nsfwPolicy = hash.nsfwPolicy
  58     this.webTorrentEnabled = hash.webTorrentEnabled
  59     this.autoPlayVideo = hash.autoPlayVideo
  60     this.createdAt = hash.createdAt
  61     this.blocked = hash.blocked
  62     this.blockedReason = hash.blockedReason
  63
  64     if (hash.account !== undefined) {
  65       this.account = new Account(hash.account)
  66     }
  67   }
  68
  69   get accountAvatarUrl () {
  70     if (!this.account) return ''
  71
  72     return this.account.avatarUrl
  73   }
  74
  75   hasRight (right: UserRight) {
  76     return hasUserRight(this.role, right)
  77   }
  78
  79   patch (obj: UserServerModel) {
  80     for (const key of Object.keys(obj)) {
  81       this[key] = obj[key]
  82     }
  83
  84     if (obj.account !== undefined) {
  85       this.account = new Account(obj.account)
  86     }
  87   }
  88
  89   updateAccountAvatar (newAccountAvatar: Avatar) {
  90     this.account.updateAvatar(newAccountAvatar)
  91   }
  92 }