diff options
Diffstat (limited to 'src/command')
| -rw-r--r-- | src/command/import.rs | 15 | ||||
| -rw-r--r-- | src/command/mod.rs | 5 | ||||
| -rw-r--r-- | src/command/stats.rs | 101 |
3 files changed, 111 insertions, 10 deletions
diff --git a/src/command/import.rs b/src/command/import.rs index 5a91b6b7..88108400 100644 --- a/src/command/import.rs +++ b/src/command/import.rs @@ -96,16 +96,11 @@ fn import_zsh(db: &mut Sqlite) -> Result<()> { let buf_size = 100; let mut buf = Vec::<History>::with_capacity(buf_size); - for i in zsh { - match i { - Ok(h) => { - buf.push(h); - } - Err(e) => { - error!("{}", e); - continue; - } - } + for i in zsh + .filter_map(Result::ok) + .filter(|x| !x.command.trim().is_empty()) + { + buf.push(i); if buf.len() == buf_size { db.save_bulk(&buf)?; diff --git a/src/command/mod.rs b/src/command/mod.rs index 2e8d4778..a5dd039e 100644 --- a/src/command/mod.rs +++ b/src/command/mod.rs @@ -7,6 +7,7 @@ use crate::local::database::Sqlite; mod history; mod import; mod server; +mod stats; #[derive(StructOpt)] pub enum AtuinCmd { @@ -22,6 +23,9 @@ pub enum AtuinCmd { #[structopt(about = "start an atuin server")] Server(server::Cmd), + #[structopt(about = "calculate statistics for your history")] + Stats(stats::Cmd), + #[structopt(about = "generates a UUID")] Uuid, } @@ -36,6 +40,7 @@ impl AtuinCmd { Self::History(history) => history.run(db), Self::Import(import) => import.run(db), Self::Server(server) => server.run(), + Self::Stats(stats) => stats.run(db), Self::Uuid => { println!("{}", uuid_v4()); 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(()) + } + } + } +} |
