#![allow(clippy::needless_range_loop)]
#![forbid(elided_lifetimes_in_paths, unsafe_code)]

use aoc23::read;
use chumsky::prelude::*;

struct History(Vec<i64>);

impl History {
	fn parser() -> impl Parser<char, Self, Error = Simple<char>> {
		one_of(['-', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9'])
			.repeated()
			.at_least(1)
			.separated_by(just(" "))
			.map(|nums: Vec<Vec<char>>| {
				Self(
					nums.into_iter()
						.map(|num| num.into_iter().collect::<String>().parse().unwrap())
						.collect()
				)
			})
	}
}

fn parser() -> impl Parser<char, Vec<History>, Error = Simple<char>> {
	History::parser()
		.then_ignore(just("\n"))
		.repeated()
		.then_ignore(end())
}

impl History {
	fn diff(&self) -> Self {
		let mut diffs = Vec::with_capacity(self.0.len() - 1);
		if self.0.len() > 1 {
			for i in 1 .. self.0.len() {
				diffs.push(self.0[i] - self.0[i - 1]);
			}
		}
		Self(diffs)
	}

	fn extend_last(&self) -> i64 {
		if self.0.iter().any(|num| *num != 0) {
			let diff = self.diff();
			let last = diff.extend_last();
			self.0.last().unwrap() + last
		} else {
			0
		}
	}

	fn extend_first(&self) -> i64 {
		if self.0.iter().any(|num| *num != 0) {
			let diff = self.diff();
			let first = diff.extend_first();
			self.0.first().unwrap() - first
		} else {
			0
		}
	}
}

fn main() -> anyhow::Result<()> {
	let histories = read("inputs/day9.txt", parser())?;

	let mut sum_last = 0;
	let mut sum_first = 0;
	for h in histories {
		sum_last += h.extend_last();
		sum_first += h.extend_first();
	}
	println!("{sum_last}");
	println!("{sum_first}");

	Ok(())
}