add --cfg=wasm_bindgen_unwrap_throw_debug flag to defer to std's un…

…wrap in unwrap_throw
rustwasm · Apr 2, 2024 · 03f7287 · 03f7287
1 parent 78463a2
commit 03f7287
Showing 1 changed file with 20 additions and 6 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -1327,7 +1327,7 @@ pub fn anyref_heap_live_count() -> u32 {
 pub trait UnwrapThrowExt<T>: Sized {
     /// Unwrap this `Option` or `Result`, but instead of panicking on failure,
     /// throw an exception to JavaScript.
-    #[cfg_attr(debug_assertions, track_caller)]
+    #[cfg_attr(any(debug_assertions, wasm_bindgen_unwrap_throw_debug), track_caller)]
     fn unwrap_throw(self) -> T {
         if cfg!(all(debug_assertions, feature = "std")) {
             let loc = core::panic::Location::caller();
@@ -1346,16 +1346,26 @@ pub trait UnwrapThrowExt<T>: Sized {
     /// Unwrap this container's `T` value, or throw an error to JS with the
     /// given message if the `T` value is unavailable (e.g. an `Option<T>` is
     /// `None`).
-    #[cfg_attr(debug_assertions, track_caller)]
+    #[cfg_attr(any(debug_assertions, wasm_bindgen_unwrap_throw_debug), track_caller)]
     fn expect_throw(self, message: &str) -> T;
 }
 
 impl<T> UnwrapThrowExt<T> for Option<T> {
-    #[cfg_attr(debug_assertions, track_caller)]
+    #[cfg(wasm_bindgen_unwrap_throw_debug)]
+    #[track_caller]
+    fn unwrap_throw(self) -> T {
+        self.unwrap()
+    }
+
+    #[cfg_attr(any(debug_assertions, wasm_bindgen_unwrap_throw_debug), track_caller)]
     fn expect_throw(self, message: &str) -> T {
         if cfg!(all(
             target_arch = "wasm32",
-            not(any(target_os = "emscripten", target_os = "wasi"))
+            not(any(
+                wasm_bindgen_unwrap_throw_debug,
+                target_os = "emscripten",
+                target_os = "wasi"
+            ))
         )) {
             match self {
                 Some(val) => val,
@@ -1371,11 +1381,15 @@ impl<T, E> UnwrapThrowExt<T> for Result<T, E>
 where
     E: core::fmt::Debug,
 {
-    #[cfg_attr(debug_assertions, track_caller)]
+    #[cfg_attr(any(debug_assertions, wasm_bindgen_unwrap_throw_debug), track_caller)]
     fn expect_throw(self, message: &str) -> T {
         if cfg!(all(
             target_arch = "wasm32",
-            not(any(target_os = "emscripten", target_os = "wasi"))
+            not(any(
+                wasm_bindgen_unwrap_throw_debug,
+                target_os = "emscripten",
+                target_os = "wasi"
+            ))
         )) {
             match self {
                 Ok(val) => val,