MessagesWSManager.swift

import Starscream

class MessagesWSManager: CFMessagesSocket {
    weak var delegate: CFMessagesSocketDelegate?
    private let socket: WebSocket
    
    init(socket: WebSocket) {
        self.socket = socket
    }
    
    func connect() {
        socket.connect()
    }
    
    func disconnect() {
        socket.disconnect()
    }
    
    func sendMessage(message: String) {
        socket.write(string: message)
    }
    
    func sendMessage(data: Data) {
        socket.write(data: data)
    }
}

extension MessagesWSManager: WebSocketDelegate {
    func websocketDidConnect(socket: WebSocketClient) {
        print("websocket did connect successfully")
    }
    
    func websocketDidDisconnect(socket: WebSocketClient, error: Error?) {
        print("websocket did disconnect successfully")
    }
    
    func websocketDidReceiveMessage(socket: WebSocketClient, text: String) {
        delegate?.didReceiveMessage(message: text)
    }
    
    func websocketDidReceiveData(socket: WebSocketClient, data: Data) {
        delegate?.didReceiveData(data: data)
    }
}