[ADD] Disconnected event

example.js

@@ -27,5 +27,9 @@ client.on('message', (msg) => {
         // Send a new message to the same chat
         client.sendMessage(msg.from, 'pong');
     }
+});
+
+client.on('disconnected', () => {
+    console.log('Client was logged out');
 })
 

src/client/Client.js

@@ -48,14 +48,21 @@ class Client extends EventEmitter {
             await page.evaluate(LoadExtraProps, model.WAppModel, model.extraFields);
         }
 
-        await page.exposeFunction('onAddMessageEvent', (msg) => {
+        await page.exposeFunction('onAddMessageEvent', msg => {
             if (msg.id.fromMe) return;
-
+            this.emit(Events.MESSAGE_CREATE, new Message(this, msg));
-            this.emit('message', new Message(this, msg));
         });
 
+        await page.exposeFunction('onConnectionChangedEvent', (conn, connected) => {
+            if (!connected) {
+                this.emit(Events.DISCONNECTED);
+                this.destroy();
+            }
+        })
+
         await page.evaluate(() => {
             Store.Msg.on('add', onAddMessageEvent);
+            Store.Conn.on('change:connected', onConnectionChangedEvent);
         })
 
         this.pupBrowser = browser;
@@ -78,7 +85,7 @@ class Client extends EventEmitter {
         let chat = await this.pupPage.evaluate(chatId => {
             return Store.Chat.get(chatId).serialize();
         }, chatId);
-        
+
         return new Chat(this, chat);
     }
 

src/util/Constants.js

@@ -20,7 +20,8 @@ exports.Events = {
     AUTHENTICATED: 'authenticated',
     READY: 'ready',
     MESSAGE_CREATE: 'message',
-    QR_RECEIVED: 'qr'
+    QR_RECEIVED: 'qr',
+    DISCONNECTED: 'disconnected'
 }
 
 exports.MessageTypes = {

src/util/Injected.js

@@ -53,7 +53,6 @@ exports.ExposeStore = () => {
  * Adds extra props to the serialization of a model
  */
 exports.LoadExtraProps = (model, props) => {
-    console.log(model, props);
     Store[model].models[0].__props = Store[model].models[0].__props.concat(props);
 }