Client improvements - disconnect, ping-pong, timeout, reconnect

Author: pvzig

SlackKit/Sources/Channel.swift

@@ -43,7 +43,7 @@ public struct Channel {
     internal(set) public var unread: Int?
     internal(set) public var unreadCountDisplay: Int?
     internal(set) public var hasPins: Bool?
-    internal(set) public var members = [String]()
+    internal(set) public var members: [String]?
     // Client use
     internal(set) public var pinnedItems = [Item]()
     internal(set) public var usersTyping = [String]()
@@ -70,11 +70,7 @@ public struct Channel {
         unread = channel?["unread_count"] as? Int
         unreadCountDisplay = channel?["unread_count_display"] as? Int
         hasPins = channel?["has_pins"] as? Bool
-        
-        if let members = channel?["members"] as? [String] {
-            self.members = members
-        }
-        
+        members = channel?["members"] as? [String]
     }
 
     internal init?(id:String?) {

SlackKit/Sources/Client.swift

@@ -53,7 +53,7 @@ public class Client: WebSocketDelegate {
     public var teamProfileEventsDelegate: TeamProfileEventsDelegate?
 
     internal var token = "SLACK_AUTH_TOKEN"
-
+    
     public func setAuthToken(token: String) {
         self.token = token
     }
@@ -63,15 +63,25 @@ public class Client: WebSocketDelegate {
     }
 
     internal var webSocket: WebSocket?
+    internal let api = NetworkInterface()
     private var dispatcher: EventDispatcher?
 
-    internal let api = NetworkInterface()
+    private let pingPongQueue = dispatch_queue_create("com.launchsoft.SlackKit", DISPATCH_QUEUE_SERIAL)
+    internal var ping: Double?
+    internal var pong: Double?
+    
+    internal var pingInterval: NSTimeInterval?
+    internal var timeout: NSTimeInterval?
+    internal var reconnect: Bool?
 
     required public init(apiToken: String) {
         self.token = apiToken
     }
 
-    public func connect() {
+    public func connect(pingInterval pingInterval: NSTimeInterval? = nil, timeout: NSTimeInterval? = nil, reconnect: Bool? = nil) {
+        self.pingInterval = pingInterval
+        self.timeout = timeout
+        self.reconnect = reconnect
         dispatcher = EventDispatcher(client: self)
         webAPI.rtmStart(success: {
             (response) -> Void in
@@ -85,6 +95,10 @@ public class Client: WebSocketDelegate {
             }, failure:nil)
     }
 
+    public func disconnect() {
+        webSocket?.disconnect()
+    }
+    
     //MARK: - Message send
     public func sendMessage(message: String, channelID: String) {
         if (connected) {
@@ -97,7 +111,7 @@ public class Client: WebSocketDelegate {
 
     private func formatMessageToSlackJsonString(message: (msg: String, channel: String)) -> NSData? {
         let json: [String: AnyObject] = [
-            "id": NSDate().timeIntervalSince1970,
+            "id": NSDate().slackTimestamp(),
             "type": "message",
             "channel": message.channel,
             "text": message.msg.slackFormatEscaping()
@@ -121,6 +135,52 @@ public class Client: WebSocketDelegate {
         sentMessages[ts!.stringValue] = Message(message: message)
     }
 
+    //MARK: - RTM Ping
+    private func pingRTMServerAtInterval(interval: NSTimeInterval) {
+        let delay = dispatch_time(DISPATCH_TIME_NOW, Int64(interval * Double(NSEC_PER_SEC)))
+        dispatch_after(delay, pingPongQueue, {
+            if self.connected && self.timeoutCheck() {
+                self.sendRTMPing()
+                self.pingRTMServerAtInterval(interval)
+            } else {
+                self.disconnect()
+            }
+        })
+    }
+    
+    private func sendRTMPing() {
+        if connected {
+            let json: [String: AnyObject] = [
+                "id": NSDate().slackTimestamp(),
+                "type": "ping",
+            ]
+            do {
+                let data = try NSJSONSerialization.dataWithJSONObject(json, options: NSJSONWritingOptions.PrettyPrinted)
+                let string = NSString(data: data, encoding: NSUTF8StringEncoding)
+                if let writePing = string as? String {
+                    ping = json["id"] as? Double
+                    webSocket?.writeString(writePing)
+                }
+            }
+            catch _ {
+                
+            }
+        }
+    }
+    
+    private func timeoutCheck() -> Bool {
+        if let pong = pong, ping = ping, timeout = timeout {
+            if pong - ping < timeout {
+                return true
+            } else {
+                return false
+            }
+        // Ping-pong or timeout not configured
+        } else {
+            return true
+        }
+    }
+    
     //MARK: - Client setup
     internal func initialSetup(json: [String: AnyObject]) {
         team = Team(team: json["team"] as? [String: AnyObject])
@@ -207,14 +267,21 @@ public class Client: WebSocketDelegate {
     }
 
     // MARK: - WebSocketDelegate
-    public func websocketDidConnect(socket: WebSocket) {}
+    public func websocketDidConnect(socket: WebSocket) {
+        if let pingInterval = pingInterval {
+            pingRTMServerAtInterval(pingInterval)
+        }
+    }
 
     public func websocketDidDisconnect(socket: WebSocket, error: NSError?) {
         connected = false
         authenticated = false
         webSocket = nil
-        if let delegate = slackEventsDelegate {
-            delegate.clientDisconnected()
+        dispatcher = nil
+        authenticatedUser = nil
+        slackEventsDelegate?.clientDisconnected()
+        if reconnect == true {
+            connect(pingInterval: pingInterval, timeout: timeout, reconnect: reconnect)
         }
     }
 
@@ -223,7 +290,9 @@ public class Client: WebSocketDelegate {
             return
         }
         do {
-            try dispatcher?.dispatch(NSJSONSerialization.JSONObjectWithData(data, options: NSJSONReadingOptions.AllowFragments) as! [String: AnyObject])
+            if let json = try NSJSONSerialization.JSONObjectWithData(data, options: NSJSONReadingOptions.AllowFragments) as? [String: AnyObject] {
+                dispatcher?.dispatch(json)
+            }
         }
         catch _ {
 

SlackKit/Sources/ClientExtensions.swift

@@ -78,8 +78,8 @@ internal extension String {
 
 public extension NSDate {
 
-    func slackTimestamp() -> String {
-        return NSNumber(double: timeIntervalSince1970).stringValue
+    func slackTimestamp() -> Double {
+        return NSNumber(double: timeIntervalSince1970).doubleValue
     }
 
 }

SlackKit/Sources/Event.swift

@@ -62,6 +62,7 @@ internal enum EventType: String {
     case FileCommentDeleted = "file_comment_deleted"
     case PinAdded = "pin_added"
     case PinRemoved = "pin_removed"
+    case Pong = "pong"
     case PresenceChange = "presence_change"
     case ManualPresenceChange = "manual_presence_change"
     case PrefChange = "pref_change"

SlackKit/Sources/EventDispatcher.swift

@@ -96,6 +96,8 @@ internal class EventDispatcher {
                 handler.pinAdded(event)
             case .PinRemoved:
                 handler.pinRemoved(event)
+            case .Pong:
+                handler.pong(event)
             case .PresenceChange:
                 handler.presenceChange(event)
             case .ManualPresenceChange:
@@ -146,7 +148,7 @@ internal class EventDispatcher {
                 // Other clients should ignore this event.
                 break
             case .TeamMigrationStarted:
-                client.connect()
+                client.connect(client.pingInterval, timeout: client.timeout, reconnect: client.reconnect)
             case .ReconnectURL:
                 // The reconnect_url event is currently unsupported and experimental.
                 break

SlackKit/Sources/EventHandler.swift

@@ -38,6 +38,11 @@ internal class EventHandler {
         }
     }
 
+    //MARK: - Pong
+    func pong(event: Event) {
+        client.pong = event.replyTo
+    }
+    
     //MARK: - Messages
     func messageSent(event: Event) {
         if let reply = event.replyTo, message = client.sentMessages[NSNumber(double: reply).stringValue], channel = message.channel, ts = message.ts {
@@ -147,8 +152,8 @@ internal class EventHandler {
 
     func channelLeft(event: Event) {
         if let channel = event.channel, id = channel.id, userID = client.authenticatedUser?.id {
-            if let index = client.channels[id]?.members.indexOf(userID) {
-                client.channels[id]?.members.removeAtIndex(index)
+            if let index = client.channels[id]?.members?.indexOf(userID) {
+                client.channels[id]?.members?.removeAtIndex(index)
 
                 if let delegate = client.channelEventsDelegate {
                     delegate.channelLeft(channel)