bumblebee-cli 0.1.1 → 0.2.0

package/python/bb_cli/commands/auth.py

@@ -1,79 +1,79 @@
-import typer
-from rich import print as rprint
-
-from ..api_client import api_get, api_post
-from ..config import get_api_url, get_project_path, is_local_config_active, load_config, save_config
-
-app = typer.Typer(help="Authentication commands")
-
-
-@app.command()
-def login(
-    email: str = typer.Option(..., prompt=True),
-    password: str = typer.Option(..., prompt=True, hide_input=True),
-):
-    """Log in to Bumblebee API."""
-    try:
-        data = api_post("/auth/login", json={"email": email, "password": password})
-        cfg = load_config()
-        cfg["token"] = data["access_token"]
-        save_config(cfg)
-        rprint("[green]Logged in successfully.[/green]")
-    except Exception as e:
-        rprint(f"[red]Login failed: {e}[/red]")
-        raise typer.Exit(1)
-
-
-@app.command()
-def register(
-    email: str = typer.Option(..., prompt=True),
-    username: str = typer.Option(..., prompt=True),
-    password: str = typer.Option(..., prompt=True, hide_input=True, confirmation_prompt=True),
-):
-    """Register a new Bumblebee account."""
-    try:
-        api_post("/auth/register", json={"email": email, "username": username, "password": password})
-        rprint("[green]Registered! You can now run [bold]bb login[/bold].[/green]")
-    except Exception as e:
-        rprint(f"[red]Registration failed: {e}[/red]")
-        raise typer.Exit(1)
-
-
-@app.command()
-def logout():
-    """Clear saved credentials."""
-    cfg = load_config()
-    cfg.pop("token", None)
-    save_config(cfg)
-    rprint("[yellow]Logged out.[/yellow]")
-
-
-@app.command()
-def whoami():
-    """Show current user info."""
-    try:
-        user = api_get("/auth/me")
-        rprint(f"[bold]{user['username']}[/bold] ({user['email']})")
-    except Exception:
-        rprint("[red]Not logged in.[/red]")
-        raise typer.Exit(1)
-
-
-@app.command(name="config")
-def show_config():
-    """Show current CLI configuration."""
-    cfg = load_config()
-    slug = cfg.get("current_project")
-    config_source = "local (.bumblebee/)" if is_local_config_active() else "global (~/.bumblebee/)"
-    rprint(f"Config source: {config_source}")
-    rprint(f"API URL: {get_api_url()}")
-    rprint(f"Logged in: {'Yes' if cfg.get('token') else 'No'}")
-    rprint(f"Current project: {slug or 'None'}")
-    if slug:
-        path = get_project_path(slug)
-        rprint(f"Source path: {path or '[dim]not linked[/dim]'}")
-    projects = cfg.get("projects", {})
-    if projects:
-        rprint("\n[bold]Project paths:[/bold]")
-        for name, info in projects.items():
-            rprint(f"  {name} -> {info.get('path', '?')}")
+import typer
+from rich import print as rprint
+
+from ..api_client import api_get, api_post
+from ..config import get_api_url, get_project_path, is_local_config_active, load_config, save_config
+
+app = typer.Typer(help="Authentication commands")
+
+
+@app.command()
+def login(
+    email: str = typer.Option(..., prompt=True),
+    password: str = typer.Option(..., prompt=True, hide_input=True),
+):
+    """Log in to Bumblebee API."""
+    try:
+        data = api_post("/auth/login", json={"email": email, "password": password})
+        cfg = load_config()
+        cfg["token"] = data["access_token"]
+        save_config(cfg)
+        rprint("[green]Logged in successfully.[/green]")
+    except Exception as e:
+        rprint(f"[red]Login failed: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@app.command()
+def register(
+    email: str = typer.Option(..., prompt=True),
+    username: str = typer.Option(..., prompt=True),
+    password: str = typer.Option(..., prompt=True, hide_input=True, confirmation_prompt=True),
+):
+    """Register a new Bumblebee account."""
+    try:
+        api_post("/auth/register", json={"email": email, "username": username, "password": password})
+        rprint("[green]Registered! You can now run [bold]bb login[/bold].[/green]")
+    except Exception as e:
+        rprint(f"[red]Registration failed: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@app.command()
+def logout():
+    """Clear saved credentials."""
+    cfg = load_config()
+    cfg.pop("token", None)
+    save_config(cfg)
+    rprint("[yellow]Logged out.[/yellow]")
+
+
+@app.command()
+def whoami():
+    """Show current user info."""
+    try:
+        user = api_get("/auth/me")
+        rprint(f"[bold]{user['username']}[/bold] ({user['email']})")
+    except Exception:
+        rprint("[red]Not logged in.[/red]")
+        raise typer.Exit(1)
+
+
+@app.command(name="config")
+def show_config():
+    """Show current CLI configuration."""
+    cfg = load_config()
+    slug = cfg.get("current_project")
+    config_source = "local (.bumblebee/)" if is_local_config_active() else "global (~/.bumblebee/)"
+    rprint(f"Config source: {config_source}")
+    rprint(f"API URL: {get_api_url()}")
+    rprint(f"Logged in: {'Yes' if cfg.get('token') else 'No'}")
+    rprint(f"Current project: {slug or 'None'}")
+    if slug:
+        path = get_project_path(slug)
+        rprint(f"Source path: {path or '[dim]not linked[/dim]'}")
+    projects = cfg.get("projects", {})
+    if projects:
+        rprint("\n[bold]Project paths:[/bold]")
+        for name, info in projects.items():
+            rprint(f"  {name} -> {info.get('path', '?')}")

package/python/bb_cli/commands/board.py

@@ -1,47 +1,47 @@
-import typer
-from rich import print as rprint
-from rich.columns import Columns
-from rich.panel import Panel
-
-from ..api_client import api_get
-from ..config import get_current_project
-
-app = typer.Typer(help="Kanban board view")
-
-COLUMNS = ["open", "in_progress", "in_review", "resolved", "closed"]
-
-
-@app.callback(invoke_without_command=True)
-def board(
-    type: str = typer.Option(None, "-t", "--type", help="Filter by type (story, task, etc.)"),
-):
-    """Show kanban board for current project work items."""
-    slug = get_current_project()
-    if not slug:
-        rprint("[red]No project selected.[/red]")
-        raise typer.Exit(1)
-
-    params = {}
-    if type:
-        params["type"] = type
-    items = api_get(f"/api/projects/{slug}/work-items", params=params)
-
-    panels = []
-    for col in COLUMNS:
-        col_items = [i for i in items if i["status"] == col]
-        content = ""
-        for i in col_items:
-            key = i.get("key") or f"#{i['number']}"
-            prio = i.get("priority", "")
-            assignee = i.get("assignee") or ""
-            content += f"[bold]{key}[/bold] [{i['type']}] {i['title']}\n"
-            if prio or assignee:
-                content += f"  [dim]{prio}[/dim]"
-                if assignee:
-                    content += f" -> {assignee}"
-                content += "\n"
-        if not content:
-            content = "[dim]Empty[/dim]"
-        panels.append(Panel(content.strip(), title=col.replace("_", " ").title(), expand=True))
-
-    rprint(Columns(panels, equal=True, expand=True))
+import typer
+from rich import print as rprint
+from rich.columns import Columns
+from rich.panel import Panel
+
+from ..api_client import api_get
+from ..config import get_current_project
+
+app = typer.Typer(help="Kanban board view")
+
+COLUMNS = ["open", "in_progress", "in_review", "resolved", "closed"]
+
+
+@app.callback(invoke_without_command=True)
+def board(
+    type: str = typer.Option(None, "-t", "--type", help="Filter by type (story, task, etc.)"),
+):
+    """Show kanban board for current project work items."""
+    slug = get_current_project()
+    if not slug:
+        rprint("[red]No project selected.[/red]")
+        raise typer.Exit(1)
+
+    params = {}
+    if type:
+        params["type"] = type
+    items = api_get(f"/api/projects/{slug}/work-items", params=params)
+
+    panels = []
+    for col in COLUMNS:
+        col_items = [i for i in items if i["status"] == col]
+        content = ""
+        for i in col_items:
+            key = i.get("key") or f"#{i['number']}"
+            prio = i.get("priority", "")
+            assignee = i.get("assignee") or ""
+            content += f"[bold]{key}[/bold] [{i['type']}] {i['title']}\n"
+            if prio or assignee:
+                content += f"  [dim]{prio}[/dim]"
+                if assignee:
+                    content += f" -> {assignee}"
+                content += "\n"
+        if not content:
+            content = "[dim]Empty[/dim]"
+        panels.append(Panel(content.strip(), title=col.replace("_", " ").title(), expand=True))
+
+    rprint(Columns(panels, equal=True, expand=True))

package/python/bb_cli/commands/comment.py

@@ -1,34 +1,34 @@
-import typer
-from rich import print as rprint
-
-from ..api_client import api_get, api_post
-from .item import _require_project, _resolve_item
-
-app = typer.Typer(help="Comment management")
-
-
-@app.command(name="list")
-def list_comments(id_or_number: str = typer.Argument(..., help="Work item UUID, number, or KEY-number")):
-    """List comments on a work item."""
-    slug = _require_project()
-    item = _resolve_item(slug, id_or_number)
-    comments = api_get(f"/api/work-items/{item['id']}/comments")
-    if not comments:
-        rprint("[dim]No comments yet.[/dim]")
-        return
-    for c in comments:
-        type_tag = f" [{c['type']}]" if c.get("type") and c["type"] != "discussion" else ""
-        rprint(f"[bold]{c['author']}[/bold]{type_tag}: {c['body']}")
-
-
-@app.command()
-def add(
-    id_or_number: str = typer.Argument(..., help="Work item UUID, number, or KEY-number"),
-    body: str = typer.Argument(...),
-    type: str = typer.Option("discussion", "-t", "--type", help="Comment type: discussion, investigation, proposal, review, agent_output"),
-):
-    """Add a comment to a work item."""
-    slug = _require_project()
-    item = _resolve_item(slug, id_or_number)
-    api_post(f"/api/work-items/{item['id']}/comments", json={"body": body, "type": type})
-    rprint("[green]Comment added.[/green]")
+import typer
+from rich import print as rprint
+
+from ..api_client import api_get, api_post
+from .item import _require_project, _resolve_item
+
+app = typer.Typer(help="Comment management")
+
+
+@app.command(name="list")
+def list_comments(id_or_number: str = typer.Argument(..., help="Work item UUID, number, or KEY-number")):
+    """List comments on a work item."""
+    slug = _require_project()
+    item = _resolve_item(slug, id_or_number)
+    comments = api_get(f"/api/work-items/{item['id']}/comments")
+    if not comments:
+        rprint("[dim]No comments yet.[/dim]")
+        return
+    for c in comments:
+        type_tag = f" [{c['type']}]" if c.get("type") and c["type"] != "discussion" else ""
+        rprint(f"[bold]{c['author']}[/bold]{type_tag}: {c['body']}")
+
+
+@app.command()
+def add(
+    id_or_number: str = typer.Argument(..., help="Work item UUID, number, or KEY-number"),
+    body: str = typer.Argument(...),
+    type: str = typer.Option("discussion", "-t", "--type", help="Comment type: discussion, investigation, proposal, review, agent_output"),
+):
+    """Add a comment to a work item."""
+    slug = _require_project()
+    item = _resolve_item(slug, id_or_number)
+    api_post(f"/api/work-items/{item['id']}/comments", json={"body": body, "type": type})
+    rprint("[green]Comment added.[/green]")

package/python/bb_cli/commands/daemon.py

@@ -0,0 +1,153 @@
+"""Long-running daemon that picks up web-initiated agent requests."""
+
+from __future__ import annotations
+
+import json
+import platform
+import signal
+import sys
+import time
+
+import typer
+from rich import print as rprint
+
+from ..api_client import api_get, api_post
+from ..config import get_api_url, get_current_project, get_project_path, get_token
+
+app = typer.Typer(help="Agent daemon for web-initiated sessions")
+
+
+def _daemon_id() -> str:
+    """Generate a unique daemon identifier."""
+    return f"{platform.node()}-{id(sys)}"
+
+
+def _poll_pending_sessions(slug: str) -> list[dict]:
+    """Poll API for pending (web-initiated) sessions."""
+    try:
+        sessions = api_get("/api/agent-sessions", params={"project_slug": slug})
+        return [s for s in sessions if s.get("status") == "pending" and not s.get("claimed_by")]
+    except Exception:
+        return []
+
+
+def _claim_session(session_id: int, daemon_id: str) -> dict | None:
+    """Attempt to claim a pending session. Returns session or None if already claimed."""
+    try:
+        return api_post(
+            f"/api/agent-sessions/{session_id}/claim",
+            params={"daemon_id": daemon_id},
+        )
+    except Exception:
+        return None
+
+
+def _run_session(slug: str, project_path: str, session: dict):
+    """Execute the work for a claimed session."""
+    from .agent import _execute_one, _suggest_one, _verify_one
+
+    session_id = session["id"]
+    work_item_id = session.get("work_item_id")
+    phase = session.get("phase") or "execute"
+
+    if not work_item_id:
+        rprint(f"  [yellow]Session {session_id}: no work_item_id, skipping.[/yellow]")
+        api_post(f"/api/agent-sessions/{session_id}/complete", json={
+            "status": "failed", "error": "No work item specified",
+        })
+        return
+
+    # Resolve item number for the worker functions
+    try:
+        item = api_get(f"/api/work-items/{work_item_id}")
+        id_or_number = item.get("key") or str(item["number"])
+    except Exception as e:
+        rprint(f"  [red]Session {session_id}: failed to resolve item {work_item_id}: {e}[/red]")
+        api_post(f"/api/agent-sessions/{session_id}/complete", json={
+            "status": "failed", "error": str(e),
+        })
+        return
+
+    rprint(f"  [cyan]Running {phase} for {id_or_number}...[/cyan]")
+
+    if phase == "suggest":
+        result = _suggest_one(slug, project_path, id_or_number)
+    elif phase == "verify":
+        # verify() is a CLI command, we use suggest as fallback
+        result = _suggest_one(slug, project_path, id_or_number)
+    else:
+        result = _execute_one(slug, project_path, id_or_number)
+
+    status = "ok" if result.get("status") == "ok" else "failed"
+    color = "green" if status == "ok" else "red"
+    rprint(f"  [{color}]{id_or_number}: {status}[/{color}]"
+           + (f" — {result.get('error', '')}" if status != "ok" else ""))
+
+
+@app.command()
+def start(
+    poll_interval: int = typer.Option(5, "--poll", "-p", help="Polling interval in seconds"),
+):
+    """Start the agent daemon — polls for web-initiated sessions and executes them.
+
+    Run this in a terminal to enable the web dashboard to spawn agent sessions
+    on your local machine.
+    """
+    slug = get_current_project()
+    if not slug:
+        rprint("[red]No project configured. Run 'bb init' or 'bb project use <slug>' first.[/red]")
+        raise typer.Exit(1)
+
+    project_path = get_project_path(slug)
+    if not project_path:
+        rprint("[red]No project path configured. Run 'bb init' first.[/red]")
+        raise typer.Exit(1)
+
+    token = get_token()
+    if not token:
+        rprint("[red]Not authenticated. Run 'bb login' first.[/red]")
+        raise typer.Exit(1)
+
+    daemon_id = _daemon_id()
+    api_url = get_api_url()
+
+    rprint(f"[bold green]Bumblebee Agent Daemon[/bold green]")
+    rprint(f"  Project: [cyan]{slug}[/cyan]")
+    rprint(f"  Path:    [cyan]{project_path}[/cyan]")
+    rprint(f"  API:     [cyan]{api_url}[/cyan]")
+    rprint(f"  Daemon:  [cyan]{daemon_id}[/cyan]")
+    rprint(f"  Poll:    every {poll_interval}s")
+    rprint(f"\n[dim]Listening for web-initiated agent requests... (Ctrl+C to stop)[/dim]\n")
+
+    # Graceful shutdown
+    running = True
+
+    def _shutdown(sig, frame):
+        nonlocal running
+        rprint("\n[yellow]Shutting down daemon...[/yellow]")
+        running = False
+
+    signal.signal(signal.SIGINT, _shutdown)
+    signal.signal(signal.SIGTERM, _shutdown)
+
+    while running:
+        pending = _poll_pending_sessions(slug)
+
+        for session in pending:
+            claimed = _claim_session(session["id"], daemon_id)
+            if claimed:
+                rprint(f"[cyan]Claimed session {session['id']}[/cyan]")
+                try:
+                    _run_session(slug, project_path, claimed)
+                except Exception as e:
+                    rprint(f"[red]Session {session['id']} failed: {e}[/red]")
+                    try:
+                        api_post(f"/api/agent-sessions/{session['id']}/complete", json={
+                            "status": "failed", "error": str(e),
+                        })
+                    except Exception:
+                        pass
+
+        time.sleep(poll_interval)
+
+    rprint("[dim]Daemon stopped.[/dim]")

package/python/bb_cli/commands/init.py

@@ -1,62 +1,83 @@
-from pathlib import Path
-
-import toml
-import typer
-from rich import print as rprint
-
-from ..config import LOCAL_CONFIG_NAME, LOCAL_DIR_NAME, find_project_root
-
-app = typer.Typer(help="Initialize project-local config")
-
-
-@app.command()
-def init(
-    project: str = typer.Option(None, "--project", "-p", help="Set current_project in local config"),
-    api_url: str = typer.Option(None, "--api-url", help="Set api_url in local config"),
-    force: bool = typer.Option(False, "--force", "-f", help="Overwrite existing config"),
-):
-    """Initialize a .bumblebee/ config directory in the project root.
-
-    Looks for the project root by walking up from cwd to find .git/.
-    Falls back to cwd if no .git/ is found.
-    """
-    root = find_project_root()
-    if root is None:
-        root = Path.cwd().resolve()
-
-    bb_dir = root / LOCAL_DIR_NAME
-    config_file = bb_dir / LOCAL_CONFIG_NAME
-    gitignore_file = bb_dir / ".gitignore"
-
-    if config_file.exists() and not force:
-        rprint(f"[yellow].bumblebee/ already exists at {root}[/yellow]")
-        rprint("[dim]Use --force to overwrite.[/dim]")
-        raise typer.Exit(1)
-
-    # Create directory
-    bb_dir.mkdir(parents=True, exist_ok=True)
-
-    # Build config
-    cfg: dict = {}
-    if project:
-        cfg["current_project"] = project
-    if api_url:
-        cfg["api_url"] = api_url
-
-    # Write config.toml
-    with open(config_file, "w") as f:
-        if cfg:
-            toml.dump(cfg, f)
-        else:
-            f.write("# Bumblebee project-local config\n# Settings here override ~/.bumblebee/config.toml\n")
-
-    # Write .gitignore to exclude local-only files
-    if not gitignore_file.exists() or force:
-        gitignore_file.write_text("*.local.toml\n")
-
-    rprint(f"[green]Initialized .bumblebee/ at {root}[/green]")
-    if project:
-        rprint(f"  current_project = {project}")
-    if api_url:
-        rprint(f"  api_url = {api_url}")
-    rprint("[dim]Tip: Add project-specific settings to .bumblebee/config.toml[/dim]")
+from pathlib import Path
+
+import toml
+import typer
+from rich import print as rprint
+
+from ..config import LOCAL_CONFIG_NAME, LOCAL_DIR_NAME, find_project_root, set_project_path
+
+app = typer.Typer(help="Initialize project-local config")
+
+
+@app.command()
+def init(
+    project: str = typer.Option(None, "--project", "-p", help="Set current_project in local config"),
+    api_url: str = typer.Option(None, "--api-url", help="Set api_url in local config"),
+    force: bool = typer.Option(False, "--force", "-f", help="Overwrite existing config"),
+):
+    """Initialize a .bumblebee/ config directory in the project root.
+
+    Looks for the project root by walking up from cwd to find .git/.
+    Falls back to cwd if no .git/ is found.
+    """
+    root = find_project_root()
+    if root is None:
+        root = Path.cwd().resolve()
+
+    bb_dir = root / LOCAL_DIR_NAME
+    config_file = bb_dir / LOCAL_CONFIG_NAME
+    gitignore_file = bb_dir / ".gitignore"
+
+    if config_file.exists() and not force:
+        rprint(f"[yellow].bumblebee/ already exists at {root}[/yellow]")
+        rprint("[dim]Use --force to overwrite.[/dim]")
+        raise typer.Exit(1)
+
+    # Create directory
+    bb_dir.mkdir(parents=True, exist_ok=True)
+
+    # Interactive project selection when --project is not provided
+    selected_name = None
+    if not project:
+        try:
+            from .project import _select_project_interactive
+
+            selected = _select_project_interactive()
+            project = selected["slug"]
+            selected_name = selected["name"]
+        except (SystemExit, KeyboardInterrupt):
+            # User cancelled or no projects — continue without setting a project
+            pass
+
+    # Build config
+    cfg: dict = {}
+    if project:
+        cfg["current_project"] = project
+    if api_url:
+        cfg["api_url"] = api_url
+
+    # Write config.toml
+    with open(config_file, "w") as f:
+        if cfg:
+            toml.dump(cfg, f)
+        else:
+            f.write("# Bumblebee project-local config\n# Settings here override ~/.bumblebee/config.toml\n")
+
+    # Write .gitignore to exclude local-only files
+    if not gitignore_file.exists() or force:
+        gitignore_file.write_text("*.local.toml\n")
+
+    # Auto-link the current directory as the project path
+    if project:
+        resolved = str(root.resolve())
+        set_project_path(project, resolved)
+
+    rprint(f"[green]Initialized .bumblebee/ at {root}[/green]")
+    if project:
+        display = f"{selected_name} ({project})" if selected_name else project
+        rprint(f"  current_project = {display}")
+        rprint(f"  linked path = {root.resolve()}")
+    if api_url:
+        rprint(f"  api_url = {api_url}")
+    if not project:
+        rprint("[dim]Tip: Add project-specific settings to .bumblebee/config.toml[/dim]")