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

   1 import { peertubeLocalStorage } from '@app/shared/misc/peertube-web-storage'
   2 import { UserRight } from '../../../../../shared/models/users/user-right.enum'
   3 import { MyUser as ServerMyUserModel, MyUserSpecialPlaylist } from '../../../../../shared/models/users/user.model'
   4 // Do not use the barrel (dependency loop)
   5 import { hasUserRight, UserRole } from '../../../../../shared/models/users/user-role'
   6 import { User } from '../../shared/users/user.model'
   7 import { NSFWPolicyType } from '../../../../../shared/models/videos/nsfw-policy.type'
   8
   9 export type TokenOptions = {
  10   accessToken: string
  11   refreshToken: string
  12   tokenType: string
  13 }
  14
  15 // Private class only used by User
  16 class Tokens {
  17   private static KEYS = {
  18     ACCESS_TOKEN: 'access_token',
  19     REFRESH_TOKEN: 'refresh_token',
  20     TOKEN_TYPE: 'token_type'
  21   }
  22
  23   accessToken: string
  24   refreshToken: string
  25   tokenType: string
  26
  27   static load () {
  28     const accessTokenLocalStorage = peertubeLocalStorage.getItem(this.KEYS.ACCESS_TOKEN)
  29     const refreshTokenLocalStorage = peertubeLocalStorage.getItem(this.KEYS.REFRESH_TOKEN)
  30     const tokenTypeLocalStorage = peertubeLocalStorage.getItem(this.KEYS.TOKEN_TYPE)
  31
  32     if (accessTokenLocalStorage && refreshTokenLocalStorage && tokenTypeLocalStorage) {
  33       return new Tokens({
  34         accessToken: accessTokenLocalStorage,
  35         refreshToken: refreshTokenLocalStorage,
  36         tokenType: tokenTypeLocalStorage
  37       })
  38     }
  39
  40     return null
  41   }
  42
  43   static flush () {
  44     peertubeLocalStorage.removeItem(this.KEYS.ACCESS_TOKEN)
  45     peertubeLocalStorage.removeItem(this.KEYS.REFRESH_TOKEN)
  46     peertubeLocalStorage.removeItem(this.KEYS.TOKEN_TYPE)
  47   }
  48
  49   constructor (hash?: TokenOptions) {
  50     if (hash) {
  51       this.accessToken = hash.accessToken
  52       this.refreshToken = hash.refreshToken
  53
  54       if (hash.tokenType === 'bearer') {
  55         this.tokenType = 'Bearer'
  56       } else {
  57         this.tokenType = hash.tokenType
  58       }
  59     }
  60   }
  61
  62   save () {
  63     peertubeLocalStorage.setItem(Tokens.KEYS.ACCESS_TOKEN, this.accessToken)
  64     peertubeLocalStorage.setItem(Tokens.KEYS.REFRESH_TOKEN, this.refreshToken)
  65     peertubeLocalStorage.setItem(Tokens.KEYS.TOKEN_TYPE, this.tokenType)
  66   }
  67 }
  68
  69 export class AuthUser extends User implements ServerMyUserModel {
  70   private static KEYS = {
  71     ID: 'id',
  72     ROLE: 'role',
  73     EMAIL: 'email',
  74     VIDEOS_HISTORY_ENABLED: 'videos-history-enabled',
  75     USERNAME: 'username',
  76     NSFW_POLICY: 'nsfw_policy',
  77     WEBTORRENT_ENABLED: 'peertube-videojs-' + 'webtorrent_enabled',
  78     AUTO_PLAY_VIDEO: 'auto_play_video'
  79   }
  80
  81   tokens: Tokens
  82   specialPlaylists: MyUserSpecialPlaylist[]
  83
  84   static load () {
  85     const usernameLocalStorage = peertubeLocalStorage.getItem(this.KEYS.USERNAME)
  86     if (usernameLocalStorage) {
  87       return new AuthUser(
  88         {
  89           id: parseInt(peertubeLocalStorage.getItem(this.KEYS.ID), 10),
  90           username: peertubeLocalStorage.getItem(this.KEYS.USERNAME),
  91           email: peertubeLocalStorage.getItem(this.KEYS.EMAIL),
  92           role: parseInt(peertubeLocalStorage.getItem(this.KEYS.ROLE), 10) as UserRole,
  93           nsfwPolicy: peertubeLocalStorage.getItem(this.KEYS.NSFW_POLICY) as NSFWPolicyType,
  94           webTorrentEnabled: peertubeLocalStorage.getItem(this.KEYS.WEBTORRENT_ENABLED) === 'true',
  95           autoPlayVideo: peertubeLocalStorage.getItem(this.KEYS.AUTO_PLAY_VIDEO) === 'true',
  96           videosHistoryEnabled: peertubeLocalStorage.getItem(this.KEYS.VIDEOS_HISTORY_ENABLED) === 'true'
  97         },
  98         Tokens.load()
  99       )
 100     }
 101
 102     return null
 103   }
 104
 105   static flush () {
 106     peertubeLocalStorage.removeItem(this.KEYS.USERNAME)
 107     peertubeLocalStorage.removeItem(this.KEYS.ID)
 108     peertubeLocalStorage.removeItem(this.KEYS.ROLE)
 109     peertubeLocalStorage.removeItem(this.KEYS.NSFW_POLICY)
 110     peertubeLocalStorage.removeItem(this.KEYS.WEBTORRENT_ENABLED)
 111     peertubeLocalStorage.removeItem(this.KEYS.VIDEOS_HISTORY_ENABLED)
 112     peertubeLocalStorage.removeItem(this.KEYS.AUTO_PLAY_VIDEO)
 113     peertubeLocalStorage.removeItem(this.KEYS.EMAIL)
 114     Tokens.flush()
 115   }
 116
 117   constructor (userHash: Partial<ServerMyUserModel>, hashTokens: TokenOptions) {
 118     super(userHash)
 119
 120     this.tokens = new Tokens(hashTokens)
 121     this.specialPlaylists = userHash.specialPlaylists
 122   }
 123
 124   getAccessToken () {
 125     return this.tokens.accessToken
 126   }
 127
 128   getRefreshToken () {
 129     return this.tokens.refreshToken
 130   }
 131
 132   getTokenType () {
 133     return this.tokens.tokenType
 134   }
 135
 136   refreshTokens (accessToken: string, refreshToken: string) {
 137     this.tokens.accessToken = accessToken
 138     this.tokens.refreshToken = refreshToken
 139   }
 140
 141   hasRight (right: UserRight) {
 142     return hasUserRight(this.role, right)
 143   }
 144
 145   canManage (user: ServerMyUserModel) {
 146     const myRole = this.role
 147
 148     if (myRole === UserRole.ADMINISTRATOR) return true
 149
 150     // I'm a moderator: I can only manage users
 151     return user.role === UserRole.USER
 152   }
 153
 154   save () {
 155     peertubeLocalStorage.setItem(AuthUser.KEYS.ID, this.id.toString())
 156     peertubeLocalStorage.setItem(AuthUser.KEYS.USERNAME, this.username)
 157     peertubeLocalStorage.setItem(AuthUser.KEYS.EMAIL, this.email)
 158     peertubeLocalStorage.setItem(AuthUser.KEYS.ROLE, this.role.toString())
 159     peertubeLocalStorage.setItem(AuthUser.KEYS.NSFW_POLICY, this.nsfwPolicy.toString())
 160     peertubeLocalStorage.setItem(AuthUser.KEYS.WEBTORRENT_ENABLED, JSON.stringify(this.webTorrentEnabled))
 161     peertubeLocalStorage.setItem(AuthUser.KEYS.AUTO_PLAY_VIDEO, JSON.stringify(this.autoPlayVideo))
 162     this.tokens.save()
 163   }
 164 }