cli: improve panic messages

RGB-WG · Feb 3, 2025 · dd2b5ab · dd2b5ab
1 parent 889da7c
commit dd2b5ab
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/cli/src/main.rs b/cli/src/main.rs
@@ -32,8 +32,22 @@ pub mod args;
 pub mod cmd;
 mod exec;
 
+use std::fmt::Display;
+use std::panic::set_hook;
+
 use clap::Parser;
 
 use crate::args::Args;
 
-fn main() -> anyhow::Result<()> { Args::parse().exec() }
+fn main() -> anyhow::Result<()> {
+    set_hook(Box::new(|info| {
+        if let Some(error) = info.payload().downcast_ref::<&dyn Display>() {
+            eprintln!("Error: {error}");
+            eprintln!("Caused by:");
+        } else {
+            eprintln!("Error: {:?}", info);
+        }
+        eprintln!("{:?}", info);
+    }));
+    Args::parse().exec()
+}