1 files changed, 101 insertions, 0 deletions
diff --git a/src/command/stats.rs b/src/command/stats.rs
new file mode 100644
index 00000000..ea5893f9
--- /dev/null
+++ b/src/command/stats.rs
@@ -0,0 +1,101 @@
+use std::collections::HashMap;
+
+use chrono::prelude::*;
+use chrono::{Duration, Utc};
+use chrono_english::{parse_date_string, Dialect};
+
+use cli_table::{format::Justify, print_stdout, Cell, Style, Table};
+use eyre::{eyre, Result};
+use structopt::StructOpt;
+
+use crate::local::database::{Database, Sqlite};
+use crate::local::history::History;
+
+#[derive(StructOpt)]
+pub enum Cmd {
+    #[structopt(
+        about="compute statistics for all of time",
+        aliases=&["d", "da"],
+    )]
+    All,
+
+    #[structopt(
+        about="compute statistics for a single day",
+        aliases=&["d", "da"],
+    )]
+    Day { words: Vec<String> },
+}
+
+fn compute_stats(history: &[History]) -> Result<()> {
+    let mut commands = HashMap::<String, i64>::new();
+
+    for i in history {
+        *commands.entry(i.command.clone()).or_default() += 1;
+    }
+
+    let most_common_command = commands.iter().max_by(|a, b| a.1.cmp(b.1));
+
+    if most_common_command.is_none() {
+        return Err(eyre!("No commands found"));
+    }
+
+    let table = vec![
+        vec![
+            "Most used command".cell(),
+            most_common_command
+                .unwrap()
+                .0
+                .cell()
+                .justify(Justify::Right),
+        ],
+        vec![
+            "Commands ran".cell(),
+            history.len().to_string().cell().justify(Justify::Right),
+        ],
+        vec![
+            "Unique commands ran".cell(),
+            commands.len().to_string().cell().justify(Justify::Right),
+        ],
+    ]
+    .table()
+    .title(vec![
+        "Statistic".cell().bold(true),
+        "Value".cell().bold(true),
+    ])
+    .bold(true);
+
+    print_stdout(table)?;
+
+    Ok(())
+}
+
+impl Cmd {
+    pub fn run(&self, db: &mut Sqlite) -> Result<()> {
+        match self {
+            Self::Day { words } => {
+                let words = if words.is_empty() {
+                    String::from("yesterday")
+                } else {
+                    words.join(" ")
+                };
+
+                let start = parse_date_string(words.as_str(), Local::now(), Dialect::Us)?;
+                let end = start + Duration::days(1);
+
+                let history = db.range(start.with_timezone(&Utc), end.with_timezone(&Utc))?;
+
+                compute_stats(&history)?;
+
+                Ok(())
+            }
+
+            Self::All => {
+                let history = db.list()?;
+
+                compute_stats(&history)?;
+
+                Ok(())
+            }
+        }
+    }
+}