robin/logsmash
1
0
Fork 0
mirror of https://codeberg.org/icewind/logsmash.git synced 2026-06-03 18:14:11 +02:00
Code Issues Projects Releases 3 Packages Wiki Activity Actions

restore terminal state on panic

Browse source
This commit is contained in:
Robin Appelman 2024-09-16 18:09:23 +02:00
commit 68282f67af
1 changed files with 23 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

27
src/ui/mod.rs
View file

@ -22,6 +22,7 @@ use ratatui::prelude::*;
use ratatui::Terminal;
use std::io;
use std::io::stdout;
use std::panic::{set_hook, take_hook};
use std::time::Duration;
mod error_list;
@ -36,10 +37,8 @@ pub mod style;
mod table;
pub fn run_ui(app: App) -> Result<(), UiError> {
enable_raw_mode()?;
stdout().execute(EnterAlternateScreen)?;
stdout().execute(EnableMouseCapture)?;
let mut terminal = Terminal::new(CrosstermBackend::new(stdout()))?;
init_panic_hook();
let mut terminal = init_tui()?;
terminal.clear().ok();
let mut ui_state = UiState::new(&app);
@ -55,6 +54,26 @@ pub fn run_ui(app: App) -> Result<(), UiError> {
}
}
Ok(())
}
pub fn init_panic_hook() {
let original_hook = take_hook();
set_hook(Box::new(move |panic_info| {
// intentionally ignore errors here since we're already in a panic
let _ = restore_tui();
original_hook(panic_info);
}));
}
pub fn init_tui() -> io::Result<Terminal<impl Backend>> {
enable_raw_mode()?;
stdout().execute(EnterAlternateScreen)?;
stdout().execute(EnableMouseCapture)?;
Terminal::new(CrosstermBackend::new(stdout()))
}
pub fn restore_tui() -> io::Result<()> {
disable_raw_mode()?;
stdout().execute(DisableMouseCapture)?;
stdout().execute(LeaveAlternateScreen)?;