From b7e2fe6160280472a625a8867ea5177b0155737c Mon Sep 17 00:00:00 2001
From: mathiasmagnusson <mathias@magnusson.space>
Date: Sun, 11 Dec 2022 23:04:39 +0100
Subject: Aoc day 6, 7 & 8

---
 aoc22/lib/src/lib.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'aoc22/lib/src/lib.rs')

diff --git a/aoc22/lib/src/lib.rs b/aoc22/lib/src/lib.rs
index ffe4e46..1ed2b06 100644
--- a/aoc22/lib/src/lib.rs
+++ b/aoc22/lib/src/lib.rs
@@ -1,9 +1,19 @@
-use std::fs;
+use std::{fs, io::Read};
 
 pub mod prelude {
     pub use std::error::Error;
 }
 
 pub fn read_input(day: usize) -> String {
-    fs::read_to_string(format!("day{}/input", day)).expect("Could not read input")
+    let mut s = String::new();
+    let path = if let Some(arg) = std::env::args().nth(1) {
+        if arg == "-" {
+            std::io::stdin().read_to_string(&mut s).unwrap();
+            return s;
+        }
+        arg
+    } else {
+        format!("day{}/input", day)
+    };
+    fs::read_to_string(path).expect("Could not read input")
 }
-- 
cgit v1.2.3