fix: Fixed stuck state after logout

wppconnect-team · Mar 13, 2022 · bb9695a · bb9695a
1 parent 880489e
commit bb9695a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/api/layers/host.layer.ts b/src/api/layers/host.layer.ts
@@ -109,14 +109,19 @@ export class HostLayer {
       );
     }
 
-    if (isValidSessionToken(sessionToken)) {
+    const isValidToken = isValidSessionToken(sessionToken);
+    if (isValidToken) {
       this.log('verbose', 'Injecting session token', { type: 'token' });
     }
 
+    const hasUserDataDir = !!this.options?.puppeteerOptions?.userDataDir;
+
+    let clear = !hasUserDataDir || (hasUserDataDir && !isValidToken);
+
     await initWhatsapp(
       this.page,
       sessionToken,
-      !this.options?.puppeteerOptions?.userDataDir,
+      clear,
       this.options.whatsappVersion
     );
 

diff --git a/src/api/whatsapp.ts b/src/api/whatsapp.ts
@@ -86,6 +86,12 @@ export class Whatsapp extends BusinessLayer {
                 this.statusFind('desconnectedMobile', session);
               } catch (error) {}
             }
+
+            if (connected) {
+              await page.evaluate(() => localStorage.clear());
+              await page.reload();
+            }
+
             connected = false;
           }, 1000);
           break;