feat: getAllChatsNewMsg

orkestral · Feb 27, 2021 · b703462 · b703462
1 parent 04f2200
commit b703462
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -512,6 +512,9 @@ const messages = await client.getAllUnreadMessages();
 // Retrieve all chats
 const chats = await client.getAllChats();
 
+//Retrieves all chats new messages
+const chatsAllNew = getAllChatsNewMsg();
+
 // Retrieve all groups
 const chats = await client.getAllChatsGroups();
 

diff --git a/src/api/layers/retriever.layer.ts b/src/api/layers/retriever.layer.ts
@@ -110,12 +110,22 @@ export class RetrieverLayer extends SenderLayer {
    * Retrieves all chats
    * @returns array of [Chat]
    */
-  public async getAllChats(withNewMessageOnly = false) {
-    if (withNewMessageOnly) {
-      return this.page.evaluate(() => WAPI.getAllChatsWithNewMsg());
-    } else {
-      return this.page.evaluate(() => WAPI.getAllChats());
-    }
+  public async getAllChats() {
+    return await this.page.evaluate(() => {
+      let chats = WAPI.getAllChats();
+      return chats;
+    });
+  }
+
+  /**
+   * Retrieves all chats new messages
+   * @returns array of [Chat]
+   */
+  public async getAllChatsNewMsg() {
+    return await this.page.evaluate(() => {
+      let chats = WAPI.getAllChatsWithNewMsg();
+      return chats;
+    });
   }
 
   /**