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
35
36
37
38
39
40
41
42
43

use crate::ic;
use std::panic;

#[cfg(target_family = "wasm")]
static mut DONE: bool = false;

#[cfg(target_family = "wasm")]
pub fn setup_hooks() {
    unsafe {
        if DONE {
            return;
        }
        DONE = true;
    }

    set_panic_hook();
}

#[cfg(not(target_family = "wasm"))]
pub fn setup_hooks() {
    set_panic_hook();
}

/// Sets a custom panic hook, uses debug.trace
fn set_panic_hook() {
    panic::set_hook(Box::new(|info| {
        let file = info.location().unwrap().file();
        let line = info.location().unwrap().line();
        let col = info.location().unwrap().column();

        let msg = match info.payload().downcast_ref::<&'static str>() {
            Some(s) => *s,
            None => match info.payload().downcast_ref::<String>() {
                Some(s) => &s[..],
                None => "Box<Any>",
            },
        };

        let err_info = format!("Panicked at '{}', {}:{}:{}", msg, file, line, col);
        ic::print(&err_info);
        ic::trap(&err_info);
    }));
}