Add force param to DashCast service #419

AlexxIT · Aug 24, 2023 · 1037c2a · 1037c2a
1 parent 6777802
commit 1037c2a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/custom_components/webrtc/__init__.py b/custom_components/webrtc/__init__.py
@@ -44,6 +44,7 @@
         vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
         vol.Exclusive("url", "url"): cv.string,
         vol.Exclusive("entity", "url"): cv.entity_id,
+        vol.Optional("force", default=False): bool,
         vol.Optional("extra"): dict,
     },
     required=True,
@@ -106,6 +107,7 @@ async def dash_cast(call: ServiceCallType):
             hass,
             call.data[ATTR_ENTITY_ID],
             f"{get_url(hass)}/webrtc/embed?" + urlencode(query),
+            call.data.get("force"),
         )
 
     hass.services.async_register(DOMAIN, "create_link", create_link, CREATE_LINK_SCHEMA)

diff --git a/custom_components/webrtc/services.yaml b/custom_components/webrtc/services.yaml
@@ -61,6 +61,11 @@ dash_cast:
       selector:
         text:
 
+    force:
+      description: Force restart DashCast application
+      selector:
+        boolean:
+
     entity:
       description: Camera entity
       example: camera.generic_stream

diff --git a/custom_components/webrtc/utils.py b/custom_components/webrtc/utils.py
@@ -169,7 +169,7 @@ async def init_resource(hass: HomeAssistant, url: str, ver: str) -> bool:
 
 
 # noinspection PyProtectedMember
-def dash_cast(hass: HomeAssistant, cast_entities: list, url: str):
+def dash_cast(hass: HomeAssistant, cast_entities: list, url: str, force=False):
     """Cast webpage to chromecast device via DashCast application."""
     try:
         entities = [
@@ -188,7 +188,7 @@ def dash_cast(hass: HomeAssistant, cast_entities: list, url: str):
                 entity._chromecast.register_handler(entity.dashcast)
 
             _LOGGER.debug(f"DashCast to {entity.entity_id}")
-            entity.dashcast.load_url(url)
+            entity.dashcast.load_url(url, force)
 
     except Exception:
         _LOGGER.exception(f"Can't DashCast to {cast_entities}")