didericis-claude
2323a9ae83
- `bottle:` in agent frontmatter is now optional; agents without it are portable and require bottles to be selected at launch. - Adds `filter_multiselect` to `tui.py`: multi-select picker with ordered selection list, Space/Enter to toggle, Ctrl-D to confirm. - `ManifestIndex` gains `all_bottle_names` and `load_for_agent` accepts `bottle_names: tuple[str, ...]` to merge bottles in order at runtime. - `merge_bottles_runtime` in `manifest_extends.py` applies the same field-merge rules as `extends:` to pre-resolved bottle objects. - `BottleSpec` gains `bottle_names`; `_validate` and `write_launch_metadata` thread it through so `resume` replays the same bottle configuration. - `cmd_start` shows the bottle multiselect after agent selection, pre-populated from the agent's `bottle:` field when present. - Existing agents with `bottle:` declared continue to work unchanged.
320 lines
11 KiB
Python
320 lines
11 KiB
Python
"""start: boot a sandboxed container for a named agent and attach an
|
|
interactive claude-code session. The container is torn down when the
|
|
session ends.
|
|
|
|
The launch core is shared with `cli.py resume <identity>` through
|
|
the private orchestrator `_launch_bottle`.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
from pathlib import Path
|
|
from typing import Callable
|
|
|
|
from ..agent_provider import runtime_for
|
|
from ..backend import (
|
|
Bottle,
|
|
BottleSpec,
|
|
enumerate_active_agents,
|
|
get_bottle_backend,
|
|
known_backend_names,
|
|
)
|
|
from ..backend.docker import util as docker_mod
|
|
from ..backend.docker.bottle_plan import DockerBottlePlan
|
|
from ..bottle_state import (
|
|
cleanup_state,
|
|
is_preserved,
|
|
mark_preserved,
|
|
)
|
|
from ..log import info
|
|
from ..manifest import ManifestIndex
|
|
from ._common import PROG, USER_CWD, read_tty_line
|
|
from . import tui
|
|
|
|
|
|
def cmd_start(argv: list[str]) -> int:
|
|
parser = argparse.ArgumentParser(prog=f"{PROG} start", add_help=True)
|
|
parser.add_argument("--dry-run", action="store_true")
|
|
parser.add_argument("--cwd", action="store_true", help="copy host cwd into the running bottle")
|
|
parser.add_argument(
|
|
"--backend",
|
|
choices=known_backend_names(),
|
|
default=None,
|
|
help=(
|
|
"backend to launch the bottle on (default: $BOT_BOTTLE_BACKEND "
|
|
"or host auto-selection). Overrides the env var when set."
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
"name",
|
|
nargs="?",
|
|
default=None,
|
|
help="agent name defined in bot-bottle.json (omit to pick interactively)",
|
|
)
|
|
args = parser.parse_args(argv)
|
|
|
|
dry_run = args.dry_run or os.environ.get("BOT_BOTTLE_DRY_RUN") == "1"
|
|
|
|
manifest = ManifestIndex.resolve(USER_CWD)
|
|
|
|
agent_name: str | None = args.name
|
|
if agent_name is None:
|
|
agent_name = tui.filter_select(
|
|
manifest.all_agent_names,
|
|
title="Select agent",
|
|
)
|
|
if agent_name is None:
|
|
return 0
|
|
|
|
backend_name: str | None = args.backend
|
|
|
|
# Bottle multiselect: always show after agent selection so operators
|
|
# can compose bottles at launch time without editing agent manifests.
|
|
available_bottles = manifest.all_bottle_names
|
|
initial_bottle = _peek_agent_bottle(manifest, agent_name)
|
|
initial_bottles = [initial_bottle] if initial_bottle else []
|
|
selected_bottles = tui.filter_multiselect(
|
|
available_bottles,
|
|
title="Select bottles",
|
|
initial=initial_bottles,
|
|
)
|
|
if selected_bottles is None:
|
|
return 0
|
|
bottle_names = tuple(selected_bottles)
|
|
|
|
label, color = tui.name_color_modal(default_label=agent_name)
|
|
label, color = _resolve_unique_label(label, color)
|
|
|
|
spec = BottleSpec(
|
|
manifest=manifest,
|
|
agent_name=agent_name,
|
|
copy_cwd=args.cwd,
|
|
user_cwd=USER_CWD,
|
|
label=label,
|
|
color=color,
|
|
bottle_names=bottle_names,
|
|
)
|
|
return _launch_bottle(
|
|
spec,
|
|
dry_run=dry_run,
|
|
backend_name=backend_name,
|
|
)
|
|
|
|
|
|
# --- Launch helpers ------------------------------------------------------
|
|
|
|
|
|
def prepare_with_preflight(
|
|
spec: BottleSpec,
|
|
*,
|
|
stage_dir: Path,
|
|
render_preflight: Callable[[DockerBottlePlan], None],
|
|
prompt_yes: Callable[[], bool],
|
|
dry_run: bool = False,
|
|
backend_name: str | None = None,
|
|
) -> tuple[DockerBottlePlan | None, str]:
|
|
"""Run `backend.prepare`, render the preflight summary via the
|
|
injected callable, prompt y/N via the injected callable.
|
|
|
|
`backend_name` selects which backend prepares the plan
|
|
(`None` → `$BOT_BOTTLE_BACKEND` → host auto-selection). The CLI
|
|
passes whatever `--backend` resolved to.
|
|
|
|
Returns `(plan, identity)`. `plan` is None on dry-run or
|
|
operator-N, but `identity` is set as soon as `backend.prepare`
|
|
returns so callers can reap the prepare-time state dir via
|
|
`settle_state(identity)` in their finally — exactly the existing
|
|
semantics."""
|
|
backend = get_bottle_backend(backend_name)
|
|
plan = backend.prepare(spec, stage_dir=stage_dir)
|
|
identity = _identity_from_plan(plan)
|
|
|
|
render_preflight(plan)
|
|
|
|
if dry_run:
|
|
info("dry-run requested; not starting container.")
|
|
return None, identity
|
|
if not prompt_yes():
|
|
info("aborted by user")
|
|
return None, identity
|
|
return plan, identity
|
|
|
|
|
|
def attach_agent(
|
|
bottle: Bottle, *, resume: bool = False,
|
|
agent_provider_template: str = "claude",
|
|
startup_args: tuple[str, ...] = (),
|
|
) -> int:
|
|
"""Run the selected provider CLI inside `bottle` as an
|
|
interactive session. Blocks until the session ends; returns the
|
|
agent process's exit code.
|
|
|
|
`resume=True` adds `--continue` so claude picks up its most
|
|
recent session non-interactively (no session-picker prompt).
|
|
First-attach paths (`./cli.py start`) leave it False.
|
|
|
|
Used as the inner step of `./cli.py start`."""
|
|
runtime = runtime_for(agent_provider_template)
|
|
info(
|
|
f"attaching interactive {agent_provider_template} session "
|
|
"(Ctrl-D or 'exit' to leave; container will be removed)"
|
|
)
|
|
agent_args = list(runtime.bypass_args)
|
|
agent_args.extend(startup_args)
|
|
if resume:
|
|
agent_args.extend(runtime.resume_args)
|
|
return bottle.exec_agent(agent_args, tty=True)
|
|
|
|
|
|
def capture_claude_session_state(identity: str, exit_code: int) -> None:
|
|
"""Inside the launch context, while the container is still
|
|
alive: snapshot the transcript and mark for preservation if
|
|
claude crashed."""
|
|
# FIXME: this captures Claude-specific session state. A follow-up
|
|
# spike should explore freezing provider-neutral container state
|
|
# instead of relying on each agent's transcript layout.
|
|
if not identity:
|
|
return
|
|
# snapshot_transcript(identity)
|
|
if exit_code != 0:
|
|
mark_preserved(identity)
|
|
|
|
|
|
def settle_state(identity: str) -> None:
|
|
"""Post-teardown housekeeping: print the resume hint if the
|
|
state was preserved, otherwise reap the per-bottle state dir."""
|
|
if not identity:
|
|
return
|
|
if is_preserved(identity):
|
|
info(f"to resume this bottle: ./cli.py resume {identity}")
|
|
return
|
|
cleanup_state(identity)
|
|
|
|
|
|
def _identity_from_plan(plan: object) -> str:
|
|
"""Backend-specific: the docker plan exposes the identity as
|
|
`.slug`. Other backends in the future would expose their own
|
|
identity attribute; for now we duck-type to keep this layer
|
|
backend-agnostic."""
|
|
return getattr(plan, "slug", "")
|
|
|
|
|
|
def _peek_agent_bottle(manifest: ManifestIndex, agent_name: str) -> str:
|
|
"""Return the `bottle:` value from the named agent's frontmatter without
|
|
fully parsing the agent file, or "" when absent or unreadable.
|
|
|
|
Used to pre-populate the bottle multiselect with the agent's default
|
|
bottle so operators who haven't removed `bottle:` from their manifests
|
|
don't need to re-select it every time."""
|
|
if manifest.home_md is None:
|
|
# Eager mode (from_json_obj): agent is pre-parsed.
|
|
if agent_name in manifest.agents:
|
|
return manifest.agents[agent_name].bottle
|
|
return ""
|
|
|
|
from ..manifest_loader import scan_agent_names
|
|
from ..yaml_subset import YamlSubsetError, parse_frontmatter
|
|
|
|
home_agents = scan_agent_names(manifest.home_md / "agents")
|
|
cwd_agents: dict[str, Path] = {}
|
|
if manifest.cwd_md is not None:
|
|
cwd_agents = scan_agent_names(manifest.cwd_md / "agents")
|
|
merged = {**home_agents, **cwd_agents}
|
|
path = merged.get(agent_name)
|
|
if path is None:
|
|
return ""
|
|
try:
|
|
fm, _ = parse_frontmatter(path.read_text())
|
|
bottle = fm.get("bottle", "")
|
|
return str(bottle) if isinstance(bottle, str) else ""
|
|
except (OSError, YamlSubsetError):
|
|
return ""
|
|
|
|
|
|
def _resolve_unique_label(label: str, color: str) -> tuple[str, str]:
|
|
"""Re-prompt with a disclaimer until the label's slug is not already
|
|
in use among running bottles. Passes through unchanged when no
|
|
collision is found on the first check."""
|
|
while True:
|
|
slug_candidate = docker_mod.slugify(label)
|
|
active_slugs = {a.slug for a in enumerate_active_agents()}
|
|
if slug_candidate not in active_slugs:
|
|
return label, color
|
|
label, color = tui.name_color_modal(
|
|
default_label=label,
|
|
disclaimer=f'"{label}" is already in use',
|
|
)
|
|
|
|
|
|
def _text_prompt_yes() -> bool:
|
|
"""Default `prompt_yes` for CLI use: reads y/N from the
|
|
controlling tty via stderr prompt + tty-line read."""
|
|
sys.stderr.write("bot-bottle: launch this agent? [y/N] ")
|
|
sys.stderr.flush()
|
|
reply = read_tty_line()
|
|
return reply in ("y", "Y", "yes", "YES")
|
|
|
|
|
|
def _text_render_preflight():
|
|
def _render(plan: DockerBottlePlan) -> None:
|
|
plan.print()
|
|
return _render
|
|
|
|
|
|
def _launch_bottle(
|
|
spec: BottleSpec,
|
|
*,
|
|
dry_run: bool,
|
|
backend_name: str | None = None,
|
|
) -> int:
|
|
"""Shared launch core for `start` and `resume`. Builds the plan,
|
|
prints / dry-runs / prompts as appropriate, brings the bottle up,
|
|
attaches claude, and prints the resume hint on session end."""
|
|
stage_dir = Path(tempfile.mkdtemp(prefix="bot-bottle-stage."))
|
|
identity = ""
|
|
try:
|
|
plan, identity = prepare_with_preflight(
|
|
spec,
|
|
stage_dir=stage_dir,
|
|
render_preflight=_text_render_preflight(),
|
|
prompt_yes=_text_prompt_yes,
|
|
dry_run=dry_run,
|
|
backend_name=backend_name,
|
|
)
|
|
if plan is None:
|
|
return 0
|
|
|
|
backend = get_bottle_backend(backend_name)
|
|
with backend.launch(plan) as bottle:
|
|
agent_provider_template = getattr(plan, "agent_provider_template", "claude")
|
|
exit_code = attach_agent(
|
|
bottle,
|
|
agent_provider_template=agent_provider_template,
|
|
startup_args=plan.agent_provision.startup_args,
|
|
)
|
|
info(
|
|
f"session ended (exit {exit_code}); "
|
|
f"container {bottle.name} will be removed"
|
|
)
|
|
# While the container is still alive: always snapshot the
|
|
# transcript and — if the agent exited non-zero — mark
|
|
# the state for preservation. This picks up crashes /
|
|
# Ctrl-Cs / OOM kills before cleanup removes the state dir.
|
|
if agent_provider_template == "claude":
|
|
capture_claude_session_state(identity, exit_code)
|
|
return 0
|
|
finally:
|
|
# PRD 0018 chunk 2: prepare now writes the bottle's bind-mount
|
|
# sources under state/<slug>/. If we never reached the
|
|
# launch context (dry-run, preflight-N, prepare exception), or
|
|
# we did but nothing requested preservation, reap them along
|
|
# with everything else. `settle_state` subsumes the prior
|
|
# post-launch settlement and the new pre-launch cleanup.
|
|
settle_state(identity)
|
|
shutil.rmtree(stage_dir, ignore_errors=True)
|