diff options
Diffstat (limited to 'src/config.rs')
| -rw-r--r-- | src/config.rs | 253 |
1 files changed, 253 insertions, 0 deletions
diff --git a/src/config.rs b/src/config.rs new file mode 100644 index 0000000..b54f8f1 --- /dev/null +++ b/src/config.rs @@ -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 +} |