Merge pull request #9 from szymonkups/add-set-mode-to-cli

Add set mode to CLI

Author: michaelarnauts

README.md

@@ -22,6 +22,7 @@ $ python -m aiocomfoconnect register --host 192.168.1.213
 
 $ python -m aiocomfoconnect set-speed away --host 192.168.1.213
 $ python -m aiocomfoconnect set-speed low --host 192.168.1.213
+$ python -m aiocomfoconnect set-mode auto --host 192.168.1.213
 $ python -m aiocomfoconnect set-speed medium --host 192.168.1.213
 $ python -m aiocomfoconnect set-speed high --host 192.168.1.213
 
@@ -140,7 +141,7 @@ if __name__ == "__main__":
 
 ### Decode network traffic
 
-You can use the `scripts/decode_pcap.py` file to decode network traffic between the Mobile App and the ComfoConnect LAN C. 
+You can use the `scripts/decode_pcap.py` file to decode network traffic between the Mobile App and the ComfoConnect LAN C.
 Make sure that the first TCP session in the capture is the connection between the bridge and the app. It's therefore recommended to start the capture before you open the app.
 
 ```shell

aiocomfoconnect/__main__.py

@@ -32,6 +32,9 @@ async def main(args):
     elif args.action == "set-speed":
         await run_set_speed(args.host, args.uuid, args.speed)
 
+    elif args.action == "set-mode":
+        await run_set_mode(args.host, args.uuid, args.mode)
+
     elif args.action == "show-sensors":
         await run_show_sensors(args.host, args.uuid)
 
@@ -110,6 +113,26 @@ async def run_set_speed(host: str, uuid: str, speed: Literal["away", "low", "med
     await comfoconnect.disconnect()
 
 
+async def run_set_mode(host: str, uuid: str, mode: Literal["auto", "manual"]):
+    """Connect to a bridge."""
+    # Discover bridge so we know the UUID
+    bridges = await discover_bridges(host)
+    if not bridges:
+        raise Exception("No bridge found")
+
+    # Connect to the bridge
+    comfoconnect = ComfoConnect(bridges[0].host, bridges[0].uuid)
+    try:
+        await comfoconnect.connect(uuid)
+    except ComfoConnectNotAllowed:
+        print("Could not connect to bridge. Please register first.")
+        sys.exit(1)
+
+    await comfoconnect.set_mode(mode)
+
+    await comfoconnect.disconnect()
+
+
 async def run_show_sensors(host: str, uuid: str):
     """Connect to a bridge."""
     # Discover bridge so we know the UUID
@@ -217,6 +240,11 @@ def sensor_callback(sensor_, value):
     p_set_speed.add_argument("--host", help="Host address of the bridge")
     p_set_speed.add_argument("--uuid", help="UUID of this app", default=DEFAULT_UUID)
 
+    p_set_mode = subparsers.add_parser("set-mode", help="set operation mode")
+    p_set_mode.add_argument("mode", help="Operation mode", choices=["auto", "manual"])
+    p_set_mode.add_argument("--host", help="Host address of the bridge")
+    p_set_mode.add_argument("--uuid", help="UUID of this app", default=DEFAULT_UUID)
+
     p_sensors = subparsers.add_parser("show-sensors", help="show the sensor values")
     p_sensors.add_argument("--host", help="Host address of the bridge")
     p_sensors.add_argument("--uuid", help="UUID of this app", default=DEFAULT_UUID)

docs/PROTOCOL.md

@@ -91,8 +91,8 @@ indicates the length of the `op` field, the rest of the data contains the `msg`.
 | Field                 | Data                                 | Remark                                                |
 |-----------------------|--------------------------------------|-------------------------------------------------------|
 | length (32 bit)       | `0x0000004f`                         | Length of the whole message excluding this field      |
-| src (12 bytes)        | `0xaf154804169043898d2da77148f886be` |                                                       |
-| dst (12 bytes)        | `0x0000000000251010800170b3d54264b4` |                                                       |
+| src (16 bytes)        | `0xaf154804169043898d2da77148f886be` |                                                       |
+| dst (16 bytes)        | `0x0000000000251010800170b3d54264b4` |                                                       |
 | op_length (16 bit     | `0x0004`                             | Length of the `op` message                            |
 | op (variable length)  | `0x08342002`                         | Message with type `GatewayOperation`                  |
 | msg (variable length) | `...`                                | Message with type that is stated in `op.type`         |