NotAShelf/microfetch
1
1
Fork 0
mirror of https://github.com/NotAShelf/microfetch.git synced 2025-12-12 16:03:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

speeding it up

Browse source
This commit is contained in:
raf 2024-08-04 12:44:12 +03:00
commit 81cdc0a281
Signed by: NotAShelf
GPG key ID: AF26552424E53993
5 changed files with 24 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

54
src/system.rs
View file

@ -1,17 +1,15 @@
use nix::sys::statvfs::statvfs;
use nix::sys::sysinfo::SysInfo;
use std::env;
use std::fs::File;
use std::io::{self, BufRead};
use std::path::Path;
use std::process::Command;
use std::io::{self};
use crate::colors::{CYAN, GREEN, RED, RESET, YELLOW};
pub fn get_username_and_hostname() -> Result<String, io::Error> {
let username = env::var("USER").unwrap_or_else(|_| "unknown_user".to_string());
let output = Command::new("hostname").output()?;
let hostname = String::from_utf8_lossy(&output.stdout).trim().to_string();
let hostname = nix::unistd::gethostname()?;
let hostname = hostname.to_string_lossy();
Ok(format!("{YELLOW}{username}{RED}@{GREEN}{hostname}"))
}
@ -34,49 +32,21 @@ pub fn get_root_disk_usage() -> Result<String, io::Error> {
))
}
pub fn get_memory_usage() -> Result<String, io::Error> {
fn parse_memory_info() -> Result<(f64, f64), io::Error> {
let path = Path::new("/proc/meminfo");
let file = File::open(path)?;
let reader = io::BufReader::new(file);
let mut total_memory_kb = 0.0;
let mut available_memory_kb = 0.0;
for line in reader.lines() {
let line = line?;
if line.starts_with("MemTotal:") {
total_memory_kb = line
.split_whitespace()
.nth(1)
.ok_or_else(|| {
io::Error::new(io::ErrorKind::InvalidData, "Failed to parse MemTotal")
})?
.parse::<f64>()
.unwrap_or(0.0);
} else if line.starts_with("MemAvailable:") {
available_memory_kb = line
.split_whitespace()
.nth(1)
.ok_or_else(|| {
io::Error::new(io::ErrorKind::InvalidData, "Failed to parse MemAvailable")
})?
.parse::<f64>()
.unwrap_or(0.0);
}
}
pub fn get_memory_usage(info: SysInfo) -> String {
#[inline(always)]
fn parse_memory_info(info: SysInfo) -> (f64, f64) {
let total_memory_kb = (info.ram_total() / 1024) as f64;
let available_memory_kb = (info.ram_unused() / 1024) as f64;
let total_memory_gb = total_memory_kb / (1024.0 * 1024.0);
let available_memory_gb = available_memory_kb / (1024.0 * 1024.0);
let used_memory_gb = total_memory_gb - available_memory_gb;
Ok((used_memory_gb, total_memory_gb))
(used_memory_gb, total_memory_gb)
}
let (used_memory, total_memory) = parse_memory_info()?;
let (used_memory, total_memory) = parse_memory_info(info);
let percentage_used = (used_memory / total_memory * 100.0).round() as u64;
Ok(format!(
"{used_memory:.2} GiB / {total_memory:.2} GiB ({CYAN}{percentage_used}%{RESET})"
))
format!("{used_memory:.2} GiB / {total_memory:.2} GiB ({CYAN}{percentage_used}%{RESET})")
}