use envision::prelude::*;
struct ButtonApp;
#[derive(Clone)]
struct State {
save: ButtonState,
cancel: ButtonState,
submit: ButtonState,
focus_index: usize,
last_pressed: Option<String>,
}
#[derive(Clone, Debug)]
enum Msg {
Save(ButtonMessage),
Cancel(ButtonMessage),
Submit(ButtonMessage),
FocusNext,
FocusPrev,
Quit,
}
const BUTTON_COUNT: usize = 3;
impl State {
fn set_focus(&mut self, index: usize) {
self.focus_index = index;
}
}
impl App for ButtonApp {
type State = State;
type Message = Msg;
fn init() -> (State, Command<Msg>) {
let save = ButtonState::new("Save");
let cancel = ButtonState::new("Cancel");
let submit = ButtonState::new("Submit");
let state = State {
save,
cancel,
submit,
focus_index: 0,
last_pressed: None,
};
(state, Command::none())
}
fn update(state: &mut State, msg: Msg) -> Command<Msg> {
match msg {
Msg::Save(m) => {
if let Some(ButtonOutput::Pressed) = Button::update(&mut state.save, m) {
state.last_pressed = Some("Save".to_string());
}
}
Msg::Cancel(m) => {
if let Some(ButtonOutput::Pressed) = Button::update(&mut state.cancel, m) {
state.last_pressed = Some("Cancel".to_string());
}
}
Msg::Submit(m) => {
if let Some(ButtonOutput::Pressed) = Button::update(&mut state.submit, m) {
state.last_pressed = Some("Submit".to_string());
}
}
Msg::FocusNext => {
let next = (state.focus_index + 1) % BUTTON_COUNT;
state.set_focus(next);
}
Msg::FocusPrev => {
let prev = (state.focus_index + BUTTON_COUNT - 1) % BUTTON_COUNT;
state.set_focus(prev);
}
Msg::Quit => return Command::quit(),
}
Command::none()
}
fn view(state: &State, frame: &mut Frame) {
let theme = Theme::default();
let area = frame.area();
let chunks = Layout::vertical([
Constraint::Length(3),
Constraint::Length(3),
Constraint::Length(3),
Constraint::Min(0),
Constraint::Length(1),
])
.split(area);
Button::view(
&state.save,
&mut RenderContext::new(frame, chunks[0], &theme),
);
Button::view(
&state.cancel,
&mut RenderContext::new(frame, chunks[1], &theme),
);
Button::view(
&state.submit,
&mut RenderContext::new(frame, chunks[2], &theme),
);
let info = match &state.last_pressed {
Some(name) => format!(" Last pressed: {}", name),
None => " No button pressed yet".to_string(),
};
let info_widget = ratatui::widgets::Paragraph::new(info).block(
ratatui::widgets::Block::default()
.borders(ratatui::widgets::Borders::ALL)
.title("Status"),
);
frame.render_widget(info_widget, chunks[3]);
let status = format!(
" Focus: {} | Tab/Shift+Tab: navigate, Enter/Space: press, q: quit",
state.focus_index
);
frame.render_widget(
ratatui::widgets::Paragraph::new(status).style(Style::default().fg(Color::DarkGray)),
chunks[4],
);
}
fn handle_event_with_state(state: &State, event: &Event) -> Option<Msg> {
if let Some(key) = event.as_key() {
match key.code {
Key::Char('q') | Key::Esc => return Some(Msg::Quit),
Key::Tab if key.modifiers.shift() => return Some(Msg::FocusPrev),
Key::Tab => return Some(Msg::FocusNext),
_ => {}
}
}
match state.focus_index {
0 => Button::handle_event(&state.save, event, &EventContext::new().focused(true))
.map(Msg::Save),
1 => Button::handle_event(&state.cancel, event, &EventContext::new().focused(true))
.map(Msg::Cancel),
_ => Button::handle_event(&state.submit, event, &EventContext::new().focused(true))
.map(Msg::Submit),
}
}
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut vt = Runtime::<ButtonApp, _>::virtual_builder(50, 16).build()?;
println!("=== Button Example ===\n");
vt.tick()?;
println!("Initial state (Save focused, Submit disabled):");
println!("{}\n", vt.display());
vt.dispatch(Msg::Save(ButtonMessage::Press));
vt.tick()?;
println!("After pressing Save:");
println!("{}\n", vt.display());
vt.dispatch(Msg::FocusNext);
vt.dispatch(Msg::Cancel(ButtonMessage::Press));
vt.tick()?;
println!("After pressing Cancel:");
println!("{}\n", vt.display());
vt.dispatch(Msg::FocusNext);
vt.dispatch(Msg::Submit(ButtonMessage::Press));
vt.tick()?;
println!("After attempting to press disabled Submit:");
println!("{}\n", vt.display());
Ok(())
}