fix: Clear all token data when disconnected (fix #1147)

wppconnect-team · Jun 1, 2022 · 4abcfed · 4abcfed
1 parent d0c0ad0
commit 4abcfed
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/src/api/whatsapp.ts b/src/api/whatsapp.ts
@@ -89,6 +89,31 @@ export class Whatsapp extends BusinessLayer {
 
             if (connected) {
               await page.evaluate(() => localStorage.clear());
+              await page.evaluate(() => {
+                const promises = [];
+                window.indexedDB
+                  .databases()
+                  .then((dbs) => {
+                    dbs.forEach((db) => {
+                      promises.push(
+                        new Promise<void>((resolve) => {
+                          const r = window.indexedDB.deleteDatabase(db.name);
+                          r.onerror = r.onblocked = function () {
+                            this.result.close();
+                            window.indexedDB.deleteDatabase(db.name);
+                            resolve();
+                          };
+                          r.onsuccess = function () {
+                            resolve();
+                          };
+                        })
+                      );
+                    });
+                  })
+                  .catch(() => null);
+
+                return Promise.all(promises);
+              });
               await page.reload();
             }
 

diff --git a/src/controllers/auth.ts b/src/controllers/auth.ts
@@ -180,6 +180,15 @@ export async function injectSessionToken(
   if (clear) {
     await page.evaluate((session) => {
       localStorage.clear();
+
+      window.indexedDB
+        .databases()
+        .then((dbs) => {
+          dbs.forEach((db) => {
+            window.indexedDB.deleteDatabase(db.name);
+          });
+        })
+        .catch(() => null);
     });
   }
   await page.evaluate((session) => {