sage-cli 0.13.57

Command-line interface for Sage Agent
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

//! UI state management for rnk app

use parking_lot::RwLock;
use sage_core::ui::bridge::state::UiSessionInfo;
use std::sync::Arc;

/// Permission mode for the UI
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum PermissionMode {
    Normal,
    Bypass,
    Plan,
}

impl PermissionMode {
    pub fn cycle(self) -> Self {
        match self {
            PermissionMode::Normal => PermissionMode::Bypass,
            PermissionMode::Bypass => PermissionMode::Plan,
            PermissionMode::Plan => PermissionMode::Normal,
        }
    }
}

/// UI state shared between render loop and background tasks
pub struct UiState {
    /// Current input text
    pub input_text: String,
    /// Permission mode
    pub permission_mode: PermissionMode,
    /// Whether agent is busy
    pub is_busy: bool,
    /// Status text
    pub status_text: String,
    /// Should quit
    pub should_quit: bool,
    /// Number of messages already printed
    pub printed_count: usize,
    /// Error already displayed (to avoid duplicate error messages)
    pub error_displayed: bool,
    /// Session info
    pub session: UiSessionInfo,
    /// Selected command suggestion index (0-based, default 0 for first match)
    pub suggestion_index: usize,
    /// Currently printed tool (to avoid duplicate tool start messages)
    pub current_tool_printed: Option<String>,
    /// Pending tool to print (waits for message to be printed first)
    pub pending_tool: Option<(String, String)>,
    /// Animation frame counter for spinner
    pub animation_frame: usize,
    /// Pending tool result to print (tool_name, success, duration)
    pub pending_tool_result: Option<(String, bool, std::time::Duration)>,
    /// Model selection mode
    pub model_select_mode: bool,
    /// Available models for selection
    pub available_models: Vec<String>,
    /// Selected model index
    pub model_select_index: usize,
}

impl Default for UiState {
    fn default() -> Self {
        Self {
            input_text: String::new(),
            permission_mode: PermissionMode::Normal,
            is_busy: false,
            status_text: String::new(),
            should_quit: false,
            printed_count: 0,
            error_displayed: false,
            session: UiSessionInfo {
                session_id: None,
                model: "unknown".to_string(),
                provider: "unknown".to_string(),
                working_dir: std::env::current_dir()
                    .unwrap_or_default()
                    .to_string_lossy()
                    .to_string(),
                git_branch: None,
                step: 0,
                max_steps: None,
            },
            suggestion_index: 0,
            current_tool_printed: None,
            pending_tool: None,
            animation_frame: 0,
            pending_tool_result: None,
            model_select_mode: false,
            available_models: Vec::new(),
            model_select_index: 0,
        }
    }
}

/// Shared state wrapper
pub type SharedState = Arc<RwLock<UiState>>;

/// Command from UI to executor
#[derive(Debug)]
pub enum UiCommand {
    Submit(String),
    Cancel,
    Quit,
}