1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
use clap::Parser;
use eyre::Result;
use interim::parse_date_string;
use time::{Duration, OffsetDateTime, Time};
use crate::atuin_client::database::ClientSqlite;
use crate::atuin_client::{database::current_context, settings::Settings, theme::Theme};
use crate::atuin_history::stats::{compute, pretty_print};
fn parse_ngram_size(s: &str) -> Result<usize, String> {
let value = s
.parse::<usize>()
.map_err(|_| format!("'{s}' is not a valid window size"))?;
if value == 0 {
return Err("ngram window size must be at least 1".to_string());
}
Ok(value)
}
#[derive(Parser, Debug)]
#[command(infer_subcommands = true)]
pub(crate) struct Cmd {
/// Compute statistics for the specified period, leave blank for statistics since the beginning. See [this](https://docs.atuin.sh/reference/stats/) for more details.
period: Vec<String>,
/// How many top commands to list
#[arg(long, short, default_value = "10")]
count: usize,
/// The number of consecutive commands to consider
#[arg(long, short, default_value = "1", value_parser = parse_ngram_size)]
ngram_size: usize,
}
impl Cmd {
pub(crate) async fn run(
&self,
db: &ClientSqlite,
settings: &Settings,
theme: &Theme,
) -> Result<()> {
let context = current_context().await?;
let words = if self.period.is_empty() {
String::from("all")
} else {
self.period.join(" ")
};
let now = OffsetDateTime::now_utc().to_offset(settings.timezone.0);
let last_night = now.replace_time(Time::MIDNIGHT);
let history = if words.as_str() == "all" {
db.list(&[], &context, None, false, false).await?
} else if words.trim() == "today" {
let start = last_night;
let end = start + Duration::days(1);
db.range(start, end).await?
} else if words.trim() == "month" {
let end = last_night;
let start = end - Duration::days(31);
db.range(start, end).await?
} else if words.trim() == "week" {
let end = last_night;
let start = end - Duration::days(7);
db.range(start, end).await?
} else if words.trim() == "year" {
let end = last_night;
let start = end - Duration::days(365);
db.range(start, end).await?
} else {
let start = parse_date_string(&words, now, settings.dialect.into())?;
let end = start + Duration::days(1);
db.range(start, end).await?
};
let stats = compute(settings, &history, self.count, self.ngram_size);
if let Some(stats) = stats {
pretty_print(stats, self.ngram_size, theme);
}
Ok(())
}
}
|
