tokenhoggers 0.1.0__tar.gz

tokenhoggers-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Token Hoggers Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

tokenhoggers-0.1.0/PKG-INFO

@@ -0,0 +1,138 @@
+Metadata-Version: 2.4
+Name: tokenhoggers
+Version: 0.1.0
+Summary: Track your AI usage across Claude Code, Cursor, ChatGPT, Gemini, and more
+Author: Vikrant Tyagi
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/AIPeek/token-hoggers
+Project-URL: Repository, https://github.com/AIPeek/token-hoggers
+Keywords: ai,analytics,leaderboard,claude,cursor,llm
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Software Development :: Libraries
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: license-file
+
+# Token Hoggers
+
+**Gamified, tool-agnostic AI usage leaderboard.** Track and compete on your AI usage across Claude Code, Cursor, Gemini, ChatGPT, Copilot, and more.
+
+Inspired by [claude_rank](https://github.com/bicro/claude_rank) but tool-agnostic, with an RPG character system built on the **AMI (AI Maturity Index)** framework.
+
+## How It Works
+
+1. **Install the CLI** — `pip install tokenhoggers`
+2. **Sign in** — Google OAuth at [tokenhoggers.com](https://web-system42.vercel.app)
+3. **Sync** — `tokenhoggers sync` reads your local AI tool stats (counts only, never content)
+4. **Compete** — See your rank, level, tier, and RPG stats on the leaderboard
+
+## Privacy-First
+
+We collect **counts and timestamps only**. Never prompt text, code, file paths, or API keys.
+
+| Collected | Never Touched |
+|---|---|
+| Token counts (in/out) | Prompt/response text |
+| Message & session counts | Project/repo names |
+| Tool call counts | File paths, code, diffs |
+| Model names | API keys, credentials |
+| Timestamps | Working directories |
+
+## RPG Character System
+
+Built on the **AMI (AI Maturity Index)** — 4 dimensions that measure how you use AI, not just how much.
+
+```
+          MST (Mastery)
+             |
+   PRC ──────+────── VRS
+             |
+          END (Endurance)
+```
+
+| Stat | Measures |
+|---|---|
+| **Mastery** | Power-user behavior (tool calls/session, model diversity) |
+| **Precision** | Output per effort (efficiency) |
+| **Versatility** | Multi-tool range |
+| **Endurance** | Consistency (streaks, active days) |
+
+**Tiers:** Bronze → Silver → Gold → Platinum → Diamond
+
+## Supported Tools
+
+| Tool | Data Source | Parser |
+|---|---|---|
+| Claude Code | `~/.claude/stats-cache.json` | CLI |
+| Cursor | `state.vscdb` (SQLite, read-only) | CLI |
+| Gemini CLI | `~/.gemini/tmp/*/logs.json` | CLI |
+| ChatGPT | Web request counting | Browser Extension |
+| Gemini Web | Web request counting | Browser Extension |
+| Claude Web | Web request counting | Browser Extension |
+
+## Quick Start
+
+### CLI
+
+```bash
+pip install tokenhoggers
+
+# Check what data is available locally
+tokenhoggers status
+
+# Sign in at the web app, copy your token, then:
+tokenhoggers config --url https://aaxkbwyolbhjgovmcael.supabase.co/functions/v1/sync
+tokenhoggers config --token <YOUR_TOKEN>
+tokenhoggers sync
+
+# Re-sync all historical data
+tokenhoggers sync --force
+```
+
+### Desktop App (macOS)
+
+Menu bar app that shows your stats at a glance.
+
+```bash
+cd desktop/src-tauri
+source ~/.cargo/env
+cargo run
+```
+
+### Browser Extension
+
+Tracks request counts on ChatGPT, Gemini, and Claude web (never captures content).
+
+```bash
+cd extension/icons && python3 generate.py
+# Chrome → chrome://extensions → Developer Mode → Load unpacked → select extension/
+```
+
+## Architecture
+
+```
+collectors/cli/        Python CLI — parses local AI tool stats
+web/                   Frontend (vanilla HTML/JS/CSS) — hosted on Vercel
+supabase/              Backend — PostgreSQL, Auth, Edge Functions, Realtime
+desktop/               Tauri 2.x macOS menu bar app
+extension/             Chrome MV3 browser extension
+```
+
+## Tech Stack
+
+- **Backend:** Supabase (PostgreSQL + Auth + Edge Functions + Realtime)
+- **Frontend:** Vanilla HTML/JS/CSS (Claude Rank-inspired design)
+- **CLI:** Python 3.9+
+- **Desktop:** Rust / Tauri 2.x
+- **Extension:** Chrome MV3
+
+## Contributing
+
+PRs welcome! See [CONTRIBUTING.md](CONTRIBUTING.md).
+
+## License
+
+MIT — see [LICENSE](LICENSE).

tokenhoggers-0.1.0/README.md

@@ -0,0 +1,120 @@
+# Token Hoggers
+
+**Gamified, tool-agnostic AI usage leaderboard.** Track and compete on your AI usage across Claude Code, Cursor, Gemini, ChatGPT, Copilot, and more.
+
+Inspired by [claude_rank](https://github.com/bicro/claude_rank) but tool-agnostic, with an RPG character system built on the **AMI (AI Maturity Index)** framework.
+
+## How It Works
+
+1. **Install the CLI** — `pip install tokenhoggers`
+2. **Sign in** — Google OAuth at [tokenhoggers.com](https://web-system42.vercel.app)
+3. **Sync** — `tokenhoggers sync` reads your local AI tool stats (counts only, never content)
+4. **Compete** — See your rank, level, tier, and RPG stats on the leaderboard
+
+## Privacy-First
+
+We collect **counts and timestamps only**. Never prompt text, code, file paths, or API keys.
+
+| Collected | Never Touched |
+|---|---|
+| Token counts (in/out) | Prompt/response text |
+| Message & session counts | Project/repo names |
+| Tool call counts | File paths, code, diffs |
+| Model names | API keys, credentials |
+| Timestamps | Working directories |
+
+## RPG Character System
+
+Built on the **AMI (AI Maturity Index)** — 4 dimensions that measure how you use AI, not just how much.
+
+```
+          MST (Mastery)
+             |
+   PRC ──────+────── VRS
+             |
+          END (Endurance)
+```
+
+| Stat | Measures |
+|---|---|
+| **Mastery** | Power-user behavior (tool calls/session, model diversity) |
+| **Precision** | Output per effort (efficiency) |
+| **Versatility** | Multi-tool range |
+| **Endurance** | Consistency (streaks, active days) |
+
+**Tiers:** Bronze → Silver → Gold → Platinum → Diamond
+
+## Supported Tools
+
+| Tool | Data Source | Parser |
+|---|---|---|
+| Claude Code | `~/.claude/stats-cache.json` | CLI |
+| Cursor | `state.vscdb` (SQLite, read-only) | CLI |
+| Gemini CLI | `~/.gemini/tmp/*/logs.json` | CLI |
+| ChatGPT | Web request counting | Browser Extension |
+| Gemini Web | Web request counting | Browser Extension |
+| Claude Web | Web request counting | Browser Extension |
+
+## Quick Start
+
+### CLI
+
+```bash
+pip install tokenhoggers
+
+# Check what data is available locally
+tokenhoggers status
+
+# Sign in at the web app, copy your token, then:
+tokenhoggers config --url https://aaxkbwyolbhjgovmcael.supabase.co/functions/v1/sync
+tokenhoggers config --token <YOUR_TOKEN>
+tokenhoggers sync
+
+# Re-sync all historical data
+tokenhoggers sync --force
+```
+
+### Desktop App (macOS)
+
+Menu bar app that shows your stats at a glance.
+
+```bash
+cd desktop/src-tauri
+source ~/.cargo/env
+cargo run
+```
+
+### Browser Extension
+
+Tracks request counts on ChatGPT, Gemini, and Claude web (never captures content).
+
+```bash
+cd extension/icons && python3 generate.py
+# Chrome → chrome://extensions → Developer Mode → Load unpacked → select extension/
+```
+
+## Architecture
+
+```
+collectors/cli/        Python CLI — parses local AI tool stats
+web/                   Frontend (vanilla HTML/JS/CSS) — hosted on Vercel
+supabase/              Backend — PostgreSQL, Auth, Edge Functions, Realtime
+desktop/               Tauri 2.x macOS menu bar app
+extension/             Chrome MV3 browser extension
+```
+
+## Tech Stack
+
+- **Backend:** Supabase (PostgreSQL + Auth + Edge Functions + Realtime)
+- **Frontend:** Vanilla HTML/JS/CSS (Claude Rank-inspired design)
+- **CLI:** Python 3.9+
+- **Desktop:** Rust / Tauri 2.x
+- **Extension:** Chrome MV3
+
+## Contributing
+
+PRs welcome! See [CONTRIBUTING.md](CONTRIBUTING.md).
+
+## License
+
+MIT — see [LICENSE](LICENSE).

tokenhoggers-0.1.0/collectors/cli/auth.py

@@ -0,0 +1,247 @@
+"""Browser-based OAuth login and token refresh for Token Hoggers CLI."""
+
+import base64
+import json
+import socket
+import sys
+import threading
+import time
+import webbrowser
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from typing import Optional
+from urllib.error import HTTPError
+from urllib.request import Request, urlopen
+
+from collectors.cli.config import Config
+
+SUPABASE_URL = "https://aaxkbwyolbhjgovmcael.supabase.co"
+SUPABASE_ANON_KEY = (
+    "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9."
+    "eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImFheGtid3lvbGJoamdvdm1jYWVsIiwi"
+    "cm9sZSI6ImFub24iLCJpYXQiOjE3NzQxODM4MTUsImV4cCI6MjA4OTc1OTgxNX0."
+    "nNDUcYm2xzpGW6ervqNQEIXQdENi1qYXpQo4Rp0JrI8"
+)
+SYNC_API_URL = f"{SUPABASE_URL}/functions/v1/sync"
+
+_PORT_RANGE = range(19876, 19900)
+_LOGIN_TIMEOUT = 120  # seconds
+
+# HTML page served at /callback — reads hash fragment and POSTs tokens back
+_CALLBACK_HTML = """<!DOCTYPE html>
+<html>
+<head>
+  <title>Token Hoggers</title>
+  <style>
+    body { font-family: -apple-system, system-ui, sans-serif; display: flex;
+           justify-content: center; align-items: center; min-height: 100vh;
+           margin: 0; background: #0f0f0f; color: #e0e0e0; }
+    .card { text-align: center; padding: 48px; border-radius: 16px;
+            background: #1a1a1a; border: 1px solid #333; max-width: 400px; }
+    h2 { margin: 0 0 8px; color: #fff; }
+    p { margin: 0; color: #888; font-size: 14px; }
+    .err { color: #ff6b6b; }
+  </style>
+</head>
+<body>
+  <div class="card">
+    <div id="status"><h2>Completing login...</h2></div>
+  </div>
+  <script>
+    (function() {
+      var hash = window.location.hash.substring(1);
+      if (!hash) {
+        document.getElementById('status').innerHTML =
+          '<h2 class="err">Login failed</h2><p>No auth tokens received. Please try again.</p>';
+        return;
+      }
+      var params = new URLSearchParams(hash);
+      var data = {
+        access_token: params.get('access_token'),
+        refresh_token: params.get('refresh_token'),
+        expires_in: parseInt(params.get('expires_in') || '3600', 10),
+        token_type: params.get('token_type')
+      };
+      if (!data.access_token) {
+        document.getElementById('status').innerHTML =
+          '<h2 class="err">Login failed</h2><p>Missing access token. Please try again.</p>';
+        return;
+      }
+      fetch('/receive-tokens', {
+        method: 'POST',
+        headers: {'Content-Type': 'application/json'},
+        body: JSON.stringify(data)
+      }).then(function(r) { return r.json(); }).then(function() {
+        document.getElementById('status').innerHTML =
+          '<h2>Login successful!</h2><p>You can close this tab and return to your terminal.</p>';
+      }).catch(function(e) {
+        document.getElementById('status').innerHTML =
+          '<h2 class="err">Error</h2><p>' + e.message + '</p>';
+      });
+    })();
+  </script>
+</body>
+</html>"""
+
+
+def _find_available_port() -> int:
+    """Find an available port in the designated range."""
+    for port in _PORT_RANGE:
+        try:
+            with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+                s.bind(("127.0.0.1", port))
+                return port
+        except OSError:
+            continue
+    raise RuntimeError(
+        f"No available port in range {_PORT_RANGE.start}-{_PORT_RANGE.stop - 1}. "
+        "Close other applications or use: tokenhoggers config --token <JWT>"
+    )
+
+
+def browser_login() -> dict:
+    """Open browser for Google OAuth and capture tokens via local HTTP server.
+
+    Returns dict with access_token, refresh_token, expires_in.
+    Raises TimeoutError if user doesn't complete login within the timeout.
+    Raises RuntimeError on other failures.
+    """
+    port = _find_available_port()
+    tokens = {}
+    token_event = threading.Event()
+
+    class CallbackHandler(BaseHTTPRequestHandler):
+        def do_GET(self):
+            if self.path.startswith("/callback"):
+                self.send_response(200)
+                self.send_header("Content-Type", "text/html")
+                self.end_headers()
+                self.wfile.write(_CALLBACK_HTML.encode())
+            else:
+                self.send_response(404)
+                self.end_headers()
+
+        def do_POST(self):
+            if self.path == "/receive-tokens":
+                length = int(self.headers.get("Content-Length", 0))
+                body = json.loads(self.rfile.read(length))
+                tokens.update(body)
+                token_event.set()
+                self.send_response(200)
+                self.send_header("Content-Type", "application/json")
+                self.end_headers()
+                self.wfile.write(b'{"ok":true}')
+            else:
+                self.send_response(404)
+                self.end_headers()
+
+        def log_message(self, format, *args):
+            pass  # suppress HTTP server logging
+
+    server = HTTPServer(("127.0.0.1", port), CallbackHandler)
+    server_thread = threading.Thread(target=server.serve_forever, daemon=True)
+    server_thread.start()
+
+    auth_url = (
+        f"{SUPABASE_URL}/auth/v1/authorize"
+        f"?provider=google"
+        f"&redirect_to=http://localhost:{port}/callback"
+    )
+
+    opened = webbrowser.open(auth_url)
+    if not opened:
+        print(f"\n  Could not open browser. Please visit this URL manually:\n")
+        print(f"  {auth_url}\n")
+
+    if not token_event.wait(timeout=_LOGIN_TIMEOUT):
+        server.shutdown()
+        raise TimeoutError("Login timed out. Please try again.")
+
+    server.shutdown()
+
+    if not tokens.get("access_token"):
+        raise RuntimeError("No access token received.")
+
+    return tokens
+
+
+def decode_jwt_exp(token: str) -> int:
+    """Decode a JWT and return the exp (expiry) claim as a Unix timestamp.
+
+    Does NOT verify the signature — only used for local expiry checks.
+    Returns 0 if the token is malformed or has no exp claim.
+    """
+    try:
+        parts = token.split(".")
+        if len(parts) != 3:
+            return 0
+        payload = parts[1]
+        # Add base64 padding
+        padding = 4 - len(payload) % 4
+        if padding != 4:
+            payload += "=" * padding
+        decoded = base64.urlsafe_b64decode(payload)
+        claims = json.loads(decoded)
+        return int(claims.get("exp", 0))
+    except Exception:
+        return 0
+
+
+def refresh_access_token(refresh_token: str) -> dict:
+    """Exchange a refresh token for a new access token pair.
+
+    Returns dict with access_token, refresh_token, expires_in.
+    Raises RuntimeError if refresh fails (token revoked/expired).
+    """
+    url = f"{SUPABASE_URL}/auth/v1/token?grant_type=refresh_token"
+    body = json.dumps({"refresh_token": refresh_token}).encode()
+    req = Request(
+        url,
+        data=body,
+        headers={
+            "apikey": SUPABASE_ANON_KEY,
+            "Content-Type": "application/json",
+        },
+        method="POST",
+    )
+    try:
+        with urlopen(req, timeout=15) as resp:
+            data = json.loads(resp.read().decode())
+            return {
+                "access_token": data["access_token"],
+                "refresh_token": data["refresh_token"],
+                "expires_in": data.get("expires_in", 3600),
+            }
+    except HTTPError as e:
+        if e.code in (400, 401, 403):
+            raise RuntimeError(
+                "Session expired. Please run `tokenhoggers login` again."
+            )
+        raise RuntimeError(f"Token refresh failed (HTTP {e.code})")
+
+
+def refresh_if_needed(config: Config) -> Config:
+    """Check if the access token is expired and refresh it if possible.
+
+    Modifies and saves config in-place if a refresh occurs.
+    Returns the (possibly updated) config.
+    """
+    if not config.token:
+        return config
+
+    exp = decode_jwt_exp(config.token)
+    now = int(time.time())
+
+    # Token still valid for more than 60 seconds
+    if exp > 0 and (exp - now) > 60:
+        return config
+
+    # No refresh token available (manual token setup)
+    if not config.refresh_token:
+        return config
+
+    tokens = refresh_access_token(config.refresh_token)
+    config.token = tokens["access_token"]
+    config.refresh_token = tokens["refresh_token"]
+    config.token_expiry = now + tokens.get("expires_in", 3600)
+    config.save()
+    return config

tokenhoggers-0.1.0/collectors/cli/config.py

@@ -0,0 +1,44 @@
+"""User configuration for the Token Hoggers CLI."""
+
+import json
+from dataclasses import dataclass
+from typing import Optional
+
+from collectors.cli.paths import config_dir
+
+
+@dataclass
+class Config:
+    api_url: str = ""
+    token: str = ""
+    refresh_token: str = ""
+    token_expiry: int = 0
+    enable_fluency: bool = False
+
+    @classmethod
+    def load(cls) -> "Config":
+        path = config_dir() / "config.json"
+        if path.exists():
+            data = json.loads(path.read_text())
+            return cls(
+                api_url=data.get("api_url", ""),
+                token=data.get("token", ""),
+                refresh_token=data.get("refresh_token", ""),
+                token_expiry=data.get("token_expiry", 0),
+                enable_fluency=data.get("enable_fluency", False),
+            )
+        return cls()
+
+    def save(self) -> None:
+        path = config_dir() / "config.json"
+        path.write_text(json.dumps({
+            "api_url": self.api_url,
+            "token": self.token,
+            "refresh_token": self.refresh_token,
+            "token_expiry": self.token_expiry,
+            "enable_fluency": self.enable_fluency,
+        }, indent=2))
+
+    @property
+    def is_configured(self) -> bool:
+        return bool(self.api_url and self.token)