diff --git a/2021/src/common.rs b/2021/src/common.rs index c79dc36..295e8d5 100644 --- a/2021/src/common.rs +++ b/2021/src/common.rs @@ -1,5 +1,13 @@ +use std::env; + pub fn read_file(day: usize) -> String { - std::fs::read_to_string(format!("inputs/day{:0>2}", day)).unwrap() + std::fs::read_to_string( + env::args() + .nth(1) + .filter(|n| n != "--bench") + .unwrap_or_else(|| format!("inputs/day{:0>2}", day)), + ) + .unwrap() } #[inline]