Impl parsing /proc/meminfo, create object with str

This commit is contained in:
Stephen Seo 2022-07-09 16:57:49 +09:00
parent ff3d53c64d
commit 0df98bc2a9
3 changed files with 93 additions and 0 deletions

View file

@ -1,3 +1,4 @@
mod proc;
mod swaybar_object;
fn main() {
@ -11,5 +12,8 @@ fn main() {
array.push_object(swaybar_object::SwaybarObject::default());
array.push_object(swaybar_object::SwaybarObject::new());
array.push_object(swaybar_object::SwaybarObject::default());
let meminfo_string = proc::get_meminfo().expect("Should be able to get meminfo");
let meminfo_object = swaybar_object::SwaybarObject::from_string(meminfo_string);
array.push_object(meminfo_object);
println!("{}", array);
}

67
src/proc.rs Normal file
View file

@ -0,0 +1,67 @@
use std::fs::File;
use std::io;
use std::io::prelude::*;
pub fn get_meminfo() -> io::Result<String> {
let mut meminfo_string = String::new();
{
let mut meminfo: File = File::open("/proc/meminfo")?;
meminfo.read_to_string(&mut meminfo_string)?;
}
let mut is_total_giga = false;
let mut total: u32 = 0;
let mut available: u32 = 0;
for line in meminfo_string.lines() {
if line.starts_with("MemTotal:") {
let line_parts = line
.split_whitespace()
.map(|s| s.to_owned())
.collect::<Vec<String>>();
total = line_parts[1]
.parse()
.map_err(|_| io::Error::new(io::ErrorKind::Other, "MemTotal: parse error"))?;
} else if line.starts_with("MemAvailable:") {
let line_parts = line
.split_whitespace()
.map(|s| s.to_owned())
.collect::<Vec<String>>();
available = line_parts[1]
.parse()
.map_err(|_| io::Error::new(io::ErrorKind::Other, "MemAvailable: parse error"))?;
}
}
let mut used = total - available;
let mut is_used_giga = false;
if total == 0 {
Ok("0".into())
} else {
if total > 1024 {
total /= 1024;
is_total_giga = true;
}
if used > 1024 {
used /= 1024;
is_used_giga = true;
}
let mut output = format!("{} ", used);
if is_used_giga {
output.push_str("GiB / ");
} else {
output.push_str("KiB / ");
}
output.push_str(&format!("{} ", total));
if is_total_giga {
output.push_str("GiB");
} else {
output.push_str("KiB");
}
Ok(output)
}
}

View file

@ -89,6 +89,28 @@ impl SwaybarObject {
markup: None,
}
}
pub fn from_string(string: String) -> Self {
Self {
full_text: string,
short_text: None,
color: None,
background: None,
border: None,
border_top: None,
border_bottom: None,
border_left: None,
border_right: None,
min_width: None,
align: None,
name: None,
instance: None,
urgent: None,
separator: None,
separator_block_width: None,
markup: None,
}
}
}
impl Default for SwaybarObject {