1 import { peertubeLocalStorage } from '@app/shared/misc/peertube-local-storage'
2 import { UserRight } from '../../../../../shared/models/users/user-right.enum'
3 // Do not use the barrel (dependency loop)
4 import { hasUserRight, UserRole } from '../../../../../shared/models/users/user-role'
5 import { User, UserConstructorHash } from '../../shared/users/user.model'
6 import { NSFWPolicyType } from '../../../../../shared/models/videos/nsfw-policy.type'
7 import { WebTorrentPolicyType } from '../../../../../shared/models/users/user-webtorrent-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 {
70 private static KEYS = {
75 NSFW_POLICY: 'nsfw_policy',
76 WEBTORRENT_POLICY: 'peertube-videojs-' + 'webtorrent_policy',
77 AUTO_PLAY_VIDEO: 'auto_play_video'
83 const usernameLocalStorage = peertubeLocalStorage.getItem(this.KEYS.USERNAME)
84 if (usernameLocalStorage) {
87 id: parseInt(peertubeLocalStorage.getItem(this.KEYS.ID), 10),
88 username: peertubeLocalStorage.getItem(this.KEYS.USERNAME),
89 email: peertubeLocalStorage.getItem(this.KEYS.EMAIL),
90 role: parseInt(peertubeLocalStorage.getItem(this.KEYS.ROLE), 10) as UserRole,
91 nsfwPolicy: peertubeLocalStorage.getItem(this.KEYS.NSFW_POLICY) as NSFWPolicyType,
92 webTorrentPolicy: peertubeLocalStorage.getItem(this.KEYS.WEBTORRENT_POLICY) as WebTorrentPolicyType,
93 autoPlayVideo: peertubeLocalStorage.getItem(this.KEYS.AUTO_PLAY_VIDEO) === 'true'
103 peertubeLocalStorage.removeItem(this.KEYS.USERNAME)
104 peertubeLocalStorage.removeItem(this.KEYS.ID)
105 peertubeLocalStorage.removeItem(this.KEYS.ROLE)
106 peertubeLocalStorage.removeItem(this.KEYS.NSFW_POLICY)
107 peertubeLocalStorage.removeItem(this.KEYS.WEBTORRENT_POLICY)
108 peertubeLocalStorage.removeItem(this.KEYS.AUTO_PLAY_VIDEO)
109 peertubeLocalStorage.removeItem(this.KEYS.EMAIL)
113 constructor (userHash: UserConstructorHash, hashTokens: TokenOptions) {
115 this.tokens = new Tokens(hashTokens)
119 return this.tokens.accessToken
123 return this.tokens.refreshToken
127 return this.tokens.tokenType
130 refreshTokens (accessToken: string, refreshToken: string) {
131 this.tokens.accessToken = accessToken
132 this.tokens.refreshToken = refreshToken
135 hasRight (right: UserRight) {
136 return hasUserRight(this.role, right)
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_POLICY, this.webTorrentPolicy.toString())
146 peertubeLocalStorage.setItem(AuthUser.KEYS.AUTO_PLAY_VIDEO, JSON.stringify(this.autoPlayVideo))