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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-cli-usage
-Version: 0.1.0
+Version: 0.1.4
 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.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gemini-cli-usage"
-version = "0.1.0"
+version = "0.1.4"
 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.4}/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:
@@ -136,14 +190,56 @@ def _write_oauth_credentials(creds: dict):
 
 
 def _get_gemini_cli_oauth2_path() -> Path | None:
+    oauth2_rels = [
+        Path("node_modules") / "@google" / "gemini-cli-core" / "dist" / "src" / "code_assist" / "oauth2.js",
+        Path("node_modules") / "@google" / "gemini-cli" / "node_modules" / "@google" / "gemini-cli-core" / "dist" / "src" / "code_assist" / "oauth2.js",
+    ]
+
+    candidates: list[Path] = []
+
+    # 1. Try shutil.which
     gemini_bin = shutil.which("gemini")
-    if not gemini_bin:
-        return None
+    if gemini_bin:
+        resolved = Path(gemini_bin).resolve()
+        candidates.extend([resolved.parent, resolved.parent.parent])
+
+    # 2. On Windows, scan fnm node version directories
+    if sys.platform == "win32":
+        fnm_dir = Path.home() / "AppData" / "Roaming" / "fnm" / "node-versions"
+        if fnm_dir.exists():
+            for version_dir in fnm_dir.iterdir():
+                candidates.append(version_dir / "installation")
+
+        # Also check AppData/Local fnm multishells
+        fnm_multi = Path.home() / "AppData" / "Local" / "fnm_multishells"
+        if fnm_multi.exists():
+            for shell_dir in fnm_multi.iterdir():
+                candidates.append(shell_dir)
+
+    # 3. Unix: check common global npm/nvm paths
+    else:
+        for p in [
+            Path("/usr/lib"),
+            Path("/usr/local/lib"),
+            Path.home() / ".nvm" / "versions",
+        ]:
+            if p.exists():
+                if "nvm" in str(p):
+                    for v in p.rglob("node"):
+                        candidates.append(v / "lib")
+                else:
+                    candidates.append(p)
+
+    for root in candidates:
+        for oauth2_rel in oauth2_rels:
+            oauth2_path = root / oauth2_rel
+            if oauth2_path.exists():
+                return oauth2_path
+            oauth2_path = root / "lib" / oauth2_rel
+            if oauth2_path.exists():
+                return oauth2_path
 
-    resolved = Path(gemini_bin).resolve()
-    package_root = resolved.parent.parent
-    oauth2_path = package_root / "node_modules" / "@google" / "gemini-cli-core" / "dist" / "src" / "code_assist" / "oauth2.js"
-    return oauth2_path if oauth2_path.exists() else None
+    return None
 
 
 def _get_gemini_cli_oauth_client_credentials() -> tuple[str, str] | None:
@@ -409,12 +505,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 +519,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 +543,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 +609,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()