Add support for stream orientation in go2rtc (home-assistant#148832)

Author: edenhaus

homeassistant/components/camera/__init__.py

@@ -81,7 +81,11 @@
 )
 from .helper import get_camera_from_entity_id
 from .img_util import scale_jpeg_camera_image
-from .prefs import CameraPreferences, DynamicStreamSettings  # noqa: F401
+from .prefs import (
+    CameraPreferences,
+    DynamicStreamSettings,  # noqa: F401
+    get_dynamic_camera_stream_settings,
+)
 from .webrtc import (
     DATA_ICE_SERVERS,
     CameraWebRTCProvider,
@@ -550,9 +554,9 @@ async def async_create_stream(self) -> Stream | None:
                     self.hass,
                     source,
                     options=self.stream_options,
-                    dynamic_stream_settings=await self.hass.data[
-                        DATA_CAMERA_PREFS
-                    ].get_dynamic_stream_settings(self.entity_id),
+                    dynamic_stream_settings=await get_dynamic_camera_stream_settings(
+                        self.hass, self.entity_id
+                    ),
                     stream_label=self.entity_id,
                 )
                 self.stream.set_update_callback(self.async_write_ha_state)
@@ -942,9 +946,7 @@ async def websocket_get_prefs(
     hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]
 ) -> None:
     """Handle request for account info."""
-    stream_prefs = await hass.data[DATA_CAMERA_PREFS].get_dynamic_stream_settings(
-        msg["entity_id"]
-    )
+    stream_prefs = await get_dynamic_camera_stream_settings(hass, msg["entity_id"])
     connection.send_result(msg["id"], asdict(stream_prefs))
 
 

homeassistant/components/camera/prefs.py

@@ -13,7 +13,7 @@
 from homeassistant.helpers.storage import Store
 from homeassistant.helpers.typing import UNDEFINED, UndefinedType
 
-from .const import DOMAIN, PREF_ORIENTATION, PREF_PRELOAD_STREAM
+from .const import DATA_CAMERA_PREFS, DOMAIN, PREF_ORIENTATION, PREF_PRELOAD_STREAM
 
 STORAGE_KEY: Final = DOMAIN
 STORAGE_VERSION: Final = 1
@@ -106,3 +106,12 @@ async def get_dynamic_stream_settings(
         )
         self._dynamic_stream_settings_by_entity_id[entity_id] = settings
         return settings
+
+
+async def get_dynamic_camera_stream_settings(
+    hass: HomeAssistant, entity_id: str
+) -> DynamicStreamSettings:
+    """Get dynamic stream settings for a camera entity."""
+    if DATA_CAMERA_PREFS not in hass.data:
+        raise HomeAssistantError("Camera integration not set up")
+    return await hass.data[DATA_CAMERA_PREFS].get_dynamic_stream_settings(entity_id)

homeassistant/components/go2rtc/__init__.py

@@ -31,7 +31,9 @@
     WebRTCSendMessage,
     async_register_webrtc_provider,
 )
+from homeassistant.components.camera.prefs import get_dynamic_camera_stream_settings
 from homeassistant.components.default_config import DOMAIN as DEFAULT_CONFIG_DOMAIN
+from homeassistant.components.stream import Orientation
 from homeassistant.config_entries import SOURCE_SYSTEM, ConfigEntry
 from homeassistant.const import CONF_URL, EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import Event, HomeAssistant, callback
@@ -57,12 +59,13 @@
 
 _LOGGER = logging.getLogger(__name__)
 
+_FFMPEG = "ffmpeg"
 _SUPPORTED_STREAMS = frozenset(
     (
         "bubble",
         "dvrip",
         "expr",
-        "ffmpeg",
+        _FFMPEG,
         "gopro",
         "homekit",
         "http",
@@ -315,6 +318,32 @@ async def _update_stream_source(self, camera: Camera) -> None:
             await self.teardown()
             raise HomeAssistantError("Stream source is not supported by go2rtc")
 
+        camera_prefs = await get_dynamic_camera_stream_settings(
+            self._hass, camera.entity_id
+        )
+        if camera_prefs.orientation is not Orientation.NO_TRANSFORM:
+            # Camera orientation manually set by user
+            if not stream_source.startswith(_FFMPEG):
+                stream_source = _FFMPEG + ":" + stream_source
+            stream_source += "#video=h264#audio=copy"
+            match camera_prefs.orientation:
+                case Orientation.MIRROR:
+                    stream_source += "#raw=-vf hflip"
+                case Orientation.ROTATE_180:
+                    stream_source += "#rotate=180"
+                case Orientation.FLIP:
+                    stream_source += "#raw=-vf vflip"
+                case Orientation.ROTATE_LEFT_AND_FLIP:
+                    # Cannot use any filter when using raw one
+                    stream_source += "#raw=-vf transpose=2,vflip"
+                case Orientation.ROTATE_LEFT:
+                    stream_source += "#rotate=-90"
+                case Orientation.ROTATE_RIGHT_AND_FLIP:
+                    # Cannot use any filter when using raw one
+                    stream_source += "#raw=-vf transpose=1,vflip"
+                case Orientation.ROTATE_RIGHT:
+                    stream_source += "#rotate=90"
+
         streams = await self._rest_client.streams.list()
 
         if (stream := streams.get(camera.entity_id)) is None or not any(

tests/components/camera/test_prefs.py

@@ -0,0 +1,76 @@
+"""Test camera helper functions."""
+
+import pytest
+
+from homeassistant.components.camera.const import DATA_CAMERA_PREFS
+from homeassistant.components.camera.prefs import (
+    CameraPreferences,
+    DynamicStreamSettings,
+    get_dynamic_camera_stream_settings,
+)
+from homeassistant.components.stream import Orientation
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+
+
+async def test_get_dynamic_camera_stream_settings_missing_prefs(
+    hass: HomeAssistant,
+) -> None:
+    """Test get_dynamic_camera_stream_settings when camera prefs are not set up."""
+    with pytest.raises(HomeAssistantError, match="Camera integration not set up"):
+        await get_dynamic_camera_stream_settings(hass, "camera.test")
+
+
+async def test_get_dynamic_camera_stream_settings_success(hass: HomeAssistant) -> None:
+    """Test successful retrieval of dynamic camera stream settings."""
+    # Set up camera preferences
+    prefs = CameraPreferences(hass)
+    await prefs.async_load()
+    hass.data[DATA_CAMERA_PREFS] = prefs
+
+    # Test with default settings
+    settings = await get_dynamic_camera_stream_settings(hass, "camera.test")
+    assert settings.orientation == Orientation.NO_TRANSFORM
+    assert settings.preload_stream is False
+
+
+async def test_get_dynamic_camera_stream_settings_with_custom_orientation(
+    hass: HomeAssistant,
+) -> None:
+    """Test get_dynamic_camera_stream_settings with custom orientation set."""
+    # Set up camera preferences
+    prefs = CameraPreferences(hass)
+    await prefs.async_load()
+    hass.data[DATA_CAMERA_PREFS] = prefs
+
+    # Set custom orientation - this requires entity registry
+    # For this test, we'll directly manipulate the internal state
+    # since entity registry setup is complex for a unit test
+    test_settings = DynamicStreamSettings(
+        orientation=Orientation.ROTATE_LEFT, preload_stream=False
+    )
+    prefs._dynamic_stream_settings_by_entity_id["camera.test"] = test_settings
+
+    settings = await get_dynamic_camera_stream_settings(hass, "camera.test")
+    assert settings.orientation == Orientation.ROTATE_LEFT
+    assert settings.preload_stream is False
+
+
+async def test_get_dynamic_camera_stream_settings_with_preload_stream(
+    hass: HomeAssistant,
+) -> None:
+    """Test get_dynamic_camera_stream_settings with preload stream enabled."""
+    # Set up camera preferences
+    prefs = CameraPreferences(hass)
+    await prefs.async_load()
+    hass.data[DATA_CAMERA_PREFS] = prefs
+
+    # Set preload stream by directly setting the dynamic stream settings
+    test_settings = DynamicStreamSettings(
+        orientation=Orientation.NO_TRANSFORM, preload_stream=True
+    )
+    prefs._dynamic_stream_settings_by_entity_id["camera.test"] = test_settings
+
+    settings = await get_dynamic_camera_stream_settings(hass, "camera.test")
+    assert settings.orientation == Orientation.NO_TRANSFORM
+    assert settings.preload_stream is True