Merge pull request #6 from wolever/patch-1

Add --connect/--disconnect options, fail with exit code if connect/disconnect fails

Author: lapfelix

BluetoothConnector.swift

@@ -3,36 +3,68 @@ import IOBluetooth
 let hardcodedAddress : String? = nil//"12-34-56-78-90-ab"
 
 func printHelp() {
-    print("Usage:\nBluetoothConnector 00-00-00-00-00-00\n\nGet the MAC address from the list below (if your device is missing, pair it with your computer first):");
+    print("Usage:\nBluetoothConnector [-c|--connect|-d|--disconnect] [00-00-00-00-00-00]\n\nGet the MAC address from the list below (if your device is missing, pair it with your computer first):\n");
     IOBluetoothDevice.pairedDevices().forEach({(device) in
         guard let device = device as? IOBluetoothDevice,
         let addressString = device.addressString,
         let deviceName = device.name
         else { return }
-        print("\(addressString) - \(deviceName)")
+        let connectedString = device.isConnected() ? "connected" : "disconnected"
+        print("\(addressString) - \(deviceName) (\(connectedString))")
     })
 }
 
-if (CommandLine.arguments.count == 2 || hardcodedAddress != nil) {
-    let deviceAddress : String? = hardcodedAddress == nil ? CommandLine.arguments[1] : nil
-    guard let bluetoothDevice = IOBluetoothDevice(addressString: hardcodedAddress ?? deviceAddress) else {
-        print("Device not found")
-        exit(-2)
-    }
-    
-    if !bluetoothDevice.isPaired() {
-        print("Not paired to device")
-        exit(-4)
-    }
-    
-    if bluetoothDevice.isConnected() {
-        bluetoothDevice.closeConnection()
-    }
-    else {
-        bluetoothDevice.openConnection()
+var macAddr : String? = nil
+var action : String? = nil
+
+for argument in CommandLine.arguments[1...] {
+    switch argument {
+        case "-c", "--connect":
+            action = "connect"
+
+        case "-d", "--disconnect":
+            action = "disconnect"
+
+        default:
+            if argument.hasPrefix("-") || macAddr != nil {
+                printHelp()
+                exit(-3)
+            }
+            macAddr = argument
     }
 }
-else {
+
+
+if (macAddr == nil) {
     printHelp()
     exit(-3)
 }
+
+let deviceAddress = hardcodedAddress ?? macAddr!
+guard let bluetoothDevice = IOBluetoothDevice(addressString: deviceAddress) else {
+    print("Device not found: \(deviceAddress)")
+    exit(-2)
+}
+
+if !bluetoothDevice.isPaired() {
+    print("Not paired to device")
+    exit(-4)
+}
+
+var error : IOReturn = -1
+
+if bluetoothDevice.isConnected() {
+    if (action == "disconnect" || action == nil) {
+        error = bluetoothDevice.closeConnection()
+    }
+}
+else {
+    if (action == "connect" || action == nil) {
+        error = bluetoothDevice.openConnection()
+    }
+}
+
+if error > 0 {
+    print("\(action ?? "toggle") failed")
+    exit(-1)
+}