Subversion Repositories Mobile Apps.GyroMouse

//
//  MouseHandler.swift
//  GyroServer
//
//  Created by Matteo Riva on 07/08/15.
//  Copyright © 2015 Matteo Riva. All rights reserved.
//

import CoreGraphics
import Cocoa
import Carbon
import CoreServices

class MouseHandler: NSObject, ServerHandlerDelegate {
    
    private let server = ServerHandler()
    
    private var activeScreen: Int {
        get {
            return UserDefaults.standard.integer(forKey: "activeScreen")
        }
    }
    private var mLoc: (x: CGFloat, y: CGFloat) = (x: 0, y: 0)
    private var isDragging = false

    override init() {
        super.init()
        resetPointerPosition(moveMouse: false)
        server.startBroadcast()
        NotificationCenter.default.addObserver(forName: ActiveScreenDidChangeNotification, object: nil, queue: OperationQueue.main) {[unowned self] (_) -> Void in
            self.resetPointerPosition(moveMouse: false)
        }
        NotificationCenter.default.addObserver(forName: ServerDidConnectNotification, object: nil, queue: OperationQueue.main) {[unowned self] (_) -> Void in
            self.resetPointerPosition(moveMouse: true)
            self.server.delegate = self
        }
        NotificationCenter.default.addObserver(forName: ServerDidDisconnectNotification, object: nil, queue: OperationQueue.main) {[unowned self] (_) -> Void in
            self.server.delegate = nil
            self.resetPointerPosition(moveMouse: false)
        }
    }
    
    deinit {
        NotificationCenter.default.removeObserver(self)
    }
    
    //MARK: - Mouse movements
    
    private func computePointerMovementWithPacket(_ packet: GyroPacket) {
         let roll = packet.roll
        let sgn: Double = roll ?? 0 < 0 ? -1 : 1
        let gate = 0.75
        let changeMovCond = roll == nil || (roll! > -gate && roll! < gate)
        
        let rotatZ = changeMovCond ? packet.rotatZ! : packet.rotatX! * -sgn
        let accZ = changeMovCond ? packet.accZ! : packet.accX!
        let rotatX = changeMovCond ? packet.rotatX! : packet.rotatZ! * sgn
        let accX = changeMovCond ? packet.accX! : packet.accZ!
        let gravZ = changeMovCond ? packet.gravZ! : packet.gravX! * sgn
        
        var dx = ((rotatZ * -1) * (1 + abs(accZ))) * packet.moveVelocity! * abs(gravZ)
        var dy = ((rotatX * -1) * (1 + abs(accX))) * packet.moveVelocity! * abs(gravZ)
        
        if packet.type == .scroll {
            dx = min(dx,1)
            dy = min(dy,1)
        }
        
        var x = mLoc.x + CGFloat(dx)
        var y = mLoc.y + CGFloat(dy)
        
        let frame = NSScreen.screens[activeScreen].frame
        let s = (o: frame.origin, w: frame.maxX - 1, h: frame.maxY - 1)

        mLoc = (x: x, y: y)
        x = max(s.o.x,min(x, s.w))
        y = max(0,min(y, s.h))
        
        let point = CGPoint(x: x, y: y)

        let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: .mouseMoved, mouseCursorPosition: point, mouseButton: .left)
        mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)
 
        if isDragging {
             let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: .leftMouseDragged, mouseCursorPosition: point, mouseButton: .left)
            mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)
        }
    }
    
    func resetPointerPosition(moveMouse: Bool) {
        let aIndex = UserDefaults.standard.integer(forKey: "activeScreen")
        let frame = NSScreen.screens[aIndex].frame
        let x = frame.midX
        let y = frame.midY - frame.origin.y
        NSLog("x: %f, y: %f, origin x: %f, y: %f, mid x: %f, y: %f", x,y,frame.origin.x, frame.origin.y, frame.midX, frame.midY)
        mLoc = (x: x, y: y)
        if moveMouse {
            let point = CGPoint(x: x, y: y)
            let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: .mouseMoved, mouseCursorPosition: point, mouseButton: .left)
            mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)
        }
    }
    
    //MARK: - Mouse actions
    
    var rollGate = 0

    private func scrollWithRoll(_ roll: Double, velocity: Double) {
        rollGate += 1
        if rollGate == 10 {
            let amount = (Int32) (roll * velocity * -50)
            let mouseEvent = CGEvent(scrollWheelEvent2Source: nil, units: .pixel, wheelCount: 1, wheel1: amount, wheel2: 0, wheel3: 0)
            mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)

            rollGate = 0
        }
    }

    private func clickButton(_ button: ButtonType, click: ClickType) {
        let mousePos = CGPoint(x: mLoc.x, y: mLoc.y)
        let mouseEventType: CGEventType
        let mouseButton: CGMouseButton

        if button == .left {
            mouseButton = .left
            mouseEventType = (click == .down ? .leftMouseDown : .leftMouseUp)
            isDragging = (click == .down)
        } else {
            mouseButton = .right
            mouseEventType = (click == .down ? .rightMouseDown : .rightMouseUp)
            isDragging = false
        }
        
        let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: mouseEventType, mouseCursorPosition: mousePos, mouseButton: mouseButton)
        mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)
    }
    
    //MARK: - AppleScript launcher
    
    private func runAppleScript(_ scriptText: String) {
        let script = NSAppleScript(source: scriptText)!
        var error: NSDictionary? = nil
        let resultMaybe = script.executeAndReturnError(&error) as NSAppleEventDescriptor?
        if (resultMaybe == nil)
        {
            let myStringDict = error as? [String:AnyObject]
            if (myStringDict?["NSAppleScriptErrorNumber"] as! Int16 == 1002) {
                let alert = NSAlert()
                alert.addButton(withTitle: "OK".localized)
                alert.messageText = "system_message".localized + "\n" + (myStringDict?["NSAppleScriptErrorBriefMessage"] as! String)
                alert.informativeText = "allow_perms".localized
                alert.alertStyle = .warning
                alert.runModal()
            }
            print (error as Any)
            return
        }
    }
    
    //MARK: - ServerHandlerDelegate
    
    func serverDidReceivePacket(_ packet: GyroPacket) {
        switch packet.type {
        case .scroll:
            scrollWithRoll(packet.roll!, velocity: packet.scrollVelocity!)
            computePointerMovementWithPacket(packet)
        case .movement:
            computePointerMovementWithPacket(packet)
        case .click:
            clickButton(packet.button!, click: packet.click!)
        case .keyTapped:
            runAppleScript("tell app \"System Events\" to keystroke \"\(packet.key!)\"")
        case .deleteBackward:
            runAppleScript ("tell app \"System Events\" to key code 51")
        case .returnTapped:
            runAppleScript("tell app \"System Events\" to keystroke return")
        case .resetPointerPosition:
            resetPointerPosition(moveMouse: true)
        }
    }
}