evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update nest to use the async WebRTC APIs (#129369)

Browse source 
* Update nest to use the new `async_handle_webrtc_offer` APIs.

* Close sessions when sessions end

* Switch to the correct close API
This commit is contained in:
Allen Porter 2024-10-28 23:18:59 -07:00 • committed by GitHub
parent 81a5722708
commit 537c95cf29
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 20 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
homeassistant/components/nest/camera.py
View file

@ -15,6 +15,7 @@ from google_nest_sdm.camera_traits import (
CameraLiveStreamTrait, CameraLiveStreamTrait,
RtspStream, RtspStream,
StreamingProtocol, StreamingProtocol,
WebRtcStream,
) )
from google_nest_sdm.device import Device from google_nest_sdm.device import Device
from google_nest_sdm.device_manager import DeviceManager from google_nest_sdm.device_manager import DeviceManager
@ -24,7 +25,9 @@ from homeassistant.components.camera import (
Camera, Camera,
CameraEntityFeature, CameraEntityFeature,
StreamType, StreamType,
WebRTCAnswer,
WebRTCClientConfiguration, WebRTCClientConfiguration,
WebRTCSendMessage,
) )
from homeassistant.components.stream import CONF_EXTRA_PART_WAIT_TIME from homeassistant.components.stream import CONF_EXTRA_PART_WAIT_TIME
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -92,6 +95,7 @@ class NestCamera(Camera):
self.stream_options[CONF_EXTRA_PART_WAIT_TIME] = 3 self.stream_options[CONF_EXTRA_PART_WAIT_TIME] = 3
# The API "name" field is a unique device identifier. # The API "name" field is a unique device identifier.
self._attr_unique_id = f"{self._device.name}-camera" self._attr_unique_id = f"{self._device.name}-camera"
self._webrtc_sessions: dict[str, WebRtcStream] = {}
@property @property
def use_stream_for_stills(self) -> bool: def use_stream_for_stills(self) -> bool:
@ -205,16 +209,29 @@ class NestCamera(Camera):
"""Return placeholder image to use when no stream is available.""" """Return placeholder image to use when no stream is available."""
return PLACEHOLDER.read_bytes() return PLACEHOLDER.read_bytes()
async def async_handle_web_rtc_offer(self, offer_sdp: str) -> str | None: async def async_handle_async_webrtc_offer(
self, offer_sdp: str, session_id: str, send_message: WebRTCSendMessage
) -> None:
"""Return the source of the stream.""" """Return the source of the stream."""
trait: CameraLiveStreamTrait = self._device.traits[CameraLiveStreamTrait.NAME] trait: CameraLiveStreamTrait = self._device.traits[CameraLiveStreamTrait.NAME]
if StreamingProtocol.WEB_RTC not in trait.supported_protocols: if StreamingProtocol.WEB_RTC not in trait.supported_protocols:
return await super().async_handle_web_rtc_offer(offer_sdp) await super().async_handle_async_webrtc_offer(
offer_sdp, session_id, send_message
)
return
try: try:
stream = await trait.generate_web_rtc_stream(offer_sdp) stream = await trait.generate_web_rtc_stream(offer_sdp)
except ApiException as err: except ApiException as err:
raise HomeAssistantError(f"Nest API error: {err}") from err raise HomeAssistantError(f"Nest API error: {err}") from err
return stream.answer_sdp self._webrtc_sessions[session_id] = stream
send_message(WebRTCAnswer(stream.answer_sdp))
@callback
def close_webrtc_session(self, session_id: str) -> None:
"""Close a WebRTC session."""
if (stream := self._webrtc_sessions.pop(session_id, None)) is not None:
self.hass.async_create_task(stream.stop_stream())
super().close_webrtc_session(session_id)
@callback @callback
def _async_get_webrtc_client_configuration(self) -> WebRTCClientConfiguration: def _async_get_webrtc_client_configuration(self) -> WebRTCClientConfiguration: