feat: Add remaining node types for wasm

Author: fl0rek

Cargo.lock

@@ -27,6 +27,7 @@ prost-types = "0.13.3"
 #libp2p = { path = "../../rust-libp2p/libp2p" }
 #libp2p-core = { path = "../../rust-libp2p/core" }
 #libp2p-swarm = { path = "../../rust-libp2p/swarm" }
+#wasm-bindgen = { path = "../wasm-bindgen" }
 
 # Uncomment this if you need debug symbols in release.
 # Also check node-wasm's `Cargo.toml`.

Cargo.toml

@@ -24,9 +24,9 @@ crate-type = ["cdylib", "rlib"]
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]
 blockstore.workspace = true
-celestia-types.workspace = true
+celestia-types = { workspace = true, features = ["wasm-bindgen"] }
 libp2p = { workspace = true, features = ["serde"] }
-lumina-node.workspace = true
+lumina-node = { workspace = true, features = ["wasm-bindgen"] }
 tendermint.workspace = true
 
 anyhow = "1.0.86"
@@ -46,7 +46,7 @@ tokio = { version = "1.38.0", features = ["sync"] }
 tracing = "0.1.40"
 tracing-subscriber = { version = "0.3.18", features = ["time"] }
 tracing-web = "0.1.3"
-wasm-bindgen = "0.2.93"
+wasm-bindgen = "0.2.99"
 wasm-bindgen-futures = "0.4.43"
 web-sys = { version = "0.3.70", features = [
   "BroadcastChannel",

node-wasm/Cargo.toml

@@ -4,15 +4,15 @@ use std::time::Duration;
 
 use js_sys::Array;
 use serde::{Deserialize, Serialize};
-use serde_wasm_bindgen::to_value;
 use tracing::{debug, error};
 use wasm_bindgen::prelude::*;
 use web_sys::BroadcastChannel;
 
+use celestia_types::ExtendedHeader;
 use lumina_node::blockstore::IndexedDbBlockstore;
 use lumina_node::network;
 use lumina_node::node::NodeBuilder;
-use lumina_node::store::IndexedDbStore;
+use lumina_node::store::{IndexedDbStore, SamplingMetadata};
 
 use crate::commands::{CheckableResponseExt, NodeCommand, SingleHeaderQuery};
 use crate::error::{Context, Result};
@@ -200,38 +200,32 @@ impl NodeClient {
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = requestHeadHeader)]
-    pub async fn request_head_header(&self) -> Result<JsValue> {
+    pub async fn request_head_header(&self) -> Result<ExtendedHeader> {
         let command = NodeCommand::RequestHeader(SingleHeaderQuery::Head);
         let response = self.worker.exec(command).await?;
-        let header = response.into_header().check_variant()?;
-
-        header.into()
+        response.into_header().check_variant()?
     }
 
     /// Request a header for the block with a given hash from the network.
     ///
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = requestHeaderByHash)]
-    pub async fn request_header_by_hash(&self, hash: &str) -> Result<JsValue> {
+    pub async fn request_header_by_hash(&self, hash: &str) -> Result<ExtendedHeader> {
         let command = NodeCommand::RequestHeader(SingleHeaderQuery::ByHash(hash.parse()?));
         let response = self.worker.exec(command).await?;
-        let header = response.into_header().check_variant()?;
-
-        header.into()
+        response.into_header().check_variant()?
     }
 
     /// Request a header for the block with a given height from the network.
     ///
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = requestHeaderByHeight)]
-    pub async fn request_header_by_height(&self, height: u64) -> Result<JsValue> {
+    pub async fn request_header_by_height(&self, height: u64) -> Result<ExtendedHeader> {
         let command = NodeCommand::RequestHeader(SingleHeaderQuery::ByHeight(height));
         let response = self.worker.exec(command).await?;
-        let header = response.into_header().check_variant()?;
-
-        header.into()
+        response.into_header().check_variant()?
     }
 
     /// Request headers in range (from, from + amount] from the network.
@@ -243,17 +237,12 @@ impl NodeClient {
     #[wasm_bindgen(js_name = requestVerifiedHeaders)]
     pub async fn request_verified_headers(
         &self,
-        from_header: JsValue,
+        from: ExtendedHeader,
         amount: u64,
-    ) -> Result<Array> {
-        let command = NodeCommand::GetVerifiedHeaders {
-            from: from_header,
-            amount,
-        };
+    ) -> Result<Vec<ExtendedHeader>> {
+        let command = NodeCommand::GetVerifiedHeaders { from, amount };
         let response = self.worker.exec(command).await?;
-        let headers = response.into_headers().check_variant()?;
-
-        headers.into()
+        response.into_headers().check_variant()?
     }
 
     /// Get current header syncing info.
@@ -271,51 +260,43 @@ impl NodeClient {
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = getNetworkHeadHeader)]
-    pub async fn get_network_head_header(&self) -> Result<JsValue> {
+    pub async fn get_network_head_header(&self) -> Result<Option<ExtendedHeader>> {
         let command = NodeCommand::LastSeenNetworkHead;
         let response = self.worker.exec(command).await?;
-        let header = response.into_last_seen_network_head().check_variant()?;
-
-        header.into()
+        response.into_last_seen_network_head().check_variant()?
     }
 
     /// Get the latest locally synced header.
     ///
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = getLocalHeadHeader)]
-    pub async fn get_local_head_header(&self) -> Result<JsValue> {
+    pub async fn get_local_head_header(&self) -> Result<ExtendedHeader> {
         let command = NodeCommand::GetHeader(SingleHeaderQuery::Head);
         let response = self.worker.exec(command).await?;
-        let header = response.into_header().check_variant()?;
-
-        header.into()
+        response.into_header().check_variant()?
     }
 
     /// Get a synced header for the block with a given hash.
     ///
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = getHeaderByHash)]
-    pub async fn get_header_by_hash(&self, hash: &str) -> Result<JsValue> {
+    pub async fn get_header_by_hash(&self, hash: &str) -> Result<ExtendedHeader> {
         let command = NodeCommand::GetHeader(SingleHeaderQuery::ByHash(hash.parse()?));
         let response = self.worker.exec(command).await?;
-        let header = response.into_header().check_variant()?;
-
-        header.into()
+        response.into_header().check_variant()?
     }
 
     /// Get a synced header for the block with a given height.
     ///
     /// Returns a javascript object with given structure:
     /// https://docs.rs/celestia-types/latest/celestia_types/struct.ExtendedHeader.html
     #[wasm_bindgen(js_name = getHeaderByHeight)]
-    pub async fn get_header_by_height(&self, height: u64) -> Result<JsValue> {
+    pub async fn get_header_by_height(&self, height: u64) -> Result<ExtendedHeader> {
         let command = NodeCommand::GetHeader(SingleHeaderQuery::ByHeight(height));
         let response = self.worker.exec(command).await?;
-        let header = response.into_header().check_variant()?;
-
-        header.into()
+        response.into_header().check_variant()?
     }
 
     /// Get synced headers from the given heights range.
@@ -335,28 +316,24 @@ impl NodeClient {
         &self,
         start_height: Option<u64>,
         end_height: Option<u64>,
-    ) -> Result<Array> {
+    ) -> Result<Vec<ExtendedHeader>> {
         let command = NodeCommand::GetHeadersRange {
             start_height,
             end_height,
         };
         let response = self.worker.exec(command).await?;
-        let headers = response.into_headers().check_variant()?;
-
-        headers.into()
+        response.into_headers().check_variant()?
     }
 
     /// Get data sampling metadata of an already sampled height.
     ///
     /// Returns a javascript object with given structure:
     /// https://docs.rs/lumina-node/latest/lumina_node/store/struct.SamplingMetadata.html
     #[wasm_bindgen(js_name = getSamplingMetadata)]
-    pub async fn get_sampling_metadata(&self, height: u64) -> Result<JsValue> {
+    pub async fn get_sampling_metadata(&self, height: u64) -> Result<Option<SamplingMetadata>> {
         let command = NodeCommand::GetSamplingMetadata { height };
         let response = self.worker.exec(command).await?;
-        let metadata = response.into_sampling_metadata().check_variant()?;
-
-        Ok(to_value(&metadata?)?)
+        response.into_sampling_metadata().check_variant()?
     }
 
     /// Returns a [`BroadcastChannel`] for events generated by [`Node`].
@@ -438,7 +415,6 @@ mod tests {
     use gloo_timers::future::sleep;
     use libp2p::{multiaddr::Protocol, Multiaddr};
     use rexie::Rexie;
-    use serde_wasm_bindgen::from_value;
     use wasm_bindgen_futures::spawn_local;
     use wasm_bindgen_test::wasm_bindgen_test;
     use web_sys::MessageChannel;
@@ -460,8 +436,7 @@ mod tests {
         assert_eq!(info.num_peers, 1);
 
         let bridge_head_header = rpc_client.header_network_head().await.unwrap();
-        let head_header: ExtendedHeader =
-            from_value(client.request_head_header().await.unwrap()).unwrap();
+        let head_header: ExtendedHeader = client.request_head_header().await.unwrap();
         assert_eq!(head_header, bridge_head_header);
         rpc_client
             .p2p_close_peer(&PeerId(

node-wasm/src/client.rs

@@ -1,23 +1,22 @@
 use std::fmt::Debug;
 
 use enum_as_inner::EnumAsInner;
-use js_sys::Array;
 use libp2p::Multiaddr;
 use libp2p::PeerId;
 use serde::{Deserialize, Serialize};
 use tracing::error;
-use wasm_bindgen::{JsError, JsValue};
+use wasm_bindgen::JsError;
 
-use celestia_types::hash::Hash;
+use celestia_types::{hash::Hash, ExtendedHeader};
 use lumina_node::node::{PeerTrackerInfo, SyncingInfo};
 use lumina_node::store::SamplingMetadata;
 
 use crate::client::WasmNodeConfig;
 use crate::error::Error;
 use crate::error::Result;
-use crate::utils::JsResult;
 use crate::wrapper::libp2p::NetworkInfoSnapshot;
 
+#[allow(clippy::large_enum_variant)]
 #[derive(Debug, Serialize, Deserialize)]
 pub(crate) enum NodeCommand {
     InternalPing,
@@ -40,8 +39,7 @@ pub(crate) enum NodeCommand {
     GetListeners,
     RequestHeader(SingleHeaderQuery),
     GetVerifiedHeaders {
-        #[serde(with = "serde_wasm_bindgen::preserve")]
-        from: JsValue,
+        from: ExtendedHeader,
         amount: u64,
     },
     GetHeadersRange {
@@ -78,9 +76,9 @@ pub(crate) enum WorkerResponse {
     SetPeerTrust(Result<()>),
     Connected(Result<()>),
     Listeners(Result<Vec<Multiaddr>>),
-    Header(JsResult<JsValue, Error>),
-    Headers(JsResult<Array, Error>),
-    LastSeenNetworkHead(JsResult<JsValue, Error>),
+    Header(Result<ExtendedHeader, Error>),
+    Headers(Result<Vec<ExtendedHeader>, Error>),
+    LastSeenNetworkHead(Result<Option<ExtendedHeader>, Error>),
     SamplingMetadata(Result<Option<SamplingMetadata>>),
 }
 

node-wasm/src/commands.rs

@@ -38,6 +38,7 @@ impl From<MessagePort> for MessagePortLike {
 #[derive(Debug, Clone, Copy)]
 pub struct ClientId(usize);
 
+#[allow(clippy::large_enum_variant)]
 pub(crate) enum ClientMessage {
     Command { id: ClientId, command: NodeCommand },
     AddConnection(JsValue),