1 import { User } from '@app/core/users/user.model'
2 import { peertubeLocalStorage } from '@app/helpers/peertube-web-storage'
5 MyUser as ServerMyUserModel,
8 User as ServerUserModel,
11 } from '@shared/models'
13 export type TokenOptions = {
19 // Private class only used by User
21 private static KEYS = {
22 ACCESS_TOKEN: 'access_token',
23 REFRESH_TOKEN: 'refresh_token',
24 TOKEN_TYPE: 'token_type'
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)
36 if (accessTokenLocalStorage && refreshTokenLocalStorage && tokenTypeLocalStorage) {
38 accessToken: accessTokenLocalStorage,
39 refreshToken: refreshTokenLocalStorage,
40 tokenType: tokenTypeLocalStorage
48 peertubeLocalStorage.removeItem(this.KEYS.ACCESS_TOKEN)
49 peertubeLocalStorage.removeItem(this.KEYS.REFRESH_TOKEN)
50 peertubeLocalStorage.removeItem(this.KEYS.TOKEN_TYPE)
53 constructor (hash?: TokenOptions) {
55 this.accessToken = hash.accessToken
56 this.refreshToken = hash.refreshToken
58 if (hash.tokenType === 'bearer') {
59 this.tokenType = 'Bearer'
61 this.tokenType = hash.tokenType
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)
73 export class AuthUser extends User implements ServerMyUserModel {
75 specialPlaylists: MyUserSpecialPlaylist[]
78 const usernameLocalStorage = peertubeLocalStorage.getItem(this.KEYS.USERNAME)
79 if (usernameLocalStorage) {
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'
99 peertubeLocalStorage.removeItem(this.KEYS.USERNAME)
100 peertubeLocalStorage.removeItem(this.KEYS.ID)
101 peertubeLocalStorage.removeItem(this.KEYS.ROLE)
102 peertubeLocalStorage.removeItem(this.KEYS.EMAIL)
106 constructor (userHash: Partial<ServerMyUserModel>, hashTokens: TokenOptions) {
109 this.tokens = new Tokens(hashTokens)
110 this.specialPlaylists = userHash.specialPlaylists
114 return this.tokens.accessToken
118 return this.tokens.refreshToken
122 return this.tokens.tokenType
125 refreshTokens (accessToken: string, refreshToken: string) {
126 this.tokens.accessToken = accessToken
127 this.tokens.refreshToken = refreshToken
130 hasRight (right: UserRight) {
131 return hasUserRight(this.role, right)
134 canManage (user: ServerUserModel) {
135 const myRole = this.role
137 if (myRole === UserRole.ADMINISTRATOR) return true
139 // I'm a moderator: I can only manage users
140 return user.role === UserRole.USER
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))