didericis-claude
5fcca2060f
Remove auto-migration from DbStore.__init__; add explicit check_migrations() and migrate() methods. The CLI now checks both stores on every startup and prompts the user to confirm before migrating. supervise.prepare() calls .migrate() directly now that __init__ no longer does it implicitly. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
110 lines
3.4 KiB
Python
110 lines
3.4 KiB
Python
"""Main CLI dispatcher.
|
|
|
|
Commands: cleanup, commit, edit, info, init, list, resume, start, supervise
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
|
|
from ..audit_store import AuditStore
|
|
from ..db_store import DbVersionError
|
|
from ..log import Die, die, error
|
|
from ..manifest import ManifestError
|
|
from ..queue_store import QueueStore
|
|
from ._common import PROG
|
|
from . import list as _list_mod
|
|
from .cleanup import cmd_cleanup
|
|
from .commit import cmd_commit
|
|
from .edit import cmd_edit
|
|
from .info import cmd_info
|
|
from .init import cmd_init
|
|
from .resume import cmd_resume
|
|
from .start import cmd_start
|
|
from .supervise import cmd_supervise
|
|
|
|
cmd_list = _list_mod.cmd_list
|
|
|
|
COMMANDS = {
|
|
"cleanup": cmd_cleanup,
|
|
"commit": cmd_commit,
|
|
"edit": cmd_edit,
|
|
"info": cmd_info,
|
|
"init": cmd_init,
|
|
"list": cmd_list,
|
|
"resume": cmd_resume,
|
|
"start": cmd_start,
|
|
"supervise": cmd_supervise,
|
|
}
|
|
|
|
|
|
def usage() -> None:
|
|
sys.stderr.write(f"usage: {PROG} <command> [args...]\n\n")
|
|
sys.stderr.write("Commands:\n")
|
|
sys.stderr.write(" cleanup stop and remove all active bot-bottle containers\n")
|
|
sys.stderr.write(" commit snapshot a running bottle's container state to a Docker image\n")
|
|
sys.stderr.write(" edit open an agent in vim for editing\n")
|
|
sys.stderr.write(" info print env, skills, and prompt details for a named agent\n")
|
|
sys.stderr.write(" init interactively create a new agent and add it to bot-bottle.json\n")
|
|
sys.stderr.write(" list list available agents or active containers\n")
|
|
sys.stderr.write(
|
|
" resume re-launch a bottle by its identity "
|
|
"(continues state from PRD 0016)\n"
|
|
)
|
|
sys.stderr.write(
|
|
" start boot a container for a named agent and "
|
|
"attach an interactive session\n"
|
|
)
|
|
sys.stderr.write(
|
|
" supervise view + approve/modify/reject pending supervise "
|
|
"proposals (PRD 0013)\n\n"
|
|
)
|
|
sys.stderr.write(f"Run '{PROG} <command> --help' for command-specific usage.\n")
|
|
|
|
|
|
def main(argv: list[str] | None = None) -> int:
|
|
if argv is None:
|
|
argv = sys.argv[1:]
|
|
if not argv:
|
|
usage()
|
|
return 2
|
|
command = argv[0]
|
|
rest = argv[1:]
|
|
if command in ("-h", "--help"):
|
|
usage()
|
|
return 0
|
|
handler = COMMANDS.get(command)
|
|
if handler is None:
|
|
usage()
|
|
die(f"unknown command: {command}")
|
|
queue_store = QueueStore("")
|
|
audit_store = AuditStore()
|
|
if not queue_store.check_migrations() or not audit_store.check_migrations():
|
|
sys.stderr.write("bot-bottle: database schema is out of date\n")
|
|
sys.stderr.write("Migrate now? [y/N] ")
|
|
sys.stderr.flush()
|
|
try:
|
|
answer = sys.stdin.readline().strip().lower()
|
|
except EOFError:
|
|
answer = ""
|
|
if answer != "y":
|
|
error("migration required — re-run and confirm to migrate")
|
|
return 1
|
|
queue_store.migrate()
|
|
audit_store.migrate()
|
|
try:
|
|
return handler(rest) or 0
|
|
except ManifestError as e:
|
|
# Manifest/config problems surface as a catchable exception;
|
|
# print the reason and exit non-zero (same UX die() used to give).
|
|
error(str(e))
|
|
return 1
|
|
except Die as e:
|
|
return e.code if isinstance(e.code, int) else 1
|
|
except KeyboardInterrupt:
|
|
return 130
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|