diff options
| -rw-r--r-- | guests_macro/src/lib.rs | 6 | ||||
| -rw-r--r-- | zkvms_host_io/input_macros/src/lib.rs | 3 |
2 files changed, 7 insertions, 2 deletions
diff --git a/guests_macro/src/lib.rs b/guests_macro/src/lib.rs index ba6a9ec..f87e8a5 100644 --- a/guests_macro/src/lib.rs +++ b/guests_macro/src/lib.rs @@ -1,9 +1,15 @@ +use std::{ fs::File, io::Write }; use proc_macro::TokenStream; mod parse_fn; #[proc_macro_attribute] pub fn proving_entrypoint(_: TokenStream, mut item: TokenStream) -> TokenStream { let (name, args, ret) = parse_fn::split_fn(&item); + + // Put the file in zkVMs-benchmarks/guests/ + let mut output = File::create("../type.txt").unwrap(); + write!(output, "{}", args); + item.extend(format!("#[macro_export] macro_rules! entrypoint_expr {{ () => {{ diff --git a/zkvms_host_io/input_macros/src/lib.rs b/zkvms_host_io/input_macros/src/lib.rs index 2585653..84c2792 100644 --- a/zkvms_host_io/input_macros/src/lib.rs +++ b/zkvms_host_io/input_macros/src/lib.rs @@ -4,9 +4,8 @@ use proc_macro::TokenStream; mod parse_fn; use crate::parse_fn::{ args_split, args_divide, group_streams }; - fn get_args() -> TokenStream { - "(graph: Vec<Vec<bool>>, colors: u32, coloring: Vec<Vec<u32>>,)".parse::<TokenStream>().unwrap() + include_str!("../../../guests/type.txt").parse::<TokenStream>().unwrap() } #[proc_macro] |