Refactor loading and processing of actions

xmtp · Sep 6, 2024 · 29cb855 · 29cb855
1 parent d3b13b7
commit 29cb855
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 13 deletions.
diff --git a/packages/js-sdk/src/Contacts.ts b/packages/js-sdk/src/Contacts.ts
@@ -167,17 +167,10 @@ export class ConsentList {
   /*
    * Process actions and update internal consent list
    */
-  processActions(actionsMap?: ActionsMap) {
+  processActions(actionsMap: ActionsMap) {
     const entries: ConsentListEntry[] = []
     // actions to process
-    const actions = actionsMap
-      ? Array.from(actionsMap.values())
-      : this.client.keystore.getPrivatePreferences()
-
-    // processing all actions, reset consent list
-    if (!actionsMap) {
-      this.reset()
-    }
+    const actions = Array.from(actionsMap.values())
 
     // update the consent list
     actions.forEach((action) => {
@@ -259,7 +252,13 @@ export class ConsentList {
       .filter(([timestampNs]) => Boolean(timestampNs)) as [string, Uint8Array][]
 
     // decode messages and save them to keystore
-    const actionsMap = await this.decodeMessages(new Map(messageEntries))
+    await this.decodeMessages(new Map(messageEntries))
+
+    // get all actions from keystore
+    const actionsMap = this.client.keystore.getPrivatePreferences()
+
+    // reset consent list
+    this.reset()
 
     // process actions and update consent list
     return this.processActions(actionsMap)

diff --git a/packages/js-sdk/src/keystore/privatePreferencesStore.ts b/packages/js-sdk/src/keystore/privatePreferencesStore.ts
@@ -112,15 +112,15 @@ export class PrivatePreferencesStore {
     })
   }
 
-  get actions(): privatePreferences.PrivatePreferencesAction[] {
+  get actions(): ActionsMap {
     // sort actions by their keys (timestamps) in ascending order
     const sortedActions = new Map(
       [...this.actionsMap.entries()].sort(
         (a, b) =>
           fromNanoString(a[0])!.getTime() - fromNanoString(b[0])!.getTime()
       )
     )
-    return Array.from(sortedActions.values())
+    return sortedActions
   }
 
   lookup(key: string): privatePreferences.PrivatePreferencesAction | undefined {

diff --git a/packages/js-sdk/src/keystore/rpcDefinitions.ts b/packages/js-sdk/src/keystore/rpcDefinitions.ts
@@ -44,7 +44,7 @@ type PrivatePreferenceKeystoreMethods = {
   createPrivatePreference: (
     action: privatePreferences.PrivatePreferencesAction
   ) => Promise<PublishParams[]>
-  getPrivatePreferences: () => privatePreferences.PrivatePreferencesAction[]
+  getPrivatePreferences: () => ActionsMap
   getPrivatePreferencesTopic: () => Promise<string>
   savePrivatePreferences: (data: ActionsMap) => Promise<void>
 }