atuin 18.22.0

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
100
101
102
103
104
105
106
107
108
109
110

use std::io::IsTerminal;

use atuin_client::record::sync::SyncError;
use colored::Colorize;
use crossterm::terminal;
use easy_cast::Conv;

/// Print a prominent error to stderr. Colored and box-bordered when stderr is
/// a TTY, plain "Error: ..." header otherwise. The description is word-wrapped
/// to the terminal width (capped at 100 columns) so the message stays readable.
pub fn print_error(title: &str, description: &str) {
    let is_tty = std::io::stderr().is_terminal();
    let width = if is_tty {
        terminal::size().map_or(80, |(w, _)| usize::conv(w))
    } else {
        80
    }
    .min(100);

    eprintln!();
    if is_tty {
        let bar = "━".repeat(width).red().bold().to_string();
        eprintln!("{bar}");
        eprintln!("  {} {}", "✗".red().bold(), title.red().bold());
        eprintln!("{bar}");
    } else {
        eprintln!("Error: {title}");
        eprintln!("{}", "-".repeat(width));
    }
    eprintln!();

    for line in wrap_text(description, width.saturating_sub(2)) {
        eprintln!("  {line}");
    }
    eprintln!();
}

/// Convert a `SyncError` into an `eyre::Report`, exiting on `WrongKey` after
/// painting the prominent banner.
pub fn format_sync_error(e: SyncError) -> eyre::Report {
    if matches!(e, SyncError::WrongKey) {
        print_error(
            "Wrong encryption key",
            "Your local encryption key cannot decrypt the data on the server. This usually means \
             another machine wrote records with a different key.\n\nTo fix this, find the correct \
             key by running `atuin key` on a machine that already syncs successfully, then run \
             `atuin store rekey <key>` here.",
        );
        std::process::exit(1);
    }
    e.into()
}

fn wrap_text(text: &str, width: usize) -> Vec<String> {
    let mut out = Vec::new();
    for paragraph in text.split('\n') {
        let mut line = String::new();
        let mut line_len = 0;
        for word in paragraph.split_whitespace() {
            let word_len = word.chars().count();
            if !line.is_empty() && line_len + 1 + word_len > width {
                out.push(std::mem::take(&mut line));
                line_len = 0;
            }
            if !line.is_empty() {
                line.push(' ');
                line_len += 1;
            }
            line.push_str(word);
            line_len += word_len;
        }
        // Push every paragraph's final line (even empty) so `\n\n` in the
        // input becomes a blank line in the output.
        out.push(line);
    }
    while out.first().is_some_and(String::is_empty) {
        out.remove(0);
    }
    while out.last().is_some_and(String::is_empty) {
        out.pop();
    }
    out
}

#[cfg(test)]
mod tests {
    use rstest::rstest;

    use super::wrap_text;

    #[rstest]
    #[case::width_20(20)]
    fn wraps_long_text(#[case] width: usize) {
        let text = "the quick brown fox jumps over the lazy dog";
        let lines = wrap_text(text, width);
        for line in &lines {
            assert!(line.chars().count() <= width, "line too long: {line:?}");
        }
        assert_eq!(lines.join(" "), text);
    }

    #[rstest]
    #[case::preserves_explicit_newlines("first line\nsecond line", 80, vec!["first line", "second line"])]
    #[case::word_longer_than_width("short superlongword more", 5, vec!["short", "superlongword", "more"])]
    #[case::blank_lines_between_paragraphs("first paragraph\n\nsecond paragraph", 80, vec!["first paragraph", "", "second paragraph"])]
    #[case::trims_leading_and_trailing_blank_lines("\n\nbody\n\n", 80, vec!["body"])]
    fn wraps_text(#[case] input: &str, #[case] width: usize, #[case] expected: Vec<&str>) {
        assert_eq!(wrap_text(input, width), expected);
    }
}