-
Notifications
You must be signed in to change notification settings - Fork 51
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
54e85c8
commit 0b49d30
Showing
7 changed files
with
133 additions
and
45 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,43 +1,95 @@ | ||
// minimal WASM logger based on https://github.com/DeMille/wasm-glue | ||
extern "C" { | ||
fn trace_js(ptr: *const u8, len: usize); | ||
fn warn_js(ptr: *const u8, len: usize); | ||
fn log_js(ptr: *const u8, len: usize); | ||
use alloc::format; | ||
use core::panic::PanicInfo; | ||
|
||
#[no_mangle] | ||
unsafe extern "C" fn diplomat_init() { | ||
#[cfg(debug_assertions)] | ||
std::panic::set_hook(Box::new(panic_handler)); | ||
#[cfg(feature = "log")] | ||
log::set_logger(&ConsoleLogger) | ||
.map(|()| log::set_max_level(log::LevelFilter::Debug)) | ||
.is_ok() | ||
} | ||
|
||
/// Throw an exception. | ||
pub fn console_trace(msg: &str) { | ||
unsafe { | ||
trace_js(msg.as_ptr(), msg.len()); | ||
|
||
fn panic_handler(info: &PanicInfo) { | ||
let msg = match info.payload().downcast_ref::<&'static str>() { | ||
Some(&s) => s, | ||
None => match info.payload().downcast_ref::<String>() { | ||
Some(s) => s.as_str(), | ||
None => "Box<Any>", | ||
}, | ||
}; | ||
|
||
let msg = match info.location() { | ||
Some(l) => format!( | ||
"wasm panicked at {}:{}:{}:\n{msg}", | ||
l.file(), | ||
l.line(), | ||
l.column(), | ||
), | ||
None => format!("wasm panicked at <unknown location>:\n{msg}"), | ||
}; | ||
|
||
extern "C" { | ||
fn throw_error_js(ptr: *const u8, len: usize); | ||
} | ||
} | ||
|
||
/// Write a message to `console.warn`. | ||
pub fn console_warn(msg: &str) { | ||
unsafe { warn_js(msg.as_ptr(), msg.len()) } | ||
unsafe { throw_error_js(msg.as_ptr(), msg.len()) } | ||
} | ||
|
||
/// Write a message to `console.log`. | ||
pub fn console_log(msg: &str) { | ||
unsafe { log_js(msg.as_ptr(), msg.len()) } | ||
} | ||
#[cfg(feature = "log")] | ||
struct ConsoleLogger; | ||
|
||
#[no_mangle] | ||
pub unsafe extern "C" fn diplomat_init() { | ||
#[cfg(debug_assertions)] | ||
// Sets a custom panic hook using `trace_js`, which by default crates a JS error | ||
std::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.as_str(), | ||
None => "Box<Any>", | ||
}, | ||
#[cfg(feature = "log")] | ||
impl log::Log for ConsoleLogger { | ||
#[inline] | ||
fn enabled(&self, metadata: &log::Metadata) -> bool { | ||
metadata.level() <= log::max_level() | ||
} | ||
|
||
fn log(&self, record: &log::Record) { | ||
if !self.enabled(record.metadata()) { | ||
return; | ||
} | ||
|
||
let out = match record.level() { | ||
log::Level::Error => { | ||
extern "C" { | ||
fn console_error_js(ptr: *const u8, len: usize); | ||
} | ||
console_error_js | ||
} | ||
log::Level::Warn => { | ||
extern "C" { | ||
fn console_warn_js(ptr: *const u8, len: usize); | ||
} | ||
console_warn_js | ||
} | ||
log::Level::Info => { | ||
extern "C" { | ||
fn console_info_js(ptr: *const u8, len: usize); | ||
} | ||
console_info_js | ||
} | ||
log::Level::Debug => { | ||
extern "C" { | ||
fn console_log_js(ptr: *const u8, len: usize); | ||
} | ||
console_log_js | ||
} | ||
log::Level::Trace => { | ||
extern "C" { | ||
fn console_debug_js(ptr: *const u8, len: usize); | ||
} | ||
console_debug_js | ||
} | ||
}; | ||
console_trace(&format!("Panicked at '{msg}', {file}:{line}:{col}")); | ||
})); | ||
|
||
let msg = alloc::format!("{}", record.args()); | ||
|
||
unsafe { out(msg.as_ptr(), msg.len()) }; | ||
} | ||
|
||
fn flush(&self) {} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters