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, User as ServerUserModel, 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'
9 export type TokenOptions = {
15 // Private class only used by User
17 private static KEYS = {
18 ACCESS_TOKEN: 'access_token',
19 REFRESH_TOKEN: 'refresh_token',
20 TOKEN_TYPE: 'token_type'
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)
32 if (accessTokenLocalStorage && refreshTokenLocalStorage && tokenTypeLocalStorage) {
34 accessToken: accessTokenLocalStorage,
35 refreshToken: refreshTokenLocalStorage,
36 tokenType: tokenTypeLocalStorage
44 peertubeLocalStorage.removeItem(this.KEYS.ACCESS_TOKEN)
45 peertubeLocalStorage.removeItem(this.KEYS.REFRESH_TOKEN)
46 peertubeLocalStorage.removeItem(this.KEYS.TOKEN_TYPE)
49 constructor (hash?: TokenOptions) {
51 this.accessToken = hash.accessToken
52 this.refreshToken = hash.refreshToken
54 if (hash.tokenType === 'bearer') {
55 this.tokenType = 'Bearer'
57 this.tokenType = hash.tokenType
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)
69 export class AuthUser extends User implements ServerMyUserModel {
71 specialPlaylists: MyUserSpecialPlaylist[]
74 const usernameLocalStorage = peertubeLocalStorage.getItem(this.KEYS.USERNAME)
75 if (usernameLocalStorage) {
78 id: parseInt(peertubeLocalStorage.getItem(this.KEYS.ID), 10),
79 username: peertubeLocalStorage.getItem(this.KEYS.USERNAME),
80 email: peertubeLocalStorage.getItem(this.KEYS.EMAIL),
81 role: parseInt(peertubeLocalStorage.getItem(this.KEYS.ROLE), 10) as UserRole,
82 nsfwPolicy: peertubeLocalStorage.getItem(this.KEYS.NSFW_POLICY) as NSFWPolicyType,
83 webTorrentEnabled: peertubeLocalStorage.getItem(this.KEYS.WEBTORRENT_ENABLED) === 'true',
84 autoPlayVideo: peertubeLocalStorage.getItem(this.KEYS.AUTO_PLAY_VIDEO) === 'true',
85 videosHistoryEnabled: peertubeLocalStorage.getItem(this.KEYS.VIDEOS_HISTORY_ENABLED) === 'true'
95 peertubeLocalStorage.removeItem(this.KEYS.USERNAME)
96 peertubeLocalStorage.removeItem(this.KEYS.ID)
97 peertubeLocalStorage.removeItem(this.KEYS.ROLE)
98 peertubeLocalStorage.removeItem(this.KEYS.EMAIL)
102 constructor (userHash: Partial<ServerMyUserModel>, hashTokens: TokenOptions) {
105 this.tokens = new Tokens(hashTokens)
106 this.specialPlaylists = userHash.specialPlaylists
110 return this.tokens.accessToken
114 return this.tokens.refreshToken
118 return this.tokens.tokenType
121 refreshTokens (accessToken: string, refreshToken: string) {
122 this.tokens.accessToken = accessToken
123 this.tokens.refreshToken = refreshToken
126 hasRight (right: UserRight) {
127 return hasUserRight(this.role, right)
130 canManage (user: ServerUserModel) {
131 const myRole = this.role
133 if (myRole === UserRole.ADMINISTRATOR) return true
135 // I'm a moderator: I can only manage users
136 return user.role === UserRole.USER
140 peertubeLocalStorage.setItem(AuthUser.KEYS.ID, this.id.toString())
141 peertubeLocalStorage.setItem(AuthUser.KEYS.USERNAME, this.username)
142 peertubeLocalStorage.setItem(AuthUser.KEYS.EMAIL, this.email)
143 peertubeLocalStorage.setItem(AuthUser.KEYS.ROLE, this.role.toString())
144 peertubeLocalStorage.setItem(AuthUser.KEYS.NSFW_POLICY, this.nsfwPolicy.toString())
145 peertubeLocalStorage.setItem(AuthUser.KEYS.WEBTORRENT_ENABLED, JSON.stringify(this.webTorrentEnabled))
146 peertubeLocalStorage.setItem(AuthUser.KEYS.AUTO_PLAY_VIDEO, JSON.stringify(this.autoPlayVideo))