istat 0.6.0

A lightweight and batteries-included status_command for i3 and sway
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
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
111
112

use std::collections::HashSet;
use std::path::PathBuf;
use std::time::Duration;

use async_trait::async_trait;
use bytesize::ByteSize;
use hex_color::HexColor;
use serde_derive::{Deserialize, Serialize};
use sysinfo::{Disk as SysDisk, DiskExt, SystemExt};

use crate::context::{BarItem, Context, StopAction};
use crate::error::Result;
use crate::i3::{I3Item, I3Markup};
use crate::theme::Theme;
use crate::util::Paginator;

#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct Disk {
    #[serde(with = "crate::human_time")]
    interval: Duration,
    #[serde(default)]
    mounts: HashSet<PathBuf>,
}

struct DiskStats {
    mount_point: String,
    available_bytes: u64,
    total_bytes: u64,
}

impl DiskStats {
    fn from_disk(disk: &SysDisk) -> DiskStats {
        DiskStats {
            mount_point: disk.mount_point().to_string_lossy().into_owned(),
            available_bytes: disk.available_space(),
            total_bytes: disk.total_space(),
        }
    }

    fn get_color(&self, theme: &Theme) -> Option<HexColor> {
        let pct = (self.available_bytes as f64 / self.total_bytes as f64) * 100.0;
        match pct as u32 {
            0..=10 => Some(theme.red),
            11..=20 => Some(theme.orange),
            21..=30 => Some(theme.yellow),
            _ => None,
        }
    }

    fn format(&self, _: &Theme) -> (String, String) {
        (
            format!(
                "󰋊 {} {}",
                self.mount_point,
                ByteSize(self.available_bytes).to_string_as(true)
            ),
            format!("{}", self.mount_point),
        )
    }
}

#[async_trait(?Send)]
impl BarItem for Disk {
    async fn start(&self, mut ctx: Context) -> Result<StopAction> {
        let mut p = Paginator::new();
        loop {
            let stats: Vec<DiskStats> = {
                ctx.state.sys.refresh_disks();
                ctx.state.sys.refresh_disks_list();
                ctx.state
                    .sys
                    .disks()
                    .iter()
                    .filter(|d| {
                        if self.mounts.is_empty() {
                            true
                        } else {
                            self.mounts.contains(d.mount_point())
                        }
                    })
                    .map(DiskStats::from_disk)
                    .collect()
            };
            let len = stats.len();
            if len > 0 {
                p.set_len(len)?;

                let disk = &stats[p.idx()];
                let theme = &ctx.config.theme;
                let (full, short) = disk.format(theme);
                let full = format!("{}{}", full, p.format(theme));

                let mut item = I3Item::new(full).short_text(short).markup(I3Markup::Pango);

                if let Some(fg) = disk.get_color(theme) {
                    item = item.color(fg);
                }

                ctx.update_item(item).await?;
            } else {
                ctx.update_item(I3Item::empty()).await?;
            }

            // cycle through disks
            ctx.delay_with_event_handler(self.interval, |event| {
                p.update(&event);
                async {}
            })
            .await;
        }
    }
}