Subversion Repositories Mobile Apps.GyroMouse

//

//  ClientHandler.swift

//  GyroMouse

//

//  Created by Matteo Riva on 28/08/15.

//  Copyright © 2015 Matteo Riva. All rights reserved.

//

import Foundation

import CocoaAsyncSocket

class ClientHandler: NSObject, NetServiceDelegate, NetServiceBrowserDelegate, GCDAsyncSocketDelegate {

    private var socket: GCDAsyncSocket?

    private var serviceBrowser: NetServiceBrowser?

    private(set) var services = [NetService]()

    deinit {

        socket?.setDelegate(nil, delegateQueue: nil)

        socket = nil

        serviceBrowser?.delegate = nil

        serviceBrowser = nil

    }

    //MARK: - Publics

    func startBrowsing() {

        services = []

        // Initialize Service Browser

        serviceBrowser = NetServiceBrowser()

        // Configure Service Browser

        serviceBrowser!.delegate = self

        serviceBrowser!.searchForServices(ofType: "_gyroserver._tcp.", inDomain:"local.")

    }

    func stopBrowsing() {

        serviceBrowser?.stop()

        serviceBrowser?.delegate = nil

        serviceBrowser = nil

        services.removeAll()

    }

    func connectToLocalService(_ service: NetService) {

        // Resolve Service

        service.delegate = self

        service.resolve(withTimeout: 30)

    }

    func sendPacket(_ type: String, keyTapped: String, isButtonDown: Bool, isRightButton: Bool, roll: Double, scrollVelocity: Double, grav: (X: Double, Y: Double, Z: Double), rotat: (X: Double, Z: Double), acc: (X: Double, Y: Double, Z: Double), moveVelocity: Double) {

        let buffer = NSMutableData()

        var int32var: Int32

        var uint32var: UInt32

        var boolvar: Bool

        var doublevar: Double

        int32var = 1; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // protocol version

        if (type == "key") { // key tapped

            if (keyTapped.isEmpty) { return } // consistency check: happens when we hit a backspace

            int32var = 1; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "key" packet type

            uint32var = keyTapped.unicodeScalars.first!.value; buffer.append(&uint32var, length: MemoryLayout<UInt32>.size)

        }

        else if (type == "backspace") { // backspace tapped

            int32var = 2; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "backspace" packet type

            // nothing to add

        }

        else if (type == "enter") { // enter tapped

            int32var = 3; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "enter" packet type

            // nothing to add

        }

        else if (type == "click") { // click

            int32var = 4; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "click" packet type

            boolvar = isRightButton; buffer.append(&boolvar, length: MemoryLayout<Bool>.size)

            boolvar = isButtonDown;  buffer.append(&boolvar, length: MemoryLayout<Bool>.size)

        }

        else if (type == "scroll") { // page scroll

            int32var = 5; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "scroll" packet type

            doublevar = roll;           buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = scrollVelocity; buffer.append(&doublevar, length: MemoryLayout<Double>.size)

        }

        else if (type == "move") { // movement

            int32var = 6; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "move" packet type

            doublevar = grav.X;       buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = grav.Y;       buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = grav.Z;       buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = rotat.X;      buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = rotat.Z;      buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = acc.X;        buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = acc.Y;        buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = acc.Z;        buffer.append(&doublevar, length: MemoryLayout<Double>.size)

            doublevar = moveVelocity; buffer.append(&doublevar, length: MemoryLayout<Double>.size)

        }

        else if (type == "reset") { // reset pointer

            int32var = 7; buffer.append(&int32var, length: MemoryLayout<Int32>.size) // "click" packet type

            // nothing to add

        }

        // Write Buffer

        let bufferData = buffer as Data

        print ("Sending \(bufferData.count) bytes packet:", terminator: "")

        for i in 0...(bufferData.count - 1) {

            let byte: UInt8 = bufferData.withUnsafeBytes { rawBuffer in rawBuffer.load(fromByteOffset: i * MemoryLayout<UInt8>.size, as: UInt8.self) }

            print (" " + String(format: "%02X", byte), terminator: "")

        }

        print ("")

        socket?.write(bufferData, withTimeout: -1, tag: 0)

    }

    func endConnection() {

        socket?.disconnect()

        socket?.setDelegate(nil, delegateQueue: nil)

        socket = nil

    }

    //MARK: - NSNetServiceBrowserDelegate

    func netServiceBrowser(_ browser: NetServiceBrowser, didFind service: NetService, moreComing: Bool) {

        // Update Services

        services.append(service)

        if !moreComing {

            NotificationCenter.default.post(Notification(name: GyroMouseShouldRefreshServerListNotification, object: self, userInfo: nil))

        }

    }

    func netServiceBrowser(_ browser: NetServiceBrowser, didRemove service: NetService, moreComing: Bool) {

        // Update Services

        services.remove(at: services.firstIndex(of: service)!)

        if !moreComing {

            NotificationCenter.default.post(Notification(name: GyroMouseShouldRefreshServerListNotification, object: self, userInfo: nil))

        }

    }

    func netServiceBrowserDidStopSearch(_ browser: NetServiceBrowser) {

        stopBrowsing()

    }

    func netServiceBrowser(_ browser: NetServiceBrowser, didNotSearch errorDict: [String : NSNumber]) {

        stopBrowsing()

    }

    //MARK: - NSNetServiceDelegate

    func netService(_ sender: NetService, didNotResolve errorDict: [String : NSNumber]) {

        sender.delegate = nil

    }

    func netServiceDidResolveAddress(_ sender: NetService) {

        // Connect With Service

        var isConnected = false

        if ((socket == nil) || (socket?.isDisconnected ?? false)) {

            // Initialize Socket

            socket = GCDAsyncSocket(delegate: self, delegateQueue: DispatchQueue.main)

            // Connect

            while (!isConnected && (sender.addresses!.count != 0)) {

                do {

                    try socket!.connect(toAddress: sender.addresses!.first!)

                    isConnected = true

                } catch {

                    isConnected = false

                    print("Unable to connect to address. Error \(error) with user info \(error.localizedDescription).", terminator: "\n")

                }

            }

        } else {

            isConnected = socket?.isConnected ?? false

        }

        if (isConnected) {

            print("Did Connect with Service: domain(\(sender.domain)) type(\(sender.type)) name(\(sender.name)) port(\(sender.port))", terminator: "\n")

        } else {

            print("Unable to Connect with Service: domain(\(sender.domain)) type(\(sender.type)) name(\(sender.name)) port(\(sender.port))", terminator: "\n")

            NotificationCenter.default.post(Notification(name: GyroMouseConnectionFailedNotification, object: self, userInfo: nil))

        }

    }

    //MARK: - GCDAsyncSocketDelegate

    func socket(_ sock: GCDAsyncSocket, didConnectToHost host: String, port: UInt16) {

        print("Socket Did Connect to Host: \(host) Port: \(port)", terminator: "\n")

        stopBrowsing()

        NotificationCenter.default.post(Notification(name: GyroMouseConnectionSuccessNotification, object: self, userInfo: nil))

    }

    func socketDidDisconnect(_ sock: GCDAsyncSocket, withError err: Error?) {

        socket?.delegate = nil

        socket = nil

        startBrowsing()

        NotificationCenter.default.post(Notification(name: GyroMouseDisconnectedNotification, object: self, userInfo: nil))

        if err != nil {

            print("Socket Did Disconnect with Error \(err!) with user info \(err!.localizedDescription).", terminator: "\n")

            //NotificationCenter.default.post(Notification(name: GyroMouseConnectionFailedNotification, object: self, userInfo: nil))

        } else {

            print("Socket Did Disconnect", terminator: "\n")

        }

    }

}