Init

src/main.rs

@@ -0,0 +1,81 @@
+use std::fs;
+use std::env;
+use sys_info;
+
+const LOW: &str = "#[fg=colour186]";
+const MID: &str = "#[fg=colour208]";
+const HIGH: &str = "#[fg=colour160]";
+const END: &str = "#[fg=colour7]";
+
+fn read_file(file_path: &str) -> String {
+    fs::read_to_string(file_path)
+        .expect("Cant read file.")
+}
+
+fn mem() {
+    let memory;
+    match sys_info::mem_info() {
+        Err(w) => panic!("{:?}", w),
+        Ok(mem_data) => memory = mem_data,
+    }
+    let mem_color: &str;
+    if memory.free+memory.cached <= memory.total / 100 {
+        mem_color = HIGH;
+    } else if memory.free+memory.cached <= memory.total / 30{
+        mem_color = MID;
+    } else {
+        mem_color = LOW;
+    }
+    let mem_total = memory.total as f32;
+    println!("MEM: {:.1}GiB avail:{}{}MiB{}",
+        mem_total/1024./1024.,
+        mem_color,
+        memory.free/1024+memory.cached/1024,
+        END);
+}
+
+fn cpu_load() {
+    let load = read_file("/proc/loadavg");
+    let load_data = load.split_whitespace().collect::<Vec<&str>>();
+    let _cpu_count = read_file("/proc/cpuinfo");
+    let cpu_count = _cpu_count.matches("model name").count();
+    let one: f32 = load_data[0].parse().unwrap();
+    let five: f32 = load_data[1].parse().unwrap();
+    let fiveteen: f32 = load_data[2].parse().unwrap();
+    let load_color: &str;
+    if one + five + fiveteen > (cpu_count * 3) as f32 {
+        load_color = HIGH;
+    } else if one + five + fiveteen > cpu_count as f32{
+        load_color = MID;
+    } else {
+        load_color = LOW;
+    }
+
+    println!("CPU: {}|{}{:.2} {:.2} {:.2}{}",
+        cpu_count,
+        load_color,
+        one,
+        five,
+        fiveteen,
+        END);
+}
+
+
+fn main() {
+  let args: Vec<String> = env::args().collect();
+  match args.len() {
+        1 => {
+            panic!("Available commands -m, -c");
+        },
+        2 => {
+            match args[1].as_ref() {
+                "-c" => cpu_load(),
+                "-m" => mem(),
+                _ => panic!("Available commands -m, -c"),
+            }
+        },
+        _ => {
+            panic!("Available commands -m, -c");
+        }
+  }
+}