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

//! Color theme and spinner animation for the TUI.

use ratatui::style::Color;

/// Spinner animation style.
#[derive(Debug, Clone, Copy, Default)]
#[expect(dead_code)]
pub(crate) enum SpinnerStyle {
    #[default]
    Stellar,
    Braille,
    Dots,
}

impl SpinnerStyle {
    pub(crate) fn frames(self) -> &'static [&'static str] {
        match self {
            Self::Stellar => &["✧", "✦", "✶", "✴", "✸", "✴", "✶", "✦"],
            Self::Braille => &["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"],
            Self::Dots => &["⣾", "⣽", "⣻", "⢿", "⡿", "⣟", "⣯", "⣷"],
        }
    }

    /// Get the current frame for a given elapsed time.
    pub(crate) fn frame_at(self, elapsed_ms: u128) -> &'static str {
        let frames = self.frames();
        let interval = 120u128;
        #[expect(clippy::arithmetic_side_effects)]
        // constant divisor, modulo by non-empty frames array
        let idx = (elapsed_ms / interval % frames.len() as u128) as usize;
        frames[idx]
    }
}

/// Color theme — primary text inherits the terminal foreground.
#[derive(Debug, Clone)]
pub(crate) struct Theme {
    /// User input text
    pub user: Color,
    /// Assistant response text
    pub assistant: Color,
    /// Muted/metadata text
    pub muted: Color,
    /// Tool names
    pub tool: Color,
    /// Success indicators
    pub success: Color,
    /// Warning indicators
    pub warning: Color,
    /// Error indicators
    pub error: Color,
    /// Thinking indicator
    pub thinking: Color,
    /// Border color
    pub border: Color,
    /// Cursor color
    pub cursor: Color,
    /// Diff added lines
    pub diff_added: Color,
    /// Diff removed lines
    pub diff_removed: Color,
    /// Diff context lines
    pub diff_context: Color,
    /// Spinner animation style
    pub spinner: SpinnerStyle,
}

impl Default for Theme {
    fn default() -> Self {
        Self {
            user: Color::Reset,
            assistant: Color::Reset,
            muted: Color::DarkGray,
            tool: Color::Cyan,
            success: Color::Green,
            warning: Color::Yellow,
            error: Color::Red,
            thinking: Color::Magenta,
            border: Color::DarkGray,
            cursor: Color::Reset,
            diff_added: Color::Green,
            diff_removed: Color::Red,
            diff_context: Color::DarkGray,
            spinner: SpinnerStyle::Stellar,
        }
    }
}

#[cfg(test)]
mod tests {
    use ratatui::style::Color;

    use super::Theme;

    #[test]
    fn primary_text_inherits_the_terminal_foreground() {
        let theme = Theme::default();

        assert_eq!(theme.user, Color::Reset);
        assert_eq!(theme.assistant, Color::Reset);
        assert_eq!(theme.cursor, Color::Reset);
    }
}