Skip to content

feat(WIP): WASM Support #991

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
+85 −29
Closed

feat(WIP): WASM Support #991

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions Cargo.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ version = "0.34.0"
bench = false

[features]
wasm = ["dep:wasm-bindgen"]
default = ["mnemonic", "serde"]
# Enables config
serde = ["dep:serde", "dep:serde_derive", "dep:serde_json"]
Expand Down Expand Up @@ -46,6 +47,8 @@ thiserror = "1.0.31"
time = "0.3.41"
tokio = { version = "1.44.2", features = ["time"] }
tonic = "0.12.3"
tonic-web-wasm-client = "0.7.0"
wasm-bindgen = { version = "0.2", optional = true }
tinystr = { version = "0.7.0", default-features = false }
arc-swap = "1.6.0"
rlp = "0.6.1"
Expand Down
108 changes: 79 additions & 29 deletions protobufs/build.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -72,12 +72,19 @@ fn main() -> anyhow::Result<()> {
fs::write(service, &*contents)?;
}

let mut cfg = tonic_build::configure()
// We have already emitted a cargo directive to trigger a rerun on the source folder
// that the copy this builds is based on. If the directives are not suppressed, the
// crate will rebuild on every compile due to the modified time stamps post-dating
// the start time of the compile action.
.emit_rerun_if_changed(false);
// Check if we're building for WASM
let target = env::var("TARGET").unwrap_or_default();
let is_wasm = target.contains("wasm");

let mut cfg = if is_wasm {
// Configure for WASM build
tonic_build::configure()
.build_server(false)
.build_transport(false)
.emit_rerun_if_changed(false)
} else {
tonic_build::configure().emit_rerun_if_changed(false)
};

// most of the protobufs in "basic types" should be Eq + Hash + Copy
// any protobufs that would typically be used as parameter, that meet the requirements of those
Expand Down Expand Up @@ -128,6 +135,11 @@ fn main() -> anyhow::Result<()> {
"]"#,
);

// Add WASM-specific attributes if building for WASM
if is_wasm {
cfg = cfg.type_attribute(".", "#[cfg_attr(target_arch = \"wasm32\", wasm_bindgen)]");
}

cfg.compile_protos(&services, &[services_tmp_path.clone()])?;

// NOTE: prost generates rust doc comments and fails to remove the leading * line
Expand All @@ -138,35 +150,64 @@ fn main() -> anyhow::Result<()> {
let mirror_out_dir = Path::new(&env::var("OUT_DIR")?).join("mirror");
create_dir_all(&mirror_out_dir)?;

tonic_build::configure()
.build_server(false)
.extern_path(".proto.Timestamp", "crate::services::Timestamp")
.extern_path(".proto.TopicID", "crate::services::TopicId")
.extern_path(".proto.FileID", "crate::services::FileId")
.extern_path(".proto.NodeAddress", "crate::services::NodeAddress")
.extern_path(
".proto.ConsensusMessageChunkInfo",
"crate::services::ConsensusMessageChunkInfo",
)
.out_dir(&mirror_out_dir)
.compile_protos(
&["./mirror/consensus_service.proto", "./mirror/mirror_network_service.proto"],
&["./mirror/", "./services/hapi/hedera-protobufs/services/"],
)?;
let mut mirror_cfg = if is_wasm {
tonic_build::configure()
.build_server(false)
.build_transport(false)
.extern_path(".proto.Timestamp", "crate::services::Timestamp")
.extern_path(".proto.TopicID", "crate::services::TopicId")
.extern_path(".proto.FileID", "crate::services::FileId")
.extern_path(".proto.NodeAddress", "crate::services::NodeAddress")
.extern_path(
".proto.ConsensusMessageChunkInfo",
"crate::services::ConsensusMessageChunkInfo",
)
.out_dir(&mirror_out_dir)
} else {
tonic_build::configure()
.build_server(false)
.extern_path(".proto.Timestamp", "crate::services::Timestamp")
.extern_path(".proto.TopicID", "crate::services::TopicId")
.extern_path(".proto.FileID", "crate::services::FileId")
.extern_path(".proto.NodeAddress", "crate::services::NodeAddress")
.extern_path(
".proto.ConsensusMessageChunkInfo",
"crate::services::ConsensusMessageChunkInfo",
)
.out_dir(&mirror_out_dir)
};

if is_wasm {
mirror_cfg =
mirror_cfg.type_attribute(".", "#[cfg_attr(target_arch = \"wasm32\", wasm_bindgen)]");
}

mirror_cfg.compile_protos(
&["./mirror/consensus_service.proto", "./mirror/mirror_network_service.proto"],
&["./mirror/", "./services/hapi/hedera-protobufs/services/"],
)?;

remove_useless_comments(&mirror_out_dir.join("proto.rs"))?;

// streams
// NOTE: must be compiled in a separate folder otherwise it will overwrite the previous build

let streams_out_dir = Path::new(&env::var("OUT_DIR")?).join("streams");
create_dir_all(&streams_out_dir)?;

// NOTE: **ALL** protobufs defined in basic_types must be specified here
let cfg = tonic_build::configure();
let cfg = builder::extern_basic_types(cfg);
let mut streams_cfg = if is_wasm {
tonic_build::configure().build_server(false).build_transport(false)
} else {
tonic_build::configure()
};
streams_cfg = builder::extern_basic_types(streams_cfg);

if is_wasm {
streams_cfg =
streams_cfg.type_attribute(".", "#[cfg_attr(target_arch = \"wasm32\", wasm_bindgen)]");
}

cfg.out_dir(&streams_out_dir).compile_protos(
streams_cfg.out_dir(&streams_out_dir).compile_protos(
&["./services/hapi/hedera-protobufs/streams/account_balance_file.proto"],
&[
"./services/hapi/hedera-protobufs/streams/",
Expand All @@ -184,8 +225,12 @@ fn main() -> anyhow::Result<()> {

// note:
// almost everything in services must be specified here.
let cfg = tonic_build::configure();
let cfg = builder::extern_basic_types(cfg)
let mut sdk_cfg = if is_wasm {
tonic_build::configure().build_server(false).build_transport(false)
} else {
tonic_build::configure()
};
sdk_cfg = builder::extern_basic_types(sdk_cfg)
.services_same("AssessedCustomFee")
.services_same("ConsensusCreateTopicTransactionBody")
.services_same("ConsensusDeleteTopicTransactionBody")
Expand Down Expand Up @@ -252,7 +297,12 @@ fn main() -> anyhow::Result<()> {
.services_same("UtilPrngTransactionBody")
.services_same("VirtualAddress");

cfg.out_dir(&sdk_out_dir).compile_protos(
if is_wasm {
sdk_cfg =
sdk_cfg.type_attribute(".", "#[cfg_attr(target_arch = \"wasm32\", wasm_bindgen)]");
}

sdk_cfg.out_dir(&sdk_out_dir).compile_protos(
&["./sdk/transaction_list.proto"],
&["./sdk/", services_tmp_path.as_os_str().to_str().unwrap()],
)?;
Expand Down Expand Up @@ -396,4 +446,4 @@ mod builder {
.services_path("ContractID", "ContractId")
.services_path("StakingInfo", "StakingInfo")
}
}
}
3 changes: 3 additions & 0 deletions protobufs/src/lib.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,9 @@ mod time_0_3;
#[cfg(feature = "fraction")]
mod fraction;

#[cfg(feature = "wasm")]
use wasm_bindgen::prelude::*;

// fixme: Do this, just, don't warn 70 times in generated code.
#[allow(clippy::derive_partial_eq_without_eq)]
pub mod services {
Expand Down
Loading