bitsreef-cli 0.1.0__py3-none-any.whl

bitsreef_cli/commands/projects.py

@@ -0,0 +1,89 @@
+"""Project commands: list, info, create."""
+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 projects():
+    """Manage projects."""
+
+
+@projects.command("list")
+@click.option("--org", type=int, default=None, help="Organization ID")
+def list_projects(org):
+    """List all projects in an organization."""
+    client = get_client()
+    org_id = context.resolve_org(client, org, context.read_link())
+    resp = client.get(f"/organizations/{org_id}/projects/")
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch projects.[/red]")
+        raise SystemExit(1)
+
+    results = resp.json().get("results", [])
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No projects.[/dim]")
+        return
+
+    table = Table(title="Projects")
+    table.add_column("ID", style="dim")
+    table.add_column("Name", style="bold")
+    table.add_column("Slug")
+    table.add_column("Services", justify="right")
+    for p in results:
+        table.add_row(str(p["id"]), p["name"], p["slug"], str(p.get("service_count", 0)))
+    console.print(table)
+
+
+@projects.command("info")
+@click.argument("project_id", required=False, default=None)
+@click.option("--org", type=int, default=None, help="Organization ID")
+def project_info(project_id, org):
+    """Show project details (id or name; linked project if omitted)."""
+    client = get_client()
+    link = context.read_link()
+    org_id = context.resolve_org(client, org, link)
+    project_id = context.resolve_project(client, project_id, link, org=org_id)
+    resp = client.get(f"/organizations/{org_id}/projects/{project_id}/")
+    if resp.status_code == 404:
+        console.print("[red]Project not found.[/red]")
+        raise SystemExit(1)
+
+    p = resp.json()
+    if output.json_enabled():
+        output.emit(p)
+        return
+    console.print(f"[bold]{p['name']}[/bold]  (id={p['id']}, slug={p['slug']})")
+    console.print(f"  Description: {p.get('description', '—')}")
+    console.print(f"  Created: {p['created_at']}")
+
+    svcs = p.get("services", [])
+    if svcs:
+        console.print(f"\n  [bold]Services ({len(svcs)}):[/bold]")
+        for s in svcs:
+            status_color = "green" if s["status"] == "running" else "yellow" if s["status"] in ("building", "deploying") else "red"
+            console.print(f"    [{status_color}]{s['status']:10}[/{status_color}] {s['name']}  (image={s.get('image', '—')})")
+
+
+@projects.command("create")
+@click.argument("name")
+@click.option("--org", type=int, default=None, help="Organization ID")
+@click.option("--description", "-d", default="", help="Project description")
+def create_project(name, org, description):
+    """Create a new project."""
+    client = get_client()
+    org_id = context.resolve_org(client, org, context.read_link())
+    resp = client.post(f"/organizations/{org_id}/projects/", json={"name": name, "description": description})
+    if resp.status_code == 201:
+        p = resp.json()
+        console.print(f"[green]Created project [bold]{p['name']}[/bold] (id={p['id']})[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/services.py

@@ -0,0 +1,149 @@
+"""Service commands: list, info, restart, stop, scale."""
+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 services():
+    """Manage services within a project."""
+
+
+@services.command("list")
+@click.argument("project_id", required=False, default=None)
+@click.option("--environment", "-E", default=None,
+              help="Filter to an environment id/name (else all environments)")
+def list_services(project_id, environment):
+    """List services in a project (linked project if omitted).
+
+    Pass --environment to show only one environment's services.
+    """
+    client = get_client()
+    project_id = context.resolve_project(client, project_id, context.read_link())
+    path = f"/projects/{project_id}/services/"
+    if environment is not None:
+        env_id = context.resolve_environment(client, project_id, environment)
+        path += f"?environment={env_id}"
+    resp = client.get(path)
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch services.[/red]")
+        raise SystemExit(1)
+
+    results = resp.json().get("results", [])
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No services.[/dim]")
+        return
+
+    table = Table(title="Services")
+    table.add_column("ID", style="dim")
+    table.add_column("Name", style="bold")
+    table.add_column("Status")
+    table.add_column("Image")
+    table.add_column("Replicas", justify="right")
+    table.add_column("URL")
+    for s in results:
+        status_color = "green" if s["status"] == "running" else "yellow" if s["status"] in ("building", "deploying") else "red" if s["status"] in ("failed", "stopped") else "dim"
+        table.add_row(
+            str(s["id"]),
+            s["name"],
+            f"[{status_color}]{s['status']}[/{status_color}]",
+            s.get("image") or s.get("repository", {}).get("repo_full_name", "—") if s.get("repository") else s.get("image", "—"),
+            str(s.get("current_replicas", 0)),
+            s.get("assigned_subdomain") or "—",
+        )
+    console.print(table)
+
+
+@services.command("info")
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+def service_info(project_id, service_id):
+    """Show detailed service info (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.get(f"/projects/{project_id}/services/{service_id}/")
+    if resp.status_code == 404:
+        console.print("[red]Service not found.[/red]")
+        raise SystemExit(1)
+
+    s = resp.json()
+    if output.json_enabled():
+        output.emit(s)
+        return
+    status_color = "green" if s["status"] == "running" else "yellow" if s["status"] in ("building", "deploying") else "red"
+    console.print(f"[bold]{s['name']}[/bold]  (id={s['id']})")
+    console.print(f"  Status: [{status_color}]{s['status']}[/{status_color}]")
+    console.print(f"  Source: {s.get('source_type', '—')}")
+    console.print(f"  Image: {s.get('image', '—')}")
+    console.print(f"  Replicas: {s.get('current_replicas', 0)} (min={s.get('min_replicas', 0)}, max={s.get('max_replicas', 1)})")
+    console.print(f"  CPU: {s.get('cpu_limit', '—')}  Memory: {s.get('memory_limit', '—')} MB")
+    if s.get("assigned_subdomain"):
+        console.print(f"  URL: {s['assigned_subdomain']}")
+    if s.get("private_domain"):
+        console.print(f"  Private: {s['private_domain']}")
+
+
+@services.command()
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+def restart(project_id, service_id):
+    """Restart 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}/restart/")
+    if resp.status_code in (200, 202):
+        console.print("[green]Restart queued.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@services.command()
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+def stop(project_id, service_id):
+    """Stop 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}/stop/")
+    if resp.status_code in (200, 202):
+        console.print("[green]Stop queued.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@services.command()
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+@click.option("--replicas", "-r", type=int, required=True, help="Target replica count")
+def scale(project_id, service_id, replicas):
+    """Scale a service to N replicas (linked project/service if omitted).
+
+    Examples: `bitsreef services scale -r 3` (linked) or
+    `bitsreef services scale my-proj web -r 3`.
+    """
+    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.patch(
+        f"/projects/{project_id}/services/{service_id}/",
+        json={"min_replicas": replicas, "max_replicas": max(replicas, 1)},
+    )
+    if resp.status_code == 200:
+        console.print(f"[green]Scaled to {replicas} replicas.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/templates.py

@@ -0,0 +1,106 @@
+"""Template commands: list, info, deploy (service templates)."""
+import click
+from rich.console import Console
+from rich.table import Table
+
+from ..client import get_client
+from .. import context, output
+
+console = Console()
+
+
+def _results(resp):
+    data = resp.json()
+    return data.get("results", []) if isinstance(data, dict) else data
+
+
+@click.group()
+def templates():
+    """Browse and deploy service templates."""
+
+
+@templates.command("list")
+def list_templates():
+    """List available service templates."""
+    client = get_client()
+    resp = client.get("/projects/templates/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch templates: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    results = _results(resp)
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No templates.[/dim]")
+        return
+
+    table = Table(title="Templates")
+    table.add_column("Slug", style="bold")
+    table.add_column("Name")
+    table.add_column("Category")
+    table.add_column("Official")
+    for t in results:
+        official = "[green]yes[/green]" if t.get("is_official") else ""
+        table.add_row(t.get("slug", "—"), t.get("name", "—"),
+                      t.get("category", "—"), official)
+    console.print(table)
+
+
+@templates.command("info")
+@click.argument("slug")
+def template_info(slug):
+    """Show a template's details and the services it deploys."""
+    client = get_client()
+    resp = client.get(f"/projects/templates/{slug}/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch template: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    t = resp.json()
+    if output.json_enabled():
+        output.emit(t)
+        return
+
+    console.print(f"[bold]{t.get('name')}[/bold]  [dim]({t.get('slug')})[/dim]")
+    if t.get("description"):
+        console.print(t["description"])
+    console.print(f"Category: {t.get('category', '—')}")
+    services = (t.get("config") or {}).get("services", [])
+    if services:
+        console.print("\nServices:")
+        for svc in services:
+            image = svc.get("image", "—")
+            console.print(f"  • [bold]{svc.get('name')}[/bold]  {image}")
+
+
+@templates.command("deploy")
+@click.argument("slug")
+@click.argument("project", required=False, default=None)
+@click.option("--var", "-v", "variables", multiple=True, metavar="KEY=VALUE",
+              help="Override a template variable (repeatable)")
+def deploy_template(slug, project, variables):
+    """Deploy a template into a project (linked project if omitted)."""
+    client = get_client()
+    project_id = context.resolve_project(client, project, context.read_link())
+
+    overrides = {}
+    for item in variables:
+        if "=" not in item:
+            raise click.ClickException(f"--var must be KEY=VALUE, got '{item}'")
+        key, value = item.split("=", 1)
+        overrides[key] = value
+
+    resp = client.post(
+        f"/projects/{project_id}/deploy-template/{slug}/",
+        json={"variable_overrides": overrides},
+    )
+    if resp.status_code == 201:
+        if output.json_enabled():
+            output.emit(resp.json())
+            return
+        console.print(f"[green]Deployed template[/green] [bold]{slug}[/bold] into project {project_id}.")
+        console.print("Run [bold]bitsreef services list[/bold] to see the new services.")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/up.py

@@ -0,0 +1,195 @@
+"""`bitsreef up` — create-or-update a service from an image and deploy it.
+
+The flagship one-shot: go from an image to a running service in a single
+command. If a service with the given name already exists in the project its
+image (and any provided resources/env) is updated; otherwise it is created.
+Then a deployment is triggered, and with --follow the command streams the
+build/deploy to completion and exits non-zero if it fails — so it works in CI.
+"""
+import json as jsonlib
+import sys
+
+import click
+from rich.console import Console
+
+from ..client import get_client
+from ..deploy_watch import follow_deployment
+from .. import context, output
+
+console = Console()
+
+
+def _parse_env(pairs: tuple[str, ...]) -> dict[str, str]:
+    """Parse `KEY=VALUE` options into a dict; exit on a malformed pair."""
+    out: dict[str, str] = {}
+    for p in pairs:
+        if "=" not in p:
+            console.print(f"[red]Invalid --env '{p}'. Use KEY=VALUE.[/red]")
+            raise SystemExit(2)
+        key, value = p.split("=", 1)
+        key = key.strip()
+        if not key:
+            console.print(f"[red]Invalid --env '{p}'. Empty key.[/red]")
+            raise SystemExit(2)
+        out[key] = value
+    return out
+
+
+def _find_service_by_name(client, project_id: int, name: str, environment_id: int | None = None) -> dict | None:
+    """Return the service dict matching `name` (and env) in the project."""
+    # Scope to the environment so the same name in another env isn't matched.
+    path = f"/projects/{project_id}/services/"
+    if environment_id is not None:
+        path += f"?environment={environment_id}"
+    resp = client.get(path)
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to list services: {resp.text}[/red]")
+        raise SystemExit(1)
+    data = resp.json()
+    while True:
+        for s in data.get("results", []):
+            if s.get("name") == name:
+                return s
+        next_url = data.get("next")
+        if not next_url:
+            return None
+        # `next` is an absolute URL; go through the client so auth headers apply.
+        r = client.get(next_url.split("/v1", 1)[-1] if "/v1" in next_url else next_url)
+        if r.status_code != 200:
+            return None
+        data = r.json()
+
+
+def _apply_env(client, service_id: int, env: dict[str, str]) -> None:
+    """Create or update each env var on the service before deploy."""
+    resp = client.get(f"/services/{service_id}/env/")
+    existing = {v["key"]: v for v in resp.json().get("results", [])} if resp.status_code == 200 else {}
+    for key, value in env.items():
+        if key in existing:
+            r = client.patch(
+                f"/services/{service_id}/env/{existing[key]['id']}/",
+                json={"value": value},
+            )
+        else:
+            r = client.post(
+                f"/services/{service_id}/env/",
+                json={"key": key, "value": value},
+            )
+        if r.status_code not in (200, 201):
+            console.print(f"[yellow]Warning: could not set env {key}: {r.text}[/yellow]")
+
+
+@click.command()
+@click.option("--project", "-p", default=None, help="Project id or name (else linked)")
+@click.option("--name", "-n", default=None, help="Service name (else linked service)")
+@click.option("--image", "-i", required=True, help="Docker image (e.g. nginx:latest)")
+@click.option("--port", type=int, default=None, help="Container port to expose (exposed_port)")
+@click.option("--env", "-e", "env_pairs", multiple=True, help="Env var KEY=VALUE (repeatable)")
+@click.option("--replicas", type=int, default=None, help="Number of replicas (min)")
+@click.option("--cpu", type=float, default=None, help="CPU limit (cores)")
+@click.option("--memory", type=int, default=None, help="Memory limit (MB)")
+@click.option("--follow", "-f", is_flag=True, help="Stream the deploy to completion (non-zero exit on failure)")
+@click.option("--environment", "-E", default=None,
+              help="Target environment id/name (else the project's default/production)")
+@click.option("--json", "json_mode", is_flag=True, help="Machine-readable output")
+def up(project, name, image, port, env_pairs, replicas, cpu, memory, environment, follow, json_mode):
+    """Create or update a service from an image and deploy it.
+
+    Project/name fall back to the linked directory (`bitsreef link`), so a
+    linked service can be redeployed with just `bitsreef up -i image:tag`.
+    With --environment the service is created in / matched within that
+    environment (the same name can exist in several environments).
+    """
+    json_mode = json_mode or output.json_enabled()
+    client = get_client()
+    env = _parse_env(env_pairs)
+    link = context.read_link()
+    project_id = context.resolve_project(client, project, link)
+    environment_id = context.resolve_environment(client, project_id, environment)
+    if not name:
+        name = link.get("service_name")
+        if not name:
+            _fail("No service name given and none linked. Pass --name or run `bitsreef link -s <service>`.", json_mode)
+
+    # Optional fields — only send what was provided so backend defaults apply.
+    fields: dict = {"image": image}
+    if port is not None:
+        fields["exposed_port"] = port
+    if replicas is not None:
+        fields["min_replicas"] = replicas
+        fields["max_replicas"] = max(replicas, 1)
+    if cpu is not None:
+        fields["cpu_limit"] = cpu
+    if memory is not None:
+        fields["memory_limit"] = memory
+
+    existing = _find_service_by_name(client, project_id, name, environment_id)
+    if existing:
+        service_id = existing["id"]
+        action = "updated"
+        resp = client.patch(f"/projects/{project_id}/services/{service_id}/", json=fields)
+        if resp.status_code != 200:
+            _fail(f"Failed to update service: {resp.text}", json_mode)
+    else:
+        action = "created"
+        payload = {"name": name, "source_type": "image", "environment": environment_id, **fields}
+        resp = client.post(f"/projects/{project_id}/services/", json=payload)
+        if resp.status_code != 201:
+            _fail(f"Failed to create service: {resp.text}", json_mode)
+        service_id = resp.json()["id"]
+
+    if env:
+        _apply_env(client, service_id, env)
+
+    if not json_mode:
+        console.print(f"[green]Service {action}[/green] [bold]{name}[/bold] (id={service_id})")
+
+    # Trigger the deployment.
+    dep = client.post(f"/projects/{project_id}/services/{service_id}/deploy/")
+    if dep.status_code not in (200, 202):
+        _fail(f"Deploy failed to queue: {dep.text}", json_mode)
+    deployment_id = dep.json().get("deployment_id")
+
+    success = True
+    final_status = "queued"
+    if follow and deployment_id:
+        success, final = follow_deployment(
+            client, service_id, deployment_id, json_mode=json_mode
+        )
+        final_status = (final or {}).get("status", "unknown")
+    elif not json_mode:
+        console.print("[green]Deployment queued.[/green]"
+                      f"  Follow with: bitsreef deploy status {project_id}")
+
+    url = _service_url(client, project_id, service_id)
+
+    if json_mode:
+        click.echo(jsonlib.dumps({
+            "service_id": service_id,
+            "action": action,
+            "deployment_id": deployment_id,
+            "status": final_status,
+            "url": url,
+            "success": success,
+        }))
+    elif url:
+        console.print(f"  URL: [link]https://{url}[/link]")
+
+    if not success:
+        sys.exit(1)
+
+
+def _service_url(client, project_id: int, service_id: int) -> str | None:
+    """Best-effort fetch of the service's assigned subdomain."""
+    r = client.get(f"/projects/{project_id}/services/{service_id}/")
+    if r.status_code == 200:
+        return r.json().get("assigned_subdomain") or None
+    return None
+
+
+def _fail(msg: str, json_mode: bool):
+    if json_mode:
+        click.echo(jsonlib.dumps({"success": False, "error": msg}))
+    else:
+        console.print(f"[red]{msg}[/red]")
+    raise SystemExit(1)

bitsreef_cli/commands/volumes.py

@@ -0,0 +1,99 @@
+"""Volume commands: list, create, 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 _results(resp):
+    data = resp.json()
+    return data.get("results", []) if isinstance(data, dict) else data
+
+
+def _volume_id(client, project_id, value):
+    """Resolve a volume id from an id or a volume name."""
+    if str(value).isdigit():
+        return int(value)
+    resp = client.get(f"/projects/{project_id}/volumes/")
+    for v in _results(resp):
+        if v.get("name") == value:
+            return v["id"]
+    raise click.ClickException(f"No volume '{value}' in this project.")
+
+
+@click.group()
+def volumes():
+    """Manage volumes within a project."""
+
+
+@volumes.command("list")
+@click.argument("project_id", required=False, default=None)
+def list_volumes(project_id):
+    """List volumes in 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}/volumes/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch volumes: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    results = _results(resp)
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No volumes.[/dim]")
+        return
+
+    table = Table(title="Volumes")
+    table.add_column("ID", style="dim")
+    table.add_column("Name", style="bold")
+    table.add_column("Mount")
+    table.add_column("Size (used/limit MB)", justify="right")
+    table.add_column("Service")
+    for v in results:
+        size = f"{v.get('current_size_mb', 0)}/{v.get('size_limit_mb', '—')}"
+        table.add_row(str(v["id"]), v["name"], v.get("mount_path", "—"),
+                      size, str(v.get("service") or "—"))
+    console.print(table)
+
+
+@volumes.command("create")
+@click.argument("name")
+@click.argument("project_id", required=False, default=None)
+@click.option("--size", type=int, required=True, help="Size limit in MB")
+@click.option("--mount", "mount_path", required=True, help="Mount path in the container")
+@click.option("--service", type=int, default=None, help="Attach to this service ID")
+def create_volume(name, project_id, size, mount_path, service):
+    """Create a volume in a project (linked project if omitted)."""
+    client = get_client()
+    project_id = context.resolve_project(client, project_id, context.read_link())
+    payload = {"name": name, "size_limit_mb": size, "mount_path": mount_path}
+    if service is not None:
+        payload["service"] = service
+    resp = client.post(f"/projects/{project_id}/volumes/", json=payload)
+    if resp.status_code == 201:
+        v = resp.json()
+        console.print(f"[green]Created volume[/green] [bold]{v['name']}[/bold] (id={v['id']})")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@volumes.command("delete")
+@click.argument("volume")
+@click.argument("project_id", required=False, default=None)
+@click.confirmation_option(prompt="Delete this volume? Data will be lost.")
+def delete_volume(volume, project_id):
+    """Delete a volume (by name or id) from a project."""
+    client = get_client()
+    project_id = context.resolve_project(client, project_id, context.read_link())
+    volume_id = _volume_id(client, project_id, volume)
+    resp = client.delete(f"/projects/{project_id}/volumes/{volume_id}/")
+    if resp.status_code == 204:
+        console.print("[green]Deleted.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")