From 8e39778a4f0ed54b1b84b19f3db3d452e707dd6e Mon Sep 17 00:00:00 2001
From: kyoto7250 <50972773+kyoto7250@users.noreply.github.com>
Date: Thu, 27 Jun 2024 12:45:38 +0900
Subject: [PATCH] set a panic hook for crossterm

---
 src/main.rs | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 4dc62af..af261e7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -17,12 +17,14 @@ use crate::app::App;
 use crate::config::Config;
 use crate::event::{Event, Key};
 use anyhow::Result;
+use crossterm::execute;
 use crossterm::{
     terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
     ExecutableCommand,
 };
 use ratatui::{backend::CrosstermBackend, Terminal};
-use std::io;
+use std::io::{self, stdout};
+use std::panic::{set_hook, take_hook};
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
@@ -68,10 +70,26 @@ async fn main() -> anyhow::Result<()> {
 
 fn setup_terminal() -> Result<()> {
     enable_raw_mode()?;
+    init_panic_hook();
     io::stdout().execute(EnterAlternateScreen)?;
     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 restore_tui() -> io::Result<()> {
+    disable_raw_mode()?;
+    execute!(stdout(), LeaveAlternateScreen)?;
+    Ok(())
+}
+
 fn shutdown_terminal() {
     let leave_screen = io::stdout().execute(LeaveAlternateScreen).map(|_f| ());