use envision::prelude::*;
use ratatui::widgets::Paragraph;
struct DiffViewerApp;
#[derive(Clone)]
struct State {
viewer: DiffViewerState,
}
#[derive(Clone, Debug)]
enum Msg {
Viewer(DiffViewerMessage),
Quit,
}
impl App for DiffViewerApp {
type State = State;
type Message = Msg;
fn init() -> (State, Command<Msg>) {
let old_code = "\
fn greet(name: &str) {
println!(\"Hello, {}!\", name);
}
fn main() {
greet(\"world\");
}";
let new_code = "\
fn greet(name: &str, greeting: &str) {
println!(\"{}, {}!\", greeting, name);
log::info!(\"Greeted {}\", name);
}
fn farewell(name: &str) {
println!(\"Goodbye, {}!\", name);
}
fn main() {
greet(\"world\", \"Hello\");
farewell(\"world\");
}";
let mut viewer = DiffViewerState::from_texts(old_code, new_code)
.with_title("Code Changes")
.with_old_label("greet.rs (old)")
.with_new_label("greet.rs (new)");
viewer.set_focused(true);
(State { viewer }, Command::none())
}
fn update(state: &mut State, msg: Msg) -> Command<Msg> {
match msg {
Msg::Viewer(m) => {
state.viewer.update(m);
}
Msg::Quit => return Command::quit(),
}
Command::none()
}
fn view(state: &State, frame: &mut Frame) {
let area = frame.area();
let chunks = Layout::vertical([Constraint::Min(0), Constraint::Length(1)]).split(area);
let theme = Theme::default();
DiffViewer::view(
&state.viewer,
frame,
chunks[0],
&theme,
&ViewContext::default(),
);
let mode_str = match state.viewer.mode() {
DiffMode::Unified => "Unified",
DiffMode::SideBySide => "Side-by-Side",
};
let status = Paragraph::new(format!(
" Hunk {}/{} | {} | +{} -{} | j/k scroll | n/p hunk | m mode | q quit",
state.viewer.current_hunk() + 1,
state.viewer.hunk_count().max(1),
mode_str,
state.viewer.added_count(),
state.viewer.removed_count(),
))
.style(Style::default().fg(Color::DarkGray));
frame.render_widget(status, chunks[1]);
}
fn handle_event_with_state(state: &State, event: &Event) -> Option<Msg> {
if let Some(key) = event.as_key() {
if matches!(key.code, KeyCode::Char('q') | KeyCode::Esc) {
return Some(Msg::Quit);
}
}
state.viewer.handle_event(event).map(Msg::Viewer)
}
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut vt = Runtime::<DiffViewerApp, _>::virtual_terminal(80, 24)?;
println!("=== DiffViewer Example ===\n");
vt.tick()?;
println!("Unified mode:");
println!("{}\n", vt.display());
vt.dispatch(Msg::Viewer(DiffViewerMessage::NextHunk));
vt.tick()?;
println!("After next hunk:");
println!("{}\n", vt.display());
vt.dispatch(Msg::Viewer(DiffViewerMessage::ToggleMode));
vt.dispatch(Msg::Viewer(DiffViewerMessage::Home));
vt.tick()?;
println!("Side-by-side mode:");
println!("{}\n", vt.display());
println!(
"Total changes: +{} -{}",
vt.state().viewer.added_count(),
vt.state().viewer.removed_count()
);
Ok(())
}