use zkvms_host_io::{
    benchmarkable, read_args,
    RunType::{Execute, Prove, Verify},
};

pub fn main() {
    let run_info = read_args();
    if run_info.run_type == Execute {
        panic!("Execution is not supported!");
    }

    let elf_path = std::env::var("ELF_PATH").expect("ELF PATH is missing");
    // guest_closures are generated by (Jolt's) wrapper_macro
    let (prove_guest, verify_guest) = guest::guest_closures(elf_path);

    match run_info.run_type {
        Execute => unreachable!(),
        Prove => benchmarkable! {
            let (output, _) = prove_guest(run_info.input.clone().into());
            println!("Prove output: {:?}", output);
        },
        Verify => benchmarkable! {
            let (_, proof) = prove_guest(run_info.input.clone().into());
            let is_valid = verify_guest(proof);
            println!("Verify is valid: {:?}", is_valid);
        },
    }
}