Reconnect on Disconnect Unless Reachability Changes

If the app is allowed to go to sleep in the background, or perhaps even if the
websocket is just closed from the remote end, it will never reconnect.

We're currently relying on the reachability to tell us when it's time to
connect again, but that's assuming that the reason we disconnected was
reachability.

That's not always the case.
So, I've started a timer on disconnect that gives reachability 1 second to
figure itself out before we try to reconnect.
We cancel the operation if reachability changes, otherwise we let it run.

Author: psycotica0

Source/PusherConnection.swift

@@ -17,15 +17,18 @@ public class PusherConnection {
     public var socket: WebSocket!
     public var URLSession: NSURLSession
     public weak var stateChangeDelegate: ConnectionStateChangeDelegate?
+    internal var reconnectOperation: NSOperation?
 
     public lazy var reachability: Reachability? = {
         let reachability = try? Reachability.reachabilityForInternetConnection()
         reachability?.whenReachable = { [unowned self] reachability in
+            self.reconnectOperation?.cancel()
             if self.connectionState == .Disconnected {
                 self.socket.connect()
             }
         }
         reachability?.whenUnreachable = { [unowned self] reachability in
+            self.reconnectOperation?.cancel()
             print("Network unreachable")
         }
         return reachability

Source/PusherWebsocketDelegate.swift

@@ -39,6 +39,17 @@ extension PusherConnection: WebSocketDelegate {
         for (_, channel) in self.channels.channels {
             channel.subscribed = false
         }
+
+        let operation = NSBlockOperation {
+            self.socket.connect()
+        }
+
+        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, Int64(NSEC_PER_SEC)), dispatch_get_main_queue()) {
+            NSOperationQueue.mainQueue().addOperation(operation)
+        }
+
+        self.reconnectOperation?.cancel()
+        self.reconnectOperation = operation
     }
 
     public func websocketDidConnect(ws: WebSocket) {}