Feats and fixed functions

src/chat/events/eventTypes.ts

@@ -14,6 +14,7 @@
  * limitations under the License.
  */
 
+import { Label } from '../../labels';
 import { ChatModel, MsgKey, MsgModel, Wid } from '../../whatsapp';
 
 export interface ChatEventTypes {
@@ -167,4 +168,14 @@ export interface ChatEventTypes {
     timestamp: number;
     sender: Wid;
   };
+
+  /**
+   * On Labels update
+   */
+  'chat.update_label': {
+    chat: ChatModel;
+    ids: string[];
+    labels: Label[];
+    type: 'add' | 'remove';
+  };
 }

src/chat/events/index.ts

@@ -22,3 +22,4 @@ import './registerPollEvent';
 import './registerPresenceChange';
 import './registerReactionsEvent';
 import './registerRevokeMessageEvent';
+import './registerLabelEvent';

src/chat/events/registerLabelEvent.ts

@@ -0,0 +1,64 @@
+/*!
+ * Copyright 2022 WPPConnect Team
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { isMainReady } from '../../conn';
+import { internalEv } from '../../eventEmitter';
+import { getLabelById, Label } from '../../labels';
+import * as webpack from '../../webpack';
+import { ChatModel } from '../../whatsapp';
+import { wrapModuleFunction } from '../../whatsapp/exportModule';
+import {
+  addToLabelCollection,
+  removeLabelFromCollection,
+} from '../../whatsapp/functions';
+import { get as getChat } from '../functions/';
+
+webpack.onInjected(() => register());
+
+function register() {
+  async function processLabelEvent(event: 'add' | 'remove', ...args: any) {
+    const data = args[0];
+    const ids = Array.isArray(data[1]) ? data[1] : [data[1]];
+    const chatId = data[0];
+
+    if (isMainReady()) {
+      const labels = [] as Label[];
+      for (const id of ids) {
+        labels.push(await getLabelById(id));
+      }
+      internalEv.emit('chat.update_label', {
+        chat: getChat(chatId) as ChatModel,
+        ids: ids,
+        labels: labels,
+        type: event,
+      });
+    }
+  }
+
+  wrapModuleFunction(addToLabelCollection, async (func, ...args) => {
+    queueMicrotask(() => {
+      processLabelEvent('add', args);
+    });
+    return func(...args);
+  });
+
+  wrapModuleFunction(removeLabelFromCollection, async (func, ...args) => {
+    queueMicrotask(() => {
+      processLabelEvent('remove', args);
+    });
+    return func(...args);
+  });
+}

src/chat/functions/getAllChats.ts

@@ -0,0 +1,69 @@
+/*!
+ * Copyright 2023 WPPConnect Team
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { ChatModel, ChatStore, Wid } from '../../whatsapp';
+import { get } from './get';
+
+export interface GetAllChatsOptions {
+  count?: number;
+  direction?: 'after' | 'before';
+  id?: Wid;
+  onlyUnread?: boolean;
+}
+/**
+ * Get all chats
+ *  * @example
+ * ```javascript
+ * // Some chats
+ * WPP.chat.getAllChats({
+ *   count: 20,
+ * });
+ *
+ * // All chats
+ * WPP.chat.getAllChats({
+ *   count: -1,
+ * });
+ *
+ * // 20 chats before specific chat
+ * WPP.chat.getMessages({
+ *   count: 20,
+ *   direction: 'before',
+ *   id: '[number]@c.us'
+ * });
+ * ```
+ * @category Chat
+ */
+export function getAllChats(options: GetAllChatsOptions = {}): ChatModel[] {
+  const count = options.count == -1 ? Infinity : options.count || 20;
+  const direction = options.direction === 'before' ? 'before' : 'after';
+  const indexChat = options?.id
+    ? get(options.id)
+    : ChatStore.getModelsArray()[0];
+  const onlyUnread = options.onlyUnread ? options.onlyUnread : false;
+  const allChats = onlyUnread
+    ? ChatStore.filter((chat: ChatModel) => chat.unreadCount)
+    : ChatStore.getModelsArray();
+
+  const startIndex = allChats.indexOf(indexChat as any);
+  if (direction === 'before') {
+    const fixStartIndex = startIndex - count < 0 ? 0 : startIndex - count;
+    const fixEndIndex =
+      fixStartIndex + count >= startIndex ? startIndex : fixStartIndex + count;
+    return allChats.slice(fixStartIndex, fixEndIndex);
+  } else {
+    return allChats.slice(startIndex, startIndex + count);
+  }
+}

src/chat/functions/index.ts

@@ -28,6 +28,7 @@ export { forwardMessage, ForwardMessagesOptions } from './forwardMessage';
 export { generateMessageID } from './generateMessageID';
 export { get } from './get';
 export { getActiveChat } from './getActiveChat';
+export { getAllChats } from './getAllChats';
 export { getLastSeen } from './getLastSeen';
 export { getMessageACK } from './getMessageACK';
 export { getMessageById } from './getMessageById';

src/group/functions/getAllGroups.ts

@@ -22,7 +22,7 @@ import { queryAllGroups } from '../../whatsapp/functions';
  *
  * @example
  * ```javascript
- * WPP.group.queryAllGroups();
+ * WPP.group.getAllGroups();
  * ```
  *
  * @category Group

src/whatsapp/functions/addToLabelCollection.ts

@@ -0,0 +1,45 @@
+/*!
+ * Copyright 2023 WPPConnect Team
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { exportModule } from '../exportModule';
+
+/** @whatsapp 388536
+ */
+export declare function addToLabelCollection(
+  e: any,
+  t: any,
+  i: any
+): Promise<any>;
+export declare function createLabelItemId(e: any, t: any, r: any): Promise<any>;
+export declare function getParentCollection(e: any): Promise<any>;
+export declare function initializeLabels(e: any): Promise<any>;
+export declare function removeLabelFromCollection(
+  e: any,
+  t: any,
+  i: any
+): Promise<any>;
+
+exportModule(
+  exports,
+  {
+    addToLabelCollection: 'addToLabelCollection',
+    createLabelItemId: 'createLabelItemId',
+    getParentCollection: 'getParentCollection',
+    initializeLabels: 'initializeLabels',
+    removeLabelFromCollection: 'removeLabelFromCollection',
+  },
+  (m) => m.addToLabelCollection
+);

src/whatsapp/functions/index.ts

@@ -15,6 +15,7 @@
  */
 
 export * from './addAndSendMsgToChat';
+export * from './addToLabelCollection';
 export * from './blockContact';
 export * from './calculateFilehashFromBlob';
 export * from './canEditMsg';