PXP-635: [CLI] Terminate the TUI when there is a panic on TUI

cli/src/commands/tui/mod.rs

@@ -1,4 +1,9 @@
-use std::{io::stdout, sync::Arc, time::Duration};
+use std::{
+    io::stdout,
+    panic::{self, PanicInfo},
+    sync::Arc,
+    time::Duration,
+};
 
 use crate::{
     config::{ApiChannel, Config},
@@ -7,6 +12,7 @@ use crate::{
 use crossterm::{
     event::{DisableMouseCapture, EnableMouseCapture},
     execute,
+    style::Print,
     terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
 };
 use ratatui::{prelude::CrosstermBackend, widgets::ListState, Terminal};
@@ -91,6 +97,9 @@ pub async fn handle_tui(channel: ApiChannel) -> Result<bool, WKCliError> {
 
     let arc_channel = Arc::new(channel);
 
+    // Set panic hook
+    panic::set_hook(Box::new(panic_hook));
+
     tokio::spawn(async move {
         while let Some(network_event) = receiver.recv().await {
             let app = Arc::clone(&app);
@@ -180,3 +189,28 @@ pub async fn start_ui(app: &Arc<Mutex<App>>) -> std::io::Result<bool> {
 
     Ok(true)
 }
+
+pub fn panic_hook(panic_info: &PanicInfo<'_>) {
+    let mut stdout = stdout();
+
+    let msg = match panic_info.payload().downcast_ref::<&'static str>() {
+        Some(s) => *s,
+        None => match panic_info.payload().downcast_ref::<String>() {
+            Some(s) => &s[..],
+            None => "Box<Any>",
+        },
+    };
+
+    let _ = disable_raw_mode();
+    let _ = execute!(stdout, DisableMouseCapture, LeaveAlternateScreen);
+
+    // Print stack trace. Must be done after!
+    if let Some(panic_info) = panic_info.location() {
+        let _ = execute!(
+            stdout,
+            Print(format!(
+                "thread '<unnamed>' panicked at '{msg}', {panic_info}",
+            )),
+        );
+    }
+}

cli/src/output/tokenizer.rs

@@ -188,7 +188,7 @@ impl OutputTokenizer {
                 }
             } else {
                 // this is more readable
-                #[allow(clippy::collapsible-else-if)]
+                #[allow(clippy::collapsible_else_if)]
                 if word.contains('(') {
                     let open_index = word.find('(').unwrap();
                     tokens.push(Token::Parentheses("(".to_string()));

sdk/src/services/vault/client.rs

@@ -57,8 +57,8 @@ impl Default for VaultClient {
 }
 
 impl VaultClient {
-    pub const FETCH_SECRETS: &str = "/v1/secret/data";
-    pub const UPDATE_SECRET: &str = "/v1/secret/data";
+    pub const FETCH_SECRETS: &'static str = "/v1/secret/data";
+    pub const UPDATE_SECRET: &'static str = "/v1/secret/data";
 
     pub fn new() -> Self {
         let client = reqwest::Client::new();