Fix: Can't detect max memory of browser via firefox

Author: sheepzh

src/database/common/storage-promise.ts

@@ -8,15 +8,24 @@ export default class StoragePromise {
         this.storage = storage
     }
 
-    public get(keys: string | string[] | Object | null): Promise<{ [key: string]: any }> {
+    get(keys?: string | string[] | Object | null): Promise<{ [key: string]: any }> {
         return new Promise(resolve => this.storage.get(keys, resolve))
     }
 
-    public set(obj: Object): Promise<void> {
+    set(obj: Object): Promise<void> {
         return new Promise<void>(resolve => this.storage.set(obj, resolve))
     }
 
-    public remove(key: string | string[]): Promise<void> {
+    remove(key: string | string[]): Promise<void> {
         return new Promise(resolve => this.storage.remove(key, resolve))
     }
+
+    async getUsedMemory(): Promise<number> {
+        if (this.storage.getBytesInUse) {
+            return new Promise(resolve => this.storage.getBytesInUse(resolve))
+        } else {
+            const store = await this.get()
+            return JSON.stringify(store).length
+        }
+    }
 }

src/database/memory-detector.ts

@@ -1,3 +1,5 @@
+import StoragePromise from "./common/storage-promise";
+
 /**
  * User memory of this extension
  */
@@ -12,15 +14,17 @@ export declare type MemoryInfo = {
     total: number
 }
 
-const total: number = chrome.storage.local.QUOTA_BYTES
+/**
+ * 'QUOTA_BYTES' Not supported in Firefox
+ */
+const total: number = chrome.storage.local.QUOTA_BYTES || 0
 
 /**
  * Get the used memory by bytes
  * 
  * @since 0.0.9
  */
-export function getUsedStorage(): Promise<MemoryInfo> {
-    return new Promise(
-        resolve => chrome.storage.local.getBytesInUse(used => resolve({ used, total }))
-    )
+export async function getUsedStorage(): Promise<MemoryInfo> {
+    const used = await new StoragePromise(chrome.storage.local).getUsedMemory()
+    return { used, total }
 }