dh-cli 0.4.0__tar.gz → 0.4.2__tar.gz

{dh_cli-0.4.0 → dh_cli-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dh-cli
-Version: 0.4.0
+Version: 0.4.2
 Summary: Dayhoff Labs developer CLI
 Author-email: Dayhoff Labs <dev@dayhofflabs.com>
 License: # PolyForm Noncommercial License 1.0.0

{dh_cli-0.4.0 → dh_cli-0.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dh-cli"
-version = "0.4.0"
+version = "0.4.2"
 description = "Dayhoff Labs developer CLI"
 requires-python = ">=3.11"
 readme = "README.md"

{dh_cli-0.4.0 → dh_cli-0.4.2}/src/dh_cli/hz/__init__.py

@@ -10,6 +10,48 @@ hz_app = typer.Typer(
     context_settings={"help_option_names": ["-h", "--help"]},
 )
 
+API_URLS = {
+    "prod": "https://api.horizyn.dayhofflabs.com",
+    "dev": "https://horizyn-api-dev.dayhofflabs.com",
+}
+
+
+@hz_app.command()
+def health(
+    detailed: bool = typer.Option(False, "-d", "--detailed", help="Include GPU and EFS details."),
+):
+    """Check API health for both prod and dev environments."""
+    import json
+    from concurrent.futures import ThreadPoolExecutor
+    from urllib.request import urlopen
+    from urllib.error import URLError
+
+    path = "/healthz/detailed" if detailed else "/healthz"
+
+    def _fetch(env: str) -> tuple[str, dict | str]:
+        try:
+            with urlopen(f"{API_URLS[env]}{path}", timeout=10) as resp:
+                return env, json.loads(resp.read())
+        except (URLError, OSError) as exc:
+            return env, f"unreachable ({exc})"
+        except json.JSONDecodeError:
+            return env, "invalid response"
+
+    with ThreadPoolExecutor(max_workers=2) as pool:
+        results = dict(pool.map(_fetch, API_URLS))
+
+    for env in ("prod", "dev"):
+        label = f"[{env}] {API_URLS[env]}"
+        data = results[env]
+        if isinstance(data, str):
+            typer.secho(f"{label}  ✗ {data}", fg=typer.colors.RED)
+        else:
+            status = data.get("status", "unknown")
+            version = data.get("version", "?")
+            color = typer.colors.GREEN if status == "healthy" else typer.colors.YELLOW
+            typer.secho(f"{label}  v{version}  {status}", fg=color)
+            typer.echo(json.dumps(data, indent=2))
+
 
 def _find_workspace_root() -> Path:
     root = os.environ.get("WORKSPACE_ROOT")

{dh_cli-0.4.0 → dh_cli-0.4.2}/src/dh_cli/hz/users.py

@@ -55,11 +55,19 @@ def list_users(
 
     typer.echo(f"\n  Horizyn users ({env})\n")
 
-    typer.echo("  Admin domains:   " + (", ".join(sorted(admin_domains)) or "(none)"))
-    typer.echo("  Admin emails:    " + (", ".join(sorted(admin_emails)) or "(none)"))
-    typer.echo("  Alpha emails:    " + (", ".join(sorted(alpha_emails)) or "(none)"))
-    typer.echo("  Allowed domains: " + (", ".join(sorted(allowed_domains)) or "(none)"))
-    typer.echo("  Allowed emails:  " + (", ".join(sorted(allowed_emails)) or "(none)"))
+    def _print_section(label: str, items: set[str]) -> None:
+        typer.echo(f"  {label}:")
+        if not items:
+            typer.echo("    (none)")
+        else:
+            for item in sorted(items):
+                typer.echo(f"    {item}")
+
+    _print_section("Admin domains", admin_domains)
+    _print_section("Admin emails", admin_emails)
+    _print_section("Alpha emails", alpha_emails)
+    _print_section("Allowed domains", allowed_domains)
+    _print_section("Allowed emails", allowed_emails)
     typer.echo()
 
 
@@ -133,3 +141,39 @@ def promote_user(
     else:
         alphas.add(email)
         _put_param(ssm, SSM_PARAMS["alpha_emails"].format(env=env), _format_csv(alphas))
+
+
+@users_app.command("find")
+def find_user(
+    email: str = typer.Argument(help="Email address to search for."),
+):
+    """Search for a user across both prod and dev environments."""
+    email = email.strip().lower()
+    ssm = _ssm_client()
+
+    domain = email.split("@")[-1] if "@" in email else ""
+
+    typer.echo(f"\n  Searching for: {email}\n")
+
+    for env in ("prod", "dev"):
+        matches: list[str] = []
+
+        for param_key in SSM_PARAMS:
+            param_name = SSM_PARAMS[param_key].format(env=env)
+            values = _parse_csv(_get_param(ssm, param_name))
+
+            if param_key.endswith("_domains"):
+                if domain and domain in values:
+                    matches.append(f"{param_key} (via @{domain})")
+            else:
+                if email in values:
+                    matches.append(param_key)
+
+        if matches:
+            typer.echo(f"  {env}: found")
+            for m in matches:
+                typer.echo(f"    - {m}")
+        else:
+            typer.echo(f"  {env}: not found")
+
+    typer.echo()