Remove ended crowdloan

Koniverse · Oct 25, 2023 · cf710f1 · cf710f1 · saltict · Oct 25, 2023
1 parent 73427c2
commit cf710f1
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/packages/extension-base/src/koni/background/handlers/State.ts b/packages/extension-base/src/koni/background/handlers/State.ts
@@ -290,6 +290,9 @@ export default class KoniState {
     this.onAccountAdd();
     this.onAccountRemove();
 
+    // TODO: consider moving this to a separate service
+    await this.dbService.stores.crowdloan.removeEndedCrowdloans();
+
     await this.startSubscription();
   }
 

diff --git a/packages/extension-base/src/services/storage-service/db-stores/Crowdloan.ts b/packages/extension-base/src/services/storage-service/db-stores/Crowdloan.ts
@@ -6,6 +6,23 @@ import BaseStoreWithAddressAndChain from '@subwallet/extension-base/services/sto
 import { ICrowdloanItem } from '../databases';
 
 export default class CrowdloanStore extends BaseStoreWithAddressAndChain<ICrowdloanItem> {
+  async removeEndedCrowdloans () {
+    const now = new Date();
+    const removeList: string[] = [];
+
+    await this.table.each((obj, cursor) => {
+      try {
+        if (!obj.endTime || new Date(obj.endTime) < now) {
+          removeList.push(cursor.primaryKey as string);
+        }
+      } catch (e) {
+        console.error(e);
+      }
+    });
+
+    await this.table.bulkDelete(removeList);
+  }
+
   getCrowdloan (address: string) {
     return this.table.where('address').equals(address).toArray();
   }