refactor: support bool | list[str] for vite_allowed_hosts

Author: milochen0418

reflex/compiler/templates.py

@@ -502,7 +502,7 @@ def vite_config_template(
     force_full_reload: bool,
     experimental_hmr: bool,
     sourcemap: bool | Literal["inline", "hidden"],
-    allowed_hosts: bool = False,
+    allowed_hosts: bool | list[str] = False,
 ):
     """Template for vite.config.js.
 
@@ -512,12 +512,17 @@ def vite_config_template(
         force_full_reload: Whether to force a full reload on changes.
         experimental_hmr: Whether to enable experimental HMR features.
         sourcemap: The sourcemap configuration.
-        allowed_hosts: Whether to allow all hosts in the Vite dev server.
+        allowed_hosts: Allow all hosts (True), specific hosts (list of strings), or only localhost (False).
 
     Returns:
         Rendered vite.config.js content as string.
     """
-    allowed_hosts_line = "\n    allowedHosts: true," if allowed_hosts else ""
+    if allowed_hosts is True:
+        allowed_hosts_line = "\n    allowedHosts: true,"
+    elif isinstance(allowed_hosts, list) and allowed_hosts:
+        allowed_hosts_line = f"\n    allowedHosts: {json.dumps(allowed_hosts)},"
+    else:
+        allowed_hosts_line = ""
     return rf"""import {{ fileURLToPath, URL }} from "url";
 import {{ reactRouter }} from "@react-router/dev/vite";
 import {{ defineConfig }} from "vite";

reflex/config.py

@@ -213,8 +213,10 @@ class BaseConfig:
         dataclasses.field(default=("*",))
     )
 
-    # Whether to allow all hosts in Vite dev server (prevents 403 errors in Docker, Codespaces, etc.).
-    vite_allowed_hosts: bool = False
+    # Allowed hosts for the Vite dev server. Set to True to allow all hosts,
+    # or provide a list of hostnames (e.g. ["myservice.local"]) to allow specific ones.
+    # Prevents 403 errors in Docker, Codespaces, reverse proxies, etc.
+    vite_allowed_hosts: bool | list[str] = False
 
     # Whether to use React strict mode.
     react_strict_mode: bool = True