aa-cli 0.0.1-alpha.8

aasm — command-line tool for Agent Assembly
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

//! Top-level CLI subcommand definitions and dispatch.

use std::process::ExitCode;

use clap::Subcommand;

use crate::config::ResolvedContext;
use crate::output::OutputFormat;

pub mod admin;
pub mod agent;
pub mod alerts;
pub mod approvals;
pub mod audit;
pub mod budget;
pub mod completion;
pub mod config;
pub mod context;
pub mod cost;
pub mod dashboard;
pub mod gateway;
pub mod gw_probe;
pub mod logs;
pub mod permissions;
pub mod pidfile;
pub mod policy;
pub mod proxy;
pub mod sandbox;
pub mod start;
pub mod status;
pub mod stop;
pub mod topology;
pub mod trace;
pub mod version;

/// Top-level subcommands for the `aasm` CLI.
#[derive(Subcommand)]
pub enum Commands {
    /// Gateway administrative operations.
    Admin(admin::AdminArgs),
    /// Manage monitored agent processes.
    Agent(agent::AgentArgs),
    /// Manage governance alerts.
    Alerts(alerts::AlertsArgs),
    /// Query audit log entries and export compliance reports.
    Audit(audit::AuditArgs),
    /// Query and stream audit log events.
    Logs(logs::LogsArgs),
    /// Manage governance policies.
    Policy(policy::PolicyArgs),
    /// Manage named API contexts (connection profiles).
    Context(context::ContextArgs),
    /// Validate an `agent-assembly.toml` runtime configuration file.
    Config(config::ConfigArgs),
    /// Generate shell completion scripts.
    Completion(completion::CompletionArgs),
    /// Show fleet health, agents, approvals, and budget at a glance.
    Status(status::StatusArgs),
    /// Show CLI and gateway version information.
    Version,
    /// Visualize a session trace (tree or timeline).
    Trace(trace::TraceArgs),
    /// Manage human-in-the-loop approval requests.
    Approvals(approvals::ApprovalsArgs),
    /// Query cost summary and forecast spending.
    Cost(cost::CostArgs),
    /// Open an interactive TUI dashboard for real-time governance monitoring.
    Dashboard(dashboard::DashboardArgs),
    /// Manage the aa-gateway governance daemon — agent registry, policy engine, audit log.
    Gateway(gateway::GatewayArgs),
    /// Run a WebAssembly tool inside the Agent Assembly sandbox (filesystem + CPU + memory + wall-clock isolation).
    Sandbox(sandbox::SandboxArgs),
    /// Visualize agent topology, trees, lineage, and statistics.
    Topology(topology::TopologyArgs),
    /// Manage the aa-proxy sidecar — lifecycle, CA trust, and log tailing.
    Proxy(proxy::ProxyArgs),
    /// Start the locally-managed Agent Assembly gateway process.
    Start(start::StartArgs),
    /// Stop the locally-managed Agent Assembly gateway process.
    Stop(stop::StopArgs),
}

/// Dispatch the parsed CLI command to the appropriate handler.
pub fn dispatch(cmd: Commands, ctx: &ResolvedContext, output: OutputFormat) -> ExitCode {
    match cmd {
        Commands::Admin(args) => admin::dispatch(args, ctx, output),
        Commands::Agent(args) => agent::dispatch(args, ctx, output),
        Commands::Alerts(args) => alerts::dispatch(args, ctx, output),
        Commands::Audit(args) => audit::dispatch(args, ctx, output),
        Commands::Logs(args) => logs::dispatch(args, ctx),
        Commands::Policy(args) => policy::dispatch(args, ctx, output),
        Commands::Context(args) => context::dispatch(args),
        Commands::Config(args) => config::dispatch(args),
        Commands::Completion(args) => completion::run(args),
        Commands::Status(args) => status::dispatch(args, ctx, output),
        Commands::Version => version::run(ctx, output),
        Commands::Trace(args) => trace::dispatch(args, ctx, output),
        Commands::Approvals(args) => approvals::dispatch(args, ctx, output),
        Commands::Cost(args) => cost::dispatch(args, ctx, output),
        Commands::Dashboard(args) => dashboard::dispatch(args, ctx),
        Commands::Gateway(args) => gateway::dispatch(args),
        Commands::Sandbox(args) => sandbox::dispatch(args),
        Commands::Topology(args) => topology::dispatch(args, ctx, output),
        Commands::Proxy(args) => proxy::dispatch(args),
        Commands::Start(args) => start::run(args),
        Commands::Stop(args) => stop::run(args),
    }
}