Make panic errors available and display them nicely for the demo

coolreader18 · coolreader18 · commit 878c131af42b · 2019-02-23T14:52:02.000-06:00
diff --git a/wasm/demo/src/main.js b/wasm/demo/src/main.js
@@ -36,14 +36,18 @@ function runCodeFromTextarea() {
         if (result !== null) {
             consoleElement.value += `\n${result}\n`;
         }
-    } catch (e) {
-        errorElement.textContent = e;
-        console.error(e);
+    } catch (err) {
+        if (err instanceof WebAssembly.RuntimeError) {
+            err = window.__RUSTPYTHON_ERROR || err;
+        }
+        errorElement.textContent = err;
+        console.error(err);
     }
 }
 
 document
     .getElementById('run-btn')
     .addEventListener('click', runCodeFromTextarea);
 
-runCodeFromTextarea(); // Run once for demo
+// Run once for demo
+runCodeFromTextarea();
diff --git a/wasm/lib/src/lib.rs b/wasm/lib/src/lib.rs
@@ -9,17 +9,34 @@ extern crate wasm_bindgen;
 extern crate web_sys;
 
 use js_sys::{Object, Reflect, TypeError};
+use std::panic;
 use wasm_bindgen::prelude::*;
 
 pub use crate::vm_class::*;
 
 const PY_EVAL_VM_ID: &str = "__py_eval_vm";
 
-extern crate console_error_panic_hook;
+fn panic_hook(info: &panic::PanicInfo) {
+    // If something errors, just ignore it; we don't want to panic in the panic hook
+    use js_sys::WebAssembly::RuntimeError;
+    let window = match web_sys::window() {
+        Some(win) => win,
+        None => return,
+    };
+    let msg = &info.to_string();
+    let _ = Reflect::set(&window, &"__RUSTPYTHON_ERROR_MSG".into(), &msg.into());
+    let error = RuntimeError::new(&msg);
+    let _ = Reflect::set(&window, &"__RUSTPYTHON_ERROR".into(), &error);
+    let stack = match Reflect::get(&error, &"stack".into()) {
+        Ok(stack) => stack,
+        Err(_) => return,
+    };
+    let _ = Reflect::set(&window, &"__RUSTPYTHON_ERROR_STACK".into(), &stack.into());
+}
 
 #[wasm_bindgen(start)]
 pub fn setup_console_error() {
-    std::panic::set_hook(Box::new(console_error_panic_hook::hook));
+    std::panic::set_hook(Box::new(panic_hook));
 }
 
 // Hack to comment out wasm-bindgen's generated typescript definitons
