guests_macro/src/lib.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

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();
    writeln!(output, "{}", args);
    write!(output, "{}", ret);

    item.extend(format!("#[macro_export]
        macro_rules! entrypoint_expr {{
            () => {{
                make_wrapper!{{{}{} -> {}}}
            }};
        }}", name, args, ret).parse::<TokenStream>());
    item
}