Remove archivement function (sheepzh#141)

Author: sheepzh

src/background/version-manager/1-3-0/archived-data-cleaner.ts

@@ -0,0 +1,28 @@
+/**
+ * Copyright (c) 2022-present Hengyang Zhang
+ *
+ * This software is released under the MIT License.
+ * https://opensource.org/licenses/MIT
+ */
+
+import ArchivedDatabase from "@db/archived-database"
+import { log } from "@src/common/logger"
+import IVersionProcessor from "../i-version-processor"
+
+/**
+ * All code will be removed at v1.4.0
+ */
+export default class ArchivedDataCleaner implements IVersionProcessor {
+    since(): string {
+        return '1.3.0'
+    }
+
+    async process(reason: chrome.runtime.OnInstalledReason): Promise<void> {
+        const db = new ArchivedDatabase(chrome.storage.local)
+        if (reason !== 'update') {
+            return
+        }
+        const count = await db.removeAll()
+        log(`Removed ${count} archived items`)
+    }
+}

src/background/version-manager/i-version-processor.ts

@@ -21,5 +21,5 @@ export default interface IVersionProcessor {
      * 
      * @param reason reason of chrome OnInstalled event
      */
-    process(reason: string): void
+    process(reason: chrome.runtime.OnInstalledReason): void
 }

src/background/version-manager/index.ts

@@ -8,6 +8,7 @@
 import IVersionProcessor from "./i-version-processor"
 import HostMergeInitializer from "./0-1-2/host-merge-initializer"
 import LocalFileInitializer from "./0-7-0/local-file-initializer"
+import ArchivedDataCleaner from "./1-3-0/archived-data-cleaner"
 
 /**
  * Version manager
@@ -18,7 +19,11 @@ class VersionManager {
     processorChain: IVersionProcessor[] = []
 
     constructor() {
-        this.processorChain.push(new HostMergeInitializer(), new LocalFileInitializer())
+        this.processorChain.push(
+            new HostMergeInitializer(),
+            new LocalFileInitializer(),
+            new ArchivedDataCleaner(),
+        )
         this.processorChain = this.processorChain.sort((a, b) => a.since() >= b.since() ? 1 : 0)
     }
 

src/database/archived-database.ts

@@ -28,6 +28,14 @@ class ArchivedDatabase extends BaseDatabase {
         return Promise.resolve(result)
     }
 
+    async removeAll(): Promise<number> {
+        const items = await this.storage.get()
+        const keys = Object.keys(items)
+            .filter(key => key.startsWith(ARCHIVED_PREFIX))
+        await this.storage.remove(keys)
+        return keys?.length || 0
+    }
+
     private generateKey(row: timer.stat.Row): string {
         return ARCHIVED_PREFIX + row.host
     }