NotAShelf/microfetch
1
1
Fork 0
mirror of https://github.com/NotAShelf/microfetch.git synced 2025-10-04 04:33:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use nix crate instead of libc

Browse source 
Provides a higher level and lighter alternative.
This commit is contained in:
raf 2024-08-03 20:56:02 +03:00
commit aae5f71004
Signed by: NotAShelf
GPG key ID: AF26552424E53993
4 changed files with 48 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/main.rs
View file

@ -9,7 +9,7 @@ use color_eyre::{Report, Result};
use crate::colors::{BLUE, CYAN, RESET};
use crate::desktop::get_desktop_info;
use crate::release::{get_os_pretty_name, get_system_info};
use crate::system::{get_disk_usage, get_memory_usage, get_username_and_hostname};
use crate::system::{get_memory_usage, get_root_disk_usage, get_username_and_hostname};
use crate::uptime::get_system_uptime;
fn main() -> Result<(), Report> {
@ -21,19 +21,19 @@ fn main() -> Result<(), Report> {
let uptime = get_system_uptime().expect("Failed to get uptime");
let window_manager = get_desktop_info().expect("Failed to get desktop info");
let memory_usage = get_memory_usage().expect("Failed to get memory usage");
let storage = get_disk_usage().expect("Failed to get storage info");
let storage = get_root_disk_usage().expect("Failed to get storage info");
// Construct the ASCII art with dynamic OS name
println!(
"
{CYAN} ▗▄ {BLUE}▗▄ ▄▖ {} ~{RESET}
{CYAN} ▄▄🬸█▄▄▄{BLUE}🬸█▛ {CYAN}▃ {CYAN} {BLUE}System{RESET} {}
{BLUE} ▟▛ ▜{CYAN}▃▟🬕 {CYAN} {BLUE}Kernel{RESET} {}
{BLUE}🬋🬋🬫█ {CYAN}█🬛🬋🬋 {CYAN} {BLUE}Uptime{RESET} {}
{BLUE} 🬷▛🮃{CYAN}▙ ▟▛ {CYAN} {BLUE}WM{RESET} {}
{BLUE} 🮃 {CYAN}▟█🬴{BLUE}▀▀▀█🬴▀▀ {CYAN}󰍛 {BLUE}Memory{RESET} {}
{CYAN} ▝▀ ▀▘ {BLUE}▀▘ {CYAN}󱥎 {BLUE}Storage{RESET} {}
{CYAN} ▄▄🬸█▄▄▄{BLUE}🬸█▛ {CYAN}▃ {CYAN} {BLUE}System{RESET} {}
{BLUE} ▟▛ ▜{CYAN}▃▟🬕 {CYAN} {BLUE}Kernel{RESET} {}
{BLUE}🬋🬋🬫█ {CYAN}█🬛🬋🬋 {CYAN} {BLUE}Uptime{RESET} {}
{BLUE} 🬷▛🮃{CYAN}▙ ▟▛ {CYAN} {BLUE}WM{RESET} {}
{BLUE} 🮃 {CYAN}▟█🬴{BLUE}▀▀▀█🬴▀▀ {CYAN}󰍛 {BLUE}Memory{RESET} {}
{CYAN} ▝▀ ▀▘ {BLUE}▀▘ {CYAN}󱥎 {BLUE}Storage (/){RESET} {}
",
user_info, os_name, kernel_version, uptime, window_manager, memory_usage, storage
);

40
src/system.rs
View file

@ -1,6 +1,6 @@
use libc::statfs as libc_statfs_struct;
use nix::sys::statvfs::statvfs;
use std::env;
use std::ffi::CString;
use std::fs::File;
use std::io::{self, BufRead};
use std::path::Path;
@ -16,34 +16,22 @@ pub fn get_username_and_hostname() -> Result<String, io::Error> {
Ok(format!("{YELLOW}{}{RED}@{GREEN}{}", username, hostname))
}
pub fn get_disk_usage() -> Result<String, io::Error> {
let path = CString::new("/").expect("CString::new failed");
pub fn get_root_disk_usage() -> Result<String, Box<dyn std::error::Error>> {
let vfs = statvfs("/")?;
let block_size = vfs.block_size() as u64;
let total_blocks = vfs.blocks();
let available_blocks = vfs.blocks_available();
let mut fs_stat: libc_statfs_struct = unsafe { std::mem::zeroed() };
let result = unsafe { libc::statfs(path.as_ptr(), &mut fs_stat) };
let total_size = block_size * total_blocks;
let used_size = total_size - (block_size * available_blocks);
if result != 0 {
return Err(io::Error::last_os_error());
}
let block_size = fs_stat.f_bsize as u64;
let total_blocks = fs_stat.f_blocks as u64;
let free_blocks = fs_stat.f_bfree as u64;
let total_size_bytes = total_blocks * block_size;
let used_size_bytes = (total_blocks - free_blocks) * block_size;
let total_size_gib = total_size_bytes as f64 / (1024.0 * 1024.0 * 1024.0);
let used_size_gib = used_size_bytes as f64 / (1024.0 * 1024.0 * 1024.0);
let percentage_used = (used_size_bytes as f64 / total_size_bytes as f64) * 100.0;
let formatted_total_size = format!("{:.2}", total_size_gib);
let formatted_used_size = format!("{:.2}", used_size_gib);
let formatted_percentage_used = format!("{:.0}", percentage_used);
let total_size_gib = total_size as f64 / (1024.0 * 1024.0 * 1024.0);
let used_size_gib = used_size as f64 / (1024.0 * 1024.0 * 1024.0);
let usage_percentage = (used_size as f64 / total_size as f64) * 100.0;
Ok(format!(
"{} GiB / {} GiB ({CYAN}{}%{RESET})",
formatted_used_size, formatted_total_size, formatted_percentage_used
"{:.2} GiB / {:.2} GiB ({CYAN}{:.0}%{RESET})",
used_size_gib, total_size_gib, usage_percentage
))
}