From 250bad526c914037d8295e565a58a1f64efe7da5 Mon Sep 17 00:00:00 2001
From: kageru <kageru@encode.moe>
Date: Wed, 1 Dec 2021 14:12:36 +0100
Subject: [PATCH] Revert "Remove env stuff from input reading"

I remember now. This was so I can `cargo run bigboy_input` without code changes
---
 2021/src/common.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

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]