bitsreef-cli 0.1.0__py3-none-any.whl

bitsreef_cli/commands/env.py

@@ -0,0 +1,138 @@
+"""Environment variable commands: list, set, 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)
+
+
+@click.group()
+def env():
+    """Manage environment variables for a service."""
+
+
+@env.command("list")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def list_vars(service):
+    """List environment variables for a service."""
+    client = get_client()
+    service_id = _service(client, service)
+    resp = client.get(f"/services/{service_id}/env/")
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch variables.[/red]")
+        raise SystemExit(1)
+
+    results = resp.json().get("results", [])
+    if output.json_enabled():
+        # Redact secret values in JSON too (reveal has its own command).
+        output.emit([{**v, "value": ("********" if v.get("is_secret") else v.get("value"))} for v in results])
+        return
+    if not results:
+        console.print("[dim]No environment variables.[/dim]")
+        return
+
+    table = Table(title="Environment Variables")
+    table.add_column("ID", style="dim")
+    table.add_column("Key", style="bold")
+    table.add_column("Value")
+    table.add_column("Secret")
+    for v in results:
+        val = "********" if v["is_secret"] else v["value"]
+        table.add_row(str(v["id"]), v["key"], val, "Yes" if v["is_secret"] else "No")
+    console.print(table)
+
+
+@env.command("set")
+@click.argument("key")
+@click.argument("value")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+@click.option("--secret", is_flag=True, help="Mark as secret")
+def set_var(key, value, service, secret):
+    """Set an environment variable (creates or updates)."""
+    client = get_client()
+    service_id = _service(client, service)
+    # Check if key already exists
+    resp = client.get(f"/services/{service_id}/env/")
+    existing = None
+    for v in resp.json().get("results", []):
+        if v["key"] == key:
+            existing = v
+            break
+
+    if existing:
+        resp = client.patch(
+            f"/services/{service_id}/env/{existing['id']}/",
+            json={"value": value, "is_secret": secret},
+        )
+        if resp.status_code == 200:
+            console.print(f"[green]Updated {key}[/green]")
+        else:
+            console.print(f"[red]Failed: {resp.text}[/red]")
+    else:
+        resp = client.post(
+            f"/services/{service_id}/env/",
+            json={"key": key, "value": value, "is_secret": secret},
+        )
+        if resp.status_code == 201:
+            console.print(f"[green]Set {key}[/green]")
+        else:
+            console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@env.command("delete")
+@click.argument("key")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def delete_var(key, service):
+    """Delete an environment variable by key."""
+    client = get_client()
+    service_id = _service(client, service)
+    resp = client.get(f"/services/{service_id}/env/")
+    target = None
+    for v in resp.json().get("results", []):
+        if v["key"] == key:
+            target = v
+            break
+
+    if not target:
+        console.print(f"[yellow]Variable '{key}' not found.[/yellow]")
+        return
+
+    resp = client.delete(f"/services/{service_id}/env/{target['id']}/")
+    if resp.status_code == 204:
+        console.print(f"[green]Deleted {key}[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@env.command("reveal")
+@click.argument("key")
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def reveal_var(key, service):
+    """Reveal a secret variable's value."""
+    client = get_client()
+    service_id = _service(client, service)
+    resp = client.get(f"/services/{service_id}/env/")
+    target = None
+    for v in resp.json().get("results", []):
+        if v["key"] == key:
+            target = v
+            break
+
+    if not target:
+        console.print(f"[yellow]Variable '{key}' not found.[/yellow]")
+        return
+
+    resp = client.get(f"/services/{service_id}/env/{target['id']}/reveal/")
+    if resp.status_code == 200:
+        console.print(f"[bold]{key}[/bold]={resp.json()['value']}")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/environments.py

@@ -0,0 +1,132 @@
+"""Environment commands: list, create (blank or clone), delete.
+
+An environment is an independently-running copy of a project's services
+(production, staging, development, …). Creating one can clone an existing
+environment's service definitions in a stopped state for you to configure and
+deploy.
+"""
+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 environments():
+    """Manage project environments (production, staging, …)."""
+
+
+@environments.command("list")
+@click.option("--project", "-p", default=None, help="Project id or name (else linked)")
+def list_environments(project):
+    """List a project's environments."""
+    client = get_client()
+    link = context.read_link()
+    project_id = context.resolve_project(client, project, link)
+    resp = client.get(f"/projects/{project_id}/environments/")
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch environments.[/red]")
+        raise SystemExit(1)
+    results = resp.json().get("results", [])
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No environments.[/dim]")
+        return
+    table = Table(title="Environments")
+    table.add_column("ID", style="dim")
+    table.add_column("Name", style="bold")
+    table.add_column("Default", justify="center")
+    table.add_column("Services", justify="right")
+    for e in results:
+        table.add_row(
+            str(e["id"]),
+            e["name"],
+            "✓" if e.get("is_default") else "",
+            str(e.get("service_count", 0)),
+        )
+    console.print(table)
+
+
+@environments.command("create")
+@click.argument("name")
+@click.option("--project", "-p", default=None, help="Project id or name (else linked)")
+@click.option(
+    "--clone-from",
+    default=None,
+    help="Source environment id/name to clone services from (stopped).",
+)
+@click.option(
+    "--blank",
+    is_flag=True,
+    help="Create an empty environment (no services). Default clones production.",
+)
+def create_environment(name, project, clone_from, blank):
+    """Create a new environment.
+
+    By default clones the project's production environment (services copied in a
+    stopped state — configure and deploy them). Use --clone-from to pick a
+    different source, or --blank for an empty environment.
+    """
+    client = get_client()
+    link = context.read_link()
+    project_id = context.resolve_project(client, project, link)
+
+    payload = {"name": name}
+    if not blank:
+        source = clone_from if clone_from is not None else None
+        source_id = context.resolve_environment(client, project_id, source)
+        payload["clone_from"] = source_id
+
+    resp = client.post(f"/projects/{project_id}/environments/", json=payload)
+    if resp.status_code not in (200, 201):
+        detail = resp.json() if resp.headers.get("content-type", "").startswith("application/json") else resp.text
+        if output.json_enabled():
+            output.emit({"error": detail})
+        else:
+            console.print(f"[red]Failed to create environment:[/red] {detail}")
+        raise SystemExit(1)
+
+    data = resp.json()
+    if output.json_enabled():
+        output.emit(data)
+        return
+    if blank:
+        console.print(f"[green]Created blank environment[/green] '{data['name']}' (id {data['id']}).")
+    else:
+        console.print(
+            f"[green]Created environment[/green] '{data['name']}' (id {data['id']}) — "
+            "services cloned in a stopped state. Configure and deploy them, e.g. "
+            f"`bitsreef services list -p {project_id} --environment {data['id']}`."
+        )
+
+
+@environments.command("delete")
+@click.argument("environment")
+@click.option("--project", "-p", default=None, help="Project id or name (else linked)")
+@click.option("--yes", "-y", is_flag=True, help="Skip confirmation")
+def delete_environment(environment, project, yes):
+    """Delete an environment (and its services). The default env can't be deleted."""
+    client = get_client()
+    link = context.read_link()
+    project_id = context.resolve_project(client, project, link)
+    env_id = context.resolve_environment(client, project_id, environment)
+    if not yes and not output.json_enabled():
+        click.confirm(
+            f"Delete environment {environment} and all its services?", abort=True
+        )
+    resp = client.delete(f"/projects/{project_id}/environments/{env_id}/")
+    if resp.status_code not in (200, 204):
+        detail = resp.text
+        try:
+            detail = resp.json()
+        except Exception:
+            pass
+        console.print(f"[red]Failed to delete environment:[/red] {detail}")
+        raise SystemExit(1)
+    console.print(f"[green]Deleted environment[/green] {environment}.")

bitsreef_cli/commands/exec_cmd.py

@@ -0,0 +1,27 @@
+"""`bitsreef exec` — open an interactive shell in a running service container."""
+import click
+from rich.console import Console
+
+from ..client import get_client
+from .. import config, context
+
+console = Console()
+
+
+@click.command("exec")
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+def exec_(project_id, service_id):
+    """Open an interactive shell (/bin/sh) in the service's running container.
+
+    Uses the linked project/service if omitted; names/slugs work in place of IDs.
+    Type `exit` (or Ctrl-D) to leave the shell.
+    """
+    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)
+
+    from ..shell import exec_shell
+    token = config.get_tokens()[0]
+    raise SystemExit(exec_shell(config.get_api_url(), token, service_id))

bitsreef_cli/commands/functions.py

@@ -0,0 +1,186 @@
+"""Function commands: list, create, invoke, delete."""
+import click
+from rich.console import Console
+from rich.table import Table
+
+from ..client import get_client
+from .. import output
+
+console = Console()
+
+
+@click.group()
+def functions():
+    """Manage serverless functions."""
+
+
+@functions.command("list")
+@click.argument("project_id", type=int)
+def list_functions(project_id):
+    """List all functions in a project."""
+    client = get_client()
+    resp = client.get(f"/projects/{project_id}/functions/")
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch functions.[/red]")
+        raise SystemExit(1)
+
+    results = resp.json().get("results", [])
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No functions.[/dim]")
+        return
+
+    table = Table(title="Functions")
+    table.add_column("ID", style="dim")
+    table.add_column("Name", style="bold")
+    table.add_column("Runtime")
+    table.add_column("Trigger")
+    table.add_column("Enabled")
+    table.add_column("Invocations", justify="right")
+    for fn in results:
+        enabled = "[green]Yes[/green]" if fn["enabled"] else "[red]No[/red]"
+        trigger = fn["cron_schedule"] if fn["trigger_type"] == "cron" else "HTTP"
+        table.add_row(
+            str(fn["id"]), fn["name"], fn["runtime"],
+            trigger, enabled, str(fn["invocation_count"]),
+        )
+    console.print(table)
+
+
+@functions.command("create")
+@click.argument("project_id", type=int)
+@click.argument("name")
+@click.option("--runtime", "-r", type=click.Choice(["bun", "deno", "python", "node"]), default="python")
+@click.option("--code", "-c", default=None, help="Inline code string")
+@click.option("--file", "-f", "code_file", type=click.Path(exists=True), help="Read code from file")
+@click.option("--trigger", "-t", type=click.Choice(["http", "cron"]), default="http")
+@click.option("--cron", default="", help="Cron schedule (for cron trigger)")
+@click.option("--timeout", default=30, type=int, help="Timeout in seconds")
+@click.option("--memory", default=128, type=int, help="Memory limit in MB")
+def create_function(project_id, name, runtime, code, code_file, trigger, cron, timeout, memory):
+    """Create a new function."""
+    if code_file:
+        with open(code_file) as f:
+            code = f.read()
+    if not code:
+        console.print("[red]Provide --code or --file.[/red]")
+        raise SystemExit(1)
+
+    client = get_client()
+    payload = {
+        "name": name,
+        "runtime": runtime,
+        "code": code,
+        "trigger_type": trigger,
+        "cron_schedule": cron if trigger == "cron" else "",
+        "timeout_seconds": timeout,
+        "memory_limit_mb": memory,
+    }
+    resp = client.post(f"/projects/{project_id}/functions/", json=payload)
+    if resp.status_code == 201:
+        fn = resp.json()
+        console.print(f"[green]Created [bold]{fn['name']}[/bold]  endpoint={fn['endpoint_path']}[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@functions.command("invoke")
+@click.argument("project_id", type=int)
+@click.argument("function_id", type=int)
+def invoke_function(project_id, function_id):
+    """Invoke a function."""
+    client = get_client()
+    resp = client.post(f"/projects/{project_id}/functions/{function_id}/invoke/")
+    if resp.status_code == 202:
+        data = resp.json()
+        console.print(f"[green]Invoked — invocation_id={data['invocation_id']}[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@functions.command("history")
+@click.argument("project_id", type=int)
+@click.argument("function_id", type=int)
+def invocation_history(project_id, function_id):
+    """Show invocation history for a function."""
+    client = get_client()
+    resp = client.get(f"/projects/{project_id}/functions/{function_id}/invocations/")
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch invocations.[/red]")
+        raise SystemExit(1)
+
+    results = resp.json().get("results", [])
+    if output.json_enabled():
+        output.emit(results)
+        return
+    if not results:
+        console.print("[dim]No invocations yet.[/dim]")
+        return
+
+    table = Table(title="Invocation History")
+    table.add_column("ID", style="dim")
+    table.add_column("Status")
+    table.add_column("Duration")
+    table.add_column("Trigger")
+    table.add_column("Started")
+    for inv in results:
+        s = inv["status"]
+        color = "green" if s == "success" else "red" if s == "failed" else "yellow" if s == "timeout" else "blue"
+        dur = f"{inv['duration_ms']}ms" if inv.get("duration_ms") else "—"
+        table.add_row(
+            str(inv["id"]), f"[{color}]{s}[/{color}]",
+            dur, inv["trigger"], inv["started_at"][:19],
+        )
+    console.print(table)
+
+
+@functions.command("delete")
+@click.argument("project_id", type=int)
+@click.argument("function_id", type=int)
+@click.confirmation_option(prompt="Delete this function?")
+def delete_function(project_id, function_id):
+    """Delete a function."""
+    client = get_client()
+    resp = client.delete(f"/projects/{project_id}/functions/{function_id}/")
+    if resp.status_code == 204:
+        console.print("[green]Deleted.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")
+
+
+@functions.command("update")
+@click.argument("project_id", type=int)
+@click.argument("function_id", type=int)
+@click.option("--code", "-c", default=None, help="New code string")
+@click.option("--file", "-f", "code_file", type=click.Path(exists=True), help="Read new code from file")
+@click.option("--enabled/--disabled", default=None, help="Enable or disable")
+@click.option("--timeout", default=None, type=int, help="Timeout in seconds")
+@click.option("--memory", default=None, type=int, help="Memory limit in MB")
+def update_function(project_id, function_id, code, code_file, enabled, timeout, memory):
+    """Update a function's configuration or code."""
+    if code_file:
+        with open(code_file) as f:
+            code = f.read()
+
+    payload = {}
+    if code is not None:
+        payload["code"] = code
+    if enabled is not None:
+        payload["enabled"] = enabled
+    if timeout is not None:
+        payload["timeout_seconds"] = timeout
+    if memory is not None:
+        payload["memory_limit_mb"] = memory
+
+    if not payload:
+        console.print("[yellow]Nothing to update.[/yellow]")
+        return
+
+    client = get_client()
+    resp = client.patch(f"/projects/{project_id}/functions/{function_id}/", json=payload)
+    if resp.status_code == 200:
+        console.print("[green]Updated.[/green]")
+    else:
+        console.print(f"[red]Failed: {resp.text}[/red]")

bitsreef_cli/commands/link.py

@@ -0,0 +1,87 @@
+"""`bitsreef link` — bind the current directory to a project/service.
+
+Once linked, commands can be run without repeating IDs: `bitsreef logs`,
+`bitsreef up -i img:tag`, `bitsreef deploy up`, etc. resolve the target from the
+nearest `.bitsreef.json`. Names/slugs work in place of IDs everywhere too.
+"""
+import click
+from rich.console import Console
+
+from ..client import get_client
+from .. import context
+
+console = Console()
+
+
+def _project_name(client, org_id, project_id):
+    r = client.get(f"/organizations/{org_id}/projects/{project_id}/")
+    return r.json().get("name") if r.status_code == 200 else None
+
+
+def _service_name(client, project_id, service_id):
+    r = client.get(f"/projects/{project_id}/services/{service_id}/")
+    return r.json().get("name") if r.status_code == 200 else None
+
+
+@click.command()
+@click.option("--project", "-p", default=None, help="Project id or name")
+@click.option("--service", "-s", default=None, help="Service id or name")
+@click.option("--org", type=int, default=None, help="Organization ID")
+@click.option("--show", is_flag=True, help="Show the current link and exit")
+def link(project, service, org, show):
+    """Link this directory to a project (and optionally a service)."""
+    if show:
+        current = context.read_link()
+        if not current:
+            console.print("[dim]No link in this directory.[/dim]")
+            return
+        f = context.find_link_file()
+        console.print(f"[bold]Linked[/bold] ({f}):")
+        console.print(f"  org={current.get('org_id')}  "
+                      f"project={current.get('project_name') or current.get('project_id')}  "
+                      f"service={current.get('service_name') or current.get('service_id') or '—'}")
+        return
+
+    client = get_client()
+    link_data = context.read_link()
+    org_id = context.resolve_org(client, org, link_data)
+
+    # Project: explicit id/name, else interactive pick.
+    if project is not None:
+        project_id = context.resolve_project(client, project, {"org_id": org_id}, org=org_id)
+    else:
+        projects = context._paginate(client, f"/organizations/{org_id}/projects/")
+        if not projects:
+            raise click.ClickException("No projects in this org — create one first.")
+        console.print("[bold]Select a project:[/bold]")
+        for i, p in enumerate(projects, 1):
+            console.print(f"  {i}. {p['name']} (id={p['id']})")
+        choice = click.prompt("Enter number", type=int, default=1)
+        project_id = int(projects[choice - 1]["id"])
+
+    data = {
+        "org_id": org_id,
+        "project_id": project_id,
+        "project_name": _project_name(client, org_id, project_id),
+    }
+
+    # Service is optional.
+    if service is not None:
+        service_id = context.resolve_service(client, project_id, service)
+        data["service_id"] = service_id
+        data["service_name"] = _service_name(client, project_id, service_id)
+
+    path = context.write_link(data)
+    console.print(f"[green]Linked[/green] → {data['project_name'] or project_id}"
+                  + (f" / {data.get('service_name') or data.get('service_id')}" if service is not None else "")
+                  + f"  [dim]({path})[/dim]")
+
+
+@click.command()
+def unlink():
+    """Remove this directory's link."""
+    removed = context.clear_link()
+    if removed:
+        console.print(f"[green]Unlinked[/green] [dim]({removed})[/dim]")
+    else:
+        console.print("[dim]No link to remove.[/dim]")

bitsreef_cli/commands/logs.py

@@ -0,0 +1,44 @@
+"""Logs commands: view service logs."""
+import click
+from rich.console import Console
+
+from ..client import get_client
+from .. import config, context, output
+
+console = Console()
+
+
+@click.command()
+@click.argument("project_id", required=False, default=None)
+@click.argument("service_id", required=False, default=None)
+@click.option("--tail", "-n", default=100, type=int, help="Number of log lines")
+@click.option("--follow", "-f", is_flag=True, help="Stream new logs live (Ctrl-C to stop)")
+def logs(project_id, service_id, tail, follow):
+    """View logs for a service. Uses the 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)
+
+    if follow:
+        from ..log_stream import follow_logs
+        token = config.get_tokens()[0]
+        raise SystemExit(follow_logs(config.get_api_url(), token, service_id))
+
+    resp = client.get(f"/projects/{project_id}/services/{service_id}/logs/", params={"tail": tail})
+    if resp.status_code != 200:
+        console.print("[red]Failed to fetch logs.[/red]")
+        raise SystemExit(1)
+
+    data = resp.json()
+    if output.json_enabled():
+        output.emit(data)
+        return
+    log_text = data.get("logs", "")
+    if not log_text:
+        console.print("[dim]No logs available.[/dim]")
+        return
+
+    # Print logs line by line
+    for line in log_text.splitlines():
+        console.print(line)

bitsreef_cli/commands/metrics.py

@@ -0,0 +1,50 @@
+"""Service metrics command — current CPU / memory / network snapshot."""
+import click
+from rich.console import Console
+
+from ..client import get_client
+from .. import context, output
+
+console = Console()
+
+
+@click.command()
+@click.option("--service", "-s", default=None, help="Service id or name (else linked)")
+def metrics(service):
+    """Show the latest resource-usage snapshot for a service."""
+    client = get_client()
+    link = context.read_link()
+    service_id = context.resolve_service(client, link.get("project_id"), service, link)
+    resp = client.get(f"/services/{service_id}/metrics/current/")
+    if resp.status_code != 200:
+        console.print(f"[red]Failed to fetch metrics: {resp.text}[/red]")
+        raise SystemExit(1)
+
+    snap = resp.json()
+    if output.json_enabled():
+        output.emit(snap)
+        return
+    if not snap:
+        console.print("[dim]No metrics recorded yet.[/dim]")
+        return
+
+    cpu = snap.get("cpu_percent")
+    used = snap.get("memory_used_mb")
+    limit = snap.get("memory_limit_mb")
+    console.print(f"[bold]Service {service_id}[/bold]  [dim]({snap.get('recorded_at', '')})[/dim]")
+    console.print(f"  CPU:      {cpu if cpu is not None else '—'}%")
+    console.print(f"  Memory:   {used if used is not None else '—'} / {limit if limit is not None else '—'} MB")
+    console.print(f"  Network:  ↓ {_h(snap.get('network_rx_bytes'))}  ↑ {_h(snap.get('network_tx_bytes'))}")
+    console.print(f"  Replicas: {snap.get('replica_count', '—')}")
+
+
+def _h(n) -> str:
+    """Human-readable bytes."""
+    if n is None:
+        return "—"
+    value = float(n)
+    for unit in ("B", "KB", "MB", "GB", "TB"):
+        if value < 1024 or unit == "TB":
+            return f"{value:.1f}{unit}"
+        value /= 1024
+    return f"{value:.1f}TB"