use anyhow::Context;
use chrono::NaiveDate;
use libgitheatmap::{self, cli::CliArgs};

fn main() {
	divan::main();
}

#[divan::bench]
fn current_repo_commits() {
	let home_dir = std::env::current_dir().unwrap();

	let args = CliArgs {
		root_dir: Some(vec![home_dir]),
		authors: None,
		char: '▩',
		repos: None,
		ignored_repos: None,
		branches: None,
		since: "2024-01-01".to_string(),
		until: None,
		split_months: false,
		months_per_row: 13,
		no_merges: false,
		..Default::default()
	};

	let since = NaiveDate::parse_from_str(&args.since, "%Y-%m-%d").unwrap();

	let until = NaiveDate::parse_from_str("2025-01-01", "%Y-%m-%d").unwrap();

	libgitheatmap::get_commits(args, since, until)
		.with_context(|| "Could not fetch commit list")
		.unwrap();
}

#[divan::bench(sample_count = 20)]
fn all_repos_commits() {
	let home_dir = std::path::PathBuf::from("/home");

	let args = CliArgs {
		root_dir: Some(vec![home_dir]),
		authors: None,
		char: '▩',
		repos: None,
		ignored_repos: None,
		branches: None,
		since: "2024-01-01".to_string(),
		until: None,
		split_months: false,
		months_per_row: 13,
		no_merges: false,
		..Default::default()
	};

	let since = NaiveDate::parse_from_str(&args.since, "%Y-%m-%d").unwrap();

	let until = NaiveDate::parse_from_str("2025-01-01", "%Y-%m-%d").unwrap();

	libgitheatmap::get_commits(args, since, until)
		.with_context(|| "Could not fetch commit list")
		.unwrap();
}