feat(zkvms_host_io): Implement using a guest's default input

1 files changed, 10 insertions, 3 deletions

diff --git a/zkvms_host_io/src/lib.rs b/zkvms_host_io/src/lib.rs
index 74c3992..191d2ad 100644
--- a/zkvms_host_io/src/lib.rs
+++ b/zkvms_host_io/src/lib.rs
@@ -1,16 +1,18 @@
 use clap::{Parser, ValueEnum};
 use num_traits::NumCast;
 use serde::{ Serialize, Deserialize };
+use std::fs::read_to_string;
 pub use input_macros::foreach_input_field;
 
+static DEFAULT_PUBLIC_INPUT: &str = include_str!(concat!(env!("INPUTS_DIR"), "/default_public_input.toml"));
+
 #[derive(Parser, Debug)]
 #[command(version, about, long_about = None)]
 struct Cli {
     /// What the ZKVM is going to do
     run_type: RunType,
 
-    #[arg(default_value = "./public_input.toml")]
-    public_input: String,
+    public_input: Option<String>,
 }
 
 #[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
@@ -31,7 +33,12 @@ input_macros::generate_input_struct!();
 pub fn read_args() -> RunWith<Input> {
     let cli = Cli::parse();
 
-    let val: Input = toml::from_str(&std::fs::read_to_string(cli.public_input).unwrap()).unwrap();
+    let contents: String = if cli.public_input.is_some() {
+            read_to_string(cli.public_input.unwrap()).unwrap()
+        } else {
+            DEFAULT_PUBLIC_INPUT.to_string()
+        };
+    let val: Input = toml::from_str(&contents).unwrap();
 
     RunWith {
         run_type: cli.run_type,