feat(wasm): Run Lumina in a Shared Worker

cli/static/run_node.js

@@ -1,6 +1,6 @@
 Error.stackTraceLimit = 99; // rust stack traces can get pretty big, increase the default
 
-import init, { Node, NodeConfig } from "/wasm/lumina_node_wasm.js";
+import init, { NodeConfig, NodeDriver } from "/wasm/lumina_node_wasm.js";
 
 async function fetch_config() {
   const response = await fetch('/cfg.json');
@@ -20,7 +20,7 @@ async function fetch_config() {
 }
 
 async function show_stats(node) {
-  if (!node) {
+  if (!node || !await node.is_running()) {
     return;
   }
   const info = await node.syncer_info();
@@ -36,11 +36,12 @@ async function show_stats(node) {
 
   document.getElementById("peers").replaceChildren(peers_ul);
 
-  const network_head = node.get_network_head_header();
+  const network_head = await node.get_network_head_header();
   if (network_head == null) {
     return
   }
 
+  console.log(network_head);
   const square_rows = network_head.dah.row_roots.length;
   const square_cols = network_head.dah.column_roots.length;
 
@@ -93,24 +94,28 @@ function bind_config(data) {
   });
 }
 
-async function start_node(config) {
-  window.node = await new Node(config);
-
-  document.getElementById("peer-id").innerText = await window.node.local_peer_id();
-  document.querySelectorAll(".status").forEach(elem => elem.style.visibility = "visible");
-}
-
 async function main(document, window) {
   await init();
 
+  window.driver = await new NodeDriver();
+
   bind_config(await fetch_config());
 
+  if (await window.driver.is_running() === true) {
+    document.querySelectorAll('.config').forEach(elem => elem.disabled = true);
+    document.getElementById("peer-id").innerText = await window.driver.local_peer_id();
+    document.querySelectorAll(".status").forEach(elem => elem.style.visibility = "visible");
+  }
+
   document.getElementById("start").addEventListener("click", async () => {
     document.querySelectorAll('.config').forEach(elem => elem.disabled = true);
-    start_node(window.config);
+
+    await window.driver.start(window.config);
+    document.getElementById("peer-id").innerText = await window.driver.local_peer_id();
+    document.querySelectorAll(".status").forEach(elem => elem.style.visibility = "visible");
   });
 
-  setInterval(async () => await show_stats(window.node), 1000)
+  setInterval(async () => await show_stats(window.driver), 1000)
 }
 
 await main(document, window);

cli/static/worker.js

@@ -0,0 +1,17 @@
+Error.stackTraceLimit = 99; // rust stack traces can get pretty big, increase the default
+
+import init, { run_worker } from "/wasm/lumina_node_wasm.js";
+
+// unfortunately, `init()` takes long enough for the first connection (from the tab which 
+// starts Shared Worker) to be missed. As a workaround, we queue connections in js and then
+// pass them to Rust, when it's ready. Rust code replaces `onconnect` handler, so this is
+// only necessary on startup.
+let queued = [];
+onconnect = (event) => {
+  console.log("Queued connection", event);
+  queued.push(event.ports[0]);
+}
+
+await init();
+await run_worker(queued);
+

node-wasm/Cargo.toml

@@ -24,18 +24,27 @@ crate-type = ["cdylib", "rlib"]
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]
 celestia-types = { workspace = true }
-libp2p = { workspace = true }
+libp2p = { workspace = true, features = ["serde"] }
 lumina-node = { workspace = true }
+blockstore = { workspace = true }
+celestia-tendermint = { workspace = true }
 
 anyhow = "1.0.71"
 console_error_panic_hook = "0.1.7"
+futures = "0.3"
+gloo-timers = "0.3"
+instant = "0.1"
 js-sys = "0.3.64"
 serde = { version = "1.0.164", features = ["derive"] }
-serde_repr = "0.1"
 serde-wasm-bindgen = "0.6.0"
+serde_repr = "0.1"
+thiserror = "1.0"
 time = { version = "0.3", features = ["wasm-bindgen"] }
+tokio =  { version = "*", features = ["sync"]}
 tracing = "0.1.37"
 tracing-subscriber = { version = "0.3.18", features = ["time"] }
 tracing-web = "0.1.2"
 wasm-bindgen = "0.2.88"
 wasm-bindgen-futures = "0.4.37"
+web-sys = { version = "0.3.69", features = ["BroadcastChannel", "MessageEvent", "Worker", "WorkerOptions", "WorkerType", "SharedWorker", "MessagePort", "SharedWorkerGlobalScope", "Blob", "BlobPropertyBag", "Url"]}
+enum-as-inner = "0.6"

node-wasm/src/lib.rs

@@ -5,7 +5,8 @@ use wasm_bindgen::JsError;
 
 pub mod node;
 pub mod utils;
+mod worker;
 mod wrapper;
 
 /// Alias for a `Result` with the error type [`JsError`].
-pub type Result<T> = std::result::Result<T, JsError>;
+pub type Result<T, E = JsError> = std::result::Result<T, E>;