Initial Home Assistant commit

2026-06-05 22:34:31 -04:00
commit 6a58b10e6c
4494 changed files with 297833 additions and 0 deletions
@@ -0,0 +1,106 @@
+import asyncio
+import os.path
+import platform
+
+import homeassistant.helpers.config_validation as cv
+import voluptuous as vol
+import yaml
+from homeassistant.config_entries import ConfigFlow
+from homeassistant.const import CONF_URL, CONF_USERNAME, CONF_PASSWORD
+
+from . import DOMAIN, utils
+
+
+class FlowHandler(ConfigFlow, domain=DOMAIN):
+    async def async_step_user(self, user_input=None):
+        # check if only one integration instance
+        if self._async_current_entries():
+            return self.async_abort(reason="single_instance_allowed")
+
+        # check if we support this arch
+        if not utils.get_arch():
+            return self.async_abort(
+                reason="arch",
+                description_placeholders={"uname": str(platform.uname())},
+            )
+
+        if user_input is not None:
+            url = user_input.get(CONF_URL)
+            if not url:
+                # create config file first time
+                config = self.hass.config.path("go2rtc.yaml")
+                if not os.path.isfile(config):
+                    return await self.async_step_config(None)
+
+            # check if go2rtc url from user input available
+            elif not await utils.check_go2rtc(self.hass, url):
+                return self.async_show_form(
+                    step_id="user",
+                    data_schema=vol.Schema(
+                        {
+                            vol.Optional(CONF_URL, default=url): cv.string,
+                        }
+                    ),
+                    errors={"base": "connect"},
+                )
+
+            return self.async_create_entry(title="WebRTC Camera", data=user_input)
+
+        # check if go2rtc already exists on same server
+        tests = await asyncio.gather(
+            utils.check_go2rtc(self.hass),
+            # if go2rtc inside frigate addon with closed public port
+            utils.check_go2rtc(self.hass, "http://ccab4aaf-frigate:1984"),
+            utils.check_go2rtc(self.hass, "http://ccab4aaf-frigate-fa:1984"),
+            utils.check_go2rtc(self.hass, "http://ccab4aaf-frigate-beta:1984"),
+        )
+
+        url = next((url for url in tests if url), vol.UNDEFINED)
+
+        return self.async_show_form(
+            step_id="user",
+            data_schema=vol.Schema(
+                {
+                    vol.Optional(CONF_URL, default=url): cv.string,
+                }
+            ),
+        )
+
+    async def async_step_config(self, user_input: dict = None):
+        path = self.hass.config.path("go2rtc.yaml")
+
+        if user_input:
+            config = {}
+
+            if not user_input["api"]:
+                config.setdefault("api", {}).setdefault("listen", "127.0.0.1:1984")
+
+            if not user_input["rtsp"]:
+                config.setdefault("rtsp", {}).setdefault("listen", "127.0.0.1:8554")
+
+            if user := user_input.get(CONF_USERNAME):
+                config.setdefault("api", {}).setdefault("username", user)
+                config.setdefault("rtsp", {}).setdefault("username", user)
+
+            if pasw := user_input.get(CONF_PASSWORD):
+                config.setdefault("api", {}).setdefault("password", pasw)
+                config.setdefault("rtsp", {}).setdefault("password", pasw)
+
+            if config:
+                with open(path, "w") as f:
+                    yaml.dump(config, f)
+
+            return self.async_create_entry(title="WebRTC Camera", data={})
+
+        return self.async_show_form(
+            step_id="config",
+            data_schema=vol.Schema(
+                {
+                    vol.Required("api", default=True): cv.boolean,
+                    vol.Required("rtsp", default=True): cv.boolean,
+                    vol.Optional(CONF_USERNAME): cv.string,
+                    vol.Optional(CONF_PASSWORD): cv.string,
+                }
+            ),
+            description_placeholders={"path": path},
+        )