cmz-core 0.2.0

Core functionality for the macros in the cmz crate
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use clap::Parser;
use cmz_core::*;
use std::io;
use std::process::ExitCode;

#[derive(Parser, Debug)]
#[clap(version, about, long_about = None)]
struct Args {
    /// use CMZ14 (as opposed to the default of µCMZ)
    #[arg(short = '1', long)]
    cmz14: bool,

    /// only generate the client-side code
    #[arg(short, long)]
    client_only: bool,

    /// only generate the issuer-side code
    #[arg(short, long)]
    issuer_only: bool,
}

fn pretty_print(code_str: &str) {
    let parsed_output = syn::parse_file(code_str).unwrap();
    let formatted_output = prettyplease::unparse(&parsed_output);
    println!("{}", formatted_output);
}

fn main() -> ExitCode {
    let args = Args::parse();
    let stdin = io::read_to_string(io::stdin()).unwrap();

    if args.client_only && args.issuer_only {
        eprintln!("client_only and issuer_only cannot be used together");
        return ExitCode::FAILURE;
    };
    let emit_client = !args.issuer_only;
    let emit_issuer = !args.client_only;
    let use_mucmz = !args.cmz14;

    let proto_spec: ProtoSpec = match syn::parse_str(&stdin) {
        Err(_) => {
            eprintln!("Could not parse stdin as a CMZ protocol specification");
            return ExitCode::FAILURE;
        }
        Ok(proto_spec) => proto_spec,
    };

    let output = cmz_core(&proto_spec, use_mucmz, emit_client, emit_issuer);
    pretty_print(&output.to_string());

    ExitCode::SUCCESS
}