add update sensor for routerboard firmware #229

custom_components/mikrotik_router/mikrotik_controller.py

@@ -1315,11 +1315,8 @@ class MikrotikControllerData:
                     {"name": "routerboard", "type": "bool"},
                     {"name": "model", "default": "unknown"},
                     {"name": "serial-number", "default": "unknown"},
-                    {
+                    {"name": "current-firmware", "default": "unknown"},
-                        "name": "firmware",
+                    {"name": "upgrade-firmware", "default": "unknown"},
-                        "source": "current-firmware",
-                        "default": "unknown",
-                    },
                 ],
             )
 

custom_components/mikrotik_router/update.py

@@ -24,7 +24,8 @@ DEVICE_UPDATE = "device_update"
 async def async_setup_entry(hass, config_entry, async_add_entities):
     """Set up entry for component"""
     dispatcher = {
-        "MikrotikUpdate": MikrotikUpdate,
+        "MikrotikRouterOSUpdate": MikrotikRouterOSUpdate,
+        "MikrotikRouterBoardFWUpdate": MikrotikRouterBoardFWUpdate,
     }
     await model_async_setup_entry(
         hass,
@@ -37,9 +38,9 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
 
 
 # ---------------------------
-#   MikrotikUpdate
+#   MikrotikRouterOSUpdate
 # ---------------------------
-class MikrotikUpdate(MikrotikEntity, UpdateEntity):
+class MikrotikRouterOSUpdate(MikrotikEntity, UpdateEntity):
     """Define an Mikrotik Controller Update entity."""
 
     TYPE = DEVICE_UPDATE
@@ -84,8 +85,6 @@ class MikrotikUpdate(MikrotikEntity, UpdateEntity):
 
         self._ctrl.execute("/system/package/update", "install", None, None)
 
-        # await self.controller.api.devices.upgrade(self.device.mac)
-
     async def async_release_notes(self) -> str:
         """Return the release notes."""
         try:
@@ -106,3 +105,49 @@ class MikrotikUpdate(MikrotikEntity, UpdateEntity):
     def release_url(self) -> str:
         """URL to the full release notes of the latest version available."""
         return "https://mikrotik.com/download/changelogs"
+
+
+# ---------------------------
+#   MikrotikRouterBoardFWUpdate
+# ---------------------------
+class MikrotikRouterBoardFWUpdate(MikrotikEntity, UpdateEntity):
+    """Define an Mikrotik Controller Update entity."""
+
+    def __init__(
+        self,
+        inst,
+        uid: "",
+        mikrotik_controller,
+        entity_description,
+    ):
+        """Set up device update entity."""
+        super().__init__(inst, uid, mikrotik_controller, entity_description)
+        _LOGGER.error(self._data)
+
+        self._attr_supported_features = UpdateEntityFeature.INSTALL
+
+    @property
+    def is_on(self) -> bool:
+        """Return true if device is on."""
+        return (
+            self.data["routerboard"]["current-firmware"]
+            != self.data["routerboard"]["upgrade-firmware"]
+        )
+
+    @property
+    def installed_version(self) -> str:
+        """Version installed and in use."""
+        return self._data["current-firmware"]
+
+    @property
+    def latest_version(self) -> str:
+        """Latest version available for install."""
+        return self._data["upgrade-firmware"]
+
+    async def options_updated(self) -> None:
+        """No action needed."""
+
+    async def async_install(self, version: str, backup: bool, **kwargs: Any) -> None:
+        """Install an update."""
+        self._ctrl.execute("/system/routerboard", "upgrade", None, None)
+        self._ctrl.execute("/system", "reboot", None, None)

custom_components/mikrotik_router/update_types.py

@@ -18,7 +18,7 @@ class MikrotikUpdateEntityDescription(UpdateEntityDescription):
     data_uid: str = ""
     data_reference: str = ""
     data_attributes_list: List = field(default_factory=lambda: [])
-    func: str = "MikrotikUpdate"
+    func: str = "MikrotikRouterOSUpdate"
 
 
 SENSOR_TYPES = {
@@ -30,7 +30,20 @@ SENSOR_TYPES = {
         data_name="",
         data_uid="",
         data_reference="",
+        func="MikrotikRouterOSUpdate",
+    ),
+    "system_rbfwupdate": MikrotikUpdateEntityDescription(
+        key="system_rbfwupdate",
+        name="RouterBoard firmware update",
+        ha_group="System",
+        data_path="routerboard",
+        data_attribute="current-firmware",
+        data_name="",
+        data_uid="",
+        data_reference="",
+        func="MikrotikRouterBoardFWUpdate",
     ),
 }
 
+
 SENSOR_SERVICES = {}