rab-agent 0.1.0

rab is a lightweight, extensible, Rust-based coding agent.
Documentation 
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

use std::cell::RefCell;

use crate::agent::ui::theme::current_theme;
use crate::tui::Component;
use crate::tui::components::r#box::TuiBox;
use crate::tui::components::markdown::{DefaultTextStyle, Markdown, MarkdownOptions};

const OSC133_ZONE_START: &str = "\x1b]133;A\x07";
const OSC133_ZONE_END: &str = "\x1b]133;B\x07";
const OSC133_ZONE_FINAL: &str = "\x1b]133;C\x07";

/// User message component — matches pi's UserMessageComponent.
/// Renders text in a Box with `userMessageBg` background, `userMessageText` color.
pub struct UserMessageComponent {
    box_component: TuiBox,
    cached_lines: RefCell<Option<Vec<String>>>,
    cached_width: RefCell<usize>,
}

impl UserMessageComponent {
    pub fn new(text: impl Into<String>) -> Self {
        let text = text.into();
        let theme = current_theme();
        let bg_ansi = theme.bg_ansi("userMessageBg").to_string();
        drop(theme);

        let mut msg_box = TuiBox::new(
            1,
            1,
            Some(std::boxed::Box::new(move |s: &str| -> String {
                format!("{}{}\x1b[49m", bg_ansi, s)
            })),
        );

        // Build the markdown renderer with userMessageText color
        let md_theme = crate::agent::ui::theme::get_markdown_theme();
        let default_style = DefaultTextStyle {
            color: Some(std::sync::Arc::new(|s: &str| -> String {
                let t = current_theme();
                t.fg("userMessageText", s)
            })),
            bg_color: None,
            bold: false,
            italic: false,
            strikethrough: false,
            underline: false,
        };
        let md = Markdown::new(
            text.clone(),
            0,
            0,
            md_theme,
            Some(default_style),
            Some(MarkdownOptions {
                preserve_ordered_list_markers: true,
            }),
        );
        msg_box.add_child(std::boxed::Box::new(md));

        Self {
            box_component: msg_box,
            cached_lines: RefCell::new(None),
            cached_width: RefCell::new(0),
        }
    }
}

impl Component for UserMessageComponent {
    fn set_expanded(&mut self, _expanded: bool) {
        // User messages are always fully visible
    }

    fn render(&self, width: usize) -> Vec<String> {
        if *self.cached_width.borrow() == width
            && let Some(ref lines) = *self.cached_lines.borrow()
        {
            return lines.clone();
        }

        let mut lines = self.box_component.render(width);
        if !lines.is_empty() {
            lines[0] = format!("{}{}", OSC133_ZONE_START, &lines[0]);
            if let Some(last) = lines.last_mut() {
                last.push_str(OSC133_ZONE_END);
                last.push_str(OSC133_ZONE_FINAL);
            }
        }

        // Cache
        let result = lines.clone();
        *self.cached_lines.borrow_mut() = Some(lines);
        *self.cached_width.borrow_mut() = width;
        result
    }

    fn invalidate(&mut self) {
        *self.cached_lines.borrow_mut() = None;
        self.box_component.invalidate();
    }
}