use crate::tasks;
use crate::error;

use std::path;
use std::collections::BTreeMap;

pub fn time_per_tag(days : u16, vault_folder : &path::Path) -> Result<(), error::Error> {

    let tasks = tasks::Task::load_all(vault_folder, true)?;

    let mut times = BTreeMap::<String, tasks::Duration>::new();

    for task in &tasks {
        if !task.data.discarded {
            let mut time = tasks::Duration::zero();

            for entry in &task.data.time_entries {
                if chrono::Utc::now().naive_local().date() - entry.logged_date < chrono::Duration::days(i64::from(days)) {
                    time = time + entry.duration;
                }
            }

            let tag_count = task.data.tags.len();
            let time_per_tag = time / tag_count;

            for tag in &task.data.tags {
                match times.get_mut(tag) {
                    Some(time) => {
                        *time = *time + time_per_tag;
                    },
                    None => {
                        times.insert(tag.clone(), time_per_tag);
                    }
                }
            }
        }
    }

    let mut table = comfy_table::Table::new();
    table
        .load_preset(comfy_table::presets::UTF8_FULL)
        .apply_modifier(comfy_table::modifiers::UTF8_ROUND_CORNERS)
        .set_content_arrangement(comfy_table::ContentArrangement::Dynamic);
    table.set_header(vec!["Tag", "Time"]);


    for (tag, duration) in times {
        table.add_row(
            vec![
                tag.clone(),
                duration.to_string(),
            ]
        );
    }

    println!("{}", table);

    Ok(())
}