Ability to disable media browser

custom_components/frigate/config_flow.py

@@ -15,6 +15,7 @@
 
 from .api import FrigateApiClient, FrigateApiClientError
 from .const import (
+    CONF_MEDIA_BROWSER_ENABLE,
     CONF_NOTIFICATION_PROXY_ENABLE,
     CONF_RTMP_URL_TEMPLATE,
     DEFAULT_HOST,
@@ -147,6 +148,13 @@ async def async_step_init(
                     True,
                 ),
             ): bool,
+            vol.Optional(
+                CONF_MEDIA_BROWSER_ENABLE,
+                default=self._config_entry.options.get(
+                    CONF_MEDIA_BROWSER_ENABLE,
+                    True,
+                ),
+            ): bool,
         }
 
         return cast(

custom_components/frigate/const.py

@@ -39,6 +39,7 @@
 
 # Configuration and options
 CONF_CAMERA_STATIC_IMAGE_HEIGHT = "camera_image_height"
+CONF_MEDIA_BROWSER_ENABLE = "media_browser_enable"
 CONF_NOTIFICATION_PROXY_ENABLE = "notification_proxy_enable"
 CONF_PASSWORD = "password"
 CONF_PATH = "path"

custom_components/frigate/media_source.py

@@ -31,7 +31,7 @@
 
 from . import get_friendly_name
 from .api import FrigateApiClient, FrigateApiClientError
-from .const import ATTR_CLIENT, DOMAIN, NAME
+from .const import ATTR_CLIENT, CONF_MEDIA_BROWSER_ENABLE, DOMAIN, NAME
 from .views import (
     get_config_entry_for_frigate_instance_id,
     get_default_config_entry,
@@ -47,7 +47,12 @@
 
 async def async_get_media_source(hass: HomeAssistant) -> MediaSource:
     """Set up Frigate media source."""
-    return FrigateMediaSource(hass)
+    config_entry = hass.config_entries.async_entries(DOMAIN)[0]
+
+    if config_entry.options.get(CONF_MEDIA_BROWSER_ENABLE, True):
+        return FrigateMediaSource(hass)
+
+    return None
 
 
 class FrigateBrowseMediaMetadata:

custom_components/frigate/translations/en.json

@@ -21,7 +21,8 @@
             "init": {
                 "data": {
                     "rtmp_url_template": "RTMP URL template (see documentation)",
-                    "notification_proxy_enable": "Enable the unauthenticated notification event proxy"
+                    "notification_proxy_enable": "Enable the unauthenticated notification event proxy",
+                    "media_browser_enable": "Enable the media browser"
                 }
             }
         },

tests/test_config_flow.py

@@ -8,6 +8,7 @@
 
 from custom_components.frigate.api import FrigateApiClientError
 from custom_components.frigate.const import (
+    CONF_MEDIA_BROWSER_ENABLE,
     CONF_NOTIFICATION_PROXY_ENABLE,
     CONF_RTMP_URL_TEMPLATE,
     DOMAIN,
@@ -174,12 +175,14 @@ async def test_options_advanced(hass: HomeAssistant) -> None:
             user_input={
                 CONF_RTMP_URL_TEMPLATE: "http://moo",
                 CONF_NOTIFICATION_PROXY_ENABLE: False,
+                CONF_MEDIA_BROWSER_ENABLE: False,
             },
         )
         await hass.async_block_till_done()
         assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
         assert result["data"][CONF_RTMP_URL_TEMPLATE] == "http://moo"
         assert not result["data"][CONF_NOTIFICATION_PROXY_ENABLE]
+        assert not result["data"][CONF_MEDIA_BROWSER_ENABLE]
 
 
 async def test_options(hass: HomeAssistant) -> None:

tests/test_media_source.py

@@ -14,15 +14,22 @@
 import pytest
 
 from custom_components.frigate.api import FrigateApiClient, FrigateApiClientError
-from custom_components.frigate.const import ATTR_CLIENT_ID, ATTR_MQTT, DOMAIN
+from custom_components.frigate.const import (
+    ATTR_CLIENT_ID,
+    ATTR_MQTT,
+    CONF_MEDIA_BROWSER_ENABLE,
+    DOMAIN,
+)
 from custom_components.frigate.media_source import (
     EventIdentifier,
     EventSearchIdentifier,
     FrigateMediaType,
     Identifier,
     RecordingIdentifier,
+    async_get_media_source,
 )
 from homeassistant.components import media_source
+from homeassistant.components.media_player.errors import BrowseError
 from homeassistant.components.media_source import const
 from homeassistant.components.media_source.error import MediaSourceError, Unresolvable
 from homeassistant.components.media_source.models import PlayMedia
@@ -79,6 +86,30 @@ def load_json(filename: str) -> Any:
     yield client
 
 
+async def test_async_disabled_browse_media(hass: HomeAssistant) -> None:
+    """Test disabled browse media."""
+
+    # Create the default test Frigate instance.
+    create_mock_frigate_config_entry(
+        hass,
+        options={CONF_MEDIA_BROWSER_ENABLE: False},
+    )
+
+    loaded = False
+
+    try:
+        await async_get_media_source(hass)
+        await media_source.async_browse_media(
+            hass,
+            f"{const.URI_SCHEME}{DOMAIN}",
+        )
+        loaded = True
+    except BrowseError:
+        loaded = False
+
+    assert not loaded
+
+
 async def test_async_browse_media_root(hass: HomeAssistant) -> None:
     """Test successful browse media root."""