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

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