feat: Added logout method

orkestral · Dec 1, 2020 · 6950e40 · 6950e40
1 parent b439abf
commit 6950e40
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
diff --git a/src/api/whatsapp.ts b/src/api/whatsapp.ts
@@ -93,6 +93,10 @@ export class Whatsapp extends ControlsLayer {
     return await this.page.evaluate(() => WAPI.takeOver());
   }
 
+  public async logout() {
+    return await this.page.evaluate(() => WAPI.logout());
+  }
+
   /**
    * Closes page and browser
    * @internal

diff --git a/src/lib/wapi/store/store-objects.js b/src/lib/wapi/store/store-objects.js
@@ -404,4 +404,9 @@ export const storeObjects = [
     conditions: (module) =>
       module.default && module.default.unobscure ? module.default : null,
   },
+  {
+    id: 'ws2',
+    conditions: (module) =>
+      module.default && module.default.destroyStorage ? module.default : null,
+  },
 ];
diff --git a/src/lib/wapi/wapi.js b/src/lib/wapi/wapi.js
@@ -532,6 +532,13 @@ if (typeof window.WAPI === 'undefined') {
     return true;
   };
 
+  window.WAPI.logout = async function () {
+    return await window.WAPI.waitForStore(['ws2'], () => {
+      window.Store.ws2.logout();
+      return true;
+    });
+  };
+
   window.WAPI.waitForStore = async function (stores, callback) {
     if (!Array.isArray(stores)) {
       stores = [stores];

diff --git a/src/types/WAPI.d.ts b/src/types/WAPI.d.ts
@@ -87,6 +87,7 @@ interface WAPI {
     includeNotifications: boolean
   ) => Message[];
   loadEarlierMessages: (contactId: string) => Message[];
+  logout: () => Promise<boolean>;
   markUnseenMessage: (messageId: string) => boolean;
   onAddedToGroup: (callback: Function) => any;
   onIncomingCall: (callback: Function) => any;