From deb7c19093a905a6cc627cca1f608ee177cc605a Mon Sep 17 00:00:00 2001
From: Conrad Ludgate <conradludgate@gmail.com>
Date: Fri, 10 Feb 2023 18:58:02 +0000
Subject: [PATCH] stats (#707)

---
 Cargo.lock                  | 11 ------
 Cargo.toml                  |  1 -
 src/command/client/stats.rs | 76 ++++++++++++++++++++-----------------
 3 files changed, 42 insertions(+), 46 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index eec9bef8263..9c4b3bc0dce 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -82,7 +82,6 @@ dependencies = [
  "chrono",
  "clap",
  "clap_complete",
- "cli-table",
  "crossbeam-channel",
  "crossterm",
  "directories",
@@ -383,16 +382,6 @@ dependencies = [
  "os_str_bytes",
 ]
 
-[[package]]
-name = "cli-table"
-version = "0.4.7"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "adfbb116d9e2c4be7011360d0c0bee565712c11e969c9609b25b619366dc379d"
-dependencies = [
- "termcolor",
- "unicode-width",
-]
-
 [[package]]
 name = "config"
 version = "0.13.2"
diff --git a/Cargo.toml b/Cargo.toml
index 73817d75443..ebff2d56ed3 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -62,7 +62,6 @@ itertools = "0.10.5"
 tokio = { version = "1", features = ["full"] }
 async-trait = "0.1.58"
 interim = { version = "0.1.0", features = ["chrono"] }
-cli-table = { version = "0.4", default-features = false }
 base64 = "0.20.0"
 crossbeam-channel = "0.5.1"
 clap = { version = "4.0.18", features = ["derive"] }
diff --git a/src/command/client/stats.rs b/src/command/client/stats.rs
index f0c774a4749..b663423368d 100644
--- a/src/command/client/stats.rs
+++ b/src/command/client/stats.rs
@@ -2,7 +2,7 @@ use std::collections::HashMap;
 
 use chrono::{prelude::*, Duration};
 use clap::Parser;
-use cli_table::{format::Justify, print_stdout, Cell, Style, Table};
+use crossterm::style::{Color, ResetColor, SetAttribute, SetForegroundColor};
 use eyre::{bail, Result};
 use interim::parse_date_string;
 
@@ -17,46 +17,54 @@ use atuin_client::{
 pub struct Cmd {
     /// compute statistics for the specified period, leave blank for statistics since the beginning
     period: Vec<String>,
-}
 
-fn compute_stats(history: &[History]) -> Result<()> {
-    let mut commands = HashMap::<String, i64>::new();
+    /// How many top commands to list
+    #[arg(long, short, default_value = "10")]
+    count: usize,
+}
 
+fn compute_stats(history: &[History], count: usize) -> Result<()> {
+    let mut commands = HashMap::<&str, usize>::new();
     for i in history {
-        *commands.entry(i.command.clone()).or_default() += 1;
+        *commands
+            .entry(i.command.split_ascii_whitespace().next().unwrap())
+            .or_default() += 1;
     }
-    let most_common_command = commands.iter().max_by(|a, b| a.1.cmp(b.1));
-
-    if most_common_command.is_none() {
+    let unique = commands.len();
+    let mut top = commands.into_iter().collect::<Vec<_>>();
+    top.sort_unstable_by_key(|x| std::cmp::Reverse(x.1));
+    top.truncate(count);
+    if top.is_empty() {
         bail!("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);
+    let max = top.iter().map(|x| x.1).max().unwrap();
+    let num_pad = max.ilog10() as usize + 1;
 
-    print_stdout(table)?;
+    for (command, count) in top {
+        let gray = SetForegroundColor(Color::Grey);
+        let bold = SetAttribute(crossterm::style::Attribute::Bold);
+
+        let in_ten = 10 * count / max;
+        print!("[");
+        print!("{}", SetForegroundColor(Color::Red));
+        for i in 0..in_ten {
+            if i == 2 {
+                print!("{}", SetForegroundColor(Color::Yellow));
+            }
+            if i == 5 {
+                print!("{}", SetForegroundColor(Color::Green));
+            }
+            print!("▮");
+        }
+        for _ in in_ten..10 {
+            print!(" ");
+        }
+
+        println!("{ResetColor}] {gray}{count:num_pad$}{ResetColor} {bold}{command}{ResetColor}");
+    }
+    println!("Total commands:   {}", history.len());
+    println!("Unique commands:  {unique}");
 
     Ok(())
 }
@@ -76,7 +84,7 @@ impl Cmd {
             let end = start + Duration::days(1);
             db.range(start.into(), end.into()).await?
         };
-        compute_stats(&history)?;
+        compute_stats(&history, self.count)?;
         Ok(())
     }
 }