Merge pull request #34 from MustafaSalih1993/dev
orginized the code to seprated mods and deleted the lib file
This commit is contained in:
commit
e38859e0f0
253
src/config.rs
Normal file
253
src/config.rs
Normal file
@ -0,0 +1,253 @@
|
|||||||
|
use crate::types::*;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::Error;
|
||||||
|
use std::io::Read;
|
||||||
|
use yaml_rust::{yaml, YamlLoader};
|
||||||
|
/*
|
||||||
|
this function is responsible to check if the rsblocks.yml file
|
||||||
|
exists to call parse_config to read it OTHERWISE
|
||||||
|
it will call load_defaults to load a hardcoded default configuration
|
||||||
|
|
||||||
|
it will always return a valid configuration inside a Result.
|
||||||
|
*/
|
||||||
|
pub fn load_config() -> Result<Config, Error> {
|
||||||
|
let yml_source = std::env::var("HOME").unwrap() + "/.config/rsblocks/rsblocks.yml";
|
||||||
|
let mut data = String::new();
|
||||||
|
let mut file = match File::open(&yml_source) {
|
||||||
|
Ok(file) => file,
|
||||||
|
Err(_) => {
|
||||||
|
eprintln!("{} file not found, loading defaults!", &yml_source);
|
||||||
|
return Ok(load_defaults());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
file.read_to_string(&mut data)?;
|
||||||
|
let yml_content = &YamlLoader::load_from_str(&data).unwrap()[0];
|
||||||
|
let config = parse_config(yml_content);
|
||||||
|
Ok(config)
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
this is simply returns a hardcoded configuration as default
|
||||||
|
*/
|
||||||
|
fn load_defaults() -> Config {
|
||||||
|
Config {
|
||||||
|
seperator: String::from("|"),
|
||||||
|
time: Time {
|
||||||
|
format: String::from("%T"),
|
||||||
|
icon: String::from(""),
|
||||||
|
delay: 1.0,
|
||||||
|
},
|
||||||
|
memory: Memory {
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: true,
|
||||||
|
delay: 2.0,
|
||||||
|
},
|
||||||
|
disk: Disk {
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 60.0,
|
||||||
|
},
|
||||||
|
volume: Volume {
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 0.17,
|
||||||
|
card: String::from("ALSA"),
|
||||||
|
},
|
||||||
|
weather: Weather {
|
||||||
|
city: String::from(""),
|
||||||
|
format: String::from("+%t"),
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 7200.0, //7200 seconds = 2 hours
|
||||||
|
},
|
||||||
|
battery: Battery {
|
||||||
|
source: String::from("BAT0"),
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 120.0,
|
||||||
|
},
|
||||||
|
cpu_temperature: CpuTemp {
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 120.0,
|
||||||
|
},
|
||||||
|
uptime: Uptime {
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 60.0,
|
||||||
|
},
|
||||||
|
mpd: Mpd {
|
||||||
|
icon: String::from(""),
|
||||||
|
host: String::from("127.0.0.1"),
|
||||||
|
port: String::from("6600"),
|
||||||
|
enabled: false,
|
||||||
|
delay: 15.0,
|
||||||
|
},
|
||||||
|
spotify: Spotify {
|
||||||
|
icon: String::from(""),
|
||||||
|
enabled: false,
|
||||||
|
delay: 15.0,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
it will read and parse the rsblocks.yml file content and return a valid configuration
|
||||||
|
IF some content is missing in the rsblocks.yml file, it will set
|
||||||
|
a default values to that.
|
||||||
|
|
||||||
|
NOTE: (get_or_set) functions job below getting the values from the configuration doc IF
|
||||||
|
a value is not exist in the config it will SET a value givin in the last argument.
|
||||||
|
*/
|
||||||
|
fn parse_config(doc: &yaml::Yaml) -> Config {
|
||||||
|
let seperator = get_or_set_string(doc, "general", "seperator", "|");
|
||||||
|
|
||||||
|
// time values
|
||||||
|
let time_icon = get_or_set_string(doc, "time", "icon", "");
|
||||||
|
let time_format = get_or_set_string(doc, "time", "format", "%T");
|
||||||
|
let time_delay = get_or_set_f64(doc, "time", "delay", 1.0);
|
||||||
|
|
||||||
|
// memory values
|
||||||
|
let mem_icon = get_or_set_string(doc, "memory", "icon", "");
|
||||||
|
let memory_enabled = get_or_set_bool(doc, "memory", "enable");
|
||||||
|
let memory_delay = get_or_set_f64(doc, "memory", "delay", 2.0);
|
||||||
|
|
||||||
|
//disk values
|
||||||
|
let disk_icon = get_or_set_string(doc, "disk", "icon", "");
|
||||||
|
let disk_enabled = get_or_set_bool(doc, "disk", "enable");
|
||||||
|
let disk_delay = get_or_set_f64(doc, "disk", "delay", 120.0);
|
||||||
|
|
||||||
|
// volume values
|
||||||
|
let volume_icon = get_or_set_string(doc, "volume", "icon", "");
|
||||||
|
let volume_enabled = get_or_set_bool(doc, "volume", "enable");
|
||||||
|
let volume_delay = get_or_set_f64(doc, "volume", "delay", 0.17);
|
||||||
|
let volume_card = get_or_set_string(doc, "volume", "card", "ALSA");
|
||||||
|
|
||||||
|
// weather values
|
||||||
|
let weather_icon = get_or_set_string(doc, "weather", "icon", "");
|
||||||
|
let weather_format = get_or_set_string(doc, "weather", "format", "%l:+%t");
|
||||||
|
let weather_city = get_or_set_string(doc, "weather", "city", "");
|
||||||
|
let weather_enabled = get_or_set_bool(doc, "weather", "enable");
|
||||||
|
let weather_delay = get_or_set_f64(doc, "weather", "delay", 7200.0);
|
||||||
|
|
||||||
|
// battery values
|
||||||
|
let battery_icon = get_or_set_string(doc, "battery", "icon", "");
|
||||||
|
let battery_enabled = get_or_set_bool(doc, "battery", "enable");
|
||||||
|
let battery_source = get_or_set_string(doc, "battery", "source", "BAT0");
|
||||||
|
let battery_delay = get_or_set_f64(doc, "battery", "delay", 120.0);
|
||||||
|
|
||||||
|
// cpu values
|
||||||
|
let cpu_temperature_icon = get_or_set_string(doc, "cpu_temperature", "icon", "");
|
||||||
|
let cpu_temperature_enabled = get_or_set_bool(doc, "cpu_temperature", "enable");
|
||||||
|
let cpu_temperature_delay = get_or_set_f64(doc, "cpu_temperature", "delay", 120.0);
|
||||||
|
|
||||||
|
// uptime values
|
||||||
|
let uptime_icon = get_or_set_string(doc, "uptime", "icon", "");
|
||||||
|
let uptime_enabled = get_or_set_bool(doc, "uptime", "enable");
|
||||||
|
let uptime_delay = get_or_set_f64(doc, "uptime", "delay", 60.0);
|
||||||
|
|
||||||
|
// mpd values
|
||||||
|
let mpd_icon = get_or_set_string(doc, "mpd", "icon", "");
|
||||||
|
let mpd_host = get_or_set_string(doc, "mpd", "host", "127.0.0.1");
|
||||||
|
let mpd_port = get_or_set_string(doc, "mpd", "port", "6600");
|
||||||
|
let mpd_enabled = get_or_set_bool(doc, "mpd", "enable");
|
||||||
|
let mpd_delay = get_or_set_f64(doc, "mpd", "delay", 15.0);
|
||||||
|
|
||||||
|
//spotify values
|
||||||
|
let spotify_icon = get_or_set_string(doc, "spotify", "icon", "");
|
||||||
|
let spotify_enabled = get_or_set_bool(doc, "spotify", "enable");
|
||||||
|
let spotify_delay = get_or_set_f64(doc, "spotify", "delay", 10.0);
|
||||||
|
|
||||||
|
Config {
|
||||||
|
seperator,
|
||||||
|
time: Time {
|
||||||
|
format: time_format,
|
||||||
|
icon: time_icon,
|
||||||
|
delay: time_delay,
|
||||||
|
},
|
||||||
|
memory: Memory {
|
||||||
|
icon: mem_icon,
|
||||||
|
enabled: memory_enabled,
|
||||||
|
delay: memory_delay,
|
||||||
|
},
|
||||||
|
disk: Disk {
|
||||||
|
icon: disk_icon,
|
||||||
|
enabled: disk_enabled,
|
||||||
|
delay: disk_delay,
|
||||||
|
},
|
||||||
|
volume: Volume {
|
||||||
|
icon: volume_icon,
|
||||||
|
enabled: volume_enabled,
|
||||||
|
delay: volume_delay,
|
||||||
|
card: volume_card,
|
||||||
|
},
|
||||||
|
weather: Weather {
|
||||||
|
city: weather_city,
|
||||||
|
format: weather_format,
|
||||||
|
icon: weather_icon,
|
||||||
|
enabled: weather_enabled,
|
||||||
|
delay: weather_delay,
|
||||||
|
},
|
||||||
|
battery: Battery {
|
||||||
|
source: battery_source,
|
||||||
|
icon: battery_icon,
|
||||||
|
enabled: battery_enabled,
|
||||||
|
delay: battery_delay,
|
||||||
|
},
|
||||||
|
cpu_temperature: CpuTemp {
|
||||||
|
icon: cpu_temperature_icon,
|
||||||
|
enabled: cpu_temperature_enabled,
|
||||||
|
delay: cpu_temperature_delay,
|
||||||
|
},
|
||||||
|
uptime: Uptime {
|
||||||
|
icon: uptime_icon,
|
||||||
|
enabled: uptime_enabled,
|
||||||
|
delay: uptime_delay,
|
||||||
|
},
|
||||||
|
mpd: Mpd {
|
||||||
|
icon: mpd_icon,
|
||||||
|
host: mpd_host,
|
||||||
|
port: mpd_port,
|
||||||
|
enabled: mpd_enabled,
|
||||||
|
delay: mpd_delay,
|
||||||
|
},
|
||||||
|
spotify: Spotify {
|
||||||
|
icon: spotify_icon,
|
||||||
|
enabled: spotify_enabled,
|
||||||
|
delay: spotify_delay,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getting a f64 value from rsblocks.yml file or set default (last argument)
|
||||||
|
fn get_or_set_f64(doc: &yaml::Yaml, parent: &str, child: &str, default: f64) -> f64 {
|
||||||
|
let val: f64 = if doc[parent][child].is_badvalue() {
|
||||||
|
default
|
||||||
|
} else {
|
||||||
|
doc[parent][child].as_f64().unwrap()
|
||||||
|
};
|
||||||
|
val
|
||||||
|
}
|
||||||
|
|
||||||
|
// getting a boolean value from rsblocks.yml file or set it false if it does not exist
|
||||||
|
fn get_or_set_bool(doc: &yaml::Yaml, parent: &str, child: &str) -> bool {
|
||||||
|
let val: bool;
|
||||||
|
if doc[parent][child].is_badvalue() {
|
||||||
|
val = false;
|
||||||
|
} else {
|
||||||
|
val = doc[parent][child].as_bool().unwrap()
|
||||||
|
}
|
||||||
|
val
|
||||||
|
}
|
||||||
|
|
||||||
|
// getting a String value from the rsblocks.yml file or set the default(last argument)
|
||||||
|
fn get_or_set_string(doc: &yaml::Yaml, parent: &str, child: &str, default_val: &str) -> String {
|
||||||
|
let val: String;
|
||||||
|
if doc[parent][child].is_badvalue() {
|
||||||
|
val = String::from(default_val)
|
||||||
|
} else {
|
||||||
|
val = String::from(doc[parent][child].as_str().unwrap());
|
||||||
|
}
|
||||||
|
|
||||||
|
val
|
||||||
|
}
|
901
src/lib.rs
901
src/lib.rs
@ -1,901 +0,0 @@
|
|||||||
//! `rsblocks` A multi threaded status bar for dwm window manager.
|
|
||||||
|
|
||||||
/*
|
|
||||||
|
|
||||||
Licence: MIT
|
|
||||||
|
|
||||||
|
|
||||||
Author: Mustafa Salih
|
|
||||||
[MustafaSalih](https://github.com/MustafaSalih1993)
|
|
||||||
|
|
||||||
|
|
||||||
Contributers:
|
|
||||||
|
|
||||||
- AdaShoelace - for her/his contributions.
|
|
||||||
[AdaShoelace](https://github.com/AdaShoelace)
|
|
||||||
|
|
||||||
|
|
||||||
Thanks to:
|
|
||||||
- wttr.in - for using their weather API.
|
|
||||||
[wttr.in](https://github.com/chubin/wttr.in)
|
|
||||||
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
use alsa::mixer::{Mixer, SelemChannelId, SelemId};
|
|
||||||
use breadx::{display::*, window::Window};
|
|
||||||
use chrono::prelude::*;
|
|
||||||
use dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
|
|
||||||
use dbus::{arg, blocking::Connection};
|
|
||||||
use mpd::{Client, Song};
|
|
||||||
use std::fs::File;
|
|
||||||
use std::io::Error;
|
|
||||||
use std::io::Read;
|
|
||||||
use std::sync::mpsc;
|
|
||||||
use std::thread;
|
|
||||||
use std::time::Duration;
|
|
||||||
use yaml_rust::{yaml, YamlLoader};
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub enum ThreadsData {
|
|
||||||
Mpd(String),
|
|
||||||
Sound(String),
|
|
||||||
Disk(String),
|
|
||||||
Memory(String),
|
|
||||||
Time(String),
|
|
||||||
Weather(String),
|
|
||||||
Battery(String),
|
|
||||||
CpuTemp(String),
|
|
||||||
Uptime(String),
|
|
||||||
Spotify(String),
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Config {
|
|
||||||
pub seperator: String,
|
|
||||||
pub time: Time,
|
|
||||||
pub memory: Memory,
|
|
||||||
pub disk: Disk,
|
|
||||||
pub volume: Volume,
|
|
||||||
pub weather: Weather,
|
|
||||||
pub battery: Battery,
|
|
||||||
pub cpu_temperature: CpuTemp,
|
|
||||||
pub uptime: Uptime,
|
|
||||||
pub mpd: Mpd,
|
|
||||||
pub spotify: Spotify,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Time {
|
|
||||||
pub format: String,
|
|
||||||
pub icon: String,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Memory {
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Disk {
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Volume {
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
pub card: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Weather {
|
|
||||||
pub city: String,
|
|
||||||
pub format: String,
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Battery {
|
|
||||||
pub source: String,
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct CpuTemp {
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Uptime {
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Mpd {
|
|
||||||
pub icon: String,
|
|
||||||
pub host: String,
|
|
||||||
pub port: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct Spotify {
|
|
||||||
pub icon: String,
|
|
||||||
pub enabled: bool,
|
|
||||||
pub delay: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct Blocks {
|
|
||||||
disp: Display<name::NameConnection>,
|
|
||||||
root: Window,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Blocks {
|
|
||||||
pub fn new() -> Self {
|
|
||||||
let disp = Display::create(None, None).expect("Failed to create x11 connection");
|
|
||||||
let root = disp.default_screen().root;
|
|
||||||
Self { disp, root }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for Blocks {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self::new()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* TODOS
|
|
||||||
|
|
||||||
TODO 1: Error handling required if rsblocks.yml file is empty.
|
|
||||||
|
|
||||||
TODO 2: This lib file growing and soon it will be annoying to move
|
|
||||||
arround, need fix soon.
|
|
||||||
|
|
||||||
TODO 3: Need a better comments in code, or no one will understand what happens.
|
|
||||||
|
|
||||||
TODO 4: Need a documentation.
|
|
||||||
|
|
||||||
TODO 5: Fix repeated code for threads in `run` function.
|
|
||||||
|
|
||||||
TODO 6: Getting song metadata by a format provided by the user.
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
/*
|
|
||||||
this function is responsible to check if the rsblocks.yml file
|
|
||||||
exists to call parse_config to read it OTHERWISE
|
|
||||||
it will call load_defaults to load a hardcoded default configuration
|
|
||||||
|
|
||||||
it will always return a valid configuration inside a Result.
|
|
||||||
*/
|
|
||||||
pub fn load_config() -> Result<Config, Error> {
|
|
||||||
let yml_source = std::env::var("HOME").unwrap() + "/.config/rsblocks/rsblocks.yml";
|
|
||||||
let mut data = String::new();
|
|
||||||
let mut file = match File::open(&yml_source) {
|
|
||||||
Ok(file) => file,
|
|
||||||
Err(_) => {
|
|
||||||
eprintln!("{} file not found, loading defaults!", &yml_source);
|
|
||||||
return Ok(load_defaults());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
file.read_to_string(&mut data)?;
|
|
||||||
let yml_content = &YamlLoader::load_from_str(&data).unwrap()[0];
|
|
||||||
let config = parse_config(yml_content);
|
|
||||||
Ok(config)
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
this is simply returns a hardcoded configuration as default
|
|
||||||
*/
|
|
||||||
fn load_defaults() -> Config {
|
|
||||||
Config {
|
|
||||||
seperator: String::from("|"),
|
|
||||||
time: Time {
|
|
||||||
format: String::from("%T"),
|
|
||||||
icon: String::from(""),
|
|
||||||
delay: 1.0,
|
|
||||||
},
|
|
||||||
memory: Memory {
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: true,
|
|
||||||
delay: 2.0,
|
|
||||||
},
|
|
||||||
disk: Disk {
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 60.0,
|
|
||||||
},
|
|
||||||
volume: Volume {
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 0.17,
|
|
||||||
card: String::from("ALSA"),
|
|
||||||
},
|
|
||||||
weather: Weather {
|
|
||||||
city: String::from(""),
|
|
||||||
format: String::from("+%t"),
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 7200.0, //7200 seconds = 2 hours
|
|
||||||
},
|
|
||||||
battery: Battery {
|
|
||||||
source: String::from("BAT0"),
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 120.0,
|
|
||||||
},
|
|
||||||
cpu_temperature: CpuTemp {
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 120.0,
|
|
||||||
},
|
|
||||||
uptime: Uptime {
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 60.0,
|
|
||||||
},
|
|
||||||
mpd: Mpd {
|
|
||||||
icon: String::from(""),
|
|
||||||
host: String::from("127.0.0.1"),
|
|
||||||
port: String::from("6600"),
|
|
||||||
enabled: false,
|
|
||||||
delay: 15.0,
|
|
||||||
},
|
|
||||||
spotify: Spotify {
|
|
||||||
icon: String::from(""),
|
|
||||||
enabled: false,
|
|
||||||
delay: 15.0,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
it will read and parse the rsblocks.yml file content and return a valid configuration
|
|
||||||
IF some content is missing in the rsblocks.yml file, it will set
|
|
||||||
a default values to that.
|
|
||||||
|
|
||||||
NOTE: (get_or_set) functions job below getting the values from the configuration doc IF
|
|
||||||
a value is not exist in the config it will SET a value givin in the last argument.
|
|
||||||
*/
|
|
||||||
fn parse_config(doc: &yaml::Yaml) -> Config {
|
|
||||||
let seperator = get_or_set_string(doc, "general", "seperator", "|");
|
|
||||||
|
|
||||||
// time values
|
|
||||||
let time_icon = get_or_set_string(doc, "time", "icon", "");
|
|
||||||
let time_format = get_or_set_string(doc, "time", "format", "%T");
|
|
||||||
let time_delay = get_or_set_f64(doc, "time", "delay", 1.0);
|
|
||||||
|
|
||||||
// memory values
|
|
||||||
let mem_icon = get_or_set_string(doc, "memory", "icon", "");
|
|
||||||
let memory_enabled = get_or_set_bool(doc, "memory", "enable");
|
|
||||||
let memory_delay = get_or_set_f64(doc, "memory", "delay", 2.0);
|
|
||||||
|
|
||||||
//disk values
|
|
||||||
let disk_icon = get_or_set_string(doc, "disk", "icon", "");
|
|
||||||
let disk_enabled = get_or_set_bool(doc, "disk", "enable");
|
|
||||||
let disk_delay = get_or_set_f64(doc, "disk", "delay", 120.0);
|
|
||||||
|
|
||||||
// volume values
|
|
||||||
let volume_icon = get_or_set_string(doc, "volume", "icon", "");
|
|
||||||
let volume_enabled = get_or_set_bool(doc, "volume", "enable");
|
|
||||||
let volume_delay = get_or_set_f64(doc, "volume", "delay", 0.17);
|
|
||||||
let volume_card = get_or_set_string(doc, "volume", "card", "ALSA");
|
|
||||||
|
|
||||||
// weather values
|
|
||||||
let weather_icon = get_or_set_string(doc, "weather", "icon", "");
|
|
||||||
let weather_format = get_or_set_string(doc, "weather", "format", "%l:+%t");
|
|
||||||
let weather_city = get_or_set_string(doc, "weather", "city", "");
|
|
||||||
let weather_enabled = get_or_set_bool(doc, "weather", "enable");
|
|
||||||
let weather_delay = get_or_set_f64(doc, "weather", "delay", 7200.0);
|
|
||||||
|
|
||||||
// battery values
|
|
||||||
let battery_icon = get_or_set_string(doc, "battery", "icon", "");
|
|
||||||
let battery_enabled = get_or_set_bool(doc, "battery", "enable");
|
|
||||||
let battery_source = get_or_set_string(doc, "battery", "source", "BAT0");
|
|
||||||
let battery_delay = get_or_set_f64(doc, "battery", "delay", 120.0);
|
|
||||||
|
|
||||||
// cpu values
|
|
||||||
let cpu_temperature_icon = get_or_set_string(doc, "cpu_temperature", "icon", "");
|
|
||||||
let cpu_temperature_enabled = get_or_set_bool(doc, "cpu_temperature", "enable");
|
|
||||||
let cpu_temperature_delay = get_or_set_f64(doc, "cpu_temperature", "delay", 120.0);
|
|
||||||
|
|
||||||
// uptime values
|
|
||||||
let uptime_icon = get_or_set_string(doc, "uptime", "icon", "");
|
|
||||||
let uptime_enabled = get_or_set_bool(doc, "uptime", "enable");
|
|
||||||
let uptime_delay = get_or_set_f64(doc, "uptime", "delay", 60.0);
|
|
||||||
|
|
||||||
// mpd values
|
|
||||||
let mpd_icon = get_or_set_string(doc, "mpd", "icon", "");
|
|
||||||
let mpd_host = get_or_set_string(doc, "mpd", "host", "127.0.0.1");
|
|
||||||
let mpd_port = get_or_set_string(doc, "mpd", "port", "6600");
|
|
||||||
let mpd_enabled = get_or_set_bool(doc, "mpd", "enable");
|
|
||||||
let mpd_delay = get_or_set_f64(doc, "mpd", "delay", 15.0);
|
|
||||||
|
|
||||||
//spotify values
|
|
||||||
let spotify_icon = get_or_set_string(doc, "spotify", "icon", "");
|
|
||||||
let spotify_enabled = get_or_set_bool(doc, "spotify", "enable");
|
|
||||||
let spotify_delay = get_or_set_f64(doc, "spotify", "delay", 10.0);
|
|
||||||
|
|
||||||
Config {
|
|
||||||
seperator,
|
|
||||||
time: Time {
|
|
||||||
format: time_format,
|
|
||||||
icon: time_icon,
|
|
||||||
delay: time_delay,
|
|
||||||
},
|
|
||||||
memory: Memory {
|
|
||||||
icon: mem_icon,
|
|
||||||
enabled: memory_enabled,
|
|
||||||
delay: memory_delay,
|
|
||||||
},
|
|
||||||
disk: Disk {
|
|
||||||
icon: disk_icon,
|
|
||||||
enabled: disk_enabled,
|
|
||||||
delay: disk_delay,
|
|
||||||
},
|
|
||||||
volume: Volume {
|
|
||||||
icon: volume_icon,
|
|
||||||
enabled: volume_enabled,
|
|
||||||
delay: volume_delay,
|
|
||||||
card: volume_card,
|
|
||||||
},
|
|
||||||
weather: Weather {
|
|
||||||
city: weather_city,
|
|
||||||
format: weather_format,
|
|
||||||
icon: weather_icon,
|
|
||||||
enabled: weather_enabled,
|
|
||||||
delay: weather_delay,
|
|
||||||
},
|
|
||||||
battery: Battery {
|
|
||||||
source: battery_source,
|
|
||||||
icon: battery_icon,
|
|
||||||
enabled: battery_enabled,
|
|
||||||
delay: battery_delay,
|
|
||||||
},
|
|
||||||
cpu_temperature: CpuTemp {
|
|
||||||
icon: cpu_temperature_icon,
|
|
||||||
enabled: cpu_temperature_enabled,
|
|
||||||
delay: cpu_temperature_delay,
|
|
||||||
},
|
|
||||||
uptime: Uptime {
|
|
||||||
icon: uptime_icon,
|
|
||||||
enabled: uptime_enabled,
|
|
||||||
delay: uptime_delay,
|
|
||||||
},
|
|
||||||
mpd: Mpd {
|
|
||||||
icon: mpd_icon,
|
|
||||||
host: mpd_host,
|
|
||||||
port: mpd_port,
|
|
||||||
enabled: mpd_enabled,
|
|
||||||
delay: mpd_delay,
|
|
||||||
},
|
|
||||||
spotify: Spotify {
|
|
||||||
icon: spotify_icon,
|
|
||||||
enabled: spotify_enabled,
|
|
||||||
delay: spotify_delay,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting a f64 value from rsblocks.yml file or set default (last argument)
|
|
||||||
fn get_or_set_f64(doc: &yaml::Yaml, parent: &str, child: &str, default: f64) -> f64 {
|
|
||||||
let val: f64 = if doc[parent][child].is_badvalue() {
|
|
||||||
default
|
|
||||||
} else {
|
|
||||||
doc[parent][child].as_f64().unwrap()
|
|
||||||
};
|
|
||||||
val
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting a boolean value from rsblocks.yml file or set it false if it does not exist
|
|
||||||
fn get_or_set_bool(doc: &yaml::Yaml, parent: &str, child: &str) -> bool {
|
|
||||||
let val: bool;
|
|
||||||
if doc[parent][child].is_badvalue() {
|
|
||||||
val = false;
|
|
||||||
} else {
|
|
||||||
val = doc[parent][child].as_bool().unwrap()
|
|
||||||
}
|
|
||||||
val
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting a String value from the rsblocks.yml file or set the default(last argument)
|
|
||||||
fn get_or_set_string(doc: &yaml::Yaml, parent: &str, child: &str, default_val: &str) -> String {
|
|
||||||
let val: String;
|
|
||||||
if doc[parent][child].is_badvalue() {
|
|
||||||
val = String::from(default_val)
|
|
||||||
} else {
|
|
||||||
val = String::from(doc[parent][child].as_str().unwrap());
|
|
||||||
}
|
|
||||||
|
|
||||||
val
|
|
||||||
}
|
|
||||||
|
|
||||||
/*######################## LOADING CONFIGS ENDS HERE ############################
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
// Running the program:
|
|
||||||
|
|
||||||
pub fn run(config: Config, mut blocks: Blocks) {
|
|
||||||
let (tx, rx) = mpsc::channel();
|
|
||||||
|
|
||||||
// spotify thread
|
|
||||||
if config.spotify.enabled {
|
|
||||||
let spotify_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let mut spotify_data = ThreadsData::Spotify(get_spotify(&configcp));
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
spotify_tx.send(spotify_data).unwrap();
|
|
||||||
spotify_data = ThreadsData::Spotify(get_spotify(&configcp));
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.spotify.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// mpd thread
|
|
||||||
if config.mpd.enabled {
|
|
||||||
let mpd_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let mut mpd_data = ThreadsData::Mpd(get_mpd_current(&configcp));
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
mpd_tx.send(mpd_data).unwrap();
|
|
||||||
mpd_data = ThreadsData::Mpd(get_mpd_current(&configcp));
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.mpd.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// volume thread
|
|
||||||
if config.volume.enabled {
|
|
||||||
let volume_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let mut vol_data = ThreadsData::Sound(get_volume(&configcp));
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
volume_tx.send(vol_data).unwrap();
|
|
||||||
vol_data = ThreadsData::Sound(get_volume(&configcp));
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.volume.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Disk thread
|
|
||||||
if config.disk.enabled {
|
|
||||||
let disk_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let mut disk_data = ThreadsData::Disk(get_disk(&configcp));
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
disk_tx.send(disk_data).unwrap();
|
|
||||||
disk_data = ThreadsData::Disk(get_disk(&configcp));
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.disk.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Memory thread
|
|
||||||
if config.memory.enabled {
|
|
||||||
let memory_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let memory_data = get_memory(&configcp).unwrap();
|
|
||||||
let mut memory_data = ThreadsData::Memory(memory_data);
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
memory_tx.send(memory_data).unwrap();
|
|
||||||
memory_data = ThreadsData::Memory(get_memory(&configcp).unwrap());
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.memory.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Weather thread
|
|
||||||
if config.weather.enabled {
|
|
||||||
let weather_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let weather_data = get_weather(&configcp);
|
|
||||||
let mut weather_data = ThreadsData::Weather(weather_data);
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
weather_tx.send(weather_data).unwrap();
|
|
||||||
weather_data = ThreadsData::Weather(get_weather(&configcp));
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.weather.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Battery thread
|
|
||||||
if config.battery.enabled {
|
|
||||||
let battery_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let battery_data = get_battery(&configcp).unwrap();
|
|
||||||
let mut battery_data = ThreadsData::Battery(battery_data);
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
battery_tx.send(battery_data).unwrap();
|
|
||||||
battery_data = ThreadsData::Battery(get_battery(&configcp).unwrap());
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.battery.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Cpu temperature thread
|
|
||||||
if config.cpu_temperature.enabled {
|
|
||||||
let cpu_temp_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let cpu_temp_data = get_cpu_temp(&configcp).unwrap();
|
|
||||||
let mut cpu_temp_data = ThreadsData::CpuTemp(cpu_temp_data);
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
cpu_temp_tx.send(cpu_temp_data).unwrap();
|
|
||||||
cpu_temp_data = ThreadsData::CpuTemp(get_cpu_temp(&configcp).unwrap());
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.cpu_temperature.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Uptime thread
|
|
||||||
if config.uptime.enabled {
|
|
||||||
let uptime_tx = tx.clone();
|
|
||||||
let configcp = config.clone();
|
|
||||||
let uptime_data = get_uptime(&configcp).unwrap();
|
|
||||||
let mut uptime_data = ThreadsData::Uptime(uptime_data);
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
uptime_tx.send(uptime_data).unwrap();
|
|
||||||
uptime_data = ThreadsData::Uptime(get_uptime(&configcp).unwrap());
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.uptime.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Time thread
|
|
||||||
{
|
|
||||||
let time_tx = tx;
|
|
||||||
let configcp = config;
|
|
||||||
let mut time_data = ThreadsData::Time(get_time(&configcp));
|
|
||||||
thread::spawn(move || loop {
|
|
||||||
time_tx.send(time_data).unwrap();
|
|
||||||
time_data = ThreadsData::Time(get_time(&configcp));
|
|
||||||
thread::sleep(Duration::from_secs_f64(configcp.time.delay))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
//Main
|
|
||||||
{
|
|
||||||
// NOTE: order matters to the final format
|
|
||||||
let mut bar: Vec<String> = vec!["".to_string(); 10];
|
|
||||||
//iterating the values recieved from the threads
|
|
||||||
for data in rx {
|
|
||||||
match data {
|
|
||||||
ThreadsData::Spotify(x) => bar[0] = x,
|
|
||||||
ThreadsData::Mpd(x) => bar[1] = x,
|
|
||||||
ThreadsData::Sound(x) => bar[2] = x,
|
|
||||||
ThreadsData::Weather(x) => bar[3] = x,
|
|
||||||
ThreadsData::Disk(x) => bar[4] = x,
|
|
||||||
ThreadsData::Memory(x) => bar[5] = x,
|
|
||||||
ThreadsData::CpuTemp(x) => bar[6] = x,
|
|
||||||
ThreadsData::Battery(x) => bar[7] = x,
|
|
||||||
ThreadsData::Uptime(x) => bar[8] = x,
|
|
||||||
ThreadsData::Time(x) => bar[9] = x,
|
|
||||||
}
|
|
||||||
|
|
||||||
// match ends here
|
|
||||||
update(&bar, &mut blocks);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn update(bar: &[String], blocks: &mut Blocks) {
|
|
||||||
let mut x = String::new();
|
|
||||||
for i in bar.iter() {
|
|
||||||
x.push_str(i.as_str());
|
|
||||||
}
|
|
||||||
|
|
||||||
blocks
|
|
||||||
.root
|
|
||||||
.set_title(&mut blocks.disp, &x)
|
|
||||||
.expect("Failed to set title of root");
|
|
||||||
}
|
|
||||||
|
|
||||||
/*############################ RUNNING THE PROGRAM ENDS HERE ###########################################*/
|
|
||||||
|
|
||||||
/*
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
############################# HELPER FUNCTIONS BELOW ###################################
|
|
||||||
*/
|
|
||||||
|
|
||||||
pub fn get_time(config: &Config) -> String {
|
|
||||||
let now = Local::now();
|
|
||||||
|
|
||||||
format!(
|
|
||||||
" {} {} {}",
|
|
||||||
config.time.icon,
|
|
||||||
now.format(&config.time.format),
|
|
||||||
config.seperator
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// will make a GET request from wttr.in
|
|
||||||
fn get_weather(config: &Config) -> String {
|
|
||||||
let format = if config.weather.format.is_empty() {
|
|
||||||
String::from("%l:+%t")
|
|
||||||
} else {
|
|
||||||
config.weather.format.clone()
|
|
||||||
};
|
|
||||||
|
|
||||||
let url = format!("http://wttr.in/{}?format=\"{}", config.weather.city, format);
|
|
||||||
let err_string = String::from("Error");
|
|
||||||
let res = match minreq::get(url).send() {
|
|
||||||
Ok(resp) => match resp.as_str() {
|
|
||||||
Ok(res_str) => res_str.trim_matches('"').to_string(),
|
|
||||||
Err(_) => err_string,
|
|
||||||
},
|
|
||||||
Err(_) => err_string,
|
|
||||||
};
|
|
||||||
|
|
||||||
format!(" {} {} {}", config.weather.icon, res, config.seperator)
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting disk usage
|
|
||||||
pub fn get_disk(config: &Config) -> String {
|
|
||||||
const GB: u64 = (1024 * 1024) * 1024;
|
|
||||||
let statvfs = nix::sys::statvfs::statvfs("/").unwrap();
|
|
||||||
let mut disk_used = String::new();
|
|
||||||
|
|
||||||
let total = (statvfs.blocks() * statvfs.fragment_size()) / GB;
|
|
||||||
let available = (statvfs.blocks_free() * statvfs.fragment_size()) / GB;
|
|
||||||
let used = total - available;
|
|
||||||
|
|
||||||
disk_used.push_str(&format!("{}G", used));
|
|
||||||
format!(
|
|
||||||
" {} {} {}",
|
|
||||||
config.disk.icon, disk_used, config.seperator
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting volume percentage
|
|
||||||
pub fn get_volume(config: &Config) -> String {
|
|
||||||
let card = if config.volume.card == "PULSE" {
|
|
||||||
"pulse"
|
|
||||||
} else {
|
|
||||||
"default"
|
|
||||||
};
|
|
||||||
|
|
||||||
let mixer = Mixer::new(card, false).expect("Failed to open mixer");
|
|
||||||
let selem_id = SelemId::new("Master", 0);
|
|
||||||
let selem = mixer.find_selem(&selem_id).expect("Couldn't find selem");
|
|
||||||
let selem_chan_id = SelemChannelId::FrontLeft;
|
|
||||||
|
|
||||||
let (min, max) = selem.get_playback_volume_range();
|
|
||||||
let mut raw_volume = selem
|
|
||||||
.get_playback_volume(selem_chan_id)
|
|
||||||
.expect("Failed to get raw_volume");
|
|
||||||
|
|
||||||
let range = max - min;
|
|
||||||
let vol = if range == 0 {
|
|
||||||
0
|
|
||||||
} else {
|
|
||||||
raw_volume -= min;
|
|
||||||
((raw_volume as f64 / range as f64) * 100.) as u64
|
|
||||||
};
|
|
||||||
|
|
||||||
format!(" {} {}% {}", config.volume.icon, vol, config.seperator)
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
mem_used = (mem_total + shmem - mem_free - mem_buffers - mem_cached - mem_srecl
|
|
||||||
thanks for htop's developer on stackoverflow for providing this algorithm to
|
|
||||||
calculate used memory.
|
|
||||||
*/
|
|
||||||
pub fn get_memory(config: &Config) -> Result<String, std::io::Error> {
|
|
||||||
let mut buf = String::new();
|
|
||||||
|
|
||||||
File::open("/proc/meminfo")?.read_to_string(&mut buf)?;
|
|
||||||
|
|
||||||
let mut mem_total: u32 = 0;
|
|
||||||
let mut shmem: u32 = 0;
|
|
||||||
let mut mem_free: u32 = 0;
|
|
||||||
let mut mem_buffers: u32 = 0;
|
|
||||||
let mut mem_cached: u32 = 0;
|
|
||||||
let mut mem_srecl: u32 = 0;
|
|
||||||
|
|
||||||
for line in buf.lines() {
|
|
||||||
if mem_total > 0
|
|
||||||
&& shmem > 0
|
|
||||||
&& mem_free > 0
|
|
||||||
&& mem_buffers > 0
|
|
||||||
&& mem_cached > 0
|
|
||||||
&& mem_srecl > 0
|
|
||||||
{
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if line.starts_with("MemTotal") {
|
|
||||||
assign_val(line, &mut mem_total);
|
|
||||||
}
|
|
||||||
if line.starts_with("SReclaimable") {
|
|
||||||
assign_val(line, &mut mem_srecl)
|
|
||||||
}
|
|
||||||
if line.starts_with("Cached") {
|
|
||||||
assign_val(line, &mut mem_cached)
|
|
||||||
}
|
|
||||||
|
|
||||||
if line.starts_with("Shmem") {
|
|
||||||
assign_val(line, &mut shmem);
|
|
||||||
}
|
|
||||||
|
|
||||||
if line.starts_with("MemFree") {
|
|
||||||
assign_val(line, &mut mem_free);
|
|
||||||
}
|
|
||||||
if line.starts_with("Buffers") {
|
|
||||||
assign_val(line, &mut mem_buffers);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let mem_used = (mem_total + shmem - mem_free - mem_buffers - mem_cached - mem_srecl) / 1024;
|
|
||||||
let result: String;
|
|
||||||
if mem_used > 1000 {
|
|
||||||
result = format!(
|
|
||||||
" {} {:.1}G {}",
|
|
||||||
config.memory.icon,
|
|
||||||
mem_used as f32 / 1000.0,
|
|
||||||
config.seperator
|
|
||||||
);
|
|
||||||
} else {
|
|
||||||
result = format!(
|
|
||||||
" {} {}M {}",
|
|
||||||
config.memory.icon, mem_used, config.seperator
|
|
||||||
);
|
|
||||||
}
|
|
||||||
Ok(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
this helper function will split the line(first argument) by the character(:)
|
|
||||||
and then parse the right splited item as u32
|
|
||||||
then assign that to the "assignable"(2nd argument).
|
|
||||||
*/
|
|
||||||
fn assign_val(line: &str, assignable: &mut u32) {
|
|
||||||
let parsed: u32 = line.split(':').collect::<Vec<&str>>()[1]
|
|
||||||
.trim()
|
|
||||||
.split(' ')
|
|
||||||
.collect::<Vec<&str>>()[0]
|
|
||||||
.parse()
|
|
||||||
.unwrap();
|
|
||||||
*assignable = parsed;
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting battery percentage
|
|
||||||
pub fn get_battery(config: &Config) -> Result<String, Error> {
|
|
||||||
let battery_full_cap_file = format!(
|
|
||||||
"/sys/class/power_supply/{}/charge_full_design",
|
|
||||||
config.battery.source
|
|
||||||
);
|
|
||||||
let battery_charge_now_file = format!(
|
|
||||||
"/sys/class/power_supply/{}/charge_now",
|
|
||||||
config.battery.source
|
|
||||||
);
|
|
||||||
|
|
||||||
let mut buf = String::new();
|
|
||||||
|
|
||||||
// FIXME: ugly error handling AGAIN fixing later, im lazy
|
|
||||||
match File::open(&battery_full_cap_file) {
|
|
||||||
Ok(mut file) => file.read_to_string(&mut buf)?,
|
|
||||||
Err(_) => return Ok(String::from("check your battery source name")),
|
|
||||||
};
|
|
||||||
let full_design = buf.trim().parse::<u32>().unwrap();
|
|
||||||
buf.clear();
|
|
||||||
|
|
||||||
// NOTE: no need to error check if passed the above match
|
|
||||||
File::open(&battery_charge_now_file)?.read_to_string(&mut buf)?;
|
|
||||||
|
|
||||||
let charge_now = buf.trim().parse::<u32>().unwrap();
|
|
||||||
|
|
||||||
let battery_percentage = (charge_now as f32 / full_design as f32) * 100.0;
|
|
||||||
let result = format!(
|
|
||||||
" {} {:.0}% {}",
|
|
||||||
config.battery.icon, battery_percentage, config.seperator
|
|
||||||
);
|
|
||||||
Ok(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting cpu temperature
|
|
||||||
pub fn get_cpu_temp(config: &Config) -> Result<String, std::io::Error> {
|
|
||||||
let mut buf = String::new();
|
|
||||||
File::open("/sys/class/thermal/thermal_zone0/temp")?.read_to_string(&mut buf)?;
|
|
||||||
let value = buf.trim().parse::<f32>().unwrap();
|
|
||||||
|
|
||||||
let result = format!(
|
|
||||||
" {} {}° {}",
|
|
||||||
config.cpu_temperature.icon,
|
|
||||||
value / 1000.0,
|
|
||||||
config.seperator
|
|
||||||
);
|
|
||||||
Ok(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_uptime(config: &Config) -> Result<String, std::io::Error> {
|
|
||||||
let mut buf = String::new();
|
|
||||||
File::open("/proc/uptime")?.read_to_string(&mut buf)?;
|
|
||||||
|
|
||||||
let buf: f32 = buf.split(' ').collect::<Vec<&str>>()[0].parse().unwrap();
|
|
||||||
|
|
||||||
let hour = buf.round() as u32 / 3600;
|
|
||||||
let rem = buf as u32 - hour * 3600;
|
|
||||||
let minutes = rem / 60;
|
|
||||||
|
|
||||||
let uptime = if hour > 0 {
|
|
||||||
format!("{}:{}", hour, minutes)
|
|
||||||
} else {
|
|
||||||
format!("{} min", minutes)
|
|
||||||
};
|
|
||||||
let result = format!(" {} {} {}", config.uptime.icon, uptime, config.seperator);
|
|
||||||
Ok(result)
|
|
||||||
}
|
|
||||||
|
|
||||||
// yes, error handling looks fucking sucks!
|
|
||||||
// getting mpd song file
|
|
||||||
pub fn get_mpd_current(config: &Config) -> String {
|
|
||||||
let stream_path = format!("{}:{}", config.mpd.host, config.mpd.port);
|
|
||||||
let empty_string = String::from("");
|
|
||||||
let mut conn = match Client::connect(&stream_path) {
|
|
||||||
Ok(connection) => connection,
|
|
||||||
_ => return empty_string,
|
|
||||||
};
|
|
||||||
let current: Song = match conn.currentsong() {
|
|
||||||
Ok(opt) => match opt {
|
|
||||||
Some(song) => song,
|
|
||||||
_ => return empty_string,
|
|
||||||
},
|
|
||||||
_ => return empty_string,
|
|
||||||
};
|
|
||||||
|
|
||||||
let result = format!(
|
|
||||||
" {} {} {}",
|
|
||||||
config.mpd.icon, current.file, config.seperator
|
|
||||||
);
|
|
||||||
|
|
||||||
result
|
|
||||||
}
|
|
||||||
|
|
||||||
// getting spotify current artist and title.
|
|
||||||
// FIXME: I know im lazy asshole, this error handling looks ugly, i dont like it too, need to fix soon.
|
|
||||||
fn get_spotify(config: &Config) -> String {
|
|
||||||
let conn = match Connection::new_session() {
|
|
||||||
Ok(conn) => conn,
|
|
||||||
_ => return String::from(""),
|
|
||||||
};
|
|
||||||
|
|
||||||
let p = conn.with_proxy(
|
|
||||||
"org.mpris.MediaPlayer2.spotify",
|
|
||||||
"/org/mpris/MediaPlayer2",
|
|
||||||
Duration::from_millis(5000),
|
|
||||||
);
|
|
||||||
|
|
||||||
let metadata: arg::PropMap = match p.get("org.mpris.MediaPlayer2.Player", "Metadata") {
|
|
||||||
Ok(data) => data,
|
|
||||||
_ => return String::from(""),
|
|
||||||
};
|
|
||||||
|
|
||||||
let title: Option<&String> = arg::prop_cast(&metadata, "xesam:title");
|
|
||||||
let artist: Option<&Vec<String>> = arg::prop_cast(&metadata, "xesam:artist");
|
|
||||||
|
|
||||||
let title = match title {
|
|
||||||
Some(title) => title,
|
|
||||||
_ => "",
|
|
||||||
};
|
|
||||||
|
|
||||||
let artist = match artist {
|
|
||||||
Some(artist_vec) => match artist_vec.first() {
|
|
||||||
Some(name) => name,
|
|
||||||
_ => "",
|
|
||||||
},
|
|
||||||
None => "",
|
|
||||||
};
|
|
||||||
|
|
||||||
format!(
|
|
||||||
" {} {} - {} {}",
|
|
||||||
config.spotify.icon, artist, title, config.seperator
|
|
||||||
)
|
|
||||||
}
|
|
11
src/main.rs
11
src/main.rs
@ -1,5 +1,10 @@
|
|||||||
|
mod config;
|
||||||
|
mod run;
|
||||||
|
mod types;
|
||||||
|
mod utils;
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let config = rsblocks::load_config().unwrap();
|
let config = config::load_config().unwrap();
|
||||||
let blocks = rsblocks::Blocks::new();
|
let blocks = types::Blocks::new();
|
||||||
rsblocks::run(config, blocks);
|
run::run(config, blocks);
|
||||||
}
|
}
|
||||||
|
171
src/run.rs
Normal file
171
src/run.rs
Normal file
@ -0,0 +1,171 @@
|
|||||||
|
use crate::types::*;
|
||||||
|
use crate::utils::*;
|
||||||
|
use std::sync::mpsc;
|
||||||
|
use std::thread;
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
|
// FIXME: The most part i hate is this, looks really ugly, fix me you dumb fuck
|
||||||
|
pub fn run(config: Config, mut blocks: Blocks) {
|
||||||
|
let (tx, rx) = mpsc::channel();
|
||||||
|
|
||||||
|
// spotify thread
|
||||||
|
if config.spotify.enabled {
|
||||||
|
let spotify_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let mut spotify_data = ThreadsData::Spotify(spotify::get_spotify(&configcp));
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
spotify_tx.send(spotify_data).unwrap();
|
||||||
|
spotify_data = ThreadsData::Spotify(spotify::get_spotify(&configcp));
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.spotify.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// mpd thread
|
||||||
|
if config.mpd.enabled {
|
||||||
|
let mpd_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let mut mpd_data = ThreadsData::Mpd(mpd::get_mpd_current(&configcp));
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
mpd_tx.send(mpd_data).unwrap();
|
||||||
|
mpd_data = ThreadsData::Mpd(mpd::get_mpd_current(&configcp));
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.mpd.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// volume thread
|
||||||
|
if config.volume.enabled {
|
||||||
|
let volume_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let mut vol_data = ThreadsData::Sound(volume::get_volume(&configcp));
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
volume_tx.send(vol_data).unwrap();
|
||||||
|
vol_data = ThreadsData::Sound(volume::get_volume(&configcp));
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.volume.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Disk thread
|
||||||
|
if config.disk.enabled {
|
||||||
|
let disk_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let mut disk_data = ThreadsData::Disk(disk::get_disk(&configcp));
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
disk_tx.send(disk_data).unwrap();
|
||||||
|
disk_data = ThreadsData::Disk(disk::get_disk(&configcp));
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.disk.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Memory thread
|
||||||
|
if config.memory.enabled {
|
||||||
|
let memory_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let memory_data = memory::get_memory(&configcp).unwrap();
|
||||||
|
let mut memory_data = ThreadsData::Memory(memory_data);
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
memory_tx.send(memory_data).unwrap();
|
||||||
|
memory_data = ThreadsData::Memory(memory::get_memory(&configcp).unwrap());
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.memory.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Weather thread
|
||||||
|
if config.weather.enabled {
|
||||||
|
let weather_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let weather_data = weather::get_weather(&configcp);
|
||||||
|
let mut weather_data = ThreadsData::Weather(weather_data);
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
weather_tx.send(weather_data).unwrap();
|
||||||
|
weather_data = ThreadsData::Weather(weather::get_weather(&configcp));
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.weather.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Battery thread
|
||||||
|
if config.battery.enabled {
|
||||||
|
let battery_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let battery_data = battery::get_battery(&configcp).unwrap();
|
||||||
|
let mut battery_data = ThreadsData::Battery(battery_data);
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
battery_tx.send(battery_data).unwrap();
|
||||||
|
battery_data = ThreadsData::Battery(battery::get_battery(&configcp).unwrap());
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.battery.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Cpu temperature thread
|
||||||
|
if config.cpu_temperature.enabled {
|
||||||
|
let cpu_temp_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let cpu_temp_data = cpu::get_cpu_temp(&configcp).unwrap();
|
||||||
|
let mut cpu_temp_data = ThreadsData::CpuTemp(cpu_temp_data);
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
cpu_temp_tx.send(cpu_temp_data).unwrap();
|
||||||
|
cpu_temp_data = ThreadsData::CpuTemp(cpu::get_cpu_temp(&configcp).unwrap());
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.cpu_temperature.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Uptime thread
|
||||||
|
if config.uptime.enabled {
|
||||||
|
let uptime_tx = tx.clone();
|
||||||
|
let configcp = config.clone();
|
||||||
|
let uptime_data = uptime::get_uptime(&configcp).unwrap();
|
||||||
|
let mut uptime_data = ThreadsData::Uptime(uptime_data);
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
uptime_tx.send(uptime_data).unwrap();
|
||||||
|
uptime_data = ThreadsData::Uptime(uptime::get_uptime(&configcp).unwrap());
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.uptime.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Time thread
|
||||||
|
{
|
||||||
|
let time_tx = tx;
|
||||||
|
let configcp = config;
|
||||||
|
let mut time_data = ThreadsData::Time(time::get_time(&configcp));
|
||||||
|
thread::spawn(move || loop {
|
||||||
|
time_tx.send(time_data).unwrap();
|
||||||
|
time_data = ThreadsData::Time(time::get_time(&configcp));
|
||||||
|
thread::sleep(Duration::from_secs_f64(configcp.time.delay))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
//Main
|
||||||
|
{
|
||||||
|
// NOTE: order matters to the final format
|
||||||
|
let mut bar: Vec<String> = vec!["".to_string(); 10];
|
||||||
|
//iterating the values recieved from the threads
|
||||||
|
for data in rx {
|
||||||
|
match data {
|
||||||
|
ThreadsData::Spotify(x) => bar[0] = x,
|
||||||
|
ThreadsData::Mpd(x) => bar[1] = x,
|
||||||
|
ThreadsData::Sound(x) => bar[2] = x,
|
||||||
|
ThreadsData::Weather(x) => bar[3] = x,
|
||||||
|
ThreadsData::Disk(x) => bar[4] = x,
|
||||||
|
ThreadsData::Memory(x) => bar[5] = x,
|
||||||
|
ThreadsData::CpuTemp(x) => bar[6] = x,
|
||||||
|
ThreadsData::Battery(x) => bar[7] = x,
|
||||||
|
ThreadsData::Uptime(x) => bar[8] = x,
|
||||||
|
ThreadsData::Time(x) => bar[9] = x,
|
||||||
|
}
|
||||||
|
|
||||||
|
// match ends here
|
||||||
|
update(&bar, &mut blocks);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn update(bar: &[String], blocks: &mut Blocks) {
|
||||||
|
let mut x = String::new();
|
||||||
|
for i in bar.iter() {
|
||||||
|
x.push_str(i.as_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
blocks
|
||||||
|
.root
|
||||||
|
.set_title(&mut blocks.disp, &x)
|
||||||
|
.expect("Failed to set title of root");
|
||||||
|
}
|
124
src/types.rs
Normal file
124
src/types.rs
Normal file
@ -0,0 +1,124 @@
|
|||||||
|
use breadx::{display::*, window::Window};
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub enum ThreadsData {
|
||||||
|
Mpd(String),
|
||||||
|
Sound(String),
|
||||||
|
Disk(String),
|
||||||
|
Memory(String),
|
||||||
|
Time(String),
|
||||||
|
Weather(String),
|
||||||
|
Battery(String),
|
||||||
|
CpuTemp(String),
|
||||||
|
Uptime(String),
|
||||||
|
Spotify(String),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Config {
|
||||||
|
pub seperator: String,
|
||||||
|
pub time: Time,
|
||||||
|
pub memory: Memory,
|
||||||
|
pub disk: Disk,
|
||||||
|
pub volume: Volume,
|
||||||
|
pub weather: Weather,
|
||||||
|
pub battery: Battery,
|
||||||
|
pub cpu_temperature: CpuTemp,
|
||||||
|
pub uptime: Uptime,
|
||||||
|
pub mpd: Mpd,
|
||||||
|
pub spotify: Spotify,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Time {
|
||||||
|
pub format: String,
|
||||||
|
pub icon: String,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Memory {
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Disk {
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Volume {
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
pub card: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Weather {
|
||||||
|
pub city: String,
|
||||||
|
pub format: String,
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Battery {
|
||||||
|
pub source: String,
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct CpuTemp {
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Uptime {
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Mpd {
|
||||||
|
pub icon: String,
|
||||||
|
pub host: String,
|
||||||
|
pub port: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Spotify {
|
||||||
|
pub icon: String,
|
||||||
|
pub enabled: bool,
|
||||||
|
pub delay: f64,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct Blocks {
|
||||||
|
pub disp: Display<name::NameConnection>,
|
||||||
|
pub root: Window,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Blocks {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
let disp = Display::create(None, None).expect("Failed to create x11 connection");
|
||||||
|
let root = disp.default_screen().root;
|
||||||
|
Self { disp, root }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Blocks {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
38
src/utils/battery.rs
Normal file
38
src/utils/battery.rs
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::Error;
|
||||||
|
use std::io::Read;
|
||||||
|
|
||||||
|
// getting battery percentage
|
||||||
|
pub fn get_battery(config: &Config) -> Result<String, Error> {
|
||||||
|
let battery_full_cap_file = format!(
|
||||||
|
"/sys/class/power_supply/{}/charge_full_design",
|
||||||
|
config.battery.source
|
||||||
|
);
|
||||||
|
let battery_charge_now_file = format!(
|
||||||
|
"/sys/class/power_supply/{}/charge_now",
|
||||||
|
config.battery.source
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut buf = String::new();
|
||||||
|
|
||||||
|
// FIXME: ugly error handling AGAIN fixing later, im lazy
|
||||||
|
match File::open(&battery_full_cap_file) {
|
||||||
|
Ok(mut file) => file.read_to_string(&mut buf)?,
|
||||||
|
Err(_) => return Ok(String::from("check your battery source name")),
|
||||||
|
};
|
||||||
|
let full_design = buf.trim().parse::<u32>().unwrap();
|
||||||
|
buf.clear();
|
||||||
|
|
||||||
|
// NOTE: no need to error check if passed the above match
|
||||||
|
File::open(&battery_charge_now_file)?.read_to_string(&mut buf)?;
|
||||||
|
|
||||||
|
let charge_now = buf.trim().parse::<u32>().unwrap();
|
||||||
|
|
||||||
|
let battery_percentage = (charge_now as f32 / full_design as f32) * 100.0;
|
||||||
|
let result = format!(
|
||||||
|
" {} {:.0}% {}",
|
||||||
|
config.battery.icon, battery_percentage, config.seperator
|
||||||
|
);
|
||||||
|
Ok(result)
|
||||||
|
}
|
18
src/utils/cpu.rs
Normal file
18
src/utils/cpu.rs
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::Read;
|
||||||
|
|
||||||
|
// getting cpu temperature
|
||||||
|
pub fn get_cpu_temp(config: &Config) -> Result<String, std::io::Error> {
|
||||||
|
let mut buf = String::new();
|
||||||
|
File::open("/sys/class/thermal/thermal_zone0/temp")?.read_to_string(&mut buf)?;
|
||||||
|
let value = buf.trim().parse::<f32>().unwrap();
|
||||||
|
|
||||||
|
let result = format!(
|
||||||
|
" {} {}° {}",
|
||||||
|
config.cpu_temperature.icon,
|
||||||
|
value / 1000.0,
|
||||||
|
config.seperator
|
||||||
|
);
|
||||||
|
Ok(result)
|
||||||
|
}
|
18
src/utils/disk.rs
Normal file
18
src/utils/disk.rs
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
|
||||||
|
// getting disk usage
|
||||||
|
pub fn get_disk(config: &Config) -> String {
|
||||||
|
const GB: u64 = (1024 * 1024) * 1024;
|
||||||
|
let statvfs = nix::sys::statvfs::statvfs("/").unwrap();
|
||||||
|
let mut disk_used = String::new();
|
||||||
|
|
||||||
|
let total = (statvfs.blocks() * statvfs.fragment_size()) / GB;
|
||||||
|
let available = (statvfs.blocks_free() * statvfs.fragment_size()) / GB;
|
||||||
|
let used = total - available;
|
||||||
|
|
||||||
|
disk_used.push_str(&format!("{}G", used));
|
||||||
|
format!(
|
||||||
|
" {} {} {}",
|
||||||
|
config.disk.icon, disk_used, config.seperator
|
||||||
|
)
|
||||||
|
}
|
85
src/utils/memory.rs
Normal file
85
src/utils/memory.rs
Normal file
@ -0,0 +1,85 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::Read;
|
||||||
|
|
||||||
|
/*
|
||||||
|
mem_used = (mem_total + shmem - mem_free - mem_buffers - mem_cached - mem_srecl
|
||||||
|
thanks for htop's developer on stackoverflow for providing this algorithm to
|
||||||
|
calculate used memory.
|
||||||
|
*/
|
||||||
|
pub fn get_memory(config: &Config) -> Result<String, std::io::Error> {
|
||||||
|
let mut buf = String::new();
|
||||||
|
|
||||||
|
File::open("/proc/meminfo")?.read_to_string(&mut buf)?;
|
||||||
|
|
||||||
|
let mut mem_total: u32 = 0;
|
||||||
|
let mut shmem: u32 = 0;
|
||||||
|
let mut mem_free: u32 = 0;
|
||||||
|
let mut mem_buffers: u32 = 0;
|
||||||
|
let mut mem_cached: u32 = 0;
|
||||||
|
let mut mem_srecl: u32 = 0;
|
||||||
|
|
||||||
|
for line in buf.lines() {
|
||||||
|
if mem_total > 0
|
||||||
|
&& shmem > 0
|
||||||
|
&& mem_free > 0
|
||||||
|
&& mem_buffers > 0
|
||||||
|
&& mem_cached > 0
|
||||||
|
&& mem_srecl > 0
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if line.starts_with("MemTotal") {
|
||||||
|
assign_val(line, &mut mem_total);
|
||||||
|
}
|
||||||
|
if line.starts_with("SReclaimable") {
|
||||||
|
assign_val(line, &mut mem_srecl)
|
||||||
|
}
|
||||||
|
if line.starts_with("Cached") {
|
||||||
|
assign_val(line, &mut mem_cached)
|
||||||
|
}
|
||||||
|
|
||||||
|
if line.starts_with("Shmem") {
|
||||||
|
assign_val(line, &mut shmem);
|
||||||
|
}
|
||||||
|
|
||||||
|
if line.starts_with("MemFree") {
|
||||||
|
assign_val(line, &mut mem_free);
|
||||||
|
}
|
||||||
|
if line.starts_with("Buffers") {
|
||||||
|
assign_val(line, &mut mem_buffers);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mem_used = (mem_total + shmem - mem_free - mem_buffers - mem_cached - mem_srecl) / 1024;
|
||||||
|
let result: String;
|
||||||
|
if mem_used > 1000 {
|
||||||
|
result = format!(
|
||||||
|
" {} {:.1}G {}",
|
||||||
|
config.memory.icon,
|
||||||
|
mem_used as f32 / 1000.0,
|
||||||
|
config.seperator
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
result = format!(
|
||||||
|
" {} {}M {}",
|
||||||
|
config.memory.icon, mem_used, config.seperator
|
||||||
|
);
|
||||||
|
}
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
this helper function will split the line(first argument) by the character(:)
|
||||||
|
and then parse the right splited item as u32
|
||||||
|
then assign that to the "assignable"(2nd argument).
|
||||||
|
*/
|
||||||
|
fn assign_val(line: &str, assignable: &mut u32) {
|
||||||
|
let parsed: u32 = line.split(':').collect::<Vec<&str>>()[1]
|
||||||
|
.trim()
|
||||||
|
.split(' ')
|
||||||
|
.collect::<Vec<&str>>()[0]
|
||||||
|
.parse()
|
||||||
|
.unwrap();
|
||||||
|
*assignable = parsed;
|
||||||
|
}
|
10
src/utils/mod.rs
Normal file
10
src/utils/mod.rs
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
pub mod battery;
|
||||||
|
pub mod cpu;
|
||||||
|
pub mod disk;
|
||||||
|
pub mod memory;
|
||||||
|
pub mod mpd;
|
||||||
|
pub mod spotify;
|
||||||
|
pub mod time;
|
||||||
|
pub mod uptime;
|
||||||
|
pub mod volume;
|
||||||
|
pub mod weather;
|
27
src/utils/mpd.rs
Normal file
27
src/utils/mpd.rs
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use mpd::{Client, Song};
|
||||||
|
|
||||||
|
// yes, error handling looks fucking sucks!
|
||||||
|
// getting mpd song file
|
||||||
|
pub fn get_mpd_current(config: &Config) -> String {
|
||||||
|
let stream_path = format!("{}:{}", config.mpd.host, config.mpd.port);
|
||||||
|
let empty_string = String::from("");
|
||||||
|
let mut conn = match Client::connect(&stream_path) {
|
||||||
|
Ok(connection) => connection,
|
||||||
|
_ => return empty_string,
|
||||||
|
};
|
||||||
|
let current: Song = match conn.currentsong() {
|
||||||
|
Ok(opt) => match opt {
|
||||||
|
Some(song) => song,
|
||||||
|
_ => return empty_string,
|
||||||
|
},
|
||||||
|
_ => return empty_string,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = format!(
|
||||||
|
" {} {} {}",
|
||||||
|
config.mpd.icon, current.file, config.seperator
|
||||||
|
);
|
||||||
|
|
||||||
|
result
|
||||||
|
}
|
45
src/utils/spotify.rs
Normal file
45
src/utils/spotify.rs
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
|
||||||
|
use dbus::{arg, blocking::Connection};
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
|
// getting spotify current artist and title.
|
||||||
|
// FIXME: I know im lazy asshole, this error handling looks ugly, i dont like it too, need to fix soon.
|
||||||
|
pub fn get_spotify(config: &Config) -> String {
|
||||||
|
let conn = match Connection::new_session() {
|
||||||
|
Ok(conn) => conn,
|
||||||
|
_ => return String::from(""),
|
||||||
|
};
|
||||||
|
|
||||||
|
let p = conn.with_proxy(
|
||||||
|
"org.mpris.MediaPlayer2.spotify",
|
||||||
|
"/org/mpris/MediaPlayer2",
|
||||||
|
Duration::from_millis(5000),
|
||||||
|
);
|
||||||
|
|
||||||
|
let metadata: arg::PropMap = match p.get("org.mpris.MediaPlayer2.Player", "Metadata") {
|
||||||
|
Ok(data) => data,
|
||||||
|
_ => return String::from(""),
|
||||||
|
};
|
||||||
|
|
||||||
|
let title: Option<&String> = arg::prop_cast(&metadata, "xesam:title");
|
||||||
|
let artist: Option<&Vec<String>> = arg::prop_cast(&metadata, "xesam:artist");
|
||||||
|
|
||||||
|
let title = match title {
|
||||||
|
Some(title) => title,
|
||||||
|
_ => "",
|
||||||
|
};
|
||||||
|
|
||||||
|
let artist = match artist {
|
||||||
|
Some(artist_vec) => match artist_vec.first() {
|
||||||
|
Some(name) => name,
|
||||||
|
_ => "",
|
||||||
|
},
|
||||||
|
None => "",
|
||||||
|
};
|
||||||
|
|
||||||
|
format!(
|
||||||
|
" {} {} - {} {}",
|
||||||
|
config.spotify.icon, artist, title, config.seperator
|
||||||
|
)
|
||||||
|
}
|
13
src/utils/time.rs
Normal file
13
src/utils/time.rs
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use chrono::prelude::*;
|
||||||
|
|
||||||
|
pub fn get_time(config: &Config) -> String {
|
||||||
|
let now = Local::now();
|
||||||
|
|
||||||
|
format!(
|
||||||
|
" {} {} {}",
|
||||||
|
config.time.icon,
|
||||||
|
now.format(&config.time.format),
|
||||||
|
config.seperator
|
||||||
|
)
|
||||||
|
}
|
22
src/utils/uptime.rs
Normal file
22
src/utils/uptime.rs
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::Read;
|
||||||
|
|
||||||
|
pub fn get_uptime(config: &Config) -> Result<String, std::io::Error> {
|
||||||
|
let mut buf = String::new();
|
||||||
|
File::open("/proc/uptime")?.read_to_string(&mut buf)?;
|
||||||
|
|
||||||
|
let buf: f32 = buf.split(' ').collect::<Vec<&str>>()[0].parse().unwrap();
|
||||||
|
|
||||||
|
let hour = buf.round() as u32 / 3600;
|
||||||
|
let rem = buf as u32 - hour * 3600;
|
||||||
|
let minutes = rem / 60;
|
||||||
|
|
||||||
|
let uptime = if hour > 0 {
|
||||||
|
format!("{}:{}", hour, minutes)
|
||||||
|
} else {
|
||||||
|
format!("{} min", minutes)
|
||||||
|
};
|
||||||
|
let result = format!(" {} {} {}", config.uptime.icon, uptime, config.seperator);
|
||||||
|
Ok(result)
|
||||||
|
}
|
31
src/utils/volume.rs
Normal file
31
src/utils/volume.rs
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
use alsa::mixer::{Mixer, SelemChannelId, SelemId};
|
||||||
|
|
||||||
|
// getting volume percentage
|
||||||
|
pub fn get_volume(config: &Config) -> String {
|
||||||
|
let card = if config.volume.card == "PULSE" {
|
||||||
|
"pulse"
|
||||||
|
} else {
|
||||||
|
"default"
|
||||||
|
};
|
||||||
|
|
||||||
|
let mixer = Mixer::new(card, false).expect("Failed to open mixer");
|
||||||
|
let selem_id = SelemId::new("Master", 0);
|
||||||
|
let selem = mixer.find_selem(&selem_id).expect("Couldn't find selem");
|
||||||
|
let selem_chan_id = SelemChannelId::FrontLeft;
|
||||||
|
|
||||||
|
let (min, max) = selem.get_playback_volume_range();
|
||||||
|
let mut raw_volume = selem
|
||||||
|
.get_playback_volume(selem_chan_id)
|
||||||
|
.expect("Failed to get raw_volume");
|
||||||
|
|
||||||
|
let range = max - min;
|
||||||
|
let vol = if range == 0 {
|
||||||
|
0
|
||||||
|
} else {
|
||||||
|
raw_volume -= min;
|
||||||
|
((raw_volume as f64 / range as f64) * 100.) as u64
|
||||||
|
};
|
||||||
|
|
||||||
|
format!(" {} {}% {}", config.volume.icon, vol, config.seperator)
|
||||||
|
}
|
22
src/utils/weather.rs
Normal file
22
src/utils/weather.rs
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
use crate::types::Config;
|
||||||
|
|
||||||
|
// will make a GET request from wttr.in
|
||||||
|
pub fn get_weather(config: &Config) -> String {
|
||||||
|
let format = if config.weather.format.is_empty() {
|
||||||
|
String::from("%l:+%t")
|
||||||
|
} else {
|
||||||
|
config.weather.format.clone()
|
||||||
|
};
|
||||||
|
|
||||||
|
let url = format!("http://wttr.in/{}?format=\"{}", config.weather.city, format);
|
||||||
|
let err_string = String::from("Error");
|
||||||
|
let res = match minreq::get(url).send() {
|
||||||
|
Ok(resp) => match resp.as_str() {
|
||||||
|
Ok(res_str) => res_str.trim_matches('"').to_string(),
|
||||||
|
Err(_) => err_string,
|
||||||
|
},
|
||||||
|
Err(_) => err_string,
|
||||||
|
};
|
||||||
|
|
||||||
|
format!(" {} {} {}", config.weather.icon, res, config.seperator)
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user