Pass CLI parameters to UI (#1530)

Author: vchan

sqlmesh/cli/main.py

@@ -445,19 +445,21 @@ def ide(
 @click.option(
     "--host",
     type=str,
+    default="127.0.0.1",
     help="Bind socket to this host. Default: 127.0.0.1",
 )
 @click.option(
     "--port",
     type=int,
+    default=8000,
     help="Bind socket to this port. Default: 8000",
 )
-@click.pass_obj
+@click.pass_context
 @error_handler
 def ui(
-    obj: Context,
-    host: t.Optional[str],
-    port: t.Optional[int],
+    ctx: click.Context,
+    host: str,
+    port: int,
 ) -> None:
     """Start a browser-based SQLMesh UI."""
     try:
@@ -467,9 +469,10 @@ def ui(
             "Missing UI dependencies. Run `pip install 'sqlmesh[web]'` to install them."
         ) from e
 
-    host = host or "127.0.0.1"
-    port = 8000 if port is None else port
-    os.environ["PROJECT_PATH"] = str(obj.path)
+    os.environ["PROJECT_PATH"] = str(ctx.obj.path)
+    if ctx.parent:
+        os.environ["CONFIG"] = ctx.parent.params.get("config") or ""
+        os.environ["GATEWAY"] = ctx.parent.params.get("gateway") or ""
     uvicorn.run(
         "web.server.main:app",
         host=host,

web/server/settings.py

@@ -26,6 +26,7 @@
 class Settings(BaseSettings):
     project_path: Path = Path("examples/sushi")
     config: str = ""
+    gateway: str = ""
 
 
 @lru_cache()
@@ -34,10 +35,10 @@ def get_settings() -> Settings:
 
 
 @lru_cache()
-def _get_context(path: str | Path, config: str) -> Context:
+def _get_context(path: str | Path, config: str, gateway: str) -> Context:
     from web.server.main import api_console
 
-    return Context(paths=str(path), config=config, console=api_console, load=False)
+    return Context(paths=str(path), config=config, console=api_console, gateway=gateway, load=False)
 
 
 @lru_cache()
@@ -56,8 +57,9 @@ def _get_path_mappings(context: Context) -> dict[Path, FileType]:
 
 
 @lru_cache()
-def _get_loaded_context(path: str | Path, config: str) -> Context:
-    context = _get_context(path, config)
+def _get_loaded_context(path: str | Path, config: str, gateway: str) -> Context:
+    print(config)
+    context = _get_context(path, config, gateway)
     context.load()
     return context
 
@@ -93,7 +95,11 @@ async def get_loaded_context(settings: Settings = Depends(get_settings)) -> Cont
     try:
         async with get_context_lock:
             return await loop.run_in_executor(
-                None, _get_loaded_context, settings.project_path, settings.config
+                None,
+                _get_loaded_context,
+                settings.project_path,
+                settings.config,
+                settings.gateway,
             )
     except Exception:
         raise ApiException(
@@ -105,7 +111,7 @@ async def get_loaded_context(settings: Settings = Depends(get_settings)) -> Cont
 async def get_context(settings: Settings = Depends(get_settings)) -> Context:
     try:
         async with get_context_lock:
-            return _get_context(settings.project_path, settings.config)
+            return _get_context(settings.project_path, settings.config, settings.gateway)
     except Exception:
         raise ApiException(
             message="Unable to create a context",