use std::env;
use std::fs;
use std::fs::File;
use std::process::Command;
use std::io::Read;
use colored::Colorize;
use libmacchina::traits::MemoryReadout as _;
use libmacchina::MemoryReadout;
use nixinfo::uptime;
fn main() {
let args: Vec<String> = env::args().collect();
let mut formatting = true;
let mut tree = false;
for arg in &args {
if arg == "-h" || arg == "--help" || arg == "--usage" {
return help();
} else if arg == "-nc" || arg == "--no-color" {
formatting = false;
} else if arg == "-t" || arg == "--tree" {
tree = true;
}
}
info(formatting, tree);
}
fn info(formatting: bool, tree: bool) {
let user = match formatting {
false => whoami(),
true => whoami().purple().to_string(),
};
let hostname = match formatting {
false => uname_n(),
true => uname_n().purple().to_string(),
};
let distroascii = match formatting {
false => get_distro_ascii(),
true => get_distro_ascii().blue().bold().to_string(),
};
let kernel = match formatting {
false => uname_r(),
true => uname_r().purple().to_string(),
};
let desktop = match formatting {
false => get_wm(),
true => get_wm().purple().to_string(),
};
let uptime = match formatting {
false => match uptime() {
Ok(string_from_uptime) => string_from_uptime,
Err(error) => {
eprintln!("Error from uptime(): {}", error);
"".to_string()
}
},
true => match uptime() {
Ok(string_from_uptime) => string_from_uptime.purple().to_string(),
Err(error) => {
eprintln!("Error from uptime(): {}", error);
"".to_string()
}
},
};
let shell = match formatting {
false => shell_name(),
true => shell_name().purple().to_string(),
};
let terminal = match formatting {
false => get_terminal(),
true => get_terminal().purple().to_string(),
};
let memory = match formatting {
false => get_mem(),
true => get_mem().purple().to_string(),
};
let os = match formatting {
false => uname_s(),
true => uname_s().purple().to_string(),
};
if tree {
println!("{}\n", distroascii);
println!(" {} ~ {}", "OS", os);
println!("┠ {} ~ {}", "hostname", hostname);
println!("┠ {} ~ {}", "kernel", kernel);
println!("┠ {} ~ {}", "memory", memory);
println!("╰ {} ~ {}\n", "uptime", uptime);
println!(" {} ~ {}", "user", user);
println!("┠ {} ~ {}", "terminal", terminal);
println!("┠ {} ~ {}", "shell", shell);
println!("╰ {} ~ {}", "de", desktop);
} else if !tree {
println!("{}\n", distroascii);
println!(" {} ~ {}@{}", "user", user, hostname);
println!(" {} ~ {}", "OS", os);
println!(" {} ~ {}", "kernel", kernel);
println!(" {} ~ {}", "memory", memory);
println!(" {} ~ {}", "uptime", uptime);
println!(" {} ~ {}", "terminal", terminal);
println!(" {} ~ {}", "shell", shell);
println!(" {} ~ {}", "de", desktop);
}
}
fn help() {
println!("{}", "Rsftch".bold());
println!("A lightning fast hardware fetch written in rust,");
println!("{}", "Written by charklie.".italic());
println!("\nUsage: rsftch [OPTION...]\n");
println!(" -h, --help Bring up this menu");
println!(" -o, --override Override distribution, changes ASCII. (not implemented yet)");
println!(" -nc, --no-color Removes all colors and formatting.");
println!(" -t, --tree Enables tree mode.");
}
fn whoami() -> String {
let output = Command::new("whoami").output().expect("whoami failed");
String::from_utf8_lossy(&output.stdout).trim().to_string()
}
fn get_os_release_pretty_name(opt: char) -> Option<String> {
if opt == 'i' {
let output = Command::new("cat").arg("/etc/lsb-release").output().ok()?;
let output_str = String::from_utf8_lossy(&output.stdout);
let lines = output_str.lines();
for line in lines {
if line.starts_with("DISTRIB_ID=") {
let parts = line.splitn(2, '=').collect::<Vec<_>>();
if parts.len() == 2 {
return Some(parts[1].trim().trim_matches('\"').to_owned());
}
}
}
return None;
} else if opt == 'p' {
let contents = fs::read_to_string("/etc/lsb-release").ok()?;
let lines = contents.lines();
for line in lines {
if line.starts_with("DISTRIB_DESCRIPTION=") {
let parts = line.splitn(2, '=').collect::<Vec<_>>();
if parts.len() == 2 {
return Some(parts[1].trim().trim_matches('\"').to_owned());
}
}
}
return None;
}
return None;
}
fn format_bytes(kbytes: u64) -> String {
const MIB: u64 = 1048576;
format!("{:.2} GiB", kbytes as f64 / MIB as f64)
}
fn get_wm() -> String {
if env::var("DISPLAY").is_err() {
return String::new();
}
for env_var in &["XDG_SESSION_DESKTOP", "XDG_CURRENT_DESKTOP", "DESKTOP_SESSION"] {
if let Ok(de) = env::var(env_var) {
return de;
}
}
let path = format!("{}/.xinitrc", env::var("HOME").unwrap_or_default());
if let Ok(mut file) = File::open(&path) {
let mut buf = String::new();
if file.read_to_string(&mut buf).is_ok() {
if let Some(last_line) = buf.lines().last() {
let last_word = last_line.split(' ').last().unwrap_or("");
return last_word.to_string();
}
}
}
String::new()
}
fn get_mem() -> String {
let mem_readout = MemoryReadout::new();
let total_mem = mem_readout.total().unwrap_or(0);
let used_mem = mem_readout.used().unwrap_or(0);
let total_mem_str = format_bytes(total_mem);
let used_mem_str = format_bytes(used_mem);
format!("{} / {}", used_mem_str, total_mem_str)
}
fn uname_r() -> String {
let output = Command::new("uname")
.arg("-r")
.output()
.expect("uname failed -r");
String::from_utf8_lossy(&output.stdout).trim().to_string()
}
fn uname_s() -> String {
let output = Command::new("uname")
.arg("-s")
.output()
.expect("uname failed -s");
String::from_utf8_lossy(&output.stdout).trim().to_string()
}
fn uname_n() -> String {
let output = Command::new("uname")
.arg("-n")
.output()
.expect("uname failed -n");
String::from_utf8_lossy(&output.stdout).trim().to_string()
}
fn shell_name() -> String {
let shell = env::var("SHELL").expect("SHELL not set");
let parts: Vec<&str> = shell.split('/').collect();
parts.last().unwrap().to_string()
}
fn get_terminal() -> String {
let term = env::var("TERM").unwrap_or("".to_string());
return term;
}
fn get_distro_ascii() -> String {
if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("arch")
{
return " ___ __ \n / _ | ____ ____ / / \n / __ | / __// __/ / _ \\\n/_/ |_|/_/ \\__/ /_//_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("debian")
{
return " ___ __ _ \n / _ \\___ / / (_)__ ____ \n / // / -_) _ \\/ / _ `/ _ \\\n/____/\\__/_.__/_/\\_,_/_//_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("fedora")
{
return " ____ __ \n / __/__ ___/ /__ _______ _\n / _// -_) _ / _ \\/ __/ _ `/\n/_/ \\__/\\_,_/\\___/_/ \\_,_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("endeavouros")
{
return " ____ __ \n / __/__ ___/ /__ ___ __ _____ __ ______\n / _// _ \\/ _ / -_) _ `/ |/ / _ \\/ // / __/\n/___/_//_/\\_,_/\\__/\\_,_/|___/\\___/\\_,_/_/ ".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("void")
{
return " _ __ _ __\n | | / /__ (_)__/ /\n | |/ / _ \\/ / _ / \n |___/\\___/_/\\_,_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("ubuntu")
{
return " __ ____ __ \n / / / / / __ _____ / /___ __\n/ /_/ / _ \\/ // / _ \\/ __/ // /\n\\____/_.__/\\_,_/_//_/\\__/\\_,_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("suse")
{
return " ____ ____ \n / __ \\___ ___ ___ / __/_ _____ ___ \n/ /_/ / _ \\/ -_) _ \\_\\ \\/ // (_-</ -_)\n\\____/ .__/\\__/_//_/___/\\_,_/___/\\__/ \n /_/ ".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("raspbian")
{
return " ___ __ _ \n / _ \\___ ____ ___ / / (_)__ ____ \n / , _/ _ `(_-</ _ \\/ _ \\/ / _ `/ _ \\\n/_/|_|\\_,_/___/ .__/_.__/_/\\_,_/_//_/\n /_/ ".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("mint")
{
return " __ ____ __ \n / |/ (_)__ / /_\n / /|_/ / / _ \\/ __/\n/_/ /_/_/_//_/\\__/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("mx")
{
return " __ ____ __ __ _ \n / |/ / |/_/ / / (_)__ __ ____ __\n / /|_/ /> < / /__/ / _ \\/ // /\\ \\ /\n/_/ /_/_/|_| /____/_/_//_/\\_,_//_\\_\\ ".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("gentoo")
{
return " _____ __ \n / ___/__ ___ / /____ ___ \n/ (_ / -_) _ \\/ __/ _ \\/ _ \\\n\\___/\\__/_//_/\\__/\\___/\\___/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("slackware")
{
return " ______ __ \n / __/ /__ _____/ /___ _____ ________ \n _\\ \\/ / _ `/ __/ '_/ |/|/ / _ `/ __/ -_)\n/___/_/\\_,_/\\__/_/\\_\\|__,__/\\_,_/_/ \\__/ ".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("uwuntu")
{
return " __ __ __ __ __ \n / / / / __/ / / /__ / /___ __\n/ /_/ / |/|/ / /_/ / _ \\/ __/ // /\n\\____/|__,__/\\____/_//_/\\__/\\_,_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("nix")
{
return " _ ___ ____ ____\n / |/ (_)_ __/ __ \\/ __/\n / / /\\ \\ / /_/ /\\ \\ \n/_/|_/_//_\\_\\\\____/___/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("vanilla")
{
return " _ __ _ ____ \n | | / /__ ____ (_) / /__ _\n | |/ / _ `/ _ \\n/ / / / _ `/\n |___/\\_,_/_//_/_/_/_/\\_,_/".to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("kali")
{
return " __ __ ___ \n / //_/__ _/ (_) \n / ,< / _ `/ / / \n/_/|_|\\_,_/_/_/"
.to_string();
} else if get_os_release_pretty_name('i')
.unwrap_or("".to_string())
.to_ascii_lowercase()
.contains("cachy")
{
return " _____ __ \n / ___/__ _____/ / __ __\n/ /__/ _ `/ __/ _ \\/ // /\n\\___/\\_,_/\\__/_//_/\\_, / \n /___/".to_string();
}
if uname_s().to_ascii_lowercase().contains("netbsd") {
return " _ __ __ ___ _______ \n / |/ /__ / /_/ _ )/ __/ _ \\\n / / -_) __/ _ |\\ \\/ // /\n/_/|_/\\__/\\__/____/___/____/ ".to_string();
} else if uname_s().to_ascii_lowercase().contains("freebsd") {
return " ___ ___ _______ \n / _/______ ___ / _ )/ __/ _ \\\n / _/ __/ -_) -_) _ |\\ \\/ // /\n/_//_/ \\__/\\__/____/___/____/".to_string();
} else {
return " ___ __ ____ __ __ \n / _ \\__ _____ / /_ / __/__ / /_____/ / \n / , _/ // (_-</ __/ / _// -_) __/ __/ _ \\\n/_/|_|\\_,_/___/\\__/ /_/ \\__/\\__/\\__/_//_/".to_string();
}
}