use std::io::Read;

mod common;
mod day01;
mod day02;
mod day03;
mod day04;
mod day05;
mod day06;
mod day07;
mod day08;
mod day09;
mod day10;

pub trait Solution {
    fn part1(&mut self, input: &mut dyn Read) -> String;

    fn part2(&mut self, _input: &mut dyn Read) -> String {
        unimplemented!("Still working on part 1");
    }
}

pub fn get_implementation(day: usize) -> Box<dyn Solution> {
    match day {
        1 => Box::new(day01::Day01::default()),
        2 => Box::new(day02::Day02::default()),
        3 => Box::new(day03::Day03::default()),
        4 => Box::new(day04::Day04::default()),
        5 => Box::new(day05::Day05::default()),
        6 => Box::new(day06::Day06::default()),
        7 => Box::new(day07::Day07::default()),
        8 => Box::new(day08::Day08::default()),
        9 => Box::new(day09::Day09::default()),
        10 => Box::new(day10::Day10::default()),
        _ => panic!("Unsupported day {}", day),
    }
}

#[cfg(test)]
#[macro_export]
macro_rules! test_implementation {
    ($impl:ident, 1, $source:ident, $output:expr) => {
        let mut implementation = $impl::default();

        let result = implementation.part1(&mut $source.as_ref());
        assert_eq!($output.to_string(), result);
    };

    ($impl:ident, 2, $source:ident, $output:expr) => {
        let mut implementation = $impl::default();

        let result = implementation.part2(&mut $source.as_ref());
        assert_eq!($output.to_string(), result);
    };
}