client/src/app/core/auth/auth-user.model.ts

   1 import { Observable, of } from 'rxjs'
   2 import { map } from 'rxjs/operators'
   3 import { User } from '@app/core/users/user.model'
   4 import {
   5   flushUserInfoFromLocalStorage,
   6   getUserInfoFromLocalStorage,
   7   saveUserInfoIntoLocalStorage,
   8   TokenOptions,
   9   Tokens
  10 } from '@root-helpers/users'
  11 import { hasUserRight } from '@shared/core-utils/users'
  12 import {
  13   MyUser as ServerMyUserModel,
  14   MyUserSpecialPlaylist,
  15   User as ServerUserModel,
  16   UserRight,
  17   UserRole,
  18   UserVideoQuota
  19 } from '@shared/models'
  20
  21 export class AuthUser extends User implements ServerMyUserModel {
  22   tokens: Tokens
  23   specialPlaylists: MyUserSpecialPlaylist[]
  24
  25   canSeeVideosLink = true
  26
  27   static load () {
  28     const userInfo = getUserInfoFromLocalStorage()
  29
  30     if (!userInfo) return null
  31
  32     return new AuthUser(userInfo, Tokens.load())
  33   }
  34
  35   static flush () {
  36     flushUserInfoFromLocalStorage()
  37
  38     Tokens.flush()
  39   }
  40
  41   constructor (userHash: Partial<ServerMyUserModel>, hashTokens: TokenOptions) {
  42     super(userHash)
  43
  44     this.tokens = new Tokens(hashTokens)
  45     this.specialPlaylists = userHash.specialPlaylists
  46   }
  47
  48   getAccessToken () {
  49     return this.tokens.accessToken
  50   }
  51
  52   getRefreshToken () {
  53     return this.tokens.refreshToken
  54   }
  55
  56   getTokenType () {
  57     return this.tokens.tokenType
  58   }
  59
  60   refreshTokens (accessToken: string, refreshToken: string) {
  61     this.tokens.accessToken = accessToken
  62     this.tokens.refreshToken = refreshToken
  63   }
  64
  65   hasRight (right: UserRight) {
  66     return hasUserRight(this.role, right)
  67   }
  68
  69   canManage (user: ServerUserModel) {
  70     const myRole = this.role
  71
  72     if (myRole === UserRole.ADMINISTRATOR) return true
  73
  74     // I'm a moderator: I can only manage users
  75     return user.role === UserRole.USER
  76   }
  77
  78   save () {
  79     saveUserInfoIntoLocalStorage({
  80       id: this.id,
  81       username: this.username,
  82       email: this.email,
  83       role: this.role,
  84       nsfwPolicy: this.nsfwPolicy,
  85       webTorrentEnabled: this.webTorrentEnabled,
  86       autoPlayVideo: this.autoPlayVideo
  87     })
  88
  89     this.tokens.save()
  90   }
  91
  92   computeCanSeeVideosLink (quotaObservable: Observable<UserVideoQuota>): Observable<boolean> {
  93     if (!this.isUploadDisabled()) {
  94       this.canSeeVideosLink = true
  95       return of(this.canSeeVideosLink)
  96     }
  97
  98     // Check if the user has videos
  99     return quotaObservable.pipe(
 100       map(({ videoQuotaUsed }) => {
 101         if (videoQuotaUsed !== 0) {
 102           // User already uploaded videos, so it can see the link
 103           this.canSeeVideosLink = true
 104         } else {
 105           // No videos, no upload so the user don't need to see the videos link
 106           this.canSeeVideosLink = false
 107         }
 108
 109         return this.canSeeVideosLink
 110       })
 111     )
 112   }
 113 }