leenfetch 0.2.4

Fast, minimal, customizable system info tool in Rust (Neofetch alternative)
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

use crate::modules::{
    enums::{DiskDisplay, DiskSubtitle},
    utils::get_bar,
};
use std::process::Command;

pub fn get_disks(
    subtitle_mode: DiskSubtitle,
    display_mode: DiskDisplay,
    _paths: Option<Vec<&str>>,
) -> Option<Vec<(String, String)>> {
    let output = Command::new("wmic")
        .args(["logicaldisk", "get", "DeviceID,Size,FreeSpace"])
        .output()
        .ok()?;

    let stdout = String::from_utf8_lossy(&output.stdout);
    Some(parse_disk_output(&stdout, subtitle_mode, display_mode))
}

fn parse_disk_output(
    stdout: &str,
    subtitle_mode: DiskSubtitle,
    display_mode: DiskDisplay,
) -> Vec<(String, String)> {
    let lines = stdout.lines().skip(1); // skip header
    let mut results = Vec::new();

    for line in lines {
        let parts: Vec<&str> = line.split_whitespace().collect();
        if parts.len() < 3 {
            continue;
        }

        let mount = parts[0]; // "C:"

        let free = match parts[1].parse::<u64>() {
            Ok(v) => v,
            Err(_) => continue,
        };
        let total = match parts[2].parse::<u64>() {
            Ok(v) => v,
            Err(_) => continue,
        };

        if total == 0 {
            continue;
        }

        let used = total - free;
        let perc = ((used as f64 / total as f64) * 100.0).round() as u8;

        let used_str = format!("{:.1}G", used as f64 / 1024.0 / 1024.0 / 1024.0);
        let total_str = format!("{:.1}G", total as f64 / 1024.0 / 1024.0 / 1024.0);
        let usage_display = format!("{} / {}", used_str, total_str);

        let bar = get_bar(perc);
        let final_str = match display_mode {
            DiskDisplay::Info => usage_display.clone(),
            DiskDisplay::Percentage => format!("{}% {}", perc, bar),
            DiskDisplay::InfoBar => format!("{} {}", usage_display, bar),
            DiskDisplay::BarInfo => format!("{} {}", bar, usage_display),
            DiskDisplay::Bar => bar,
        };

        let subtitle = match subtitle_mode {
            DiskSubtitle::Name => mount.to_string(),
            DiskSubtitle::Dir => mount.trim_end_matches(':').to_string(),
            DiskSubtitle::Mount => mount.to_string(),
            DiskSubtitle::None => String::new(),
        };

        let label = if subtitle.is_empty() {
            "Disk".to_string()
        } else {
            format!("Disk ({})", subtitle)
        };

        results.push((label, final_str));
    }

    results
}