From 8eff36b744c78f91e052c0321949ae8963bd6694 Mon Sep 17 00:00:00 2001
From: bdistin <bdistin@gmail.com>
Date: Mon, 5 Dec 2016 15:06:49 -0600
Subject: [PATCH] Fix reconnect when using WS (#952)

* Fix reconnect when using WS

* Add disconnect status (fix reconnect with WS)
---
 src/client/websocket/WebSocketManager.js | 1 +
 src/util/Constants.js                    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/client/websocket/WebSocketManager.js b/src/client/websocket/WebSocketManager.js
index 4fc55bc1a..3da6ecda1 100644
--- a/src/client/websocket/WebSocketManager.js
+++ b/src/client/websocket/WebSocketManager.js
@@ -216,6 +216,7 @@ class WebSocketManager extends EventEmitter {
   eventClose(event) {
     this.emit('close', event);
     this.client.clearInterval(this.client.manager.heartbeatInterval);
+    this.status = Constants.Status.DISCONNECTED;
     /**
      * Emitted whenever the client websocket is disconnected
      * @event Client#disconnect
diff --git a/src/util/Constants.js b/src/util/Constants.js
index 9fe2f9d98..f404b97c7 100644
--- a/src/util/Constants.js
+++ b/src/util/Constants.js
@@ -149,6 +149,7 @@ exports.Status = {
   RECONNECTING: 2,
   IDLE: 3,
   NEARLY: 4,
+  DISCONNECTED: 5,
 };
 
 exports.ChannelTypes = {