derive_codec_sv2: refactor to std logic

stratum-mining · Oct 20, 2024 · 03a2dae · 03a2dae
1 parent 8f5d7ae
commit 03a2dae
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/protocols/v2/binary-sv2/no-serde-sv2/derive_codec/Cargo.toml b/protocols/v2/binary-sv2/no-serde-sv2/derive_codec/Cargo.toml
@@ -16,6 +16,9 @@ keywords = ["stratum", "mining", "bitcoin", "protocol"]
 [dependencies]
 binary_codec_sv2 = {version = "^1.0.0", path="../codec"}
 
+[features]
+std = []
+
 [lib]
 proc-macro = true
 

diff --git a/protocols/v2/binary-sv2/no-serde-sv2/derive_codec/src/lib.rs b/protocols/v2/binary-sv2/no-serde-sv2/derive_codec/src/lib.rs
@@ -1,4 +1,13 @@
+#![cfg_attr(not(feature = "std"), no_std)]
+
 extern crate proc_macro;
+#[macro_use]
+extern crate alloc;
+
+use alloc::{
+    string::{String, ToString},
+    vec::Vec,
+};
 use core::iter::FromIterator;
 use proc_macro::{Group, TokenStream, TokenTree};