implement some features of Venus OS

custom_components/victron_mk2_mqtt/number.py

@@ -21,7 +21,16 @@ async def async_setup_platform(
 ) -> None:
     """Set up Victron number entities."""
     bridge: VictronMqttBridge = hass.data[DOMAIN][DATA_BRIDGE]
-    async_add_entities([VictronRemotePanelCurrentLimitNumber(bridge)])
+    entities: list[NumberEntity] = [VictronRemotePanelCurrentLimitNumber(bridge)]
+    if bridge.venus_guide_compat:
+        entities.extend(
+            [
+                VictronESSGridSetpointNumber(bridge),
+                VictronESSMaxChargePowerNumber(bridge),
+                VictronESSMaxDischargePowerNumber(bridge),
+            ]
+        )
+    async_add_entities(entities)
 
 
 class VictronRemotePanelCurrentLimitNumber(VictronMqttEntity, NumberEntity):
@@ -52,3 +61,75 @@ class VictronRemotePanelCurrentLimitNumber(VictronMqttEntity, NumberEntity):
         await self.bridge.async_publish_command(
             {"kind": "panel_state", "current_limit": float(value)}
         )
+
+
+class _VictronESSNumberBase(VictronMqttEntity, NumberEntity):
+    """Base class for ESS compatibility numbers."""
+
+    _attr_mode = NumberMode.BOX
+    _attr_native_step = 1.0
+    _attr_native_min_value = -20000.0
+    _attr_native_max_value = 20000.0
+    _attr_native_unit_of_measurement = "W"
+    _attr_icon = "mdi:transmission-tower-export"
+
+    @property
+    def available(self) -> bool:
+        return bool(self.bridge.command_topic and self.bridge.venus_guide_compat)
+
+
+class VictronESSGridSetpointNumber(_VictronESSNumberBase):
+    """Guide-compatible ESS AC power setpoint."""
+
+    _attr_name = "ESS Grid Setpoint"
+
+    def __init__(self, bridge: VictronMqttBridge) -> None:
+        super().__init__(bridge)
+        self._attr_unique_id = f"{bridge.topic_root}_ess_grid_setpoint"
+
+    @property
+    def native_value(self) -> float | None:
+        return self.bridge.ess_setpoint
+
+    async def async_set_native_value(self, value: float) -> None:
+        await self.bridge.async_publish_command({"kind": "ess_setpoint", "value": float(value)})
+
+
+class VictronESSMaxChargePowerNumber(_VictronESSNumberBase):
+    """Guide-compatible ESS max charge power."""
+
+    _attr_name = "ESS Max Charge Power"
+    _attr_native_min_value = 0.0
+
+    def __init__(self, bridge: VictronMqttBridge) -> None:
+        super().__init__(bridge)
+        self._attr_unique_id = f"{bridge.topic_root}_ess_max_charge_power"
+
+    @property
+    def native_value(self) -> float | None:
+        return self.bridge.ess_max_charge_power
+
+    async def async_set_native_value(self, value: float) -> None:
+        await self.bridge.async_publish_command(
+            {"kind": "ess_max_charge_power", "value": float(value)}
+        )
+
+
+class VictronESSMaxDischargePowerNumber(_VictronESSNumberBase):
+    """Guide-compatible ESS max discharge power."""
+
+    _attr_name = "ESS Max Discharge Power"
+    _attr_native_min_value = 0.0
+
+    def __init__(self, bridge: VictronMqttBridge) -> None:
+        super().__init__(bridge)
+        self._attr_unique_id = f"{bridge.topic_root}_ess_max_discharge_power"
+
+    @property
+    def native_value(self) -> float | None:
+        return self.bridge.ess_max_discharge_power
+
+    async def async_set_native_value(self, value: float) -> None:
+        await self.bridge.async_publish_command(
+            {"kind": "ess_max_discharge_power", "value": float(value)}
+        )