aa-cli 0.0.1-alpha.9

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

//! Confirmation dialog overlay for approve/reject actions.

use ratatui::layout::{Constraint, Direction, Layout, Rect};
use ratatui::style::{Color, Modifier, Style};
use ratatui::text::Line;
use ratatui::widgets::{Block, Borders, Clear, Paragraph};
use ratatui::Frame;

use crate::commands::status::models::ApprovalResponse;

/// Which action the dialog is confirming.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum DialogAction {
    Approve,
    Reject,
}

/// Render a centered confirmation dialog over the dashboard.
pub fn draw_confirm_dialog(f: &mut Frame, approval: &ApprovalResponse, action: DialogAction) {
    let area = centered_rect(50, 30, f.area());

    // Clear the area behind the dialog.
    f.render_widget(Clear, area);

    let (title, color) = match action {
        DialogAction::Approve => ("Approve?", Color::Green),
        DialogAction::Reject => ("Reject?", Color::Red),
    };

    let block = Block::default()
        .title(format!(" {title} "))
        .borders(Borders::ALL)
        .border_style(Style::default().fg(color));

    let inner = block.inner(area);
    f.render_widget(block, area);

    let chunks = Layout::default()
        .direction(Direction::Vertical)
        .constraints([Constraint::Min(3), Constraint::Length(2)])
        .split(inner);

    // Summary of what is being approved/rejected.
    let summary = Paragraph::new(vec![
        Line::from(format!("ID:     {}", approval.id)),
        Line::from(format!("Agent:  {}", approval.agent_id)),
        Line::from(format!("Action: {}", approval.action)),
        Line::from(format!("Reason: {}", approval.reason)),
    ]);
    f.render_widget(summary, chunks[0]);

    // Instruction line.
    let instruction = Paragraph::new(Line::from(vec![
        ratatui::text::Span::styled("y", Style::default().add_modifier(Modifier::BOLD)),
        ratatui::text::Span::raw(" confirm  "),
        ratatui::text::Span::styled("n", Style::default().add_modifier(Modifier::BOLD)),
        ratatui::text::Span::raw("/"),
        ratatui::text::Span::styled("Esc", Style::default().add_modifier(Modifier::BOLD)),
        ratatui::text::Span::raw(" cancel"),
    ]))
    .style(Style::default().fg(Color::DarkGray));
    f.render_widget(instruction, chunks[1]);
}

/// Compute a centered rectangle within `area`.
fn centered_rect(percent_x: u16, percent_y: u16, area: Rect) -> Rect {
    let vertical = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Percentage((100 - percent_y) / 2),
            Constraint::Percentage(percent_y),
            Constraint::Percentage((100 - percent_y) / 2),
        ])
        .split(area);

    let horizontal = Layout::default()
        .direction(Direction::Horizontal)
        .constraints([
            Constraint::Percentage((100 - percent_x) / 2),
            Constraint::Percentage(percent_x),
            Constraint::Percentage((100 - percent_x) / 2),
        ])
        .split(vertical[1]);

    horizontal[1]
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn centered_rect_produces_inner_rect() {
        let area = Rect::new(0, 0, 100, 50);
        let center = centered_rect(50, 40, area);
        // Should be roughly centered.
        assert!(center.x > 0);
        assert!(center.y > 0);
        assert!(center.width > 0);
        assert!(center.height > 0);
        assert!(center.x + center.width <= area.width);
        assert!(center.y + center.height <= area.height);
    }
}