438 lines
16 KiB
Rust
438 lines
16 KiB
Rust
|
use std::io;
|
|||
|
|
|
|||
|
|
use crossterm::event::{self, Event, KeyCode, KeyEvent, KeyModifiers};
|
|||
|
|
use crossterm::terminal::{self, EnterAlternateScreen, LeaveAlternateScreen};
|
|||
|
|
use crossterm::execute;
|
|||
|
|
use ratatui::backend::CrosstermBackend;
|
|||
|
|
use ratatui::layout::{Constraint, Layout, Rect};
|
|||
|
|
use ratatui::style::{Color, Modifier, Style};
|
|||
|
|
use ratatui::text::{Line, Span, Text};
|
|||
|
|
use ratatui::widgets::{Block, Borders, Paragraph, Wrap};
|
|||
|
|
use ratatui::Terminal;
|
|||
|
|
|
|||
|
|
// ── Sol color palette ──────────────────────────────────────────────────────
|
|||
|
|
|
|||
|
|
const SOL_YELLOW: Color = Color::Rgb(245, 197, 66);
|
|||
|
|
const SOL_AMBER: Color = Color::Rgb(232, 168, 64);
|
|||
|
|
const SOL_BLUE: Color = Color::Rgb(108, 166, 224);
|
|||
|
|
const SOL_RED: Color = Color::Rgb(224, 88, 88);
|
|||
|
|
const SOL_DIM: Color = Color::Rgb(138, 122, 90);
|
|||
|
|
const SOL_GRAY: Color = Color::Rgb(112, 112, 112);
|
|||
|
|
const SOL_FAINT: Color = Color::Rgb(80, 80, 80);
|
|||
|
|
const SOL_STATUS: Color = Color::Rgb(106, 96, 80);
|
|||
|
|
const SOL_APPROVAL_BG: Color = Color::Rgb(50, 42, 20);
|
|||
|
|
const SOL_APPROVAL_CMD: Color = Color::Rgb(200, 180, 120);
|
|||
|
|
|
|||
|
|
// ── Message types for the conversation log ─────────────────────────────────
|
|||
|
|
|
|||
|
|
#[derive(Clone)]
|
|||
|
|
pub enum LogEntry {
|
|||
|
|
UserInput(String),
|
|||
|
|
AssistantText(String),
|
|||
|
|
ToolSuccess { name: String, detail: String },
|
|||
|
|
ToolExecuting { name: String, detail: String },
|
|||
|
|
ToolFailed { name: String, detail: String },
|
|||
|
|
ToolOutput { lines: Vec<String>, collapsed: bool },
|
|||
|
|
Status(String),
|
|||
|
|
Error(String),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ── Approval state ─────────────────────────────────────────────────────────
|
|||
|
|
|
|||
|
|
pub struct ApprovalPrompt {
|
|||
|
|
pub tool_name: String,
|
|||
|
|
pub command: String,
|
|||
|
|
pub options: Vec<String>,
|
|||
|
|
pub selected: usize,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ── App state ──────────────────────────────────────────────────────────────
|
|||
|
|
|
|||
|
|
pub struct App {
|
|||
|
|
pub log: Vec<LogEntry>,
|
|||
|
|
pub input: String,
|
|||
|
|
pub cursor_pos: usize,
|
|||
|
|
pub scroll_offset: u16,
|
|||
|
|
pub project_name: String,
|
|||
|
|
pub branch: String,
|
|||
|
|
pub model: String,
|
|||
|
|
pub input_tokens: u32,
|
|||
|
|
pub output_tokens: u32,
|
|||
|
|
pub approval: Option<ApprovalPrompt>,
|
|||
|
|
pub is_thinking: bool,
|
|||
|
|
pub should_quit: bool,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
impl App {
|
|||
|
|
pub fn new(project_name: &str, branch: &str, model: &str) -> Self {
|
|||
|
|
Self {
|
|||
|
|
log: Vec::new(),
|
|||
|
|
input: String::new(),
|
|||
|
|
cursor_pos: 0,
|
|||
|
|
scroll_offset: 0,
|
|||
|
|
project_name: project_name.into(),
|
|||
|
|
branch: branch.into(),
|
|||
|
|
model: model.into(),
|
|||
|
|
input_tokens: 0,
|
|||
|
|
output_tokens: 0,
|
|||
|
|
approval: None,
|
|||
|
|
is_thinking: false,
|
|||
|
|
should_quit: false,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
pub fn push_log(&mut self, entry: LogEntry) {
|
|||
|
|
self.log.push(entry);
|
|||
|
|
// Auto-scroll to bottom
|
|||
|
|
self.scroll_offset = u16::MAX;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ── Rendering ──────────────────────────────────────────────────────────────
|
|||
|
|
|
|||
|
|
pub fn draw(frame: &mut ratatui::Frame, app: &App) {
|
|||
|
|
let area = frame.area();
|
|||
|
|
|
|||
|
|
// Layout: title (1) + log (flex) + input (3) + status (1)
|
|||
|
|
let chunks = Layout::vertical([
|
|||
|
|
Constraint::Length(1), // title bar
|
|||
|
|
Constraint::Min(5), // conversation log
|
|||
|
|
Constraint::Length(3), // input area
|
|||
|
|
Constraint::Length(1), // status bar
|
|||
|
|
])
|
|||
|
|
.split(area);
|
|||
|
|
|
|||
|
|
draw_title_bar(frame, chunks[0], app);
|
|||
|
|
draw_log(frame, chunks[1], app);
|
|||
|
|
|
|||
|
|
if let Some(ref approval) = app.approval {
|
|||
|
|
draw_approval(frame, chunks[2], approval);
|
|||
|
|
} else {
|
|||
|
|
draw_input(frame, chunks[2], app);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
draw_status_bar(frame, chunks[3], app);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn draw_title_bar(frame: &mut ratatui::Frame, area: Rect, app: &App) {
|
|||
|
|
let left = vec![
|
|||
|
|
Span::styled("sunbeam code", Style::default().fg(SOL_YELLOW).add_modifier(Modifier::BOLD)),
|
|||
|
|
Span::styled(" · ", Style::default().fg(SOL_FAINT)),
|
|||
|
|
Span::raw(&app.project_name),
|
|||
|
|
Span::styled(" · ", Style::default().fg(SOL_FAINT)),
|
|||
|
|
Span::styled(&app.branch, Style::default().fg(SOL_DIM)),
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
let right = Span::styled(&app.model, Style::default().fg(SOL_DIM));
|
|||
|
|
|
|||
|
|
// Render left-aligned title and right-aligned model
|
|||
|
|
let title_line = Line::from(left);
|
|||
|
|
frame.render_widget(Paragraph::new(title_line), area);
|
|||
|
|
|
|||
|
|
let right_area = Rect {
|
|||
|
|
x: area.width.saturating_sub(right.width() as u16 + 1),
|
|||
|
|
y: area.y,
|
|||
|
|
width: right.width() as u16 + 1,
|
|||
|
|
height: 1,
|
|||
|
|
};
|
|||
|
|
frame.render_widget(Paragraph::new(Line::from(right)), right_area);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn draw_log(frame: &mut ratatui::Frame, area: Rect, app: &App) {
|
|||
|
|
let mut lines: Vec<Line> = Vec::new();
|
|||
|
|
|
|||
|
|
for entry in &app.log {
|
|||
|
|
match entry {
|
|||
|
|
LogEntry::UserInput(text) => {
|
|||
|
|
lines.push(Line::from(""));
|
|||
|
|
lines.push(Line::from(vec![
|
|||
|
|
Span::styled("> ", Style::default().fg(SOL_DIM)),
|
|||
|
|
Span::raw(text.as_str()),
|
|||
|
|
]));
|
|||
|
|
lines.push(Line::from(""));
|
|||
|
|
}
|
|||
|
|
LogEntry::AssistantText(text) => {
|
|||
|
|
for line in text.lines() {
|
|||
|
|
lines.push(Line::from(Span::styled(line, Style::default().fg(SOL_YELLOW))));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
LogEntry::ToolSuccess { name, detail } => {
|
|||
|
|
lines.push(Line::from(vec![
|
|||
|
|
Span::styled(" ✓ ", Style::default().fg(SOL_BLUE)),
|
|||
|
|
Span::styled(name.as_str(), Style::default().fg(SOL_AMBER)),
|
|||
|
|
Span::styled(format!(" {detail}"), Style::default().fg(SOL_DIM)),
|
|||
|
|
]));
|
|||
|
|
}
|
|||
|
|
LogEntry::ToolExecuting { name, detail } => {
|
|||
|
|
lines.push(Line::from(vec![
|
|||
|
|
Span::styled(" ● ", Style::default().fg(SOL_AMBER)),
|
|||
|
|
Span::styled(name.as_str(), Style::default().fg(SOL_AMBER)),
|
|||
|
|
Span::styled(format!(" {detail}"), Style::default().fg(SOL_DIM)),
|
|||
|
|
]));
|
|||
|
|
}
|
|||
|
|
LogEntry::ToolFailed { name, detail } => {
|
|||
|
|
lines.push(Line::from(vec![
|
|||
|
|
Span::styled(" ✗ ", Style::default().fg(SOL_RED)),
|
|||
|
|
Span::styled(name.as_str(), Style::default().fg(SOL_RED)),
|
|||
|
|
Span::styled(format!(" {detail}"), Style::default().fg(SOL_DIM)),
|
|||
|
|
]));
|
|||
|
|
}
|
|||
|
|
LogEntry::ToolOutput { lines: output_lines, collapsed } => {
|
|||
|
|
let show = if *collapsed { 5 } else { output_lines.len() };
|
|||
|
|
for line in output_lines.iter().take(show) {
|
|||
|
|
lines.push(Line::from(Span::styled(
|
|||
|
|
format!(" {line}"),
|
|||
|
|
Style::default().fg(SOL_GRAY),
|
|||
|
|
)));
|
|||
|
|
}
|
|||
|
|
if *collapsed && output_lines.len() > 5 {
|
|||
|
|
lines.push(Line::from(Span::styled(
|
|||
|
|
format!(" … +{} lines (ctrl+o to expand)", output_lines.len() - 5),
|
|||
|
|
Style::default().fg(SOL_FAINT),
|
|||
|
|
)));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
LogEntry::Status(msg) => {
|
|||
|
|
lines.push(Line::from(Span::styled(
|
|||
|
|
format!(" [{msg}]"),
|
|||
|
|
Style::default().fg(SOL_DIM),
|
|||
|
|
)));
|
|||
|
|
}
|
|||
|
|
LogEntry::Error(msg) => {
|
|||
|
|
lines.push(Line::from(Span::styled(
|
|||
|
|
format!(" error: {msg}"),
|
|||
|
|
Style::default().fg(SOL_RED),
|
|||
|
|
)));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if app.is_thinking {
|
|||
|
|
lines.push(Line::from(""));
|
|||
|
|
lines.push(Line::from(Span::styled(
|
|||
|
|
" thinking...",
|
|||
|
|
Style::default().fg(SOL_DIM).add_modifier(Modifier::ITALIC),
|
|||
|
|
)));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let total_lines = lines.len() as u16;
|
|||
|
|
let visible = area.height;
|
|||
|
|
let max_scroll = total_lines.saturating_sub(visible);
|
|||
|
|
let scroll = if app.scroll_offset == u16::MAX {
|
|||
|
|
max_scroll
|
|||
|
|
} else {
|
|||
|
|
app.scroll_offset.min(max_scroll)
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
let log_widget = Paragraph::new(Text::from(lines))
|
|||
|
|
.wrap(Wrap { trim: false })
|
|||
|
|
.scroll((scroll, 0));
|
|||
|
|
|
|||
|
|
frame.render_widget(log_widget, area);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn draw_input(frame: &mut ratatui::Frame, area: Rect, app: &App) {
|
|||
|
|
let input_block = Block::default()
|
|||
|
|
.borders(Borders::TOP)
|
|||
|
|
.border_style(Style::default().fg(SOL_FAINT));
|
|||
|
|
|
|||
|
|
let input_text = Line::from(vec![
|
|||
|
|
Span::styled("> ", Style::default().fg(SOL_DIM)),
|
|||
|
|
Span::raw(&app.input),
|
|||
|
|
]);
|
|||
|
|
|
|||
|
|
let input_widget = Paragraph::new(input_text)
|
|||
|
|
.block(input_block)
|
|||
|
|
.wrap(Wrap { trim: false });
|
|||
|
|
|
|||
|
|
frame.render_widget(input_widget, area);
|
|||
|
|
|
|||
|
|
// Position cursor
|
|||
|
|
let cursor_x = area.x + 2 + app.cursor_pos as u16;
|
|||
|
|
let cursor_y = area.y + 1;
|
|||
|
|
frame.set_cursor_position((cursor_x, cursor_y));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn draw_approval(frame: &mut ratatui::Frame, area: Rect, approval: &ApprovalPrompt) {
|
|||
|
|
let block = Block::default()
|
|||
|
|
.borders(Borders::TOP)
|
|||
|
|
.border_style(Style::default().fg(SOL_FAINT));
|
|||
|
|
|
|||
|
|
let mut lines = vec![
|
|||
|
|
Line::from(vec![
|
|||
|
|
Span::styled(" ⚠ ", Style::default().fg(SOL_YELLOW)),
|
|||
|
|
Span::styled(&approval.tool_name, Style::default().fg(SOL_YELLOW).add_modifier(Modifier::BOLD)),
|
|||
|
|
Span::styled(format!(" {}", approval.command), Style::default().fg(SOL_APPROVAL_CMD)),
|
|||
|
|
]),
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
for (i, opt) in approval.options.iter().enumerate() {
|
|||
|
|
let prefix = if i == approval.selected { " › " } else { " " };
|
|||
|
|
let style = if i == approval.selected {
|
|||
|
|
Style::default().fg(SOL_YELLOW)
|
|||
|
|
} else {
|
|||
|
|
Style::default().fg(SOL_DIM)
|
|||
|
|
};
|
|||
|
|
lines.push(Line::from(Span::styled(format!("{prefix}{opt}"), style)));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let widget = Paragraph::new(Text::from(lines))
|
|||
|
|
.block(block)
|
|||
|
|
.style(Style::default().bg(SOL_APPROVAL_BG));
|
|||
|
|
|
|||
|
|
frame.render_widget(widget, area);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn draw_status_bar(frame: &mut ratatui::Frame, area: Rect, app: &App) {
|
|||
|
|
let status = Line::from(vec![
|
|||
|
|
Span::styled(
|
|||
|
|
format!(" ~/…/{}", app.project_name),
|
|||
|
|
Style::default().fg(SOL_STATUS),
|
|||
|
|
),
|
|||
|
|
Span::styled(
|
|||
|
|
format!(" {} ±", app.branch),
|
|||
|
|
Style::default().fg(SOL_STATUS),
|
|||
|
|
),
|
|||
|
|
Span::styled(
|
|||
|
|
format!(" {}k in · {}k out", app.input_tokens / 1000, app.output_tokens / 1000),
|
|||
|
|
Style::default().fg(SOL_STATUS),
|
|||
|
|
),
|
|||
|
|
]);
|
|||
|
|
|
|||
|
|
frame.render_widget(Paragraph::new(status), area);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ── Terminal setup/teardown ────────────────────────────────────────────────
|
|||
|
|
|
|||
|
|
pub fn setup_terminal() -> io::Result<Terminal<CrosstermBackend<io::Stdout>>> {
|
|||
|
|
terminal::enable_raw_mode()?;
|
|||
|
|
let mut stdout = io::stdout();
|
|||
|
|
execute!(stdout, EnterAlternateScreen)?;
|
|||
|
|
let backend = CrosstermBackend::new(stdout);
|
|||
|
|
Terminal::new(backend)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
pub fn restore_terminal(terminal: &mut Terminal<CrosstermBackend<io::Stdout>>) -> io::Result<()> {
|
|||
|
|
terminal::disable_raw_mode()?;
|
|||
|
|
execute!(terminal.backend_mut(), LeaveAlternateScreen)?;
|
|||
|
|
terminal.show_cursor()?;
|
|||
|
|
Ok(())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[cfg(test)]
|
|||
|
|
mod tests {
|
|||
|
|
use super::*;
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_app_creation() {
|
|||
|
|
let app = App::new("sol", "mainline", "devstral-2");
|
|||
|
|
assert_eq!(app.project_name, "sol");
|
|||
|
|
assert!(!app.should_quit);
|
|||
|
|
assert!(app.log.is_empty());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_push_log_auto_scrolls() {
|
|||
|
|
let mut app = App::new("sol", "main", "devstral-2");
|
|||
|
|
app.scroll_offset = 0;
|
|||
|
|
app.push_log(LogEntry::Status("test".into()));
|
|||
|
|
assert_eq!(app.scroll_offset, u16::MAX); // auto-scroll to bottom
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_color_constants() {
|
|||
|
|
assert!(matches!(SOL_YELLOW, Color::Rgb(245, 197, 66)));
|
|||
|
|
assert!(matches!(SOL_AMBER, Color::Rgb(232, 168, 64)));
|
|||
|
|
assert!(matches!(SOL_BLUE, Color::Rgb(108, 166, 224)));
|
|||
|
|
assert!(matches!(SOL_RED, Color::Rgb(224, 88, 88)));
|
|||
|
|
// No green in the palette
|
|||
|
|
assert!(!matches!(SOL_YELLOW, Color::Rgb(_, 255, _)));
|
|||
|
|
assert!(!matches!(SOL_BLUE, Color::Rgb(_, 255, _)));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_log_entries_all_variants() {
|
|||
|
|
let mut app = App::new("test", "main", "devstral-2");
|
|||
|
|
app.push_log(LogEntry::UserInput("hello".into()));
|
|||
|
|
app.push_log(LogEntry::AssistantText("response".into()));
|
|||
|
|
app.push_log(LogEntry::ToolSuccess { name: "file_read".into(), detail: "src/main.rs".into() });
|
|||
|
|
app.push_log(LogEntry::ToolExecuting { name: "bash".into(), detail: "cargo test".into() });
|
|||
|
|
app.push_log(LogEntry::ToolFailed { name: "grep".into(), detail: "no matches".into() });
|
|||
|
|
app.push_log(LogEntry::ToolOutput { lines: vec!["line 1".into(), "line 2".into()], collapsed: true });
|
|||
|
|
app.push_log(LogEntry::Status("thinking".into()));
|
|||
|
|
app.push_log(LogEntry::Error("connection lost".into()));
|
|||
|
|
assert_eq!(app.log.len(), 8);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_tool_output_collapse_threshold() {
|
|||
|
|
// Collapsed output shows max 5 lines + "... +N lines"
|
|||
|
|
let lines: Vec<String> = (0..20).map(|i| format!("line {i}")).collect();
|
|||
|
|
let entry = LogEntry::ToolOutput { lines: lines.clone(), collapsed: true };
|
|||
|
|
if let LogEntry::ToolOutput { lines, collapsed } = &entry {
|
|||
|
|
assert!(lines.len() > 5);
|
|||
|
|
assert!(*collapsed);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_approval_prompt() {
|
|||
|
|
let approval = ApprovalPrompt {
|
|||
|
|
tool_name: "bash".into(),
|
|||
|
|
command: "cargo test".into(),
|
|||
|
|
options: vec![
|
|||
|
|
"Yes".into(),
|
|||
|
|
"Yes, always allow bash".into(),
|
|||
|
|
"No".into(),
|
|||
|
|
],
|
|||
|
|
selected: 0,
|
|||
|
|
};
|
|||
|
|
assert_eq!(approval.options.len(), 3);
|
|||
|
|
assert_eq!(approval.selected, 0);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_approval_navigation() {
|
|||
|
|
let mut approval = ApprovalPrompt {
|
|||
|
|
tool_name: "bash".into(),
|
|||
|
|
command: "rm -rf".into(),
|
|||
|
|
options: vec!["Yes".into(), "No".into()],
|
|||
|
|
selected: 0,
|
|||
|
|
};
|
|||
|
|
// Navigate down
|
|||
|
|
approval.selected = (approval.selected + 1).min(approval.options.len() - 1);
|
|||
|
|
assert_eq!(approval.selected, 1);
|
|||
|
|
// Navigate down again (clamped)
|
|||
|
|
approval.selected = (approval.selected + 1).min(approval.options.len() - 1);
|
|||
|
|
assert_eq!(approval.selected, 1);
|
|||
|
|
// Navigate up
|
|||
|
|
approval.selected = approval.selected.saturating_sub(1);
|
|||
|
|
assert_eq!(approval.selected, 0);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_thinking_state() {
|
|||
|
|
let mut app = App::new("sol", "main", "devstral-2");
|
|||
|
|
assert!(!app.is_thinking);
|
|||
|
|
app.is_thinking = true;
|
|||
|
|
assert!(app.is_thinking);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_input_cursor() {
|
|||
|
|
let mut app = App::new("sol", "main", "devstral-2");
|
|||
|
|
app.input = "hello world".into();
|
|||
|
|
app.cursor_pos = 5;
|
|||
|
|
assert_eq!(&app.input[..app.cursor_pos], "hello");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[test]
|
|||
|
|
fn test_token_tracking() {
|
|||
|
|
let mut app = App::new("sol", "main", "devstral-2");
|
|||
|
|
app.input_tokens = 1200;
|
|||
|
|
app.output_tokens = 340;
|
|||
|
|
assert_eq!(app.input_tokens / 1000, 1);
|
|||
|
|
assert_eq!(app.output_tokens / 1000, 0);
|
|||
|
|
}
|
|||
|
|
}
|