didericis
a21f2358c6
tracker-policy-pr / check-pr (pull_request) Successful in 16s
test / integration-docker (pull_request) Successful in 21s
lint / lint (push) Failing after 58s
test / unit (pull_request) Successful in 2m11s
test / integration-firecracker (pull_request) Successful in 3m27s
test / coverage (pull_request) Successful in 16s
test / publish-infra (pull_request) Has been skipped
_common had rotted into a junk drawer. Trim it to what actually
justifies a shared leaf module and rename for legibility:
* REPO_DIR: deleted (dead — nothing imported it).
* read_tty_line: dropped the pointless re-export; cleanup/start/init
now import it straight from bot_bottle.util.
* USER_CWD: deleted. It captured os.getcwd() at import time, but
nothing chdirs and no test patched it, so it was equivalent to a
live os.getcwd() — inlined at the six call sites.
* PROG: kept, now the sole member. It's still a leaf (both the
dispatcher and the commands it imports need it) so it can't move
into __init__ without a circular import.
_common.py -> constants.py. Full unit suite green (2243); CLI dispatch
verified.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""list: list available agents or active bottles."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
|
|
from ...backend import enumerate_active_agents
|
|
from ...manifest import ManifestIndex
|
|
from ..constants import PROG
|
|
|
|
_ANSI_COLOR_CODES: dict[str, str] = {
|
|
"red": "\033[91m",
|
|
"green": "\033[92m",
|
|
"yellow": "\033[93m",
|
|
"blue": "\033[94m",
|
|
"magenta": "\033[95m",
|
|
}
|
|
_ANSI_RESET = "\033[0m"
|
|
|
|
|
|
def _ansi_label(text: str, color: str) -> str:
|
|
if not color:
|
|
return text
|
|
if not sys.stdout.isatty():
|
|
return text
|
|
term = os.environ.get("TERM", "")
|
|
if term in ("dumb", ""):
|
|
return text
|
|
code = _ANSI_COLOR_CODES.get(color)
|
|
if not code:
|
|
return text
|
|
return f"{code}{text}{_ANSI_RESET}"
|
|
|
|
|
|
def cmd_list(argv: list[str]) -> int:
|
|
parser = argparse.ArgumentParser(prog=f"{PROG} list", add_help=True)
|
|
parser.add_argument("scope", choices=["available", "active"])
|
|
args = parser.parse_args(argv)
|
|
|
|
if args.scope == "available":
|
|
manifest = ManifestIndex.resolve(os.getcwd())
|
|
for name in manifest.all_agent_names:
|
|
print(name)
|
|
return 0
|
|
|
|
# `active` enumerates every backend (docker, firecracker,
|
|
# macos-container) so non-docker bottles aren't hidden behind
|
|
# the env var.
|
|
active = enumerate_active_agents()
|
|
if not active:
|
|
print("no active bot-bottle bottles", file=sys.stderr)
|
|
return 0
|
|
# One line per bottle: `<backend>\t<slug>\t<label>\t<services>`.
|
|
# Tab-separated keeps the format stable for shell pipelines.
|
|
for b in active:
|
|
services = ",".join(b.services) if b.services else "-"
|
|
display_name = f"{b.label} ({b.agent_name})" if b.label else b.agent_name
|
|
colored_name = _ansi_label(display_name, b.color)
|
|
print(f"{b.backend_name}\t{b.slug}\t{colored_name}\t{services}")
|
|
return 0
|