From 8fc6e6098c88e7280074e4a9bdaabedc0646589e Mon Sep 17 00:00:00 2001
From: tofers <acerkot@mail.ru>
Date: Sun, 25 Jun 2023 22:25:49 +0300
Subject: [PATCH] feat: add or remove labels (#1915)

* add or remove labels

* Strings use singlequote

* add event chat change labels

* Revert "add event chat change labels"
---
 example.js             | 12 ++++++++++++
 index.d.ts             |  5 +++++
 src/Client.js          | 29 +++++++++++++++++++++++++++++
 src/structures/Chat.js |  9 +++++++++
 4 files changed, 55 insertions(+)

diff --git a/example.js b/example.js
index 529a8c4..4885398 100644
--- a/example.js
+++ b/example.js
@@ -210,6 +210,18 @@ client.on('message', async msg => {
                 msg.reply('I can only edit my own messages');
             }
         }
+    } else if (msg.body === '!updatelabels') {
+        const chat = await msg.getChat();
+        await chat.changeLabels([0, 1]);
+    } else if (msg.body === '!addlabels') {
+        const chat = await msg.getChat();
+        let labels = (await chat.getLabels()).map(l => l.id);
+        labels.push('0');
+        labels.push('1');
+        await chat.changeLabels(labels);
+    } else if (msg.body === '!removelabels') {
+        const chat = await msg.getChat();
+        await chat.changeLabels([]);
     }
 });
 
diff --git a/index.d.ts b/index.d.ts
index a9864d4..7e2b86c 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -71,6 +71,9 @@ declare namespace WAWebJS {
 
         /** Get all current Labels  */
         getLabels(): Promise<Label[]>
+        
+        /** Change labels in chats  */
+        addOrRemoveLabels(labelIds: Array<number|string>, chatIds: Array<string>): Promise<void>
 
         /** Get Label instance by ID */
         getLabelById(labelId: string): Promise<Label>
@@ -1162,6 +1165,8 @@ declare namespace WAWebJS {
         markUnread: () => Promise<void>
         /** Returns array of all Labels assigned to this Chat */
         getLabels: () => Promise<Label[]>
+        /** Add or remove labels to this Chat */
+        changeLabels: (labelIds: Array<string | number>) => Promise<void>
     }
 
     export interface MessageSearchOptions {
diff --git a/src/Client.js b/src/Client.js
index 0a5b5d1..e5d45c3 100644
--- a/src/Client.js
+++ b/src/Client.js
@@ -1332,6 +1332,35 @@ class Client extends EventEmitter {
 
         return success;
     }
+    
+    /**
+     * Change labels in chats
+     * @param {Array<number|string>} labelIds
+     * @param {Array<string>} chatIds
+     * @returns {Promise<void>}
+     */
+    async addOrRemoveLabels(labelIds, chatIds) {
+
+        return this.pupPage.evaluate(async (labelIds, chatIds) => {
+            if (['smba', 'smbi'].indexOf(window.Store.Conn.platform) === -1) {
+                throw '[LT01] Only Whatsapp business';
+            }
+            const labels = window.WWebJS.getLabels().filter(e => labelIds.find(l => l == e.id) !== undefined);
+            const chats = window.Store.Chat.filter(e => chatIds.includes(e.id._serialized));
+
+            let actions = labels.map(label => ({id: label.id, type: 'add'}));
+
+            chats.forEach(chat => {
+                (chat.labels || []).forEach(n => {
+                    if (!actions.find(e => e.id == n)) {
+                        actions.push({id: n, type: 'remove'});
+                    }
+                });
+            });
+
+            return await window.Store.Label.addOrRemoveLabels(actions, chats);
+        }, labelIds, chatIds);
+    }
 }
 
 module.exports = Client;
diff --git a/src/structures/Chat.js b/src/structures/Chat.js
index 0e0ae78..f60aea1 100644
--- a/src/structures/Chat.js
+++ b/src/structures/Chat.js
@@ -261,6 +261,15 @@ class Chat extends Base {
     async getLabels() {
         return this.client.getChatLabels(this.id._serialized);
     }
+
+    /**
+     * Add or remove labels to this Chat
+     * @param {Array<number|string>} labelIds
+     * @returns {Promise<void>}
+     */
+    async changeLabels(labelIds) {
+        return this.client.addOrRemoveLabels(labelIds, [this.id._serialized]);
+    }
 }
 
 module.exports = Chat;