statusline 0.23.0

Simple and fast bash PS1 line with useful features
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

use crate::{Block, Color, Environment, Icon, IconMode, Pretty, Style, WithStyle as _};
use std::path::PathBuf;

pub struct UnseenMail {
    count: usize,
}

super::register_block!(UnseenMail);

impl Block for UnseenMail {
    fn new(environ: &Environment) -> Option<Self> {
        let maildir_path =
            std::env::var("MAIL").unwrap_or_else(|_| format!("/var/spool/mail/{}", environ.user));
        let maildir = PathBuf::from(maildir_path);
        let unseen_count = ignore_errors(maildir.join("new").read_dir()).count();
        let unread_count = ignore_errors(maildir.join("cur").read_dir())
            .filter(|entry| entry.file_name().as_encoded_bytes().ends_with(b":2,"))
            .count();
        let count = unseen_count + unread_count;
        (count > 0).then_some(UnseenMail { count })
    }
}

fn ignore_errors<T, E>(
    res: Result<impl Iterator<Item = Result<T, E>>, E>,
) -> impl Iterator<Item = T> {
    res.into_iter().flatten().flatten()
}

impl Pretty for UnseenMail {
    fn pretty(&self, f: &mut std::fmt::Formatter<'_>, mode: IconMode) -> std::fmt::Result {
        f.with_style(Color::YELLOW, Style::empty(), |f| {
            write!(f, "[{} {}]", self.icon(mode), self.count)
        })
    }
}

impl Icon for UnseenMail {
    fn icon(&self, mode: IconMode) -> &'static str {
        use IconMode::*;
        match &mode {
            Text => "eml",
            Icons | MinimalIcons => "󰇰",
        }
    }
}