astrid 0.10.1

Command-line interface for Astrid secure agent runtime
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

//! `astrid ps` — list loaded capsules and their lifecycle state.
//!
//! Reads `KernelRequest::GetCapsuleMetadata` for the loaded list. Per-
//! capsule resource accounting (memory, IPC/sec, active calls, uptime)
//! requires telemetry that isn't fully wired (#639) — columns we can't
//! fill yet are marked `—`. We do not fabricate values.

use std::process::ExitCode;

use anyhow::Result;
use astrid_core::kernel_api::{KernelRequest, KernelResponse};
use clap::Args;
use colored::Colorize;
use serde::Serialize;

use crate::theme::Theme;
use crate::value_formatter::{ValueFormat, emit_structured};

#[derive(Args, Debug, Clone)]
pub(crate) struct PsArgs {
    /// Output format.
    #[arg(long, default_value = "pretty")]
    pub format: String,
}

/// JSON/YAML/TOML record for one capsule row.
#[derive(Debug, Clone, Serialize)]
pub(crate) struct CapsuleRow {
    /// Capsule name.
    pub capsule: String,
    /// Lifecycle state: `ready`, `loading`, `error`. Today only
    /// `ready` is exposed by `GetCapsuleMetadata` so other states are
    /// inferred as `unknown`.
    pub state: String,
}

/// Entry point for `astrid ps`.
pub(crate) async fn run(args: PsArgs) -> Result<ExitCode> {
    let format = ValueFormat::parse(&args.format);
    let socket_path = crate::socket_client::proxy_socket_path();
    if !socket_path.exists() {
        if format.is_pretty() {
            println!("{}", Theme::info("No Astrid daemon is running."));
        } else {
            emit_structured(&Vec::<CapsuleRow>::new(), format)?;
        }
        return Ok(ExitCode::SUCCESS);
    }
    let mut client = match crate::socket_client::connect_kernel_for_workspace(None).await {
        Ok(client) => client,
        Err(error) => {
            eprintln!(
                "{}",
                Theme::error(&format!("Failed to connect to daemon: {error:#}"))
            );
            return Ok(ExitCode::from(1));
        },
    };
    let entries = match client.request(KernelRequest::GetCapsuleMetadata).await {
        Ok(KernelResponse::CapsuleMetadata(list)) => list,
        Ok(KernelResponse::Error(message)) => {
            eprintln!(
                "{}",
                Theme::error(&format!(
                    "Daemon rejected capsule metadata request: {message}"
                ))
            );
            return Ok(ExitCode::from(1));
        },
        Ok(_) => {
            eprintln!(
                "{}",
                Theme::error("Unexpected response from daemon while listing capsules")
            );
            return Ok(ExitCode::from(1));
        },
        Err(error) => {
            eprintln!(
                "{}",
                Theme::error(&format!("Failed to query daemon: {error}"))
            );
            return Ok(ExitCode::from(1));
        },
    };
    let mut rows: Vec<CapsuleRow> = entries
        .into_iter()
        .map(|e| CapsuleRow {
            capsule: e.name,
            state: "ready".into(),
        })
        .collect();
    rows.sort_by(|a, b| a.capsule.cmp(&b.capsule));
    if !format.is_pretty() {
        emit_structured(&rows, format)?;
        return Ok(ExitCode::SUCCESS);
    }
    if rows.is_empty() {
        println!("{}", Theme::info("(no capsules loaded)"));
        return Ok(ExitCode::SUCCESS);
    }
    println!(
        "{:<28}  {:<8}  {:<10}  {:<8}  {}",
        "CAPSULE".bold(),
        "STATE".bold(),
        "MEM".bold(),
        "CALLS".bold(),
        "UPTIME".bold()
    );
    for r in &rows {
        println!(
            "{:<28}  {:<8}  {:<10}  {:<8}  {}",
            r.capsule,
            r.state.green(),
            "—".dimmed(),
            "—".dimmed(),
            "—".dimmed()
        );
    }
    println!(
        "\n{}",
        Theme::info(
            "Memory / call / uptime columns require per-capsule telemetry (#639) — empty until that lands."
        )
    );
    Ok(ExitCode::SUCCESS)
}