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