43 lines
1.6 KiB
Rust
43 lines
1.6 KiB
Rust
pub mod input;
|
|
pub mod error;
|
|
|
|
pub mod days;
|
|
|
|
use days::*;
|
|
|
|
use std::env;
|
|
|
|
use crate::error::AdventError;
|
|
|
|
fn main() -> Result<(), AdventError> {
|
|
let days = vec![
|
|
(day1::one as fn() -> Result<u32, AdventError>, day1::two as fn() -> Result<u32, AdventError>),
|
|
(day2::one as fn() -> Result<u32, AdventError>, day2::two as fn() -> Result<u32, AdventError>),
|
|
(day3::one as fn() -> Result<u32, AdventError>, day3::two as fn() -> Result<u32, AdventError>),
|
|
(day4::one as fn() -> Result<u32, AdventError>, day4::two as fn() -> Result<u32, AdventError>),
|
|
(day5::one as fn() -> Result<u32, AdventError>, day5::two as fn() -> Result<u32, AdventError>),
|
|
(day6::one as fn() -> Result<u32, AdventError>, day6::two as fn() -> Result<u32, AdventError>),
|
|
];
|
|
|
|
if env::args().len() != 3 {
|
|
Ok(days.into_iter().map(|(one, two)| {
|
|
let one_result = one()?;
|
|
let two_result = two()?;
|
|
Ok((one_result, two_result))
|
|
}).enumerate().map(|(index, r): (usize, Result<(u32, u32), AdventError>)| {
|
|
match r {
|
|
Ok((res1, res2)) => println!("{}: ({}, {})", index + 1, res1, res2),
|
|
Err(e) => println!("Error: {}", e.0)
|
|
};
|
|
}).fold((), |acc, _| { acc }))
|
|
} else {
|
|
let (one, two) = days[env::args().nth(1).ok_or(AdventError("Failed to parse day".into()))?.parse::<usize>()? - 1];
|
|
match env::args().nth(2).unwrap().parse::<usize>()? {
|
|
1 => Ok(println!("1: {}", one()?)),
|
|
2 => Ok(println!("2: {}", two()?)),
|
|
_ => return Err(AdventError("part must be either 1 or 2".into()))
|
|
}
|
|
}
|
|
|
|
}
|