Add canSupportAsMotor(ioType:) method

bricklife · Jul 23, 2018 · 826e4aa · 826e4aa
1 parent ca10def
commit 826e4aa
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/Sources/BoostBLEKit/Hub/Boost.swift b/Sources/BoostBLEKit/Hub/Boost.swift
@@ -32,6 +32,15 @@ public final class Boost {
             .AB:    0x39,
             ]
 
+        public func canSupportAsMotor(ioType: IOType) -> Bool {
+            switch ioType {
+            case .interactiveMotor, .builtInMotor, .mediumMotor, .trainMotor:
+                return true
+            default:
+                return false
+            }
+        }
+
         public func motorPowerCommand(port: Port, power: Int8) -> Command? {
             guard let portId = portId(for: port) else { return nil }
             guard let ioType = connectedIOs[portId] else { return nil }

diff --git a/Sources/BoostBLEKit/Hub/Hub.swift b/Sources/BoostBLEKit/Hub/Hub.swift
@@ -16,6 +16,8 @@ public protocol Hub {
     func port(for portId: PortId) -> Port?
     func portId(for port: Port) -> PortId?
 
+    func canSupportAsMotor(ioType: IOType) -> Bool
+
     func motorPowerCommand(port: Port, power: Int8) -> Command?
     func rgbLightColorCommand(color: RGBLightColorCommand.Color) -> Command?
 }

diff --git a/Sources/BoostBLEKit/Hub/PoweredUp.swift b/Sources/BoostBLEKit/Hub/PoweredUp.swift
@@ -26,6 +26,15 @@ public final class PoweredUp {
             .AB:    0x39,
             ]
 
+        public func canSupportAsMotor(ioType: IOType) -> Bool {
+            switch ioType {
+            case .mediumMotor, .trainMotor:
+                return true
+            default:
+                return false
+            }
+        }
+
         public func motorPowerCommand(port: Port, power: Int8) -> Command? {
             guard let portId = portId(for: port) else { return nil }
             guard let ioType = connectedIOs[portId] else { return nil }