feat: add heat-status CLI set command for TUI parity

Add _set_heat_status() to cli.py following the existing _set_* pattern,
dispatching from cmd_set() for "heat-status" on/off. Update
_SET_PARAM_NAMES and argparse help text to include heat-status. Add
tests for on, off, invalid values, and dispatch integration.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: deviantintegral

src/flameconnect/cli.py

@@ -130,9 +130,9 @@
 
 _SET_PARAM_NAMES = (
     "mode, flame-speed, brightness, pulsating, flame-color,"
-    " media-theme, heat-mode, heat-temp, timer, temp-unit,"
-    " flame-effect, media-light, media-color, overhead-light,"
-    " overhead-color, ambient-sensor"
+    " media-theme, heat-status, heat-mode, heat-temp, timer,"
+    " temp-unit, flame-effect, media-light, media-color,"
+    " overhead-light, overhead-color, ambient-sensor"
 )
 
 
@@ -481,6 +481,9 @@ async def cmd_set(
     if param == "ambient-sensor":
         await _set_ambient_sensor(client, fire_id, value)
         return
+    if param == "heat-status":
+        await _set_heat_status(client, fire_id, value)
+        return
 
     print(f"Error: unknown parameter '{param}'. Valid: {_SET_PARAM_NAMES}.")
     sys.exit(1)
@@ -810,6 +813,28 @@ async def _set_ambient_sensor(
     print(f"Ambient sensor set to {value}.")
 
 
+async def _set_heat_status(
+    client: FlameConnectClient, fire_id: str, value: str
+) -> None:
+    """Set the heater on or off."""
+    from flameconnect.models import HeatStatus
+
+    lookup: dict[str, HeatStatus] = {"on": HeatStatus.ON, "off": HeatStatus.OFF}
+    if value not in lookup:
+        valid = ", ".join(lookup)
+        print(f"Error: heat-status must be one of: {valid}.")
+        sys.exit(1)
+    heat_status = lookup[value]
+    overview = await client.get_fire_overview(fire_id)
+    current = _find_param(overview.parameters, HeatParam)
+    if current is None:
+        print("Error: no HeatSettings parameter found.")
+        sys.exit(1)
+    new_param = replace(current, heat_status=heat_status)
+    await client.write_parameters(fire_id, [new_param])
+    print(f"Heat status set to {value}.")
+
+
 async def cmd_tui(*, verbose: bool = False) -> None:
     """Launch the TUI, showing install message if missing."""
     try:
@@ -866,9 +891,10 @@ def build_parser() -> argparse.ArgumentParser:
         "param",
         help=(
             "Parameter name: mode, flame-speed, brightness, pulsating,"
-            " flame-color, media-theme, heat-mode, heat-temp, timer,"
-            " temp-unit, flame-effect, media-light, media-color,"
-            " overhead-light, overhead-color, ambient-sensor"
+            " flame-color, media-theme, heat-status, heat-mode,"
+            " heat-temp, timer, temp-unit, flame-effect, media-light,"
+            " media-color, overhead-light, overhead-color,"
+            " ambient-sensor"
         ),
     )
     sp_set.add_argument("value", help="Value to set")

tests/test_cli_set.py

@@ -17,6 +17,7 @@
     _set_flame_effect,
     _set_flame_speed,
     _set_heat_mode,
+    _set_heat_status,
     _set_heat_temp,
     _set_media_color,
     _set_media_light,
@@ -573,6 +574,16 @@ async def test_dispatch_temp_unit(self, mock_api, token_auth):
         key = ("POST", URL(WRITE_URL))
         assert len(mock_api.requests[key]) == 1
 
+    async def test_dispatch_heat_status(self, mock_api, token_auth, overview_payload):
+        mock_api.get(OVERVIEW_URL, payload=overview_payload)
+        mock_api.post(WRITE_URL, payload={})
+
+        async with FlameConnectClient(token_auth) as client:
+            await cmd_set(client, FIRE_ID, "heat-status", "on")
+
+        key = ("POST", URL(WRITE_URL))
+        assert len(mock_api.requests[key]) == 1
+
     async def test_dispatch_unknown_param(self, mock_api, token_auth, capsys):
         async with FlameConnectClient(token_auth) as client:
             with pytest.raises(SystemExit):
@@ -729,6 +740,47 @@ async def test_set_ambient_sensor_invalid(self, mock_api, token_auth, capsys):
         assert "Error" in captured.out
 
 
+# ---------------------------------------------------------------------------
+# _set_heat_status
+# ---------------------------------------------------------------------------
+
+
+class TestSetHeatStatus:
+    """Tests for the _set_heat_status CLI command."""
+
+    async def test_set_heat_status_on(self, mock_api, token_auth, overview_payload):
+        mock_api.get(OVERVIEW_URL, payload=overview_payload)
+        mock_api.post(WRITE_URL, payload={})
+
+        async with FlameConnectClient(token_auth) as client:
+            await _set_heat_status(client, FIRE_ID, "on")
+
+        key = ("POST", URL(WRITE_URL))
+        calls = mock_api.requests[key]
+        assert len(calls) == 1
+        body = calls[0].kwargs["json"]
+        assert body["FireId"] == FIRE_ID
+        assert body["Parameters"][0]["ParameterId"] == 323
+
+    async def test_set_heat_status_off(self, mock_api, token_auth, overview_payload):
+        mock_api.get(OVERVIEW_URL, payload=overview_payload)
+        mock_api.post(WRITE_URL, payload={})
+
+        async with FlameConnectClient(token_auth) as client:
+            await _set_heat_status(client, FIRE_ID, "off")
+
+        key = ("POST", URL(WRITE_URL))
+        calls = mock_api.requests[key]
+        assert len(calls) == 1
+
+    async def test_set_heat_status_invalid(self, mock_api, token_auth, capsys):
+        async with FlameConnectClient(token_auth) as client:
+            with pytest.raises(SystemExit):
+                await _set_heat_status(client, FIRE_ID, "maybe")
+        captured = capsys.readouterr()
+        assert "Error" in captured.out
+
+
 # ---------------------------------------------------------------------------
 # _set_media_color
 # ---------------------------------------------------------------------------