fix: Fixed live location event register and added a option

wppconnect-team · Jan 21, 2022 · 08949a9 · 08949a9
1 parent 8e26a00
commit 08949a9
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/src/chat/events/registerLiveLocationUpdateEvent.ts b/src/chat/events/registerLiveLocationUpdateEvent.ts
@@ -16,6 +16,7 @@
 
 import Debug from 'debug';
 
+import { config } from '../../config';
 import * as webpack from '../../webpack';
 import {
   ChatStore,
@@ -108,7 +109,8 @@ function registerLiveLocationUpdateEvent() {
    * Start for all active chats
    */
   ChatStore.once('collection_has_synced', () => {
-    ChatStore.forEach((chat) => {
+    const chats = ChatStore.models.slice(0, config.liveLocationLimit);
+    chats.forEach((chat) => {
       LiveLocationStore.update(chat.id)
         .then((liveLocation) => {
           liveLocation.startViewingMap();

diff --git a/src/config.ts b/src/config.ts
@@ -21,7 +21,8 @@
  * ```javascript
  * // Global variable before injection
  * WPPConfig = {
- *   deviceName: 'WPPConnect'
+ *   deviceName: 'WPPConnect',
+ *   liveLocationLimit: 10
  * };
  * ```
  */
@@ -31,10 +32,16 @@ export interface Config {
    * @default 'WPPConnect'
    */
   deviceName: string | false;
+
+  /**
+   * Number of last chats to check live location after a page reload
+   */
+  liveLocationLimit: number;
 }
 
 export const defaultConfig: Config = {
   deviceName: false,
+  liveLocationLimit: 10,
 };
 
 export const config: Config = defaultConfig;

diff --git a/src/status/events/registerSyncedEvent.ts b/src/status/events/registerSyncedEvent.ts
@@ -20,7 +20,7 @@ import * as webpack from '../../webpack';
 import { StatusV3Store } from '../../whatsapp';
 import { eventEmitter } from '../eventEmitter';
 
-const debug = Debug('WA-JS:blocklist');
+const debug = Debug('WA-JS:status');
 
 webpack.onInjected(() => registerSyncedEvent());