cortexctl 0.1.0.dev20260811135705__py3-none-any.whl

cortex_cli/config.py

@@ -0,0 +1,88 @@
+from __future__ import annotations
+
+import json
+import os
+import tempfile
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any
+
+DEFAULT_API_URL = "http://127.0.0.1:8000/api/v1"
+
+
+@dataclass
+class CliState:
+    api_url: str = DEFAULT_API_URL
+    access_token: str | None = None
+    refresh_token: str | None = None
+    default_workspace_id: str | None = None
+
+
+class StateStore:
+    def __init__(self, path: Path | None = None) -> None:
+        self.path = path or default_state_path()
+
+    def load(self) -> CliState:
+        if not self.path.exists():
+            return CliState(api_url=default_api_url())
+
+        data = json.loads(self.path.read_text(encoding="utf-8"))
+        return CliState(
+            api_url=str(data.get("api_url") or default_api_url()),
+            access_token=data.get("access_token"),
+            refresh_token=data.get("refresh_token"),
+            default_workspace_id=data.get("default_workspace_id"),
+        )
+
+    def save(self, state: CliState) -> None:
+        self.path.parent.mkdir(parents=True, exist_ok=True)
+        payload: dict[str, Any] = {
+            "api_url": state.api_url,
+            "access_token": state.access_token,
+            "refresh_token": state.refresh_token,
+            "default_workspace_id": state.default_workspace_id,
+        }
+        fd = -1
+        tmp_path: Path | None = None
+        try:
+            fd, tmp_name = tempfile.mkstemp(
+                prefix=f".{self.path.name}.",
+                dir=self.path.parent,
+            )
+            tmp_path = Path(tmp_name)
+            os.fchmod(fd, 0o600)
+            with os.fdopen(fd, "w", encoding="utf-8") as handle:
+                fd = -1
+                json.dump(payload, handle, indent=2, sort_keys=True)
+                handle.write("\n")
+            os.replace(tmp_path, self.path)
+            tmp_path = None
+        finally:
+            if fd != -1:
+                os.close(fd)
+            if tmp_path is not None:
+                tmp_path.unlink(missing_ok=True)
+        self.path.chmod(0o600)
+
+    def clear_token(self) -> CliState:
+        state = self.load()
+        state.access_token = None
+        state.refresh_token = None
+        self.save(state)
+        return state
+
+
+def default_api_url() -> str:
+    return os.environ.get("CORTEX_API_URL", DEFAULT_API_URL).rstrip("/")
+
+
+def default_state_path() -> Path:
+    configured = os.environ.get("CORTEX_CLI_STATE")
+    if configured:
+        return Path(configured).expanduser()
+
+    config_home = os.environ.get("XDG_CONFIG_HOME")
+    if config_home:
+        return Path(config_home).expanduser() / "cortex" / "state.json"
+
+    return Path.home() / ".config" / "cortex" / "state.json"