slab-cli 0.1.0__py3-none-any.whl

slab_cli/context.py

@@ -0,0 +1,50 @@
+"""Shared runtime context for CLI commands.
+
+Created once per invocation in main(). Commands receive it instead of
+calling _client()/_require_collector() themselves."""
+
+from __future__ import annotations
+
+import sys
+
+from .client import SlabClient
+from .config import get_api_key, get_base_url, get_collector_uuid
+from .display import console
+
+
+class MissingCollector(Exception):
+    """Raised when a command needs a collector but SLAB_COLLECTOR is not set."""
+
+
+class CliContext:
+    """Holds the API client and resolved collector UUID for a CLI session."""
+
+    def __init__(self) -> None:
+        self._client: SlabClient | None = None
+        self._collector: str | None = None
+
+    @property
+    def client(self) -> SlabClient:
+        if self._client is None:
+            self._client = SlabClient(get_base_url(), get_api_key())
+        return self._client
+
+    @property
+    def collector(self) -> str:
+        """The active collector UUID. Resolution order: SLAB_COLLECTOR env var, then the account's
+        default collector (so a solo user only needs their API key). Raises MissingCollector if
+        neither yields one. Cached per invocation."""
+        if self._collector:
+            return self._collector
+        uuid = get_collector_uuid()
+        if not uuid:
+            # Solo zero-config: fall back to the calling account's default collector.
+            uuid = self.client.get_account_context().default_collector_uuid
+        if not uuid:
+            raise MissingCollector
+        self._collector = uuid
+        return uuid
+
+
+# Module-level singleton — set by main() before commands run.
+ctx = CliContext()