wisp/components/
plan_view.rs

1use agent_client_protocol::{PlanEntry, PlanEntryStatus};
2
3use tui::{Line, Style, ViewContext};
4
5const CHECKBOX_EMPTY: &str = "\u{2610}"; // Ballot Box
6const CHECKBOX_FILLED: &str = "\u{2611}"; // Ballot Box with Check
7
8/// Renders the agent's task plan as a compact checklist.
9///
10/// ```text
11/// Plan
12///   ☑ ~~Research AI agent patterns~~
13///   ☑ Implement task tracking
14///   ☐ Write integration tests
15/// ```
16pub struct PlanView<'a> {
17    pub entries: &'a [PlanEntry],
18}
19
20impl PlanView<'_> {
21    pub fn render(&self, context: &ViewContext) -> Vec<Line> {
22        if self.entries.is_empty() {
23            return vec![];
24        }
25
26        let mut lines = Vec::with_capacity(self.entries.len() + 2);
27        lines.push(Line::default());
28
29        let mut header = Line::default();
30        header.push_styled("Plan".to_string(), context.theme.muted());
31        lines.push(header);
32
33        for entry in self.entries {
34            let mut line = Line::default();
35            match entry.status {
36                PlanEntryStatus::Completed => {
37                    line.push_styled(format!("  {CHECKBOX_FILLED} "), context.theme.muted());
38                    let completed_style = Style::fg(context.theme.muted()).strikethrough();
39                    line.push_with_style(entry.content.clone(), completed_style);
40                }
41                PlanEntryStatus::InProgress => {
42                    line.push_styled(format!("  {CHECKBOX_FILLED} "), context.theme.primary());
43                    line.push_text(entry.content.clone());
44                }
45                _ => {
46                    line.push_styled(format!("  {CHECKBOX_EMPTY} "), context.theme.muted());
47                    line.push_styled(entry.content.clone(), context.theme.muted());
48                }
49            }
50            lines.push(line);
51        }
52
53        lines
54    }
55}
56
57#[cfg(test)]
58mod tests {
59    use super::*;
60    use agent_client_protocol::{PlanEntry, PlanEntryPriority, PlanEntryStatus};
61
62    fn ctx() -> ViewContext {
63        ViewContext::new((80, 24))
64    }
65
66    fn entry(content: &str, status: PlanEntryStatus) -> PlanEntry {
67        PlanEntry::new(content.to_string(), PlanEntryPriority::Medium, status)
68    }
69
70    #[test]
71    fn empty_entries_render_nothing() {
72        let view = PlanView { entries: &[] };
73        assert!(view.render(&ctx()).is_empty());
74    }
75
76    #[test]
77    fn renders_header_plus_entries() {
78        let entries = vec![
79            entry("Research", PlanEntryStatus::Completed),
80            entry("Implement", PlanEntryStatus::InProgress),
81            entry("Test", PlanEntryStatus::Pending),
82        ];
83        let view = PlanView { entries: &entries };
84        let lines = view.render(&ctx());
85        assert_eq!(lines.len(), 5);
86        assert_eq!(lines[0].plain_text(), "");
87        assert_eq!(lines[1].plain_text(), "Plan");
88    }
89
90    #[test]
91    fn completed_entry_has_filled_checkbox() {
92        let entries = vec![entry("Done task", PlanEntryStatus::Completed)];
93        let view = PlanView { entries: &entries };
94        let lines = view.render(&ctx());
95        let text = lines[2].plain_text();
96        assert!(text.contains(CHECKBOX_FILLED));
97        assert!(text.contains("Done task"));
98    }
99
100    #[test]
101    fn completed_entry_has_strikethrough() {
102        let entries = vec![entry("Done task", PlanEntryStatus::Completed)];
103        let view = PlanView { entries: &entries };
104        let lines = view.render(&ctx());
105        let spans = lines[2].spans();
106        let text_span = &spans[1];
107        assert!(text_span.style().strikethrough);
108    }
109
110    #[test]
111    fn in_progress_entry_has_filled_checkbox() {
112        let entries = vec![entry("Working", PlanEntryStatus::InProgress)];
113        let view = PlanView { entries: &entries };
114        let lines = view.render(&ctx());
115        let text = lines[2].plain_text();
116        assert!(text.contains(CHECKBOX_FILLED));
117        assert!(text.contains("Working"));
118    }
119
120    #[test]
121    fn pending_entry_has_empty_checkbox() {
122        let entries = vec![entry("Todo", PlanEntryStatus::Pending)];
123        let view = PlanView { entries: &entries };
124        let lines = view.render(&ctx());
125        let text = lines[2].plain_text();
126        assert!(text.contains(CHECKBOX_EMPTY));
127        assert!(text.contains("Todo"));
128    }
129}
wisp/components/plan_view.rs

wisp/components/
plan_view.rs
