atuin 18.24.1

atuin - magical shell history
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
88
89
90
91
92
93
94
95
96
97
98
99

use atuin_client::database::{Sqlite, current_context};
use atuin_client::settings::{FilterMode, Settings};
use atuin_client::theme::Theme;
use atuin_history::stats::{compute, pretty_print};
use clap::Parser;
use eyre::Result;
use interim::parse_date_string;
use time::{Duration, OffsetDateTime, Time};
use tracing::instrument;

use crate::i18n::fl;

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 struct Cmd {
    #[arg(
        help = fl!("arg-stats-period"),
        long_help = fl!("arg-stats-period", "long", url = atuin_common::docs::url("reference/stats/"))
    )]
    period: Vec<String>,

    #[arg(long, short, default_value = "10", help = fl!("arg-stats-count"))]
    count: usize,

    #[arg(
        long,
        short,
        default_value = "1",
        value_parser = parse_ngram_size,
        help = fl!("arg-stats-ngram-size")
    )]
    ngram_size: usize,

    #[arg(long = "filter-mode", help = fl!("arg-stats-filter-mode"))]
    filter_mode: Option<FilterMode>,
}

impl Cmd {
    #[instrument(level = "trace", skip_all, err)]
    pub async fn run(&self, db: &Sqlite, settings: &Settings, theme: &Theme) -> Result<()> {
        let context = current_context(settings).await?;
        let words = if self.period.is_empty() {
            String::from("all")
        } else {
            self.period.join(" ")
        };

        // A single filter mode, or none. `list` takes an iterator of modes so it can
        // OR several, but stats only ever scopes to one at a time.
        let filter = self.filter_mode.map(|f| vec![f]).unwrap_or_default();

        let now = OffsetDateTime::now_utc().to_offset(settings.timezone.0);
        let last_night = now.replace_time(Time::MIDNIGHT);

        let range = if words.as_str() == "all" {
            None
        } else if words.trim() == "today" {
            let start = last_night;
            let end = start + Duration::days(1);
            Some((start, end))
        } else if words.trim() == "month" {
            let end = last_night;
            let start = end - Duration::days(31);
            Some((start, end))
        } else if words.trim() == "week" {
            let end = last_night;
            let start = end - Duration::days(7);
            Some((start, end))
        } else if words.trim() == "year" {
            let end = last_night;
            let start = end - Duration::days(365);
            Some((start, end))
        } else {
            let start = parse_date_string(&words, now, settings.dialect.into())?;
            let end = start + Duration::days(1);
            Some((start, end))
        };

        let history = db.list(filter, &context, None, false, false, range).await?;

        let stats = compute(settings, &history, self.count, self.ngram_size);

        if let Some(stats) = stats {
            pretty_print(stats, self.ngram_size, theme);
        }

        Ok(())
    }
}