diff options
| -rw-r--r-- | 2023/tpl.rs | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/2023/tpl.rs b/2023/tpl.rs index 041681e..8bd3cfb 100644 --- a/2023/tpl.rs +++ b/2023/tpl.rs @@ -1,17 +1,21 @@ #![allow(dead_code)] #![allow(unused_variables)] #![allow(unused_mut)] +use std::env; use std::fs::File; use std::io::Read; fn main() { - let filename = "in/day .ref"; - // let filename = "in/day .pzl"; - + let args: Vec<String> = env::args().collect(); + let filename = if args.len() == 1 { + "in/".to_owned() + args[0].split('/').last().unwrap() + ".pzl" + } else { + args[1].clone() + }; let mut f = File::open(filename).expect("cannot open file"); let mut content = String::new(); f.read_to_string(&mut content).expect("cannot read file"); - let lines: Vec<_> = content.trim_end().split('\n').collect(); + let lines = content.trim_end().split("\n"); println!("{:?}", lines); let mut res1 = 0; |
