feat(UserStore): implement store()

cAttte · cAttte · commit 873038eecd54 · 2020-10-22T23:47:20.000-03:00
diff --git a/src/struct/stores/UserStore.ts b/src/struct/stores/UserStore.ts
@@ -24,6 +24,26 @@ export default class UserStore extends Map<string, User> {
         for (const [k, v] of entries) this.set(k, v)
     }
 
+    /**
+     * Store and get a user by their username.
+     * @param data The data obtained from the API
+     * @param username The user's username.
+     */
+    store(
+        data: any,
+        username: string = data.username,
+        Type: typeof User | typeof ClientUser = User
+    ): User {
+        let existing = this.get(username)
+        if (existing) {
+            existing._apply(data)
+        } else {
+            existing = new Type(this.client, data)
+            this.set(username, existing)
+        }
+        return existing
+    }
+
     /**
      * Fetch a user by their username, and store them in the cache.
      * @param username The user's username
@@ -57,8 +77,7 @@ export default class UserStore extends Map<string, User> {
             UserConstructor = ClientUser
         }
 
-        const user = new UserConstructor(this.client, data)
-        this.set(username, user)
+        const user = this.store(data, data.username, UserConstructor)
         return user
     }
 }
