Base framework

1 files changed, 74 insertions, 0 deletions

diff --git a/aoc/src/lib.rs b/aoc/src/lib.rs
new file mode 100644
index 0000000..5fad72d
--- /dev/null
+++ b/aoc/src/lib.rs
@@ -0,0 +1,74 @@
+use clap::Parser;
+use std::fs::File;
+use std::io;
+use std::io::{BufRead, BufReader, Read};
+
+#[derive(Parser, Debug)]
+#[command(version, about, long_about = None)]
+struct Args {
+    #[arg(short, long, default_value = "-")]
+    file: String,
+
+    #[arg(short, long, default_value_t = 1)]
+    part: u8,
+}
+
+pub trait Day {
+    fn example_input(&self) -> &'static str;
+    fn example_result_part_1(&self) -> &'static str;
+    fn example_result_part_2(&self) -> &'static str;
+
+    fn read_lines(&self, input:  BufReader<Box<dyn Read>>) -> Vec<String> {
+        let mut lines = Vec::new();
+
+        for line in input.lines() {
+            match line {
+                Ok(content) => lines.push(content),
+                Err(e) => eprintln!("Error reading line: {}", e),
+            }
+        }
+
+        lines
+    }
+
+    fn part_1(&self, input: BufReader<Box<dyn Read>>) -> String;
+    fn part_2(&self, input: BufReader<Box<dyn Read>>) -> String;
+}
+
+pub fn main(day: &dyn Day) {
+    let args = Args::parse();
+
+    let reader: Box<dyn Read> = if args.file == "-" {
+        Box::new(io::stdin())
+    } else {
+        Box::new(File::open(&args.file).expect("Failed to open file"))
+    };
+
+    let buffer = BufReader::new(reader);
+
+    let result: String;
+    if args.part == 1 {
+        result = day.part_1(buffer);
+    } else {
+        result = day.part_2(buffer);
+    }
+
+    println!("Result: {result}")
+}
+
+pub fn test_day<T: Day>(day: &T) {
+    use std::io::Cursor;
+    use std::io::{BufReader, Read};
+
+    let example_input = day.example_input();
+
+    // Test Part 1
+    let input = BufReader::new(Box::new(Cursor::new(example_input)) as Box<dyn Read>);
+    let output = day.part_1(input);
+    assert_eq!(output.trim(), day.example_result_part_1(), "Part 1 failed");
+
+    // Test Part 2
+    let input = BufReader::new(Box::new(Cursor::new(example_input)) as Box<dyn Read>);
+    let output = day.part_2(input);
+    assert_eq!(output.trim(), day.example_result_part_2(), "Part 2 failed");
+}
\ No newline at end of file