Cleanup tokens logic in embed

4 files changed, 134 insertions, 0 deletions

diff --git a/client/src/root-helpers/users/index.ts b/client/src/root-helpers/users/index.ts
new file mode 100644
index 000000000..8fbaca9e3
--- /dev/null
+++ b/client/src/root-helpers/users/index.ts
@@ -0,0 +1,3 @@
+export * from './user-local-storage-keys'
+export * from './user-local-storage-manager'
+export * from './user-tokens'
diff --git a/client/src/root-helpers/users/user-local-storage-keys.ts b/client/src/root-helpers/users/user-local-storage-keys.ts
new file mode 100644
index 000000000..5f915899c
--- /dev/null
+++ b/client/src/root-helpers/users/user-local-storage-keys.ts
@@ -0,0 +1,15 @@
+export const UserLocalStorageKeys = {
+  ID: 'id',
+  ROLE: 'role',
+  EMAIL: 'email',
+  VIDEOS_HISTORY_ENABLED: 'videos-history-enabled',
+  USERNAME: 'username',
+  NSFW_POLICY: 'nsfw_policy',
+  WEBTORRENT_ENABLED: 'peertube-videojs-' + 'webtorrent_enabled',
+  AUTO_PLAY_VIDEO: 'auto_play_video',
+  SESSION_STORAGE_AUTO_PLAY_NEXT_VIDEO: 'auto_play_next_video',
+  AUTO_PLAY_VIDEO_PLAYLIST: 'auto_play_video_playlist',
+  THEME: 'theme',
+  LAST_ACTIVE_THEME: 'last_active_theme',
+  VIDEO_LANGUAGES: 'video_languages'
+}
diff --git a/client/src/root-helpers/users/user-local-storage-manager.ts b/client/src/root-helpers/users/user-local-storage-manager.ts
new file mode 100644
index 000000000..c75cea127
--- /dev/null
+++ b/client/src/root-helpers/users/user-local-storage-manager.ts
@@ -0,0 +1,55 @@
+import { NSFWPolicyType, UserRole } from '@shared/models'
+import { peertubeLocalStorage } from '../peertube-web-storage'
+import { UserLocalStorageKeys } from './user-local-storage-keys'
+
+function getUserInfoFromLocalStorage () {
+  const usernameLocalStorage = peertubeLocalStorage.getItem(UserLocalStorageKeys.USERNAME)
+
+  if (!usernameLocalStorage) return undefined
+
+  return {
+    id: parseInt(peertubeLocalStorage.getItem(UserLocalStorageKeys.ID), 10),
+    username: peertubeLocalStorage.getItem(UserLocalStorageKeys.USERNAME),
+    email: peertubeLocalStorage.getItem(UserLocalStorageKeys.EMAIL),
+    role: parseInt(peertubeLocalStorage.getItem(UserLocalStorageKeys.ROLE), 10) as UserRole,
+    nsfwPolicy: peertubeLocalStorage.getItem(UserLocalStorageKeys.NSFW_POLICY) as NSFWPolicyType,
+    webTorrentEnabled: peertubeLocalStorage.getItem(UserLocalStorageKeys.WEBTORRENT_ENABLED) === 'true',
+    autoPlayVideo: peertubeLocalStorage.getItem(UserLocalStorageKeys.AUTO_PLAY_VIDEO) === 'true',
+    videosHistoryEnabled: peertubeLocalStorage.getItem(UserLocalStorageKeys.VIDEOS_HISTORY_ENABLED) === 'true'
+  }
+}
+
+function flushUserInfoFromLocalStorage () {
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.ID)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.USERNAME)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.EMAIL)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.ROLE)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.NSFW_POLICY)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.WEBTORRENT_ENABLED)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.AUTO_PLAY_VIDEO)
+  peertubeLocalStorage.removeItem(UserLocalStorageKeys.VIDEOS_HISTORY_ENABLED)
+}
+
+function saveUserInfoIntoLocalStorage (info: {
+  id: number
+  username: string
+  email: string
+  role: UserRole
+  nsfwPolicy: NSFWPolicyType
+  webTorrentEnabled: boolean
+  autoPlayVideo: boolean
+}) {
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.ID, info.id.toString())
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.USERNAME, info.username)
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.EMAIL, info.email)
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.ROLE, info.role.toString())
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.NSFW_POLICY, info.nsfwPolicy.toString())
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.WEBTORRENT_ENABLED, JSON.stringify(info.webTorrentEnabled))
+  peertubeLocalStorage.setItem(UserLocalStorageKeys.AUTO_PLAY_VIDEO, JSON.stringify(info.autoPlayVideo))
+}
+
+export {
+  getUserInfoFromLocalStorage,
+  saveUserInfoIntoLocalStorage,
+  flushUserInfoFromLocalStorage
+}
diff --git a/client/src/root-helpers/users/user-tokens.ts b/client/src/root-helpers/users/user-tokens.ts
new file mode 100644
index 000000000..d42e1c8f3
--- /dev/null
+++ b/client/src/root-helpers/users/user-tokens.ts
@@ -0,0 +1,61 @@
+import { peertubeLocalStorage } from '../peertube-web-storage'
+
+export type TokenOptions = {
+  accessToken: string
+  refreshToken: string
+  tokenType: string
+}
+
+// Private class only used by User
+export class Tokens {
+  private static KEYS = {
+    ACCESS_TOKEN: 'access_token',
+    REFRESH_TOKEN: 'refresh_token',
+    TOKEN_TYPE: 'token_type'
+  }
+
+  accessToken: string
+  refreshToken: string
+  tokenType: string
+
+  static load () {
+    const accessTokenLocalStorage = peertubeLocalStorage.getItem(this.KEYS.ACCESS_TOKEN)
+    const refreshTokenLocalStorage = peertubeLocalStorage.getItem(this.KEYS.REFRESH_TOKEN)
+    const tokenTypeLocalStorage = peertubeLocalStorage.getItem(this.KEYS.TOKEN_TYPE)
+
+    if (accessTokenLocalStorage && refreshTokenLocalStorage && tokenTypeLocalStorage) {
+      return new Tokens({
+        accessToken: accessTokenLocalStorage,
+        refreshToken: refreshTokenLocalStorage,
+        tokenType: tokenTypeLocalStorage
+      })
+    }
+
+    return null
+  }
+
+  static flush () {
+    peertubeLocalStorage.removeItem(this.KEYS.ACCESS_TOKEN)
+    peertubeLocalStorage.removeItem(this.KEYS.REFRESH_TOKEN)
+    peertubeLocalStorage.removeItem(this.KEYS.TOKEN_TYPE)
+  }
+
+  constructor (hash?: TokenOptions) {
+    if (hash) {
+      this.accessToken = hash.accessToken
+      this.refreshToken = hash.refreshToken
+
+      if (hash.tokenType === 'bearer') {
+        this.tokenType = 'Bearer'
+      } else {
+        this.tokenType = hash.tokenType
+      }
+    }
+  }
+
+  save () {
+    peertubeLocalStorage.setItem(Tokens.KEYS.ACCESS_TOKEN, this.accessToken)
+    peertubeLocalStorage.setItem(Tokens.KEYS.REFRESH_TOKEN, this.refreshToken)
+    peertubeLocalStorage.setItem(Tokens.KEYS.TOKEN_TYPE, this.tokenType)
+  }
+}