treadstone-cli 0.2.1__tar.gz

treadstone_cli-0.2.1/.gitignore

@@ -0,0 +1,34 @@
+__pycache__/
+*.py[cod]
+*$py.class
+*.egg-info/
+dist/
+build/
+*.egg
+
+.venv/
+.env
+.env.test
+.env.local
+.env.demo
+.env.prod
+
+.mypy_cache/
+.pytest_cache/
+.ruff_cache/
+.coverage
+htmlcov/
+
+*.db
+*.sqlite3
+
+openapi.json
+reports/
+
+.DS_Store
+Thumbs.db
+
+.idea/
+.vscode/
+*.swp
+*.swo

treadstone_cli-0.2.1/PKG-INFO

@@ -0,0 +1,222 @@
+Metadata-Version: 2.4
+Name: treadstone-cli
+Version: 0.2.1
+Summary: CLI for the Treadstone sandbox service.
+Requires-Python: >=3.12
+Requires-Dist: click>=8.1
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: rich>=14.0
+Description-Content-Type: text/markdown
+
+# Treadstone CLI
+
+Command-line interface for the Treadstone sandbox service.
+
+## Installation
+
+### From PyPI
+
+```bash
+pip install treadstone
+```
+
+### Pre-built binary
+
+Download the latest release from [GitHub Releases](https://github.com/earayu/treadstone/releases). The binary is self-contained and does not require Python.
+
+```bash
+# macOS / Linux
+chmod +x treadstone
+sudo mv treadstone /usr/local/bin/
+```
+
+## Quick start
+
+```bash
+# Check that the server is reachable
+treadstone health
+
+# Register an account (first time only)
+treadstone auth register
+
+# Log in
+treadstone auth login
+
+# Create an API key for non-interactive use
+treadstone api-keys create --name my-key
+# Save the key — it is shown only once
+
+# Store the key in config so you don't have to pass it every time
+treadstone config set api_key ts_live_xxxxxxxxxxxx
+
+# Create a sandbox
+treadstone sandboxes create --template default --name my-sandbox
+
+# List sandboxes
+treadstone sb list
+```
+
+## Configuration
+
+The CLI reads configuration from three sources, in order of priority:
+
+| Priority | Source | Example |
+|----------|--------|---------|
+| 1 (highest) | CLI flags | `--base-url https://... --api-key ts_...` |
+| 2 | Environment variables | `TREADSTONE_BASE_URL`, `TREADSTONE_API_KEY` |
+| 3 (lowest) | Config file | `~/.config/treadstone/config.toml` |
+
+### Config file
+
+Location: `~/.config/treadstone/config.toml`
+
+```toml
+[default]
+base_url = "https://your-server.example.com"
+api_key  = "ts_live_xxxxxxxxxxxx"
+```
+
+You can manage this file with the `config` subcommand:
+
+```bash
+# Set a value
+treadstone config set base_url https://your-server.example.com
+treadstone config set api_key ts_live_xxxxxxxxxxxx
+
+# View current settings (api_key is partially masked)
+treadstone config get
+
+# View a single key
+treadstone config get base_url
+
+# Remove a value
+treadstone config unset api_key
+
+# Show config file path
+treadstone config path
+```
+
+### Environment variables
+
+| Variable | Description | Default |
+|----------|-------------|---------|
+| `TREADSTONE_BASE_URL` | Base URL of the Treadstone server | `https://api.treadstone-ai.dev` |
+| `TREADSTONE_API_KEY` | API key for authentication | (none) |
+
+### Global flags
+
+| Flag | Description |
+|------|-------------|
+| `--base-url URL` | Override the server URL |
+| `--api-key KEY` | Override the API key |
+| `--json` | Output responses as JSON (useful for scripting) |
+| `--version` | Show CLI version |
+| `--help` | Show help message |
+
+## Commands
+
+### `health`
+
+Check if the server is reachable and healthy.
+
+```bash
+treadstone health
+# Server is healthy
+```
+
+### `auth`
+
+Authentication and user management.
+
+```bash
+treadstone auth register                 # Create a new account
+treadstone auth login                    # Log in interactively
+treadstone auth logout                   # Log out
+treadstone auth whoami                   # Show current user info
+treadstone auth change-password          # Change your password
+treadstone auth invite --email user@example.com   # Invite a user (admin)
+treadstone auth users                    # List all users (admin)
+treadstone auth delete-user <user-id>    # Delete a user (admin)
+```
+
+### `sandboxes` (alias: `sb`)
+
+Create and manage sandboxes.
+
+```bash
+treadstone sb create --template default --name my-box
+treadstone sb create --template python --label env:dev --persist
+treadstone sb list
+treadstone sb list --label env:prod --limit 10
+treadstone sb get <sandbox-id>
+treadstone sb start <sandbox-id>
+treadstone sb stop <sandbox-id>
+treadstone sb delete <sandbox-id>
+treadstone sb token <sandbox-id>                    # Get an access token
+treadstone sb token <sandbox-id> --expires-in 7200  # Custom TTL
+```
+
+### `templates`
+
+List available sandbox templates.
+
+```bash
+treadstone templates list
+```
+
+### `api-keys`
+
+Manage long-lived API keys.
+
+```bash
+treadstone api-keys create --name ci-bot
+treadstone api-keys create --name temp --expires-in 86400  # 24h
+treadstone api-keys list
+treadstone api-keys delete <key-id>
+```
+
+### `config`
+
+Manage local CLI configuration.
+
+```bash
+treadstone config set base_url https://...
+treadstone config set api_key ts_...
+treadstone config get
+treadstone config get base_url
+treadstone config unset api_key
+treadstone config path
+```
+
+## JSON output
+
+Pass `--json` before any command to get machine-readable JSON output:
+
+```bash
+treadstone --json sb list
+treadstone --json health
+treadstone --json api-keys list
+```
+
+## Error handling
+
+The CLI displays user-friendly error messages instead of stack traces:
+
+```
+Error: Connection refused.
+  Possible causes:
+    - The Treadstone server is not running
+    - The --base-url or TREADSTONE_BASE_URL is incorrect
+    - A firewall or proxy is blocking the connection
+  Detail: ...
+```
+
+Common errors:
+
+| Error | Cause | Fix |
+|-------|-------|-----|
+| Connection refused | Server not running or wrong URL | Check `treadstone config get base_url` |
+| Request timed out | Server slow or unreachable | Retry, or check network |
+| No API key configured | Missing authentication | `treadstone config set api_key <key>` |
+| HTTP 401 | Invalid or expired API key | Create a new key with `treadstone api-keys create` |
+| HTTP 404 | Resource not found | Verify the ID is correct |

treadstone_cli-0.2.1/README.md

@@ -0,0 +1,212 @@
+# Treadstone CLI
+
+Command-line interface for the Treadstone sandbox service.
+
+## Installation
+
+### From PyPI
+
+```bash
+pip install treadstone
+```
+
+### Pre-built binary
+
+Download the latest release from [GitHub Releases](https://github.com/earayu/treadstone/releases). The binary is self-contained and does not require Python.
+
+```bash
+# macOS / Linux
+chmod +x treadstone
+sudo mv treadstone /usr/local/bin/
+```
+
+## Quick start
+
+```bash
+# Check that the server is reachable
+treadstone health
+
+# Register an account (first time only)
+treadstone auth register
+
+# Log in
+treadstone auth login
+
+# Create an API key for non-interactive use
+treadstone api-keys create --name my-key
+# Save the key — it is shown only once
+
+# Store the key in config so you don't have to pass it every time
+treadstone config set api_key ts_live_xxxxxxxxxxxx
+
+# Create a sandbox
+treadstone sandboxes create --template default --name my-sandbox
+
+# List sandboxes
+treadstone sb list
+```
+
+## Configuration
+
+The CLI reads configuration from three sources, in order of priority:
+
+| Priority | Source | Example |
+|----------|--------|---------|
+| 1 (highest) | CLI flags | `--base-url https://... --api-key ts_...` |
+| 2 | Environment variables | `TREADSTONE_BASE_URL`, `TREADSTONE_API_KEY` |
+| 3 (lowest) | Config file | `~/.config/treadstone/config.toml` |
+
+### Config file
+
+Location: `~/.config/treadstone/config.toml`
+
+```toml
+[default]
+base_url = "https://your-server.example.com"
+api_key  = "ts_live_xxxxxxxxxxxx"
+```
+
+You can manage this file with the `config` subcommand:
+
+```bash
+# Set a value
+treadstone config set base_url https://your-server.example.com
+treadstone config set api_key ts_live_xxxxxxxxxxxx
+
+# View current settings (api_key is partially masked)
+treadstone config get
+
+# View a single key
+treadstone config get base_url
+
+# Remove a value
+treadstone config unset api_key
+
+# Show config file path
+treadstone config path
+```
+
+### Environment variables
+
+| Variable | Description | Default |
+|----------|-------------|---------|
+| `TREADSTONE_BASE_URL` | Base URL of the Treadstone server | `https://api.treadstone-ai.dev` |
+| `TREADSTONE_API_KEY` | API key for authentication | (none) |
+
+### Global flags
+
+| Flag | Description |
+|------|-------------|
+| `--base-url URL` | Override the server URL |
+| `--api-key KEY` | Override the API key |
+| `--json` | Output responses as JSON (useful for scripting) |
+| `--version` | Show CLI version |
+| `--help` | Show help message |
+
+## Commands
+
+### `health`
+
+Check if the server is reachable and healthy.
+
+```bash
+treadstone health
+# Server is healthy
+```
+
+### `auth`
+
+Authentication and user management.
+
+```bash
+treadstone auth register                 # Create a new account
+treadstone auth login                    # Log in interactively
+treadstone auth logout                   # Log out
+treadstone auth whoami                   # Show current user info
+treadstone auth change-password          # Change your password
+treadstone auth invite --email user@example.com   # Invite a user (admin)
+treadstone auth users                    # List all users (admin)
+treadstone auth delete-user <user-id>    # Delete a user (admin)
+```
+
+### `sandboxes` (alias: `sb`)
+
+Create and manage sandboxes.
+
+```bash
+treadstone sb create --template default --name my-box
+treadstone sb create --template python --label env:dev --persist
+treadstone sb list
+treadstone sb list --label env:prod --limit 10
+treadstone sb get <sandbox-id>
+treadstone sb start <sandbox-id>
+treadstone sb stop <sandbox-id>
+treadstone sb delete <sandbox-id>
+treadstone sb token <sandbox-id>                    # Get an access token
+treadstone sb token <sandbox-id> --expires-in 7200  # Custom TTL
+```
+
+### `templates`
+
+List available sandbox templates.
+
+```bash
+treadstone templates list
+```
+
+### `api-keys`
+
+Manage long-lived API keys.
+
+```bash
+treadstone api-keys create --name ci-bot
+treadstone api-keys create --name temp --expires-in 86400  # 24h
+treadstone api-keys list
+treadstone api-keys delete <key-id>
+```
+
+### `config`
+
+Manage local CLI configuration.
+
+```bash
+treadstone config set base_url https://...
+treadstone config set api_key ts_...
+treadstone config get
+treadstone config get base_url
+treadstone config unset api_key
+treadstone config path
+```
+
+## JSON output
+
+Pass `--json` before any command to get machine-readable JSON output:
+
+```bash
+treadstone --json sb list
+treadstone --json health
+treadstone --json api-keys list
+```
+
+## Error handling
+
+The CLI displays user-friendly error messages instead of stack traces:
+
+```
+Error: Connection refused.
+  Possible causes:
+    - The Treadstone server is not running
+    - The --base-url or TREADSTONE_BASE_URL is incorrect
+    - A firewall or proxy is blocking the connection
+  Detail: ...
+```
+
+Common errors:
+
+| Error | Cause | Fix |
+|-------|-------|-----|
+| Connection refused | Server not running or wrong URL | Check `treadstone config get base_url` |
+| Request timed out | Server slow or unreachable | Retry, or check network |
+| No API key configured | Missing authentication | `treadstone config set api_key <key>` |
+| HTTP 401 | Invalid or expired API key | Create a new key with `treadstone api-keys create` |
+| HTTP 404 | Resource not found | Verify the ID is correct |

treadstone_cli-0.2.1/pyproject.toml

@@ -0,0 +1,25 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "treadstone-cli"
+version = "0.2.1"
+description = "CLI for the Treadstone sandbox service."
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "click>=8.1",
+    "httpx>=0.28.1",
+    "rich>=14.0",
+]
+
+[project.scripts]
+treadstone = "treadstone_cli.main:cli"
+
+[tool.ruff]
+target-version = "py312"
+line-length = 120
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "UP"]

treadstone_cli-0.2.1/treadstone_cli/__main__.py

@@ -0,0 +1,6 @@
+"""Entry point for `python -m treadstone_cli` and PyInstaller builds."""
+
+from treadstone_cli.main import cli
+
+if __name__ == "__main__":
+    cli()

treadstone_cli-0.2.1/treadstone_cli/_client.py

@@ -0,0 +1,77 @@
+"""HTTP client factory for CLI commands.
+
+Priority: CLI flags > environment variables > config file.
+"""
+
+from __future__ import annotations
+
+import os
+import sys
+from pathlib import Path
+
+import click
+import httpx
+
+CONFIG_DIR = Path.home() / ".config" / "treadstone"
+CONFIG_FILE = CONFIG_DIR / "config.toml"
+
+_DEFAULT_BASE_URL = "https://api.treadstone-ai.dev"
+
+
+def _read_config() -> dict[str, str]:
+    if not CONFIG_FILE.exists():
+        return {}
+    try:
+        import tomllib
+    except ModuleNotFoundError:
+        import tomli as tomllib  # type: ignore[no-redef]
+    with open(CONFIG_FILE, "rb") as f:
+        data = tomllib.load(f)
+    return data.get("default", {})
+
+
+def get_base_url(ctx: click.Context) -> str:
+    url = ctx.obj.get("base_url") or _read_config().get("base_url") or _DEFAULT_BASE_URL
+    return url.rstrip("/")
+
+
+def get_api_key(ctx: click.Context) -> str | None:
+    return ctx.obj.get("api_key") or _read_config().get("api_key")
+
+
+def effective_base_url() -> tuple[str, str]:
+    """Return (url, source) without requiring a Click context.
+
+    Source is one of: 'env', 'file', 'default'.
+    Used for displaying configuration in help text.
+    """
+    env = os.environ.get("TREADSTONE_BASE_URL")
+    if env:
+        return env.rstrip("/"), "env"
+    cfg = _read_config().get("base_url")
+    if cfg:
+        return cfg.rstrip("/"), "file"
+    return _DEFAULT_BASE_URL, "default"
+
+
+def effective_api_key() -> str | None:
+    """Return API key without requiring a Click context."""
+    return os.environ.get("TREADSTONE_API_KEY") or _read_config().get("api_key")
+
+
+def build_client(ctx: click.Context) -> httpx.Client:
+    base_url = get_base_url(ctx)
+    api_key = get_api_key(ctx)
+    headers: dict[str, str] = {}
+    if api_key:
+        headers["Authorization"] = f"Bearer {api_key}"
+    return httpx.Client(base_url=base_url, headers=headers, timeout=30.0)
+
+
+def require_auth(ctx: click.Context) -> httpx.Client:
+    """Build client and abort if no API key is configured."""
+    client = build_client(ctx)
+    if "Authorization" not in client.headers:
+        click.echo("Error: No API key configured. Set TREADSTONE_API_KEY or use --api-key.", err=True)
+        sys.exit(1)
+    return client

treadstone_cli-0.2.1/treadstone_cli/_output.py

@@ -0,0 +1,102 @@
+"""Output formatting utilities for CLI — table (Rich) or JSON mode."""
+
+from __future__ import annotations
+
+import json
+import sys
+from typing import Any
+
+import click
+import httpx
+from rich.console import Console
+from rich.table import Table
+
+console = Console()
+
+
+def friendly_exception_handler(cli_main: click.Group) -> click.Group:
+    """Wrap a Click group so unhandled exceptions print user-friendly messages instead of tracebacks."""
+    original_main = cli_main.main
+
+    def _patched_main(*args: Any, **kwargs: Any) -> Any:
+        try:
+            return original_main(*args, standalone_mode=False, **kwargs)
+        except click.exceptions.Abort:
+            click.echo("\nAborted.", err=True)
+            sys.exit(130)
+        except click.exceptions.Exit as exc:
+            sys.exit(exc.exit_code)
+        except click.UsageError as exc:
+            exc.show()
+            sys.exit(exc.exit_code if exc.exit_code is not None else 2)
+        except SystemExit:
+            raise
+        except KeyboardInterrupt:
+            click.echo("\nInterrupted.", err=True)
+            sys.exit(130)
+        except httpx.ConnectError as exc:
+            _print_network_hint(str(exc), "Connection refused")
+            sys.exit(1)
+        except httpx.TimeoutException:
+            click.echo("Error: Request timed out. The server may be slow or unreachable.", err=True)
+            sys.exit(1)
+        except httpx.HTTPStatusError as exc:
+            click.echo(f"Error: HTTP {exc.response.status_code} — {exc.response.text[:200]}", err=True)
+            sys.exit(1)
+        except httpx.HTTPError as exc:
+            click.echo(f"Error: {type(exc).__name__} — {exc}", err=True)
+            sys.exit(1)
+        except Exception as exc:
+            click.echo(f"Error: {exc}", err=True)
+            sys.exit(1)
+
+    cli_main.main = _patched_main  # type: ignore[assignment]
+    return cli_main
+
+
+def _print_network_hint(detail: str, summary: str) -> None:
+    click.echo(f"Error: {summary}.", err=True)
+    click.echo("  Possible causes:", err=True)
+    click.echo("    - The Treadstone server is not running", err=True)
+    click.echo("    - The --base-url or TREADSTONE_BASE_URL is incorrect", err=True)
+    click.echo("    - A firewall or proxy is blocking the connection", err=True)
+    click.echo(f"  Detail: {detail}", err=True)
+
+
+def is_json_mode(ctx: click.Context) -> bool:
+    return ctx.obj.get("json_output", False)
+
+
+def print_json(data: Any) -> None:
+    click.echo(json.dumps(data, indent=2, default=str))
+
+
+def print_table(columns: list[str], rows: list[list[Any]], title: str | None = None) -> None:
+    table = Table(title=title, show_header=True, header_style="bold cyan")
+    for col in columns:
+        table.add_column(col)
+    for row in rows:
+        table.add_row(*[str(v) if v is not None else "" for v in row])
+    console.print(table)
+
+
+def print_detail(data: dict[str, Any], title: str | None = None) -> None:
+    table = Table(title=title, show_header=False)
+    table.add_column("Field", style="bold")
+    table.add_column("Value")
+    for k, v in data.items():
+        table.add_row(k, str(v) if v is not None else "")
+    console.print(table)
+
+
+def handle_error(resp: Any) -> None:
+    """Print error and exit if response is not 2xx."""
+    if resp.status_code >= 400:
+        try:
+            body = resp.json()
+            err = body.get("error", body)
+            msg = err.get("message", str(err)) if isinstance(err, dict) else str(err)
+        except Exception:
+            msg = resp.text
+        click.echo(f"Error ({resp.status_code}): {msg}", err=True)
+        sys.exit(1)