codex-cli-usage 0.1.8__tar.gz → 0.1.10__tar.gz

codex_cli_usage-0.1.10/.github/workflows/publish.yml

@@ -0,0 +1,29 @@
+name: Publish to PyPI
+
+on:
+  push:
+    tags:
+      - "v*"
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      contents: read
+      id-token: write
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v7.0.1
+      - name: Install uv
+        uses: astral-sh/setup-uv@v9.0.0
+        with:
+          enable-cache: false
+      - name: Verify version tag
+        run: test "$GITHUB_REF_NAME" = "v$(uv --no-config version --short)"
+      - name: Test
+        run: uv --no-config run --locked python -m unittest discover -s tests
+      - name: Build
+        run: uv --no-config build --no-sources
+      - name: Publish
+        run: uv publish --trusted-publishing always

{codex_cli_usage-0.1.8 → codex_cli_usage-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: codex-cli-usage
-Version: 0.1.8
+Version: 0.1.10
 Summary: Codex CLI usage monitor — fetches rate limits through the Codex app-server
 Project-URL: Homepage, https://github.com/wakamex/codex-cli-usage
 Project-URL: Source, https://github.com/wakamex/codex-cli-usage
@@ -100,7 +100,10 @@ window actually classifies as 5-hour or weekly. Readers should prefer
 
 ### Authentication
 
-The OAuth tokens live at `~/.codex/auth.json`, written by the Codex CLI on login.
+The OAuth tokens live at `~/.codex/auth.json` by default and are written by the
+Codex CLI on login. When `CODEX_HOME` is set, the direct compatibility fallback
+reads `$CODEX_HOME/auth.json`. The preferred app-server path leaves credential
+resolution to Codex.
 
 Codex owns automatic token refresh and credential persistence. The compatibility
 fallback reads `auth.json` but never modifies credentials.

{codex_cli_usage-0.1.8 → codex_cli_usage-0.1.10}/README.md

@@ -91,7 +91,10 @@ window actually classifies as 5-hour or weekly. Readers should prefer
 
 ### Authentication
 
-The OAuth tokens live at `~/.codex/auth.json`, written by the Codex CLI on login.
+The OAuth tokens live at `~/.codex/auth.json` by default and are written by the
+Codex CLI on login. When `CODEX_HOME` is set, the direct compatibility fallback
+reads `$CODEX_HOME/auth.json`. The preferred app-server path leaves credential
+resolution to Codex.
 
 Codex owns automatic token refresh and credential persistence. The compatibility
 fallback reads `auth.json` but never modifies credentials.

{codex_cli_usage-0.1.8 → codex_cli_usage-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codex-cli-usage"
-version = "0.1.8"
+version = "0.1.10"
 description = "Codex CLI usage monitor — fetches rate limits through the Codex app-server"
 readme = "README.md"
 requires-python = ">=3.12"

{codex_cli_usage-0.1.8 → codex_cli_usage-0.1.10}/src/codex_cli_usage/__init__.py

@@ -16,6 +16,8 @@ Usage:
 
 import argparse
 import json
+import math
+import os
 import queue
 import shutil
 import signal
@@ -66,8 +68,15 @@ def _find_codex_path(filename: str) -> Path:
     return native
 
 
+def _find_codex_auth_path() -> Path:
+    codex_home = os.environ.get("CODEX_HOME")
+    if codex_home:
+        return Path(codex_home) / "auth.json"
+    return _find_codex_path("auth.json")
+
+
 CODEX_DIR = Path.home() / ".codex"
-AUTH_FILE = _find_codex_path("auth.json")
+AUTH_FILE = _find_codex_auth_path()
 USAGE_FILE = _find_codex_path("usage-limits.json")
 DAEMON_INTERVAL = 300  # 5 minutes
 USAGE_URL = "https://chatgpt.com/backend-api/codex/usage"
@@ -323,7 +332,7 @@ def _fetch_usage_direct() -> dict:
     """
     auth = get_auth()
     if not auth:
-        raise RuntimeError("No auth at ~/.codex/auth.json — run `codex` first")
+        raise RuntimeError(f"No auth at {AUTH_FILE}. Run `codex` first")
 
     tokens = auth.get("tokens", {})
     access_token = tokens.get("access_token")
@@ -381,13 +390,36 @@ def classify_window(seconds: int | float | None, kind: str) -> tuple[str, str]:
 
 
 def _build_window(kind: str, window: dict) -> dict:
+    used_percent = window.get("used_percent")
     seconds = window.get("limit_window_seconds")
-    key, label = classify_window(seconds, kind)
     reset_at = window.get("reset_at")
+    for value in (used_percent, seconds, reset_at):
+        if value is not None and (
+            not isinstance(value, int | float)
+            or isinstance(value, bool)
+            or (isinstance(value, float) and not math.isfinite(value))
+        ):
+            raise UsageServiceError("Invalid rate limit response")
+    if seconds is not None and seconds <= 0:
+        raise UsageServiceError("Invalid rate limit response")
+
+    try:
+        key, label = classify_window(seconds, kind)
+    except OverflowError:
+        raise UsageServiceError("Invalid rate limit response") from None
+    try:
+        resets_at = (
+            datetime.fromtimestamp(reset_at, tz=timezone.utc).isoformat()
+            if reset_at is not None
+            else None
+        )
+    except (OSError, OverflowError, ValueError):
+        raise UsageServiceError("Invalid rate limit response") from None
+
     return {
         "kind": kind,
-        "pct": window.get("used_percent"),
-        "resets_at": datetime.fromtimestamp(reset_at, tz=timezone.utc).isoformat() if reset_at else None,
+        "pct": used_percent,
+        "resets_at": resets_at,
         "window_secs": seconds,
         "key": key,
         "label": label,
@@ -442,11 +474,13 @@ def build_usage_json(api_data: dict) -> dict:
     # Code review limits
     cr = api_data.get("code_review_rate_limit") or {}
     cr_primary = cr.get("primary_window")
-    if cr_primary and cr_primary.get("used_percent", 0) > 0:
-        result["code_review"] = {
-            "pct": cr_primary["used_percent"],
-            "resets_at": datetime.fromtimestamp(cr_primary["reset_at"], tz=timezone.utc).isoformat() if cr_primary.get("reset_at") else None,
-        }
+    if isinstance(cr_primary, dict):
+        window = _build_window("code_review", cr_primary)
+        if window["pct"] is not None and window["pct"] > 0:
+            result["code_review"] = {
+                "pct": window["pct"],
+                "resets_at": window["resets_at"],
+            }
 
     # Credits
     credits = api_data.get("credits") or {}

{codex_cli_usage-0.1.8 → codex_cli_usage-0.1.10}/tests/test_auth.py

@@ -1,4 +1,5 @@
 import json
+import os
 import tempfile
 import unittest
 import urllib.error
@@ -65,6 +66,14 @@ class AuthenticationRecoveryTests(unittest.TestCase):
     def write_auth(self, auth):
         self.auth_file.write_text(json.dumps(auth))
 
+    def test_codex_home_sets_direct_auth_path(self):
+        with tempfile.TemporaryDirectory() as tmp, patch.dict(
+            os.environ, {"CODEX_HOME": tmp}, clear=True
+        ):
+            self.assertEqual(
+                codex_cli_usage._find_codex_auth_path(), Path(tmp) / "auth.json"
+            )
+
     def test_transient_403_retries_same_token_without_refresh(self):
         requests = []
 

{codex_cli_usage-0.1.8 → codex_cli_usage-0.1.10}/tests/test_usage.py

@@ -6,6 +6,7 @@ from pathlib import Path
 from unittest.mock import patch
 
 from codex_cli_usage import (
+    UsageServiceError,
     build_usage_json,
     classify_window,
     cmd_status,
@@ -86,6 +87,20 @@ class UsageWindowTests(unittest.TestCase):
             "window_secs": None,
         })
 
+    def test_malformed_window_values_are_unavailable(self):
+        invalid_windows = (
+            {"used_percent": float("nan")},
+            {"used_percent": 1, "limit_window_seconds": float("inf")},
+            {"used_percent": 1, "reset_at": "not-a-timestamp"},
+            {"used_percent": 1, "reset_at": 10**30},
+        )
+
+        for window in invalid_windows:
+            with self.subTest(window=window), self.assertRaisesRegex(
+                UsageServiceError, "Invalid rate limit response"
+            ):
+                build_usage_json({"rate_limit": {"primary_window": window}})
+
     def test_status_shows_one_weekly_row_and_no_spark_bucket(self):
         output = StringIO()
         with patch(
@@ -101,7 +116,11 @@ class UsageWindowTests(unittest.TestCase):
     def test_statusline_enumerates_structured_windows(self):
         usage = build_usage_json(fixture("primary_5h_secondary_weekly.json"))
         output = StringIO()
-        with patch("codex_cli_usage._get_cached_usage", return_value=usage), redirect_stdout(output):
+        with (
+            patch("codex_cli_usage._get_cached_usage", return_value=usage),
+            patch("codex_cli_usage._TTY", False),
+            redirect_stdout(output),
+        ):
             cmd_statusline()
 
         self.assertIn("5h:12%", output.getvalue())

codex_cli_usage-0.1.10/uv.lock

@@ -0,0 +1,8 @@
+version = 1
+revision = 3
+requires-python = ">=3.12"
+
+[[package]]
+name = "codex-cli-usage"
+version = "0.1.10"
+source = { editable = "." }

codex_cli_usage-0.1.8/.github/workflows/publish.yml

@@ -1,18 +0,0 @@
-name: Publish to PyPI
-
-on:
-  release:
-    types: [published]
-
-permissions:
-  id-token: write
-
-jobs:
-  publish:
-    runs-on: ubuntu-latest
-    environment: pypi
-    steps:
-      - uses: actions/checkout@v4
-      - uses: astral-sh/setup-uv@v4
-      - run: uv build
-      - uses: pypa/gh-action-pypi-publish@release/v1

codex_cli_usage-0.1.8/uv.lock

@@ -1,20 +0,0 @@
-version = 1
-revision = 3
-requires-python = ">=3.12"
-
-[options]
-exclude-newer = "2026-07-05T20:06:09.848919716Z"
-exclude-newer-span = "P7D"
-
-[options.exclude-newer-package]
-clanker-analytics = false
-ghp = false
-search-claude-history = false
-gemini-cli-usage = false
-ccusage = false
-codex-cli-usage = false
-
-[[package]]
-name = "codex-cli-usage"
-version = "0.1.8"
-source = { editable = "." }