2019-09-29 05:55:49 +00:00
|
|
|
use byte_unit::{Byte, ByteUnit};
|
|
|
|
use sysinfo::{RefreshKind, SystemExt};
|
|
|
|
|
2020-01-26 22:37:18 +00:00
|
|
|
use super::{Context, Module, RootModuleConfig, Shell};
|
2019-10-20 15:26:04 +00:00
|
|
|
|
|
|
|
use crate::configs::memory_usage::MemoryConfig;
|
2020-07-07 22:45:32 +00:00
|
|
|
use crate::formatter::StringFormatter;
|
2019-10-20 15:26:04 +00:00
|
|
|
|
|
|
|
fn format_kib(n_kib: u64) -> String {
|
|
|
|
let byte = Byte::from_unit(n_kib as f64, ByteUnit::KiB).unwrap_or_else(|_| Byte::from_bytes(0));
|
|
|
|
let mut display_bytes = byte.get_appropriate_unit(true).format(0);
|
|
|
|
display_bytes.retain(|c| c != ' ');
|
|
|
|
display_bytes
|
|
|
|
}
|
|
|
|
|
2020-07-07 22:45:32 +00:00
|
|
|
fn format_pct(pct_number: f64, pct_sign: &str) -> String {
|
|
|
|
format!("{:.0}{}", pct_number, pct_sign)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn format_usage_total(usage: u64, total: u64) -> String {
|
|
|
|
format!("{}/{}", format_kib(usage), format_kib(total))
|
|
|
|
}
|
|
|
|
|
2019-09-29 05:55:49 +00:00
|
|
|
/// Creates a module with system memory usage information
|
|
|
|
pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
|
|
|
|
let mut module = context.new_module("memory_usage");
|
2019-10-20 15:26:04 +00:00
|
|
|
let config = MemoryConfig::try_load(module.config);
|
|
|
|
|
|
|
|
// TODO: Update when v1.0 printing refactor is implemented to only
|
|
|
|
// print escapes in a prompt context.
|
2020-07-07 22:45:32 +00:00
|
|
|
let pct_sign = match context.shell {
|
2020-01-26 22:37:18 +00:00
|
|
|
Shell::Zsh => "%%", // % is an escape in zsh, see PROMPT in `man zshmisc`
|
2019-10-20 15:26:04 +00:00
|
|
|
_ => "%",
|
|
|
|
};
|
2019-09-29 05:55:49 +00:00
|
|
|
|
2020-09-26 07:35:41 +00:00
|
|
|
// As we default to disabled=true, we have to check here after loading our config module,
|
|
|
|
// before it was only checking against whatever is in the config starship.toml
|
2019-10-20 15:26:04 +00:00
|
|
|
if config.disabled {
|
2019-10-05 10:03:48 +00:00
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
2019-12-12 20:32:46 +00:00
|
|
|
let system = sysinfo::System::new_with_specifics(RefreshKind::new().with_memory());
|
2019-09-29 05:55:49 +00:00
|
|
|
let used_memory_kib = system.get_used_memory();
|
|
|
|
let total_memory_kib = system.get_total_memory();
|
2020-07-07 22:45:32 +00:00
|
|
|
let ram_used = (used_memory_kib as f64 / total_memory_kib as f64) * 100.;
|
|
|
|
let ram_pct = format_pct(ram_used, pct_sign);
|
2019-09-29 05:55:49 +00:00
|
|
|
|
2019-10-20 15:26:04 +00:00
|
|
|
let threshold = config.threshold;
|
2020-07-07 22:45:32 +00:00
|
|
|
if ram_used.round() < threshold as f64 {
|
2019-09-29 05:55:49 +00:00
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
2020-07-07 22:45:32 +00:00
|
|
|
let ram = format_usage_total(used_memory_kib, total_memory_kib);
|
2019-10-20 15:26:04 +00:00
|
|
|
let total_swap_kib = system.get_total_swap();
|
2020-07-07 22:45:32 +00:00
|
|
|
let used_swap_kib = system.get_used_swap();
|
|
|
|
let percent_swap_used = (used_swap_kib as f64 / total_swap_kib as f64) * 100.;
|
|
|
|
let swap_pct = format_pct(percent_swap_used, pct_sign);
|
|
|
|
let swap = format_usage_total(used_swap_kib, total_swap_kib);
|
|
|
|
|
|
|
|
let parsed = StringFormatter::new(config.format).and_then(|formatter| {
|
|
|
|
formatter
|
|
|
|
.map_meta(|var, _| match var {
|
|
|
|
"symbol" => Some(config.symbol),
|
|
|
|
_ => None,
|
|
|
|
})
|
|
|
|
.map_style(|variable| match variable {
|
|
|
|
"style" => Some(Ok(config.style)),
|
|
|
|
_ => None,
|
|
|
|
})
|
|
|
|
.map(|variable| match variable {
|
|
|
|
"ram" => Some(Ok(&ram)),
|
|
|
|
"ram_pct" => Some(Ok(&ram_pct)),
|
|
|
|
// swap only shown if there is swap on the system
|
|
|
|
"swap" if total_swap_kib > 0 => Some(Ok(&swap)),
|
|
|
|
"swap_pct" if total_swap_kib > 0 => Some(Ok(&swap_pct)),
|
|
|
|
_ => None,
|
|
|
|
})
|
|
|
|
.parse(None)
|
|
|
|
});
|
|
|
|
|
|
|
|
module.set_segments(match parsed {
|
|
|
|
Ok(segments) => segments,
|
|
|
|
Err(error) => {
|
|
|
|
log::warn!("Error in module `memory_usage`:\n{}", error);
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
});
|
2019-09-29 05:55:49 +00:00
|
|
|
|
|
|
|
Some(module)
|
|
|
|
}
|