fix: Improved onAnyMessage for ciphertext messages

wppconnect-team · Jul 11, 2022 · 7eb9e59 · 7eb9e59
1 parent e2720ed
commit 7eb9e59
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/api/layers/listener.layer.ts b/src/api/layers/listener.layer.ts
@@ -107,7 +107,9 @@ export class ListenerLayer extends ProfileLayer {
                 return;
               }
               const serialized = WAPI.processMessageObj(msg, false, false);
-              window['onMessage'](serialized);
+              if (serialized) {
+                window['onMessage'](serialized);
+              }
             });
 
             window['onMessage'].exposed = true;
@@ -125,7 +127,12 @@ export class ListenerLayer extends ProfileLayer {
         }
         try {
           if (!window['onAnyMessage'].exposed) {
-            window.WAPI.allNewMessagesListener(window['onAnyMessage']);
+            WPP.on('chat.new_message', (msg) => {
+              const serialized = WAPI.processMessageObj(msg, true, false);
+              if (serialized) {
+                window['onAnyMessage'](serialized);
+              }
+            });
             window['onAnyMessage'].exposed = true;
           }
         } catch (error) {