1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! A wrapper for panics using Bevy's plugin system.
//!
//! On supported platforms (windows, macos, linux) will produce a popup using the `msgbox` crate in addition to writing via `log::error!`, or if `bevy::log::LogPlugin` is not enabled, `stderr`.

use bevy::prelude::*;

pub struct PanicHandler;

impl Plugin for PanicHandler {
    fn build(&self, app: &mut App) {
        let use_log = app.is_plugin_added::<bevy::log::LogPlugin>();
        std::panic::set_hook(Box::new(move |info| {
            let info = format!(
                "Unhandled panic @ {}:\n{}",
                info.location()
                    .map_or("Unknown Location".to_owned(), ToString::to_string),
                info.payload().downcast_ref::<String>().unwrap_or(
                    &((*info.payload().downcast_ref::<&str>().unwrap_or(&"No Info")).to_string())
                )
            );

            if use_log {
                // Known limitations: Logging in tests prints to stdout immediately.
                bevy::log::error!("{info}");
            } else {
                eprintln!("{info}");
            }

            // Dont interrupt test execution with a popup, and dont try on unsupported platforms.
            #[cfg(all(not(test), any(target_os = "windows", target_os = "macos", target_os = "linux")))]
            { _ = msgbox::create("Fatal Error", &info, msgbox::IconType::Error); }
        }));
    }
}