update template

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;