modularize

This commit is contained in:
raf 2024-08-03 20:42:48 +03:00
parent 198fd107d7
commit 29e4ccc806
Signed by: NotAShelf
GPG key ID: AF26552424E53993
6 changed files with 223 additions and 203 deletions

6
src/colors.rs Normal file
View file

@ -0,0 +1,6 @@
pub const RESET: &str = "\x1b[0m";
pub const BLUE: &str = "\x1b[34m";
pub const CYAN: &str = "\x1b[36m";
pub const GREEN: &str = "\x1b[32m";
pub const YELLOW: &str = "\x1b[33m";
pub const RED: &str = "\x1b[31m";

13
src/desktop.rs Normal file
View file

@ -0,0 +1,13 @@
use std::{env, io};
pub fn get_desktop_info() -> Result<String, io::Error> {
let desktop_env = env::var("XDG_CURRENT_DESKTOP").unwrap_or_default();
let display_backend = env::var("XDG_SESSION_TYPE").unwrap_or_default();
// Trim "none+" from the start of desktop_env if present
// XXX: This is a workaround for NixOS modules that set XDG_CURRENT_DESKTOP to "none+foo"
// instead of just "foo"
let desktop_env = desktop_env.trim_start_matches("none+");
Ok(format!("{} ({})", desktop_env, display_backend))
}

View file

@ -1,18 +1,16 @@
use color_eyre::{Report, Result};
use libc::statfs as libc_statfs_struct;
use std::env;
use std::ffi::CString;
use std::fs::{self, File};
use std::io::{self, BufRead, Read};
use std::path::Path;
use std::process::Command;
mod colors;
mod desktop;
mod release;
mod system;
mod uptime;
const RESET: &str = "\x1b[0m";
const BLUE: &str = "\x1b[34m";
const CYAN: &str = "\x1b[36m";
const GREEN: &str = "\x1b[32m";
const YELLOW: &str = "\x1b[33m";
const RED: &str = "\x1b[31m";
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::uptime::get_system_uptime;
fn main() -> Result<(), Report> {
color_eyre::install()?;
@ -42,192 +40,3 @@ fn main() -> Result<(), Report> {
Ok(())
}
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();
Ok(format!("{YELLOW}{}{RED}@{GREEN}{}", username, hostname))
}
fn get_disk_usage() -> Result<String, io::Error> {
let path = CString::new("/").expect("CString::new failed");
let mut fs_stat: libc_statfs_struct = unsafe { std::mem::zeroed() };
let result = unsafe { libc::statfs(path.as_ptr(), &mut fs_stat) };
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);
Ok(format!(
"{} GiB / {} GiB ({CYAN}{}%{RESET})",
formatted_used_size, formatted_total_size, formatted_percentage_used
))
}
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);
}
}
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))
}
let (used_memory, total_memory) = parse_memory_info()?;
let percentage_used = (used_memory / total_memory * 100.0).round() as u64;
Ok(format!(
"{:.2} GiB / {:.2} GiB ({CYAN}{}%{RESET})",
used_memory, total_memory, percentage_used
))
}
fn get_desktop_info() -> Result<String, io::Error> {
let desktop_env = env::var("XDG_CURRENT_DESKTOP").unwrap_or_default();
let display_backend = env::var("XDG_SESSION_TYPE").unwrap_or_default();
// Trim "none+" from the start of desktop_env if present
// XXX: This is a workaround for NixOS modules that set XDG_CURRENT_DESKTOP to "none+foo"
// instead of just "foo"
let desktop_env = desktop_env.trim_start_matches("none+");
Ok(format!("{} ({})", desktop_env, display_backend))
}
fn get_os_pretty_name() -> Option<String> {
let os_release_content = fs::read_to_string("/etc/os-release").ok()?;
let os_release_lines: Vec<&str> = os_release_content.lines().collect();
let pretty_name = os_release_lines
.iter()
.find(|line| line.starts_with("PRETTY_NAME="))
.map(|line| {
line.trim_start_matches("PRETTY_NAME=")
.trim_matches('"')
.to_string()
});
pretty_name
}
fn get_system_uptime() -> Result<String, io::Error> {
let path = Path::new("/proc/uptime");
let file = File::open(path)?;
let reader = io::BufReader::new(file);
let line = reader
.lines()
.next()
.ok_or_else(|| io::Error::new(io::ErrorKind::Other, "Failed to read uptime"))??;
let uptime_seconds: f64 = line
.split_whitespace()
.next()
.ok_or_else(|| io::Error::new(io::ErrorKind::Other, "Failed to parse uptime"))?
.parse()
.map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?;
// calculate days, hours, and minutes
let total_minutes = (uptime_seconds / 60.0).round() as u64;
let days = total_minutes / (60 * 24);
let hours = (total_minutes % (60 * 24)) / 60;
let minutes = total_minutes % 60;
let mut parts = Vec::new();
if days > 0 {
parts.push(format!("{} days", days));
}
if hours > 0 || days > 0 {
parts.push(format!("{} hours", hours));
}
if minutes > 0 || hours > 0 || days > 0 {
parts.push(format!("{} minutes", minutes));
}
Ok(parts.join(", "))
}
fn get_system_info() -> Result<String, io::Error> {
// Try to detect OS type as accurately as possible and without depending on uname.
// /etc/os-release should generally imply Linux, and /etc/bsd-release would imply BSD system.
fn detect_os() -> Result<String, io::Error> {
if fs::metadata("/etc/os-release").is_ok() || fs::metadata("/usr/lib/os-release").is_ok() {
Ok("Linux".to_string())
} else if fs::metadata("/etc/rc.conf").is_ok() || fs::metadata("/etc/bsd-release").is_ok() {
Ok("BSD".to_string())
} else {
Ok("Unknown".to_string())
}
}
let system = detect_os()?;
let mut kernel_release = String::new();
fs::File::open("/proc/sys/kernel/osrelease")?.read_to_string(&mut kernel_release)?;
let kernel_release = kernel_release.trim().to_string(); // Remove any trailing newline
let mut cpuinfo = String::new();
fs::File::open("/proc/cpuinfo")?.read_to_string(&mut cpuinfo)?;
let architecture = if let Some(line) = cpuinfo.lines().find(|line| line.contains("flags")) {
if line.contains("lm") {
"x86_64".to_string()
} else {
"unknown".to_string()
}
} else {
"unknown".to_string()
};
let result = format!("{} {} ({})", system, kernel_release, architecture);
Ok(result)
}

54
src/release.rs Normal file
View file

@ -0,0 +1,54 @@
use color_eyre::Result;
use std::fs::{self, read_to_string};
use std::io::{self, Read};
// Try to detect OS type as accurately as possible and without depending on uname.
// /etc/os-release should generally imply Linux, and /etc/bsd-release would imply BSD system.
fn detect_os() -> Result<String, io::Error> {
if fs::metadata("/etc/os-release").is_ok() || fs::metadata("/usr/lib/os-release").is_ok() {
Ok("Linux".to_string())
} else if fs::metadata("/etc/rc.conf").is_ok() || fs::metadata("/etc/bsd-release").is_ok() {
Ok("BSD".to_string())
} else {
Ok("Unknown".to_string())
}
}
fn get_architecture() -> Result<String, io::Error> {
// Read architecture from /proc/sys/kernel/arch
let mut arch = String::new();
fs::File::open("/proc/sys/kernel/arch")?.read_to_string(&mut arch)?;
let arch = arch.trim().to_string();
Ok(arch)
}
pub fn get_system_info() -> Result<String, io::Error> {
let system = detect_os()?;
let mut kernel_release = String::new();
fs::File::open("/proc/sys/kernel/osrelease")?.read_to_string(&mut kernel_release)?;
let kernel_release = kernel_release.trim().to_string();
let mut cpuinfo = String::new();
fs::File::open("/proc/cpuinfo")?.read_to_string(&mut cpuinfo)?;
let architecture = get_architecture()?;
let result = format!("{} {} ({})", system, kernel_release, architecture);
Ok(result)
}
pub fn get_os_pretty_name() -> Option<String> {
let os_release_content = read_to_string("/etc/os-release").ok()?;
let os_release_lines: Vec<&str> = os_release_content.lines().collect();
let pretty_name = os_release_lines
.iter()
.find(|line| line.starts_with("PRETTY_NAME="))
.map(|line| {
line.trim_start_matches("PRETTY_NAME=")
.trim_matches('"')
.to_string()
});
pretty_name
}

96
src/system.rs Normal file
View file

@ -0,0 +1,96 @@
use libc::statfs as libc_statfs_struct;
use std::env;
use std::ffi::CString;
use std::fs::File;
use std::io::{self, BufRead};
use std::path::Path;
use std::process::Command;
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();
Ok(format!("{YELLOW}{}{RED}@{GREEN}{}", username, hostname))
}
pub fn get_disk_usage() -> Result<String, io::Error> {
let path = CString::new("/").expect("CString::new failed");
let mut fs_stat: libc_statfs_struct = unsafe { std::mem::zeroed() };
let result = unsafe { libc::statfs(path.as_ptr(), &mut fs_stat) };
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);
Ok(format!(
"{} GiB / {} GiB ({CYAN}{}%{RESET})",
formatted_used_size, formatted_total_size, formatted_percentage_used
))
}
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);
}
}
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))
}
let (used_memory, total_memory) = parse_memory_info()?;
let percentage_used = (used_memory / total_memory * 100.0).round() as u64;
Ok(format!(
"{:.2} GiB / {:.2} GiB ({CYAN}{}%{RESET})",
used_memory, total_memory, percentage_used
))
}

42
src/uptime.rs Normal file
View file

@ -0,0 +1,42 @@
use std::fs::File;
use std::io::{self, BufRead};
use std::path::Path;
pub fn get_system_uptime() -> Result<String, io::Error> {
let path = Path::new("/proc/uptime");
let file = File::open(path)?;
let reader = io::BufReader::new(file);
let line = reader
.lines()
.next()
.ok_or_else(|| io::Error::new(io::ErrorKind::Other, "Failed to read uptime"))??;
let uptime_seconds: f64 = line
.split_whitespace()
.next()
.ok_or_else(|| io::Error::new(io::ErrorKind::Other, "Failed to parse uptime"))?
.parse()
.map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?;
// calculate days, hours, and minutes
let total_minutes = (uptime_seconds / 60.0).round() as u64;
let days = total_minutes / (60 * 24);
let hours = (total_minutes % (60 * 24)) / 60;
let minutes = total_minutes % 60;
let mut parts = Vec::new();
if days > 0 {
parts.push(format!("{} days", days));
}
if hours > 0 || days > 0 {
parts.push(format!("{} hours", hours));
}
if minutes > 0 || hours > 0 || days > 0 {
parts.push(format!("{} minutes", minutes));
}
Ok(parts.join(", "))
}