From 10aa05c244642e808a17c29d5d4e3010ff28142d Mon Sep 17 00:00:00 2001
From: Jixun Wu <jixun.moe@gmail.com>
Date: Mon, 20 Dec 2021 22:19:44 +0000
Subject: [PATCH] refactor: storage factory + singleton

- Make storage easier.

(cherry picked from commit ed84a4732d7dd3ce6b2c22f30553ab5c59f85dbb)
---
 src/component/ConfigDialog.vue              |  2 +-
 src/decrypt/joox.ts                         |  2 +-
 src/utils/storage.ts                        |  8 ++----
 src/utils/storage/BaseStorage.ts            |  2 ++
 src/utils/storage/BrowserNativeStorage.ts   | 18 ++++++++++++
 src/utils/storage/ChromeExtensionStorage.ts | 12 ++++++++
 src/utils/storage/InMemoryStorage.ts        | 32 +++++++++++++++++++++
 src/utils/storage/StorageFactory.ts         | 13 +++++++++
 8 files changed, 81 insertions(+), 8 deletions(-)
 create mode 100644 src/utils/storage/InMemoryStorage.ts
 create mode 100644 src/utils/storage/StorageFactory.ts

diff --git a/src/component/ConfigDialog.vue b/src/component/ConfigDialog.vue
index e0a05ce..fe280c7 100644
--- a/src/component/ConfigDialog.vue
+++ b/src/component/ConfigDialog.vue
@@ -13,7 +13,7 @@
 </template>
 
 <script>
-import storage from '../utils/storage';
+import { storage } from '@/utils/storage';
 
 export default {
   components: {},
diff --git a/src/decrypt/joox.ts b/src/decrypt/joox.ts
index e7bbb52..60e4da9 100644
--- a/src/decrypt/joox.ts
+++ b/src/decrypt/joox.ts
@@ -2,8 +2,8 @@ import { DecryptResult } from './entity';
 import { AudioMimeType, GetArrayBuffer, SniffAudioExt } from './utils';
 
 import jooxFactory from '@unlock-music/joox-crypto';
-import storage from '@/utils/storage';
 import { MergeUint8Array } from '@/utils/MergeUint8Array';
+import { storage } from '@/utils/storage';
 
 export async function Decrypt(file: Blob, raw_filename: string, raw_ext: string): Promise<DecryptResult> {
   const uuid = await storage.loadJooxUUID('');
diff --git a/src/utils/storage.ts b/src/utils/storage.ts
index 74bfe1d..4d9cac0 100644
--- a/src/utils/storage.ts
+++ b/src/utils/storage.ts
@@ -1,7 +1,3 @@
-import BaseStorage from './storage/BaseStorage';
-import BrowserNativeStorage from './storage/BrowserNativeStorage';
-import ChromeExtensionStorage from './storage/ChromeExtensionStorage';
+import storageFactory from './storage/StorageFactory';
 
-const storage: BaseStorage = ChromeExtensionStorage.works ? new ChromeExtensionStorage() : new BrowserNativeStorage();
-
-export default storage;
+export const storage = storageFactory();
diff --git a/src/utils/storage/BaseStorage.ts b/src/utils/storage/BaseStorage.ts
index 8a5dafc..2e163ca 100644
--- a/src/utils/storage/BaseStorage.ts
+++ b/src/utils/storage/BaseStorage.ts
@@ -3,6 +3,8 @@ const KEY_JOOX_UUID = 'joox.uuid';
 export default abstract class BaseStorage {
   protected abstract save<T>(name: string, value: T): Promise<void>;
   protected abstract load<T>(name: string, defaultValue: T): Promise<T>;
+  public abstract getAll(): Promise<Record<string, any>>;
+  public abstract setAll(obj: Record<string, any>): Promise<void>;
 
   public saveJooxUUID(uuid: string): Promise<void> {
     return this.save(KEY_JOOX_UUID, uuid);
diff --git a/src/utils/storage/BrowserNativeStorage.ts b/src/utils/storage/BrowserNativeStorage.ts
index b17c048..2deae0c 100644
--- a/src/utils/storage/BrowserNativeStorage.ts
+++ b/src/utils/storage/BrowserNativeStorage.ts
@@ -1,6 +1,10 @@
 import BaseStorage from './BaseStorage';
 
 export default class BrowserNativeStorage extends BaseStorage {
+  public static get works() {
+    return typeof localStorage !== 'undefined' && localStorage.getItem;
+  }
+
   protected async load<T>(name: string, defaultValue: T): Promise<T> {
     const result = localStorage.getItem(name);
     if (result === null) {
@@ -12,4 +16,18 @@ export default class BrowserNativeStorage extends BaseStorage {
   protected async save<T>(name: string, value: T): Promise<void> {
     localStorage.setItem(name, JSON.stringify(value));
   }
+
+  public async getAll(): Promise<Record<string, any>> {
+    const result = {};
+    for (const [key, value] of Object.entries(localStorage)) {
+      Object.assign(result, { [key]: JSON.parse(value) });
+    }
+    return result;
+  }
+
+  public async setAll(obj: Record<string, any>): Promise<void> {
+    for (const [key, value] of Object.entries(obj)) {
+      localStorage.setItem(key, JSON.stringify(value));
+    }
+  }
 }
diff --git a/src/utils/storage/ChromeExtensionStorage.ts b/src/utils/storage/ChromeExtensionStorage.ts
index 8fa6a5e..b2b48be 100644
--- a/src/utils/storage/ChromeExtensionStorage.ts
+++ b/src/utils/storage/ChromeExtensionStorage.ts
@@ -24,4 +24,16 @@ export default class ChromeExtensionStorage extends BaseStorage {
       chrome.storage.local.set({ [name]: value }, resolve);
     });
   }
+
+  public async getAll(): Promise<Record<string, any>> {
+    return new Promise((resolve) => {
+      chrome.storage.local.get(null, resolve);
+    });
+  }
+
+  public async setAll(obj: Record<string, any>): Promise<void> {
+    return new Promise((resolve) => {
+      chrome.storage.local.set(obj, resolve);
+    });
+  }
 }
diff --git a/src/utils/storage/InMemoryStorage.ts b/src/utils/storage/InMemoryStorage.ts
new file mode 100644
index 0000000..b44f26f
--- /dev/null
+++ b/src/utils/storage/InMemoryStorage.ts
@@ -0,0 +1,32 @@
+import BaseStorage from './BaseStorage';
+
+export default class InMemoryStorage extends BaseStorage {
+  private values = new Map<string, any>();
+  protected async load<T>(name: string, defaultValue: T): Promise<T> {
+    if (this.values.has(name)) {
+      return this.values.get(name);
+    }
+
+    return defaultValue;
+  }
+
+  protected async save<T>(name: string, value: T): Promise<void> {
+    this.values.set(name, value);
+  }
+
+  public async getAll(): Promise<Record<string, any>> {
+    const result = {};
+    this.values.forEach((value, key) => {
+      Object.assign(result, {
+        [key]: value,
+      });
+    });
+    return result;
+  }
+
+  public async setAll(obj: Record<string, any>): Promise<void> {
+    for (const [key, value] of Object.entries(obj)) {
+      this.values.set(key, value);
+    }
+  }
+}
diff --git a/src/utils/storage/StorageFactory.ts b/src/utils/storage/StorageFactory.ts
new file mode 100644
index 0000000..9fa9b30
--- /dev/null
+++ b/src/utils/storage/StorageFactory.ts
@@ -0,0 +1,13 @@
+import BaseStorage from './BaseStorage';
+import BrowserNativeStorage from './BrowserNativeStorage';
+import ChromeExtensionStorage from './ChromeExtensionStorage';
+import InMemoryStorage from './InMemoryStorage';
+
+export default function storageFactory(): BaseStorage {
+  if (ChromeExtensionStorage.works) {
+    return new ChromeExtensionStorage();
+  } else if (BrowserNativeStorage.works) {
+    return new BrowserNativeStorage();
+  }
+  return new InMemoryStorage();
+}