bitsreef-cli 0.1.0__py3-none-any.whl

bitsreef_cli/__init__.py

@@ -0,0 +1,2 @@
+"""BitsReef CLI — command-line interface for the BitsReef PaaS platform."""
+__version__ = "0.1.0"

bitsreef_cli/client.py

@@ -0,0 +1,74 @@
+"""HTTP client for BitsReef API with automatic token refresh."""
+import sys
+
+import click
+import httpx
+
+from . import config
+
+
+class BitsReefClient:
+    """Authenticated HTTP client for the BitsReef API."""
+
+    def __init__(self):
+        self.base_url = config.get_api_url()
+        access, refresh = config.get_tokens()
+        self.access_token = access
+        self.refresh_token = refresh
+
+    def _headers(self) -> dict:
+        h = {"Content-Type": "application/json"}
+        if self.access_token:
+            h["Authorization"] = f"Bearer {self.access_token}"
+        return h
+
+    def _refresh(self) -> bool:
+        """Attempt to refresh the access token. Returns True on success."""
+        if not self.refresh_token:
+            return False
+        try:
+            resp = httpx.post(
+                f"{self.base_url}/account/token/refresh/",
+                json={"refresh": self.refresh_token},
+                timeout=10,
+            )
+            if resp.status_code == 200:
+                data = resp.json()
+                self.access_token = data["access"]
+                config.save_tokens(self.access_token, self.refresh_token)
+                return True
+        except httpx.HTTPError:
+            pass
+        return False
+
+    def request(self, method: str, path: str, **kwargs) -> httpx.Response:
+        """Make an authenticated request. Auto-refreshes token on 401."""
+        url = f"{self.base_url}{path}"
+        resp = httpx.request(method, url, headers=self._headers(), timeout=30, **kwargs)
+        if resp.status_code == 401 and self._refresh():
+            resp = httpx.request(method, url, headers=self._headers(), timeout=30, **kwargs)
+        if resp.status_code == 401:
+            click.secho("Session expired. Please run: bitsreef auth login", fg="red")
+            sys.exit(1)
+        return resp
+
+    def get(self, path: str, **kwargs) -> httpx.Response:
+        return self.request("GET", path, **kwargs)
+
+    def post(self, path: str, **kwargs) -> httpx.Response:
+        return self.request("POST", path, **kwargs)
+
+    def patch(self, path: str, **kwargs) -> httpx.Response:
+        return self.request("PATCH", path, **kwargs)
+
+    def delete(self, path: str, **kwargs) -> httpx.Response:
+        return self.request("DELETE", path, **kwargs)
+
+
+def get_client() -> BitsReefClient:
+    """Return an authenticated client, exiting if not logged in."""
+    access, _ = config.get_tokens()
+    if not access:
+        click.secho("Not logged in. Run: bitsreef auth login", fg="red")
+        sys.exit(1)
+    return BitsReefClient()

bitsreef_cli/commands/auth.py

@@ -0,0 +1,91 @@
+"""Auth commands: login, logout, whoami."""
+import sys
+
+import click
+import httpx
+from rich.console import Console
+
+from .. import config, output
+
+console = Console()
+
+
+@click.group()
+def auth():
+    """Authenticate with BitsReef."""
+
+
+@auth.command()
+@click.option("--username", "-u", default=None, help="Username (else prompted)")
+@click.option("--password-stdin", "password_stdin", is_flag=True,
+              help="Read the password from stdin (non-interactive, for CI)")
+@click.option("--token", default=None,
+              help="Store a pre-issued access token instead of logging in")
+def login(username, password_stdin, token):
+    """Log in with username/password, or store a token non-interactively.
+
+    The CLI always talks to the hosted BitsReef API — there is no server URL to
+    set.
+
+    CI examples:
+        echo "$PASS" | bitsreef auth login -u ci --password-stdin
+        bitsreef auth login --token "$BITSREEF_TOKEN"
+    (Or skip login entirely and set BITSREEF_TOKEN.)
+    """
+    api_url = config.get_api_url()
+
+    # Non-interactive: store a pre-issued access token (no refresh token).
+    if token:
+        config.save_tokens(token, "")
+        console.print(f"[green]Token stored[/green] for {api_url}")
+        return
+
+    if not username:
+        username = click.prompt("Username")
+    if password_stdin:
+        password = sys.stdin.readline().rstrip("\n")
+    else:
+        password = click.prompt("Password", hide_input=True)
+
+    try:
+        resp = httpx.post(
+            f"{api_url}/account/token/",
+            json={"username": username, "password": password},
+            timeout=15,
+        )
+    except httpx.ConnectError:
+        console.print(f"[red]Cannot connect to {api_url}[/red]")
+        raise SystemExit(1)
+
+    if resp.status_code != 200:
+        console.print("[red]Login failed — check username/password.[/red]")
+        raise SystemExit(1)
+
+    data = resp.json()
+    config.save_tokens(data["access"], data["refresh"])
+    console.print(f"[green]Logged in as [bold]{username}[/bold][/green]")
+
+
+@auth.command()
+def logout():
+    """Log out and clear stored tokens."""
+    config.clear_tokens()
+    console.print("[green]Logged out.[/green]")
+
+
+@auth.command()
+def whoami():
+    """Show the current logged-in user."""
+    from ..client import get_client
+
+    client = get_client()
+    resp = client.get("/account/me/")
+    if resp.status_code != 200:
+        console.print("[red]Could not fetch user info.[/red]")
+        raise SystemExit(1)
+
+    user = resp.json()
+    if output.json_enabled():
+        output.emit(user)
+        return
+    console.print(f"[bold]{user['username']}[/bold]  ({user['email']})")

bitsreef_cli/commands/completion.py

@@ -0,0 +1,29 @@
+"""Shell-completion command — emit a completion script for bash/zsh/fish."""
+import click
+from click.shell_completion import get_completion_class
+
+_INSTALL_HINTS = {
+    "bash": 'eval "$(bitsreef completion bash)"  # add to ~/.bashrc',
+    "zsh": 'eval "$(bitsreef completion zsh)"    # add to ~/.zshrc',
+    "fish": "bitsreef completion fish > ~/.config/fish/completions/bitsreef.fish",
+}
+
+
+@click.command()
+@click.argument("shell", type=click.Choice(["bash", "zsh", "fish"]))
+def completion(shell):
+    """Print a shell-completion script for SHELL (bash, zsh, or fish).
+
+    Enable it by sourcing the output, e.g.:
+
+        eval "$(bitsreef completion bash)"
+
+    Add that line to your shell's rc file to make it permanent.
+    """
+    from ..main import cli as root
+
+    comp_cls = get_completion_class(shell)
+    if comp_cls is None:
+        raise click.ClickException(f"Completion is not supported for {shell}.")
+    comp = comp_cls(root, {}, "bitsreef", "_BITSREEF_COMPLETE")
+    click.echo(comp.source())

bitsreef_cli/commands/cron.py

@@ -0,0 +1,166 @@
+"""Cron-job commands: list, create, trigger, runs, delete."""
+import click
+from rich.console import Console
+from rich.table import Table
+
+from ..client import get_client
+from .. import context, output
+
+console = Console()
+
+
+def _service(client, service):
+    """Resolve --service (id/name) or the linked service."""
+    link = context.read_link()
+    return context.resolve_service(client, link.get("project_id"), service, link)
+
+
+def _results(resp):
+    data = resp.json()
+    return data.get("results", []) if isinstance(data, dict) else data
+
+
+def _job_id(client, service_id, value):
+    """Resolve a cron-job id from an id or a job name."""
+    if str(value).isdigit():
+        return int(value)
+    resp = client.get(f"/services/{service_id}/cron-jobs/")
+    for j in _results(resp):
+        if j.get("name") == value:
+            return j["id"]
+    raise click.ClickException(f"No cron job '{value}' on this service.")
+
+
+@click.group()
+def cron():
+    """Manage scheduled (cron) jobs for a service."""
+
+
+@cron.command("list")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def list_jobs(service):
+    """List cron jobs on a service."""
+    client = get_client()
+    service_id = _service(client, service)
+    resp = client.get(f"/services/{service_id}/cron-jobs/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch cron jobs: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    results = _results(resp)
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No cron jobs.[/dim]")
+        return
+
+    table = Table(title="Cron Jobs")
+    table.add_column("ID", style="dim")
+    table.add_column("Name", style="bold")
+    table.add_column("Schedule")
+    table.add_column("Enabled")
+    table.add_column("Last run")
+    table.add_column("Next run")
+    for j in results:
+        enabled = "[green]yes[/green]" if j.get("enabled") else "[red]no[/red]"
+        status = j.get("last_run_status")
+        last = (j.get("last_run") or "—")[:19]
+        if status:
+            last = f"{last} ({status})"
+        table.add_row(
+            str(j["id"]), j["name"], j.get("schedule", "—"),
+            enabled, last, (j.get("next_run") or "—")[:19],
+        )
+    console.print(table)
+
+
+@cron.command("create")
+@click.argument("name")
+@click.option("--schedule", required=True, help="Cron expression, e.g. '0 * * * *'")
+@click.option("--command", required=True, help="Command to run inside the container")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+@click.option("--disabled", is_flag=True, help="Create the job disabled")
+def create_job(name, schedule, command, service, disabled):
+    """Create a cron job on a service."""
+    client = get_client()
+    service_id = _service(client, service)
+    payload = {
+        "name": name, "schedule": schedule,
+        "command": command, "enabled": not disabled,
+    }
+    resp = client.post(f"/services/{service_id}/cron-jobs/", json=payload)
+    if resp.status_code == 201:
+        j = resp.json()
+        console.print(f"[green]Created[/green] [bold]{j['name']}[/bold] (id={j['id']})  next={j.get('next_run') or '—'}")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@cron.command("trigger")
+@click.argument("job")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def trigger_job(job, service):
+    """Run a cron job now (by name or id)."""
+    client = get_client()
+    service_id = _service(client, service)
+    job_id = _job_id(client, service_id, job)
+    resp = client.post(f"/services/{service_id}/cron-jobs/{job_id}/trigger/")
+    if resp.status_code == 202:
+        console.print("[green]Triggered.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@cron.command("runs")
+@click.argument("job")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def job_runs(job, service):
+    """Show recent run history for a cron job (by name or id)."""
+    client = get_client()
+    service_id = _service(client, service)
+    job_id = _job_id(client, service_id, job)
+    resp = client.get(f"/services/{service_id}/cron-jobs/{job_id}/runs/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch runs: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    results = _results(resp)
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No runs yet.[/dim]")
+        return
+
+    table = Table(title="Cron Runs")
+    table.add_column("ID", style="dim")
+    table.add_column("Status")
+    table.add_column("Exit")
+    table.add_column("Started")
+    table.add_column("Completed")
+    for r in results:
+        s = r.get("status", "")
+        color = "green" if s == "success" else "red" if s == "failed" else "yellow"
+        table.add_row(
+            str(r["id"]), f"[{color}]{s}[/{color}]",
+            str(r.get("exit_code") if r.get("exit_code") is not None else "—"),
+            (r.get("started_at") or "—")[:19], (r.get("completed_at") or "—")[:19],
+        )
+    console.print(table)
+
+
+@cron.command("delete")
+@click.argument("job")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+@click.confirmation_option(prompt="Delete this cron job?")
+def delete_job(job, service):
+    """Delete a cron job (by name or id)."""
+    client = get_client()
+    service_id = _service(client, service)
+    job_id = _job_id(client, service_id, job)
+    resp = client.delete(f"/services/{service_id}/cron-jobs/{job_id}/")
+    if resp.status_code == 204:
+        console.print("[green]Deleted.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/deploy.py

@@ -0,0 +1,124 @@
+"""Deploy commands: up, status, redeploy."""
+import click
+from rich.console import Console
+from rich.table import Table
+
+from ..client import get_client
+from .. import context, output
+
+console = Console()
+
+
+@click.group()
+def deploy():
+    """Deploy and manage service deployments."""
+
+
+@deploy.command("up")
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+@click.option("--follow", "-f", is_flag=True, help="Stream the deploy to completion (non-zero exit on failure)")
+def deploy_up(project_id, service_id, follow):
+    """Deploy (or redeploy) a service (linked project/service if omitted)."""
+    client = get_client()
+    link = context.read_link()
+    project_id = context.resolve_project(client, project_id, link)
+    service_id = context.resolve_service(client, project_id, service_id, link)
+    resp = client.post(f"/projects/{project_id}/services/{service_id}/deploy/")
+    if resp.status_code not in (200, 202):
+        console.print(f"[red]Failed: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    data = resp.json()
+    deployment_id = data.get("deployment_id")
+    console.print("[green]Deployment queued.[/green]")
+    if deployment_id:
+        console.print(f"  deployment_id={deployment_id}")
+
+    if follow and deployment_id:
+        from ..deploy_watch import follow_deployment
+
+        success, _ = follow_deployment(client, service_id, deployment_id)
+        if not success:
+            raise SystemExit(1)
+
+
+@deploy.command("rebuild")
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+def deploy_rebuild(project_id, service_id):
+    """Force rebuild and deploy a repo-linked service (linked if omitted)."""
+    client = get_client()
+    link = context.read_link()
+    project_id = context.resolve_project(client, project_id, link)
+    service_id = context.resolve_service(client, project_id, service_id, link)
+    resp = client.post(f"/projects/{project_id}/services/{service_id}/rebuild/")
+    if resp.status_code in (200, 202):
+        console.print("[green]Rebuild queued.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@deploy.command("status")
+@click.argument("project_id", required=False, default=None)
+@click.option("--limit", "-n", default=10, type=int, help="Number of deployments to show")
+def deploy_status(project_id, limit):
+    """Show recent deployments for a project (linked project if omitted)."""
+    client = get_client()
+    project_id = context.resolve_project(client, project_id, context.read_link())
+    resp = client.get(f"/projects/{project_id}/activity/")
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch deployments.[/red]")
+        raise SystemExit(1)
+
+    results = resp.json().get("results", [])[:limit]
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No deployments.[/dim]")
+        return
+
+    table = Table(title="Recent Deployments")
+    table.add_column("ID", style="dim")
+    table.add_column("Service")
+    table.add_column("Status")
+    table.add_column("Trigger")
+    table.add_column("Image Tag")
+    table.add_column("Created")
+    for d in results:
+        s = d["status"]
+        color = "green" if s == "running" else "red" if s == "failed" else "yellow" if s in ("building", "deploying", "pushing") else "dim"
+        table.add_row(
+            str(d["id"]),
+            d.get("service_name", "—"),
+            f"[{color}]{s}[/{color}]",
+            d.get("trigger_type", "—"),
+            d.get("image_tag", "—")[:40],
+            d["created_at"][:19],
+        )
+    console.print(table)
+
+
+@deploy.command("rollback")
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+@click.option("--to", "-d", "deployment_id", type=int, required=True, help="Deployment ID to roll back to")
+def deploy_rollback(project_id, service_id, deployment_id):
+    """Roll back a service to a previous deployment (linked if omitted).
+
+    Example: `bitsreef deploy rollback -d 42` or
+    `bitsreef deploy rollback my-proj web -d 42`.
+    """
+    client = get_client()
+    link = context.read_link()
+    project_id = context.resolve_project(client, project_id, link)
+    service_id = context.resolve_service(client, project_id, service_id, link)
+    resp = client.post(
+        f"/projects/{project_id}/services/{service_id}/rollback/",
+        json={"deployment_id": deployment_id},
+    )
+    if resp.status_code in (200, 202):
+        console.print("[green]Rollback queued.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/domains.py

@@ -0,0 +1,119 @@
+"""Custom domain commands: list, add, remove, verify."""
+import click
+from rich.console import Console
+from rich.table import Table
+
+from ..client import get_client
+from .. import context, output
+
+console = Console()
+
+
+def _service(client, service):
+    """Resolve --service (id/name) or the linked service."""
+    link = context.read_link()
+    return context.resolve_service(client, link.get("project_id"), service, link)
+
+
+def _results(resp):
+    data = resp.json()
+    return data.get("results", []) if isinstance(data, dict) else data
+
+
+def _domain_id(client, service_id, value):
+    """Resolve a domain id from an id or a domain name."""
+    if str(value).isdigit():
+        return int(value)
+    resp = client.get(f"/services/{service_id}/domains/")
+    for d in _results(resp):
+        if d.get("domain_name") == value:
+            return d["id"]
+    raise click.ClickException(f"No domain '{value}' on this service.")
+
+
+@click.group()
+def domains():
+    """Manage custom domains for a service."""
+
+
+@domains.command("list")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def list_domains(service):
+    """List custom domains attached to a service."""
+    client = get_client()
+    service_id = _service(client, service)
+    resp = client.get(f"/services/{service_id}/domains/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch domains: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    results = _results(resp)
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No domains.[/dim]")
+        return
+
+    table = Table(title="Domains")
+    table.add_column("ID", style="dim")
+    table.add_column("Domain", style="bold")
+    table.add_column("SSL")
+    table.add_column("DNS verified")
+    for d in results:
+        verified = "[green]yes[/green]" if d.get("dns_verified") else "[yellow]no[/yellow]"
+        table.add_row(str(d["id"]), d["domain_name"], d.get("ssl_status", "—"), verified)
+    console.print(table)
+
+
+@domains.command("add")
+@click.argument("domain")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def add_domain(domain, service):
+    """Attach a custom domain to a service."""
+    client = get_client()
+    service_id = _service(client, service)
+    resp = client.post(f"/services/{service_id}/domains/", json={"domain_name": domain})
+    if resp.status_code == 201:
+        d = resp.json()
+        console.print(f"[green]Added[/green] {d['domain_name']} (id={d['id']})")
+        if d.get("verification_token"):
+            console.print(f"  [dim]DNS verification token: {d['verification_token']}[/dim]")
+        console.print("  Run [bold]bitsreef domains verify[/bold] once DNS is configured.")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@domains.command("remove")
+@click.argument("domain")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def remove_domain(domain, service):
+    """Remove a custom domain (by name or id)."""
+    client = get_client()
+    service_id = _service(client, service)
+    domain_id = _domain_id(client, service_id, domain)
+    resp = client.delete(f"/services/{service_id}/domains/{domain_id}/")
+    if resp.status_code == 204:
+        console.print("[green]Removed.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@domains.command("verify")
+@click.argument("domain")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def verify_domain(domain, service):
+    """Trigger DNS verification for a custom domain (by name or id)."""
+    client = get_client()
+    service_id = _service(client, service)
+    domain_id = _domain_id(client, service_id, domain)
+    resp = client.post(f"/services/{service_id}/domains/{domain_id}/verify/")
+    if resp.status_code in (200, 202):
+        d = resp.json() if resp.text else {}
+        verified = d.get("dns_verified")
+        if verified:
+            console.print("[green]Verified.[/green]")
+        else:
+            console.print("[yellow]Verification pending — DNS may not have propagated yet.[/yellow]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")