From 409f6fe1df95d67120aa6d47e3e0b9ef747ec6fd Mon Sep 17 00:00:00 2001
From: Fenrir <fenrirwolf@gmail.com>
Date: Fri, 23 Feb 2024 18:25:43 -0700
Subject: [PATCH] Add error applet panic hook

---
 ctru-rs/src/applets/error.rs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/ctru-rs/src/applets/error.rs b/ctru-rs/src/applets/error.rs
index c60c4a8b..b6edd512 100644
--- a/ctru-rs/src/applets/error.rs
+++ b/ctru-rs/src/applets/error.rs
@@ -79,6 +79,34 @@ impl PopUp {
     }
 }
 
+/// Sets a custom panic hook that uses the error applet to display panic messages. You can also choose to have
+/// messages printed over stderr along with the pop-up display.
+///
+/// SAFETY: The error applet requires that both the [`Apt`] and [`Gfx`] services are active whenever it launches.
+/// By calling this function, you promise that you will keep those services alive until either the program ends or
+/// you unregister this hook with [`std::panic::take_hook`](https://doc.rust-lang.org/std/panic/fn.take_hook.html).
+pub unsafe fn set_panic_hook(use_stderr: bool) {
+    std::panic::set_hook(Box::new(|panic_info| {
+        let mut popup = PopUp::new(Kind::Top);
+
+        let thread = std::thread::current();
+
+        let name = thread.name().unwrap_or("<unnamed>");
+
+        let payload = format!("thread '{name}' {panic_info}");
+
+        popup.set_text(&payload);
+
+        if use_stderr {
+            eprintln!("{payload}");
+        }
+
+        unsafe {
+            ctru_sys::errorDisp(popup.state.as_mut());
+        }
+    }))
+}
+
 impl std::fmt::Display for Error {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {