Files
toru/src/stats.rs
aaron-jack-manning f5fb7e1a49 fixed non-local time
2022-09-23 08:45:57 +10:00

96 lines
2.7 KiB
Rust
Executable File

use crate::tasks;
use crate::error;
use std::path;
use std::collections::BTreeMap;
use chrono::SubsecRound;
pub fn completed_tasks(days : u16, vault_folder : &path::Path) -> Result<(), error::Error> {
let tasks = tasks::Task::load_all(vault_folder, true)?;
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!["Task", "Completed"]);
for task in tasks {
if let Some(completed_date) = task.data.completed {
let time_diff = chrono::Local::now().naive_local() - completed_date;
if time_diff < chrono::Duration::days(i64::from(days)) && time_diff > chrono::Duration::zero() {
table.add_row(vec![
task.data.name.clone(),
completed_date.round_subsecs(0).to_string()
]);
}
}
}
println!("{}", table);
Ok(())
}
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 {
let mut time = tasks::Duration::zero();
for entry in &task.data.time_entries {
if chrono::Local::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"]);
let mut total_duration = tasks::Duration::zero();
for (tag, duration) in times {
table.add_row(
vec![
tag.clone(),
duration.to_string(),
]
);
total_duration = total_duration + duration;
}
table.add_row(
vec![
String::from("Total"),
total_duration.to_string(),
]
);
println!("{}", table);
Ok(())
}