From db0c1b91f44394e2056a30283c1376cc003e3f5f Mon Sep 17 00:00:00 2001 From: Kamen Mladenov Date: Tue, 14 Jan 2025 14:04:38 +0200 Subject: feat(zkvms): Add sp1 guest and it's macro --- zkvms/sp1/wrapper_macro/src/lib.rs | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 zkvms/sp1/wrapper_macro/src/lib.rs (limited to 'zkvms/sp1/wrapper_macro/src/lib.rs') diff --git a/zkvms/sp1/wrapper_macro/src/lib.rs b/zkvms/sp1/wrapper_macro/src/lib.rs new file mode 100644 index 0000000..eb24c33 --- /dev/null +++ b/zkvms/sp1/wrapper_macro/src/lib.rs @@ -0,0 +1,25 @@ +use proc_macro::TokenStream; + +#[path = "../../../../guests_macro/src/parse_fn.rs"] +mod parse_fn; +use crate::parse_fn::{ split_fn, args_split, args_divide, group_streams }; + +#[proc_macro] +pub fn make_wrapper(item: TokenStream) -> TokenStream { + let (name, args, ret) = split_fn(&item); + let args_split = args_split(&args); + + let mut out = TokenStream::new(); + for arg in args_split { + out.extend(format!("let {} = read();", arg).parse::()); + } + + let (patterns, _) = args_divide(&args); + let ts_patterns = group_streams(&patterns); + + out.extend(format!("commit(&zkp::{}{});", name, ts_patterns).parse::()); + + let mut block = TokenStream::new(); + block.extend(format!("{{ {} }}", out).parse::()); + block +} -- cgit v1.2.3