Add a service to reboot the device

Author: rroller

README.md

@@ -198,6 +198,7 @@ Service | Parameters | Description
 `dahua.vto_open_door` | `target`: camera.cam13_main <br /> `door_id`: The door ID to open, e.g.: 1 <br /> Opens a door via a VTO
 `dahua.vto_cancel_call` | `target`: camera.cam13_main <br />Cancels a call on a VTO device (Doorbell)
 `dahua.set_video_in_day_night_mode` | `target`: camera.cam13_main <br /> `config_type`: The config type: general, day, night <br /> `mode`: The mode: Auto, Color, BlackWhite. Note Auto is also known as Brightness by Dahua|Set the camera's Day/Night Mode. For example, Color, BlackWhite, or Auto
+`dahua.reboot` | `target`: camera.cam13_main <br />Reboots the device 
 
 
 ## Camera

custom_components/dahua/camera.py

@@ -34,6 +34,7 @@
 SERVICE_VTO_OPEN_DOOR = "vto_open_door"
 SERVICE_VTO_CANCEL_CALL = "vto_cancel_call"
 SERVICE_SET_DAY_NIGHT_MODE = "set_video_in_day_night_mode"
+SERVICE_REBOOT = "reboot"
 
 
 async def async_setup_entry(hass: HomeAssistant, config_entry, async_add_entities):
@@ -177,6 +178,12 @@ async def async_setup_entry(hass: HomeAssistant, config_entry, async_add_entitie
         "async_set_video_in_day_night_mode"
     )
 
+    platform.async_register_entity_service(
+        SERVICE_REBOOT,
+        {},
+        "async_reboot"
+    )
+
     platform.async_register_entity_service(
         SERVICE_SET_RECORD_MODE,
         {
@@ -274,6 +281,10 @@ async def async_set_video_in_day_night_mode(self, config_type: str, mode: str):
         await self._coordinator.client.async_set_video_in_day_night_mode(channel, config_type, mode)
         await self._coordinator.async_refresh()
 
+    async def async_reboot(self):
+        """ Handles the service call from SERVICE_REBOOT to reboot the device """
+        await self._coordinator.client.reboot()
+
     async def async_set_record_mode(self, mode: str):
         """ Handles the service call from SERVICE_SET_RECORD_MODE to set the record mode """
         channel = self._coordinator.get_channel()

custom_components/dahua/client.py

@@ -104,6 +104,10 @@ async def get_vendor(self) -> dict:
         """ get_vendor returns the vendor. Example response: vendor=Dahua """
         return await self.get("/cgi-bin/magicBox.cgi?action=getVendor")
 
+    async def reboot(self) -> dict:
+        """ Reboots the device """
+        return await self.get("/cgi-bin/magicBox.cgi?action=reboot")
+
     async def get_max_extra_streams(self) -> int:
         """ get_max_extra_streams returns the max number of sub streams supported by the camera """
         try:

custom_components/dahua/services.yaml

@@ -286,6 +286,14 @@ set_video_in_day_night_mode:
             - "Color"
             - "BlackWhite"
 
+reboot:
+  name: Reboots the device
+  description: "Reboots the device"
+  target:
+    entity:
+      integration: dahua
+      domain: camera
+
 set_record_mode:
   name: Set Record Mode on Dahua Camera
   description: Sets the record mode (on/off or auto). On is always on recording. Off is always off. Auto based on motion settings, etc.