bar-rubble 0.5.2

Quickly generate output for viewing in swaybar or similar
Documentation 
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

use crate::RubbleResult;
use clap::Args;
use glob::glob;
use serde_json::json;
use std::env;
use std::path::Path;

#[derive(Args)]
pub struct MailArgs {
    /// User to check mailcount of
    #[arg(short, long)]
    pub user: Option<String>,
}

impl MailArgs {
    pub fn empty() -> Self {
        Self { user: None }
    }
}

pub fn count(args: &MailArgs) -> RubbleResult {
    let path = if args.user.is_some() {
        format!("/home/{}", args.user.as_ref().unwrap())
    } else {
        format!("/home/{}", whoami::username())
    };

    let home = Path::new(&path);
    let dir_change = env::set_current_dir(home);
    let mut count = 0;
    let r = match dir_change {
        Err(_) => Err("Could not find home dir for user".into()),
        Ok(_) => {
            for entry in glob(".local/share/mail/*/INBOX/new/*").expect("Failed to read glob") {
                match entry {
                    Ok(_path) => count += 1,
                    Err(_) => count = count, //this needs cleaning up
                }
            }
            Ok(json!({ "count": count }))
        }
    };
    RubbleResult::new(r, default_format())
}

crate::avaliable_formattings!("{{count}}", "count");

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

    #[test]
    fn test_default_format() {
        let result = default_format();
        assert_eq!(result, "{{count}}");
    }
}