client/src/root-helpers/peertube-web-storage.ts

   1 // Thanks: https://github.com/capaj/localstorage-polyfill
   2
   3 const valuesMap = new Map()
   4
   5 function proxify (instance: MemoryStorage) {
   6   return new Proxy(instance, {
   7     set: function (obj, prop: string | symbol, value) {
   8       if (Object.prototype.hasOwnProperty.call(MemoryStorage, prop)) {
   9         instance[prop] = value
  10       } else {
  11         instance.setItem(prop, value)
  12       }
  13
  14       return true
  15     },
  16     get: function (target, name: string | symbol | number) {
  17       if (typeof instance[name] === 'function') {
  18         return instance[name]
  19       } else if (valuesMap.has(name)) {
  20         return instance.getItem(name)
  21       }
  22     }
  23   })
  24 }
  25
  26 class MemoryStorage implements Storage {
  27   [key: string | symbol]: any
  28
  29   getItem (key: any) {
  30     const stringKey = String(key)
  31     if (valuesMap.has(key)) {
  32       return String(valuesMap.get(stringKey))
  33     }
  34
  35     return null
  36   }
  37
  38   setItem (key: any, val: any) {
  39     valuesMap.set(String(key), String(val))
  40   }
  41
  42   removeItem (key: any) {
  43     valuesMap.delete(key)
  44   }
  45
  46   clear () {
  47     valuesMap.clear()
  48   }
  49
  50   key (i: any) {
  51     if (arguments.length === 0) {
  52       throw new TypeError('Failed to execute "key" on "Storage": 1 argument required, but only 0 present.')
  53     }
  54
  55     const arr = Array.from(valuesMap.keys())
  56     return arr[i]
  57   }
  58
  59   get length () {
  60     return valuesMap.size
  61   }
  62 }
  63
  64 let peertubeLocalStorage: Storage
  65 let peertubeSessionStorage: Storage
  66
  67 function reinitStorage () {
  68   const instanceLocalStorage = new MemoryStorage()
  69   const instanceSessionStorage = new MemoryStorage()
  70
  71   peertubeLocalStorage = proxify(instanceLocalStorage)
  72   peertubeSessionStorage = proxify(instanceSessionStorage)
  73 }
  74
  75 try {
  76   peertubeLocalStorage = localStorage
  77   peertubeSessionStorage = sessionStorage
  78 } catch (err) {
  79   // support Firefox and other browsers using an exception rather than null
  80   reinitStorage()
  81 }
  82
  83 // support Brave and other browsers using null rather than an exception
  84 if (!peertubeLocalStorage || !peertubeSessionStorage) {
  85   reinitStorage()
  86 }
  87
  88 export {
  89   peertubeLocalStorage,
  90   peertubeSessionStorage
  91 }