agy-swap 0.1.0__tar.gz

agy_swap-0.1.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: agy-swap
+Version: 0.1.0
+Summary: Multi-account switcher for Antigravity CLI (agy)
+Project-URL: Homepage, https://github.com/pr656d/agy-swap
+Project-URL: Repository, https://github.com/pr656d/agy-swap
+Requires-Python: >=3.10
+Requires-Dist: rich>=13.0
+Provides-Extra: tui
+Requires-Dist: textual>=1.0; extra == "tui"

agy_swap-0.1.0/README.md

@@ -0,0 +1,178 @@
+# agy-swap
+
+Multi-account switcher for Antigravity CLI (agy). Easily switch between multiple Google/Gemini accounts without logging out. Track Gemini API quota for every account in a live TUI dashboard.
+
+## Installation
+
+### Using uv (recommended)
+
+```bash
+uv tool install agy-swap
+```
+
+### Using pipx
+
+```bash
+pipx install agy-swap
+```
+
+### With TUI dashboard
+
+```bash
+uv tool install 'agy-swap[tui]'       # uv
+pipx install 'agy-swap[tui]'          # pipx
+```
+
+### From source
+
+```bash
+git clone https://github.com/pr656d/agy-swap.git
+cd agy-swap
+uv sync
+uv run agyswap help
+```
+
+### Updating
+
+```bash
+uv tool upgrade agy-swap       # uv
+pipx upgrade agy-swap          # pipx
+```
+
+## Usage
+
+### Add your first account
+
+Log into agy with your first account, then:
+
+```bash
+agyswap add
+```
+
+### Add more accounts
+
+Log in with another account, then:
+
+```bash
+agyswap add
+```
+
+### Switch accounts
+
+Rotate to the next enabled account:
+
+```bash
+agyswap switch
+```
+
+Or switch to a specific account:
+
+```bash
+agyswap switch 2
+agyswap switch user@gmail.com
+agyswap switch work               # by alias, once set with `agyswap alias 2 work`
+```
+
+See them all at a glance:
+
+```bash
+agyswap list
+```
+
+### Interactive dashboard (TUI)
+
+Run `agyswap tui` for the full-screen dashboard — every account's 5-hour and weekly usage with live progress bars, reset times, and switching, all keyboard-driven. `agyswap watch` opens it straight to the live monitor.
+
+Keyboard shortcuts from the dashboard:
+
+| Key | Action |
+|-----|--------|
+| `s` | Open account picker to switch |
+| `w` | Switch to watch screen |
+| `a` | Add current account |
+| `f` | Refresh |
+| `j`/`k` | Navigate menu |
+| `q` | Quit |
+
+### Aliases
+
+```bash
+agyswap alias 2 work            # give account 2 the alias "work"
+agyswap alias 2 --unset         # remove alias
+agyswap alias                   # list all aliases
+```
+
+### Disable / enable
+
+Hold an account out of rotation without removing its credentials:
+
+```bash
+agyswap disable 2
+agyswap enable 2
+```
+
+### Remove an account
+
+```bash
+agyswap remove 2
+```
+
+## Quota data
+
+Quota usage appears in `agyswap list` and the TUI. Two ways to populate it:
+
+- **Automatic (TUI only)** — when agy is running, the TUI probes its local language server via ConnectRPC every 3 seconds and writes quota to cache. No setup needed.
+- **Statusline** — run `agy` with the statusline script to capture quota on every prompt:
+
+  ```bash
+  agy -S "python3 -m agyswap.statusline"
+  ```
+
+  Add it as a shell alias:
+
+  ```bash
+  alias agy='agy -S "python3 -m agyswap.statusline"'
+  ```
+
+  The script writes to `~/.agy-swap/cache/quota.json` which the TUI and CLI read.
+
+## How it works
+
+- Backs up OAuth tokens when you add an account (`~/.agy-swap/credentials/`)
+- Swaps only the account-specific login by replacing the active token and Google accounts file
+- Account credentials stored securely using platform-appropriate methods (macOS Keychain + file fallback)
+- Quota fetched from agy's local gRPC language server when running, or captured via the `-S` statusline mechanism
+
+## Data locations
+
+| Data | Path |
+|------|------|
+| Credential backups | `~/.agy-swap/credentials/` |
+| Account registry | `~/.agy-swap/sequence.json` |
+| Quota cache | `~/.agy-swap/cache/quota.json` |
+
+## Uninstall
+
+Remove all data:
+
+```bash
+rm -rf ~/.agy-swap
+```
+
+Then uninstall the tool:
+
+```bash
+uv tool uninstall agy-swap
+# or
+pipx uninstall agy-swap
+```
+
+## Requirements
+
+- Python 3.10+
+- Antigravity CLI (agy) installed and logged in
+- macOS, Linux
+
+## License
+
+MIT

agy_swap-0.1.0/agy_swap.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: agy-swap
+Version: 0.1.0
+Summary: Multi-account switcher for Antigravity CLI (agy)
+Project-URL: Homepage, https://github.com/pr656d/agy-swap
+Project-URL: Repository, https://github.com/pr656d/agy-swap
+Requires-Python: >=3.10
+Requires-Dist: rich>=13.0
+Provides-Extra: tui
+Requires-Dist: textual>=1.0; extra == "tui"

agy_swap-0.1.0/agy_swap.egg-info/SOURCES.txt

@@ -0,0 +1,23 @@
+README.md
+pyproject.toml
+agy_swap.egg-info/PKG-INFO
+agy_swap.egg-info/SOURCES.txt
+agy_swap.egg-info/dependency_links.txt
+agy_swap.egg-info/entry_points.txt
+agy_swap.egg-info/requires.txt
+agy_swap.egg-info/top_level.txt
+agyswap/__init__.py
+agyswap/__main__.py
+agyswap/cli.py
+agyswap/models.py
+agyswap/paths.py
+agyswap/statusline.py
+agyswap/switcher.py
+agyswap/tui/__init__.py
+agyswap/tui/agy.tcss
+agyswap/tui/app.py
+agyswap/tui/dashboard.py
+agyswap/tui/quota_probe.py
+agyswap/tui/theme.py
+agyswap/tui/watch.py
+agyswap/tui/widgets.py

agy_swap-0.1.0/agy_swap.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

agy_swap-0.1.0/agy_swap.egg-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+agyswap = agyswap.cli:main
+agyswap-statusline = agyswap.statusline:main

agy_swap-0.1.0/agy_swap.egg-info/requires.txt

@@ -0,0 +1,4 @@
+rich>=13.0
+
+[tui]
+textual>=1.0

agy_swap-0.1.0/agy_swap.egg-info/top_level.txt

@@ -0,0 +1,3 @@
+agyswap
+build
+dist

agy_swap-0.1.0/agyswap/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

agy_swap-0.1.0/agyswap/__main__.py

@@ -0,0 +1,4 @@
+from agyswap.cli import main
+
+if __name__ == "__main__":
+    main()

agy_swap-0.1.0/agyswap/cli.py

@@ -0,0 +1,246 @@
+import argparse
+import sys
+
+from agyswap import __version__
+from agyswap.switcher import AgySwitcher, AccountNotFoundError, AgyError
+from agyswap.paths import GOOGLE_ACCOUNTS_PATH
+
+from rich.console import Console
+from rich.table import Table
+from rich import box
+
+console = Console()
+
+
+def cmd_list(sw: AgySwitcher, args) -> None:
+    rows = sw.list_accounts()
+    if not rows:
+        console.print("[dim]No accounts managed yet.[/dim]")
+        console.print("Login with [bold]agy[/bold] first, then run [bold]agyswap add[/bold].")
+        return
+
+    table = Table(box=box.ROUNDED)
+    table.add_column("#", style="bold")
+    table.add_column("Email")
+    table.add_column("Alias")
+    table.add_column("Status")
+    table.add_column("Token")
+
+    for r in rows:
+        num = str(r["number"])
+        email = r["email"]
+        alias = r["alias"] or ""
+        status_parts = []
+        if r["active"]:
+            status_parts.append("[green]active[/green]")
+        if r["disabled"]:
+            status_parts.append("[yellow]disabled[/yellow]")
+        status = " ".join(status_parts) if status_parts else "[dim]—[/dim]"
+        token = "[green]✓[/green]" if r["has_token"] else "[red]✗[/red]"
+        table.add_row(num, email, alias, status, token)
+
+    console.print(table)
+    console.print(f"\n[dim]Active account tracked at: {GOOGLE_ACCOUNTS_PATH}[/dim]")
+
+
+def cmd_status(sw: AgySwitcher, args) -> None:
+    info = sw.status()
+    if info["active"] is None:
+        console.print("[yellow]No active agy login detected.[/yellow]")
+        console.print("Run [bold]agy[/bold] to login, then [bold]agyswap add[/bold].")
+        return
+    managed = "yes" if info["managed"] else "[yellow]not managed[/yellow]"
+    console.print(f"Active account: [bold]{info['active']}[/bold]")
+    console.print(f"Managed: {managed}")
+    if info["number"]:
+        console.print(f"Slot: #{info['number']}")
+
+
+def cmd_add(sw: AgySwitcher, args) -> None:
+    acc = sw.add_current(email_override=args.email)
+    console.print(f"[green]Added[/green] account #{acc.number}: {acc.email}")
+
+
+def cmd_remove(sw: AgySwitcher, args) -> None:
+    try:
+        acc = sw._resolve_account(args.identifier)
+        sw.remove_account(args.identifier)
+        console.print(f"[red]Removed[/red] account #{acc.number}: {acc.email}")
+    except AccountNotFoundError as e:
+        console.print(f"[red]Error:[/red] {e}")
+        sys.exit(1)
+
+
+def cmd_switch(sw: AgySwitcher, args) -> None:
+    try:
+        result = sw.switch(args.identifier)
+        prev = result["previous"] or "(none)"
+        console.print(
+            f"Switched: [dim]{prev}[/dim] → [bold green]{result['active']}[/bold green]"
+        )
+    except (AgyError, AccountNotFoundError) as e:
+        console.print(f"[red]Error:[/red] {e}")
+        sys.exit(1)
+
+
+def cmd_disable(sw: AgySwitcher, args) -> None:
+    try:
+        acc = sw.set_disabled(args.identifier, True)
+        console.print(f"[yellow]Disabled[/yellow] account #{acc.number}: {acc.email}")
+    except AccountNotFoundError as e:
+        console.print(f"[red]Error:[/red] {e}")
+        sys.exit(1)
+
+
+def cmd_enable(sw: AgySwitcher, args) -> None:
+    try:
+        acc = sw.set_disabled(args.identifier, False)
+        console.print(f"[green]Enabled[/green] account #{acc.number}: {acc.email}")
+    except AccountNotFoundError as e:
+        console.print(f"[red]Error:[/red] {e}")
+        sys.exit(1)
+
+
+def cmd_alias(sw: AgySwitcher, args) -> None:
+    try:
+        if args.unset:
+            acc = sw.unalias(args.identifier)
+            console.print(f"Removed alias for #{acc.number}: {acc.email}")
+        elif args.name:
+            acc = sw.alias(args.identifier, args.name)
+            console.print(f"Set alias [bold]'{args.name}'[/bold] for #{acc.number}: {acc.email}")
+        else:
+            rows = sw.list_accounts()
+            found = False
+            for r in rows:
+                if r["alias"]:
+                    console.print(f"  {r['number']}: {r['alias']} [dim]({r['email']})[/dim]")
+                    found = True
+            if not found:
+                console.print("[dim]No aliases set.[/dim]")
+    except AccountNotFoundError as e:
+        console.print(f"[red]Error:[/red] {e}")
+        sys.exit(1)
+
+
+def cmd_tui(sw: AgySwitcher, args) -> None:
+    try:
+        from agyswap.tui.app import run_tui
+        run_tui(sw, start_watch=False)
+    except ImportError:
+        console.print("[red]TUI requires textual. Install:[/red]")
+        console.print("  pip install 'agy-swap[tui]'")
+        sys.exit(1)
+
+
+def cmd_watch(sw: AgySwitcher, args) -> None:
+    try:
+        from agyswap.tui.app import run_tui
+        run_tui(sw, start_watch=True)
+    except ImportError:
+        console.print("[red]TUI requires textual. Install:[/red]")
+        console.print("  pip install 'agy-swap[tui]'")
+        sys.exit(1)
+
+
+def cmd_statusline(sw, args) -> None:
+    if args.setup:
+        module = "agyswap.statusline"
+        console.print("Add this alias to your shell:")
+        console.print(f"  alias agy='agy -S \"python3 -m {module}\" ")
+        console.print()
+        console.print("Or use it once:")
+        console.print(f"  agy -S \"python3 -m {module}\"")
+        console.print()
+        console.print("[dim]The statusline script captures quota data to[/dim]")
+        console.print(f"[dim]  ~/.agy-swap/cache/quota.json[/dim]")
+        console.print("[dim]so agyswap TUI can display it.[/dim]")
+        return
+
+    from agyswap.statusline import main as sl_main
+    sl_main()
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        prog="agyswap",
+        description="Multi-account switcher for Antigravity CLI (agy)",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""Examples:
+  agyswap add              capture current login as a managed account
+  agyswap list             list managed accounts
+  agyswap switch           rotate to next enabled account
+  agyswap switch 2         switch to account #2
+  agyswap switch email     switch to account by email
+  agyswap disable 2        hold an account out of rotation
+  agyswap enable 2         return an account to rotation
+  agyswap remove 2         remove an account
+  agyswap alias 2 work     set an alias
+  agyswap alias 2 --unset  remove an alias
+  agyswap tui              interactive dashboard
+  agyswap watch            dashboard on the watch page
+""",
+    )
+    parser.add_argument("--version", action="version", version=f"%(prog)s {__version__}")
+
+    sub = parser.add_subparsers(dest="command")
+
+    p_list = sub.add_parser("list", help="list managed accounts")
+    p_status = sub.add_parser("status", help="show current active account")
+    p_add = sub.add_parser("add", help="capture current agy login as a managed account")
+    p_add.add_argument("--email", help="email for the account (auto-detected if omitted)")
+
+    p_switch = sub.add_parser("switch", help="switch accounts")
+    p_switch.add_argument("identifier", nargs="?", help="account number, email, or alias")
+
+    p_remove = sub.add_parser("remove", aliases=["rm"], help="remove an account")
+    p_remove.add_argument("identifier", help="account number, email, or alias")
+
+    p_disable = sub.add_parser("disable", help="hold account out of rotation")
+    p_disable.add_argument("identifier", help="account number, email, or alias")
+
+    p_enable = sub.add_parser("enable", help="return account to rotation")
+    p_enable.add_argument("identifier", help="account number, email, or alias")
+
+    p_alias = sub.add_parser("alias", help="manage account aliases")
+    p_alias.add_argument("identifier", nargs="?", help="account number, email, or alias")
+    p_alias.add_argument("name", nargs="?", help="alias name")
+    p_alias.add_argument("--unset", action="store_true", help="remove alias")
+
+    sub.add_parser("tui", help="launch interactive dashboard")
+    sub.add_parser("watch", help="dashboard on the watch page")
+
+    p_sl = sub.add_parser("statusline", help="capture quota data from agy -S")
+    p_sl.add_argument("--setup", action="store_true",
+                      help="print setup instructions")
+
+    args = parser.parse_args()
+
+    sw = AgySwitcher()
+
+    if not args.command:
+        cmd_tui(sw, args)
+        return
+
+    dispatch = {
+        "list": cmd_list,
+        "status": cmd_status,
+        "add": cmd_add,
+        "switch": cmd_switch,
+        "remove": cmd_remove,
+        "rm": cmd_remove,
+        "disable": cmd_disable,
+        "enable": cmd_enable,
+        "alias": cmd_alias,
+        "tui": cmd_tui,
+        "watch": cmd_watch,
+        "statusline": cmd_statusline,
+    }
+
+    handler = dispatch.get(args.command)
+    if handler:
+        handler(sw, args)
+
+
+if __name__ == "__main__":
+    main()

agy_swap-0.1.0/agyswap/models.py

@@ -0,0 +1,62 @@
+from dataclasses import dataclass, field
+from datetime import datetime, timezone
+from typing import Optional
+
+
+def now_iso() -> str:
+    return datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
+
+
+@dataclass
+class Account:
+    number: int
+    email: str
+    disabled: bool = False
+    alias: Optional[str] = None
+    created_at: str = field(default_factory=now_iso)
+    updated_at: str = field(default_factory=now_iso)
+
+
+@dataclass
+class SequenceData:
+    active_email: Optional[str] = None
+    sequence: list[int] = field(default_factory=list)
+    accounts: dict[int, Account] = field(default_factory=dict)
+    last_updated: str = field(default_factory=now_iso)
+
+    def to_dict(self) -> dict:
+        return {
+            "activeEmail": self.active_email,
+            "lastUpdated": self.last_updated,
+            "sequence": self.sequence,
+            "accounts": {
+                str(num): {
+                    "email": acc.email,
+                    "disabled": acc.disabled,
+                    "alias": acc.alias,
+                    "createdAt": acc.created_at,
+                    "updatedAt": acc.updated_at,
+                }
+                for num, acc in self.accounts.items()
+            },
+        }
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "SequenceData":
+        accounts = {}
+        for num_str, acc_data in data.get("accounts", {}).items():
+            num = int(num_str)
+            accounts[num] = Account(
+                number=num,
+                email=acc_data.get("email", ""),
+                disabled=acc_data.get("disabled", False),
+                alias=acc_data.get("alias"),
+                created_at=acc_data.get("createdAt", ""),
+                updated_at=acc_data.get("updatedAt", ""),
+            )
+        return cls(
+            active_email=data.get("activeEmail"),
+            sequence=data.get("sequence", []),
+            accounts=accounts,
+            last_updated=data.get("lastUpdated", ""),
+        )

agy_swap-0.1.0/agyswap/paths.py

@@ -0,0 +1,10 @@
+from pathlib import Path
+
+GEMINI_DIR = Path.home() / ".gemini"
+AGY_CLI_DIR = GEMINI_DIR / "antigravity-cli"
+AGY_TOKEN_PATH = AGY_CLI_DIR / "antigravity-oauth-token"
+GOOGLE_ACCOUNTS_PATH = GEMINI_DIR / "google_accounts.json"
+BACKUP_ROOT = Path.home() / ".agy-swap"
+SEQUENCE_FILE = BACKUP_ROOT / "sequence.json"
+CREDENTIALS_DIR = BACKUP_ROOT / "credentials"
+QUOTA_CACHE = BACKUP_ROOT / "cache" / "quota.json"

agy_swap-0.1.0/agyswap/statusline.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+"""Statusline capture script for agy -S.
+
+Run: agy -S "python3 -m agyswap.statusline"
+
+Captures the statusline JSON from stdin and writes quota data to the
+cache file so agyswap TUI can display it.
+
+Also prints a compact statusline to stdout.
+"""
+import json
+import sys
+import time
+from pathlib import Path
+
+CACHE = Path.home() / ".agy-swap" / "cache" / "quota.json"
+
+
+def main() -> None:
+    raw = sys.stdin.read()
+    if not raw.strip():
+        return
+
+    try:
+        data = json.loads(raw)
+    except json.JSONDecodeError:
+        return
+
+    context = data.get("context_window", {})
+    rate = data.get("rate_limits", {})
+    model = data.get("model", {}).get("display_name", "?")
+    email_claim = ""
+    for token_key in ("access_token", "id_token"):
+        full = data.get(token_key, "")
+        if not full:
+            continue
+        try:
+            payload_b64 = full.split(".")[1]
+            padding = 4 - len(payload_b64) % 4
+            if padding != 4:
+                payload_b64 += "=" * padding
+            import base64
+            payload = json.loads(base64.urlsafe_b64decode(payload_b64))
+            email_claim = payload.get("email", "")
+            if email_claim:
+                break
+        except Exception:
+            continue
+
+    quota = {
+        "accounts": {},
+        "model": model,
+        "captured_at": int(time.time()),
+    }
+
+    email = data.get("email", email_claim)
+
+    five = rate.get("five_hour", {})
+    seven = rate.get("seven_day", {})
+
+    account_quota = {}
+    if five:
+        account_quota["five_hour"] = five.get("used_percentage")
+        account_quota["five_resets_at"] = five.get("resets_at")
+    if seven:
+        account_quota["seven_day"] = seven.get("used_percentage")
+        account_quota["seven_resets_at"] = seven.get("resets_at")
+
+    if email:
+        quota["accounts"][email] = account_quota
+    elif account_quota:
+        quota["accounts"]["__unknown__"] = account_quota
+
+    CACHE.parent.mkdir(parents=True, exist_ok=True)
+    CACHE.write_text(json.dumps(quota, indent=2))
+
+    ctx_tokens = context.get("total_input_tokens", 0)
+    ctx_size = context.get("context_window_size", 0)
+    five_pct = five.get("used_percentage", "")
+    seven_pct = seven.get("used_percentage", "")
+
+    parts = [model]
+    if ctx_tokens and ctx_size:
+        parts.append(f"Ctx: {ctx_tokens // 1000}k/{ctx_size // 1000}k")
+    if five_pct != "":
+        parts.append(f"5h: {five_pct}%")
+    if seven_pct != "":
+        parts.append(f"7d: {seven_pct}%")
+    print("  ".join(parts), flush=True)
+
+
+if __name__ == "__main__":
+    main()