upd yaml

bartvanbenthem · Jul 4, 2023 · f81a591 · f81a591
1 parent de3cd3e
commit f81a591
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -17,6 +17,7 @@ struct Config {
     template_file: String,
     output_file: String,
     json_file: String,
+    yaml_file: String,
 }
 
 fn main() ->  Result<(), Box<dyn Error>> {
@@ -78,18 +79,27 @@ fn get_args() -> Result<Config, Box<dyn Error>> {
         )
         .arg(
             Arg::with_name("variables_json")
-                .short("v")
-                .long("variables")
+                .short("j")
+                .long("variables-json")
                 .required(false)
                 .takes_value(true)
                 .help("Path to the JSON file"),
         )
+        .arg(
+            Arg::with_name("variables_yaml")
+                .short("y")
+                .long("variables-yaml")
+                .required(false)
+                .takes_value(true)
+                .help("Path to the YAML file"),
+        )
         .get_matches();
 
     Ok(Config {
         template_file: matches.value_of("template_file").unwrap().to_string(),
         output_file: matches.value_of("output_file").unwrap_or("").to_string(),
         json_file: matches.value_of("variables_json").unwrap_or("").to_string(),
+        yaml_file: matches.value_of("variables_yaml").unwrap_or("").to_string(),
     })
 }