cursor-cli-usage 0.1.0__tar.gz → 0.1.1__tar.gz

{cursor_cli_usage-0.1.0 → cursor_cli_usage-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: cursor-cli-usage
-Version: 0.1.0
+Version: 0.1.1
 Summary: Cursor CLI account quota monitor using local Cursor authentication
 Project-URL: Homepage, https://github.com/wakamex/cursor-cli-usage
 Project-URL: Source, https://github.com/wakamex/cursor-cli-usage
@@ -59,9 +59,9 @@ uv tool install .
 For every live request, the tool first asks the installed Cursor CLI for its
 current authentication state with `agent status --format json`. If the CLI is
 unavailable, it falls back to `cursorAuth/accessToken` in Cursor IDE's local
-`state.vscdb`. It sends only the current access token to Cursor's dashboard
-usage service. It never refreshes, rotates, or modifies Cursor-owned
-credentials.
+`state.vscdb`. On Linux, the fallback follows `XDG_CONFIG_HOME` when it is set.
+It sends only the current access token to Cursor's dashboard usage service. It
+never refreshes, rotates, or modifies Cursor-owned credentials.
 
 The normalized JSON includes:
 

{cursor_cli_usage-0.1.0 → cursor_cli_usage-0.1.1}/README.md

@@ -50,9 +50,9 @@ uv tool install .
 For every live request, the tool first asks the installed Cursor CLI for its
 current authentication state with `agent status --format json`. If the CLI is
 unavailable, it falls back to `cursorAuth/accessToken` in Cursor IDE's local
-`state.vscdb`. It sends only the current access token to Cursor's dashboard
-usage service. It never refreshes, rotates, or modifies Cursor-owned
-credentials.
+`state.vscdb`. On Linux, the fallback follows `XDG_CONFIG_HOME` when it is set.
+It sends only the current access token to Cursor's dashboard usage service. It
+never refreshes, rotates, or modifies Cursor-owned credentials.
 
 The normalized JSON includes:
 

{cursor_cli_usage-0.1.0 → cursor_cli_usage-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cursor-cli-usage"
-version = "0.1.0"
+version = "0.1.1"
 description = "Cursor CLI account quota monitor using local Cursor authentication"
 readme = "README.md"
 requires-python = ">=3.12"

{cursor_cli_usage-0.1.0 → cursor_cli_usage-0.1.1}/src/cursor_cli_usage/__init__.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 
 import argparse
 import json
+import math
 import os
 import platform
 import shutil
@@ -74,7 +75,9 @@ def get_state_db() -> Path:
     if platform.system() == "Windows":
         app_data = Path(os.environ.get("APPDATA", Path.home() / "AppData" / "Roaming"))
         return app_data / "Cursor" / "User" / "globalStorage" / "state.vscdb"
-    return Path.home() / ".config" / "Cursor" / "User" / "globalStorage" / "state.vscdb"
+    config_home = os.environ.get("XDG_CONFIG_HOME")
+    base = Path(config_home).expanduser() if config_home else Path.home() / ".config"
+    return base / "Cursor" / "User" / "globalStorage" / "state.vscdb"
 
 
 def get_usage_file() -> Path:
@@ -149,30 +152,41 @@ def get_access_token() -> str:
 def _number(value: object) -> int | float | None:
     if isinstance(value, bool):
         return None
-    if isinstance(value, int | float):
+    if isinstance(value, int):
         return value
+    if isinstance(value, float):
+        return value if math.isfinite(value) else None
     if isinstance(value, str):
         try:
             parsed = float(value)
         except ValueError:
             return None
+        if not math.isfinite(parsed):
+            return None
         return int(parsed) if parsed.is_integer() else parsed
     return None
 
 
 def _usage_bucket(value: object) -> dict | None:
-    if not isinstance(value, dict):
+    if value is None:
         return None
-    return {
-        key: parsed
-        for key, source in (
-            ("limit", "limit"),
-            ("used", "used"),
-            ("remaining", "remaining"),
-            ("percent_used", "totalPercentUsed"),
-        )
-        if (parsed := _number(value.get(source))) is not None
-    }
+    if not isinstance(value, dict):
+        raise RuntimeError("Cursor usage response contained invalid usage data")
+    bucket = {}
+    for key, source in (
+        ("limit", "limit"),
+        ("used", "used"),
+        ("remaining", "remaining"),
+        ("percent_used", "totalPercentUsed"),
+    ):
+        raw = value.get(source)
+        if raw is None:
+            continue
+        parsed = _number(raw)
+        if parsed is None:
+            raise RuntimeError("Cursor usage response contained invalid numeric usage")
+        bucket[key] = parsed
+    return bucket
 
 
 def fetch_usage() -> dict:
@@ -208,12 +222,15 @@ def fetch_usage() -> dict:
 
 def _normalize(payload: dict) -> dict:
     models = payload.get("autoBucketModels")
+    plan_usage = _usage_bucket(payload.get("planUsage"))
+    if not plan_usage or plan_usage.get("percent_used") is None:
+        raise RuntimeError("Cursor usage response did not contain plan usage")
     return {
         "billing_cycle": {
             "start": payload.get("billingCycleStart"),
             "end": payload.get("billingCycleEnd"),
         },
-        "plan_usage": _usage_bucket(payload.get("planUsage")),
+        "plan_usage": plan_usage,
         "spend_limit_usage": _usage_bucket(payload.get("spendLimitUsage")),
         "auto_models": models if isinstance(models, list) else [],
     }

{cursor_cli_usage-0.1.0 → cursor_cli_usage-0.1.1}/tests/test_cursor_cli_usage.py

@@ -26,6 +26,25 @@ class FakeResponse:
 
 
 class CursorUsageTests(unittest.TestCase):
+    def test_linux_state_database_follows_xdg_config_home(self):
+        with (
+            tempfile.TemporaryDirectory() as temporary,
+            mock.patch.dict(
+                cursor_usage.os.environ,
+                {"XDG_CONFIG_HOME": temporary},
+                clear=True,
+            ),
+            mock.patch.object(cursor_usage.platform, "system", return_value="Linux"),
+        ):
+            self.assertEqual(
+                cursor_usage.get_state_db(),
+                Path(temporary)
+                / "Cursor"
+                / "User"
+                / "globalStorage"
+                / "state.vscdb",
+            )
+
     def test_access_token_is_read_from_database_without_writes(self):
         with tempfile.TemporaryDirectory() as temporary:
             database = Path(temporary) / "state.vscdb"
@@ -111,6 +130,23 @@ class CursorUsageTests(unittest.TestCase):
         self.assertEqual(data["spend_limit_usage"]["used"], 100)
         self.assertEqual(data["auto_models"], ["auto"])
 
+    def test_missing_plan_usage_is_unavailable(self):
+        with mock.patch.object(cursor_usage, "fetch_usage", return_value={}):
+            data = cursor_usage.build_usage_json()
+        self.assertEqual(data["status"], "unavailable")
+        self.assertIn("plan usage", data["error"])
+
+    def test_nonfinite_usage_is_unavailable(self):
+        for value in (float("nan"), "Infinity"):
+            with self.subTest(value=value):
+                payload = {"planUsage": {"totalPercentUsed": value}}
+                with mock.patch.object(
+                    cursor_usage, "fetch_usage", return_value=payload
+                ):
+                    data = cursor_usage.build_usage_json()
+                self.assertEqual(data["status"], "unavailable")
+                json.dumps(data, allow_nan=False)
+
     def test_unauthorized_never_attempts_refresh(self):
         error = urllib.error.HTTPError(cursor_usage.USAGE_URL, 401, "", {}, None)
         with (

{cursor_cli_usage-0.1.0 → cursor_cli_usage-0.1.1}/uv.lock

@@ -4,5 +4,5 @@ requires-python = ">=3.12"
 
 [[package]]
 name = "cursor-cli-usage"
-version = "0.1.0"
+version = "0.1.1"
 source = { editable = "." }