gemini-cli-usage 0.1.0__tar.gz → 0.1.3__tar.gz

{gemini_cli_usage-0.1.0 → gemini_cli_usage-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-cli-usage
-Version: 0.1.0
+Version: 0.1.3
 Summary: Gemini CLI quota monitor — fetches Code Assist quota data from Google's backend
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

{gemini_cli_usage-0.1.0 → gemini_cli_usage-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gemini-cli-usage"
-version = "0.1.0"
+version = "0.1.3"
 description = "Gemini CLI quota monitor — fetches Code Assist quota data from Google's backend"
 readme = "README.md"
 requires-python = ">=3.12"

{gemini_cli_usage-0.1.0 → gemini_cli_usage-0.1.3}/src/gemini_cli_usage/__init__.py

@@ -30,10 +30,56 @@ import urllib.request
 from datetime import UTC, datetime
 from pathlib import Path
 
-GEMINI_DIR = Path.home() / ".gemini"
-OAUTH_FILE = GEMINI_DIR / "oauth_creds.json"
-SETTINGS_FILE = GEMINI_DIR / "settings.json"
-DEFAULT_USAGE_FILE = GEMINI_DIR / "usage-limits.json"
+_NATIVE_GEMINI_DIR = Path.home() / ".gemini"
+
+
+def _wsl_gemini_dirs() -> list[Path]:
+    """Return candidate .gemini dirs from WSL distros (Windows only)."""
+    if sys.platform != "win32":
+        return []
+    import subprocess
+
+    try:
+        result = subprocess.run(
+            ["wsl", "-l", "-q"],
+            capture_output=True,
+            timeout=5,
+        )
+        raw = result.stdout.decode("utf-16-le", errors="replace")
+        distros = [d.strip() for d in raw.splitlines() if d.strip()]
+    except Exception:
+        return []
+
+    dirs: list[Path] = []
+    for distro in distros:
+        wsl_home = Path(f"//wsl$/{distro}/home")
+        try:
+            if wsl_home.is_dir():
+                for user_dir in wsl_home.iterdir():
+                    candidate = user_dir / ".gemini"
+                    if candidate.is_dir():
+                        dirs.append(candidate)
+        except OSError:
+            continue
+    return dirs
+
+
+def _find_gemini_file(filename: str) -> Path:
+    """Return the first existing .gemini/<filename> from native + WSL paths."""
+    native = _NATIVE_GEMINI_DIR / filename
+    if native.exists():
+        return native
+    for wsl_dir in _wsl_gemini_dirs():
+        candidate = wsl_dir / filename
+        if candidate.exists():
+            return candidate
+    return native  # default even if missing
+
+
+GEMINI_DIR = _NATIVE_GEMINI_DIR
+OAUTH_FILE = _find_gemini_file("oauth_creds.json")
+SETTINGS_FILE = _find_gemini_file("settings.json")
+DEFAULT_USAGE_FILE = _find_gemini_file("usage-limits.json")
 
 DAEMON_INTERVAL = 300  # 5 minutes
 CACHE_MAX_AGE = 300
@@ -53,6 +99,14 @@ AUTH_LABELS = {
 OAUTH_CLIENT_ID_PATTERN = re.compile(r"const OAUTH_CLIENT_ID = '([^']+)';")
 OAUTH_CLIENT_SECRET_PATTERN = re.compile(r"const OAUTH_CLIENT_SECRET = '([^']+)';")
 
+# ANSI color codes — disabled when stdout is not a terminal (e.g. piped or redirected).
+_TTY = sys.stdout.isatty()
+_RED = "\033[0;31m" if _TTY else ""
+_YELLOW = "\033[0;33m" if _TTY else ""
+_GREEN = "\033[0;32m" if _TTY else ""
+_DIM = "\033[0;90m" if _TTY else ""
+_RESET = "\033[0m" if _TTY else ""
+
 
 def _read_json(path: Path) -> dict | list | None:
     try:
@@ -409,12 +463,8 @@ def _color_pct(pct: float | int | None) -> str:
     if pct is None:
         return "?"
     p = float(pct)
-    red = "\033[0;31m"
-    yellow = "\033[0;33m"
-    green = "\033[0;32m"
-    reset = "\033[0m"
-    color = red if p >= 70 else yellow if p >= 40 else green
-    return f"{color}{_format_pct(p)}{reset}"
+    color = _RED if p >= 70 else _YELLOW if p >= 40 else _GREEN
+    return f"{color}{_format_pct(p)}{_RESET}"
 
 
 def _format_pct(pct: float | int | None) -> str:
@@ -427,9 +477,6 @@ def _format_pct(pct: float | int | None) -> str:
 
 
 def _print_status(data: dict):
-    dim = "\033[0;90m"
-    reset = "\033[0m"
-
     print(f"Project: {Path(data['project_root']).name}")
     print(f"Auth: {data.get('auth_label', 'unknown')}")
 
@@ -454,14 +501,14 @@ def _print_status(data: dict):
             )
             print(
                 f"  {label:{name_width}s} {_color_pct(bucket.get('used_pct'))} used"
-                f"{remain_part}{dim}{reset_part}{reset}"
+                f"{remain_part}{_DIM}{reset_part}{_RESET}"
             )
 
         for bucket in quota.get("buckets", []):
             model = bucket.get("model") or "unknown"
             print_bucket_line(model, bucket)
     elif data.get("quota_error"):
-        print(f"  {'Quota':20s} {dim}{data['quota_error']}{reset}")
+        print(f"  {'Quota':20s} {_DIM}{data['quota_error']}{_RESET}")
 
 
 def _statusline_text(data: dict) -> str:
@@ -520,7 +567,8 @@ def cmd_json(args):
 
 def cmd_daemon(args):
     signal.signal(signal.SIGINT, lambda *_: sys.exit(0))
-    signal.signal(signal.SIGTERM, lambda *_: sys.exit(0))
+    if hasattr(signal, "SIGTERM"):
+        signal.signal(signal.SIGTERM, lambda *_: sys.exit(0))
 
     root = Path(args.root).resolve() if args.root else None
     usage_file = get_usage_file()