fix: improve storage API

datagram1 · Dec 28, 2020 · f34d64c · f34d64c
1 parent f7012c5
commit f34d64c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/app/assets/javascripts/services/localStorage.ts b/app/assets/javascripts/services/localStorage.ts
@@ -2,12 +2,16 @@ export enum StorageKey {
   DisableErrorReporting = 'DisableErrorReporting',
 }
 
+export type StorageValue = {
+  [StorageKey.DisableErrorReporting]: boolean;
+}
+
 export const storage = {
-  get(key: StorageKey) {
+  get<K extends StorageKey>(key: K): StorageValue[K] | null {
     const value = localStorage.getItem(key);
     return value ? JSON.parse(value) : null;
   },
-  set(key: StorageKey, value: unknown) {
+  set<K extends StorageKey>(key: K, value: StorageValue[K]) {
     localStorage.setItem(key, JSON.stringify(value));
   },
   remove(key: StorageKey) {