aboutsummaryrefslogtreecommitdiff
path: root/src/utils
diff options
context:
space:
mode:
Diffstat (limited to 'src/utils')
-rw-r--r--src/utils/battery.rs26
-rw-r--r--src/utils/cpu.rs15
-rw-r--r--src/utils/disk.rs8
-rw-r--r--src/utils/load_average.rs10
-rw-r--r--src/utils/memory.rs15
-rw-r--r--src/utils/mpd.rs13
-rw-r--r--src/utils/netspeed.rs8
-rw-r--r--src/utils/pub_ip.rs12
-rw-r--r--src/utils/spotify.rs16
-rw-r--r--src/utils/time.rs9
-rw-r--r--src/utils/uptime.rs15
-rw-r--r--src/utils/volume.rs6
-rw-r--r--src/utils/weather.rs6
13 files changed, 90 insertions, 69 deletions
diff --git a/src/utils/battery.rs b/src/utils/battery.rs
index 1278303..041965a 100644
--- a/src/utils/battery.rs
+++ b/src/utils/battery.rs
@@ -1,10 +1,11 @@
use crate::config::CONFIG;
-use std::fs::File;
-use std::io::Error;
-use std::io::Read;
+use crate::types::ThreadsData;
+use std::fs::read_to_string;
// getting battery percentage
-pub fn get_battery() -> Result<String, Error> {
+pub fn get_battery() -> ThreadsData {
+ let error = ThreadsData::Battery(String::from("check your battery source name"));
+
let battery_full_cap_file = format!(
"/sys/class/power_supply/{}/charge_full_design",
CONFIG.battery.source
@@ -14,17 +15,16 @@ pub fn get_battery() -> Result<String, Error> {
CONFIG.battery.source
);
- let mut buf = String::new();
-
- 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 buf = match read_to_string(battery_full_cap_file) {
+ Ok(file) => file,
+ Err(_) => return error,
};
let full_design = buf.trim().parse::<u32>().unwrap();
- buf.clear();
- // No need to error check if passed the above match
- File::open(&battery_charge_now_file)?.read_to_string(&mut buf)?;
+ let buf = match read_to_string(&battery_charge_now_file) {
+ Ok(data) => data,
+ _ => return error,
+ };
let charge_now = buf.trim().parse::<u32>().unwrap();
@@ -33,5 +33,5 @@ pub fn get_battery() -> Result<String, Error> {
" {} {:.0}% {}",
CONFIG.battery.icon, battery_percentage, CONFIG.seperator
);
- Ok(result)
+ ThreadsData::Battery(result)
}
diff --git a/src/utils/cpu.rs b/src/utils/cpu.rs
index f8ae595..2047a07 100644
--- a/src/utils/cpu.rs
+++ b/src/utils/cpu.rs
@@ -1,11 +1,14 @@
use crate::config::CONFIG;
-use std::fs::File;
-use std::io::Read;
+use crate::types::ThreadsData;
+use std::fs::read_to_string;
// getting cpu temperature
-pub fn get_cpu_temp() -> Result<String, std::io::Error> {
- let mut buf = String::new();
- File::open("/sys/class/thermal/thermal_zone0/temp")?.read_to_string(&mut buf)?;
+pub fn get_cpu_temp() -> ThreadsData {
+ let buf = match read_to_string("/sys/class/thermal/thermal_zone0/temp") {
+ Ok(data) => data,
+ _ => return ThreadsData::CpuTemp(String::from("Error reading temp")),
+ };
+
let value = buf.trim().parse::<f32>().unwrap();
let result = format!(
@@ -14,5 +17,5 @@ pub fn get_cpu_temp() -> Result<String, std::io::Error> {
value / 1000.0,
CONFIG.seperator
);
- Ok(result)
+ ThreadsData::CpuTemp(result)
}
diff --git a/src/utils/disk.rs b/src/utils/disk.rs
index f082ef1..2250911 100644
--- a/src/utils/disk.rs
+++ b/src/utils/disk.rs
@@ -1,6 +1,7 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
-pub fn get_disk() -> String {
+pub fn get_disk() -> ThreadsData {
const GB: u64 = (1024 * 1024) * 1024;
let statvfs = nix::sys::statvfs::statvfs("/").unwrap();
let mut disk_used = String::new();
@@ -10,8 +11,9 @@ pub fn get_disk() -> String {
let used = total - available;
disk_used.push_str(&format!("{}G", used));
- format!(
+ let data = format!(
" {} {} {}",
CONFIG.disk.icon, disk_used, CONFIG.seperator
- )
+ );
+ ThreadsData::Disk(data)
}
diff --git a/src/utils/load_average.rs b/src/utils/load_average.rs
index 4fd38f7..70e4a72 100644
--- a/src/utils/load_average.rs
+++ b/src/utils/load_average.rs
@@ -1,16 +1,18 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
use std::fs::File;
use std::io::Read;
-pub fn get_load_avg() -> String {
+pub fn get_load_avg() -> ThreadsData {
let mut buf = String::new();
match File::open("/proc/loadavg") {
Ok(mut file) => match file.read_to_string(&mut buf) {
Ok(data) => data,
- _ => return String::from(""),
+ _ => return ThreadsData::LoadAvg(String::from("")),
},
- _ => return String::from("Error"),
+ _ => return ThreadsData::LoadAvg(String::from("Error")),
};
let buf = buf.split_whitespace().collect::<Vec<_>>()[0];
- format!(" {} {} {}", CONFIG.loadavg.icon, buf, CONFIG.seperator)
+ let data = format!(" {} {} {}", CONFIG.loadavg.icon, buf, CONFIG.seperator);
+ ThreadsData::LoadAvg(data)
}
diff --git a/src/utils/memory.rs b/src/utils/memory.rs
index 6283527..3d51fd5 100644
--- a/src/utils/memory.rs
+++ b/src/utils/memory.rs
@@ -1,16 +1,17 @@
use crate::config::CONFIG;
-use std::fs::File;
-use std::io::Read;
+use crate::types::ThreadsData;
+use std::fs::read_to_string;
/*
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() -> Result<String, std::io::Error> {
- let mut buf = String::new();
-
- File::open("/proc/meminfo")?.read_to_string(&mut buf)?;
+pub fn get_memory() -> ThreadsData {
+ let buf = match read_to_string("/proc/meminfo") {
+ Ok(data) => data,
+ _ => return ThreadsData::Memory(String::from("Error Reading memory!")),
+ };
let mut mem_total: u32 = 0;
let mut shmem: u32 = 0;
@@ -66,7 +67,7 @@ pub fn get_memory() -> Result<String, std::io::Error> {
CONFIG.memory.icon, mem_used, CONFIG.seperator
);
}
- Ok(result)
+ ThreadsData::Memory(result)
}
/*
diff --git a/src/utils/mpd.rs b/src/utils/mpd.rs
index b7fdbd6..3752387 100644
--- a/src/utils/mpd.rs
+++ b/src/utils/mpd.rs
@@ -1,21 +1,22 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
use mpd::{Client, Song};
// yes, error handling looks fucking sucks!
// getting mpd song file
-pub fn get_mpd_current() -> String {
+pub fn get_mpd_current() -> ThreadsData {
let stream_path = format!("{}:{}", CONFIG.mpd.host, CONFIG.mpd.port);
- let empty_string = String::from("");
+ let empty_data = ThreadsData::Mpd(String::from(""));
let mut conn = match Client::connect(&stream_path) {
Ok(connection) => connection,
- _ => return empty_string,
+ _ => return empty_data,
};
let current: Song = match conn.currentsong() {
Ok(opt) => match opt {
Some(song) => song,
- _ => return empty_string,
+ _ => return empty_data,
},
- _ => return empty_string,
+ _ => return empty_data,
};
let result = format!(
@@ -23,5 +24,5 @@ pub fn get_mpd_current() -> String {
CONFIG.mpd.icon, current.file, CONFIG.seperator
);
- result
+ ThreadsData::Mpd(result)
}
diff --git a/src/utils/netspeed.rs b/src/utils/netspeed.rs
index 59f740f..fe172a3 100644
--- a/src/utils/netspeed.rs
+++ b/src/utils/netspeed.rs
@@ -1,9 +1,10 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
use std::fs::read_to_string;
use std::thread;
use std::time::Duration;
-pub fn get_netspeed() -> String {
+pub fn get_netspeed() -> ThreadsData {
let tx1: u64 = parse_speed_file("tx_bytes");
let rx1: u64 = parse_speed_file("rx_bytes");
thread::sleep(Duration::from_secs(1));
@@ -16,10 +17,11 @@ pub fn get_netspeed() -> String {
let tx = calculate(tx_bps);
let rx = calculate(rx_bps);
- format!(
+ let data = format!(
" {} {} {} {} {}",
CONFIG.netspeed.recieve_icon, rx, CONFIG.netspeed.transmit_icon, tx, CONFIG.seperator
- )
+ );
+ ThreadsData::NetSpeed(data)
}
fn parse_speed_file(pth: &str) -> u64 {
diff --git a/src/utils/pub_ip.rs b/src/utils/pub_ip.rs
index ce2da92..c0e6e4c 100644
--- a/src/utils/pub_ip.rs
+++ b/src/utils/pub_ip.rs
@@ -1,15 +1,17 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
-pub fn get_pub_ip() -> String {
+pub fn get_pub_ip() -> ThreadsData {
let url = format!("http://api.ipify.org");
- let err_string = String::from("Error");
+ let _err = String::from("Error");
let res = match minreq::get(url).send() {
Ok(resp) => match resp.as_str() {
Ok(res_str) => res_str.trim().to_string(),
- Err(_) => err_string,
+ Err(_) => _err,
},
- Err(_) => err_string,
+ Err(_) => _err,
};
- format!(" {} {} {}", CONFIG.pub_ip.icon, res, CONFIG.seperator)
+ let data = format!(" {} {} {}", CONFIG.pub_ip.icon, res, CONFIG.seperator);
+ ThreadsData::PubIp(data)
}
diff --git a/src/utils/spotify.rs b/src/utils/spotify.rs
index a73128d..9e5c5da 100644
--- a/src/utils/spotify.rs
+++ b/src/utils/spotify.rs
@@ -1,14 +1,15 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
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() -> String {
+
+pub fn get_spotify() -> ThreadsData {
+ let empty_data = ThreadsData::Spotify(String::from(""));
let conn = match Connection::new_session() {
Ok(conn) => conn,
- _ => return String::from(""),
+ _ => return empty_data,
};
let p = conn.with_proxy(
@@ -19,7 +20,7 @@ pub fn get_spotify() -> String {
let metadata: arg::PropMap = match p.get("org.mpris.MediaPlayer2.Player", "Metadata") {
Ok(data) => data,
- _ => return String::from(""),
+ _ => return empty_data,
};
let title: Option<&String> = arg::prop_cast(&metadata, "xesam:title");
@@ -38,8 +39,9 @@ pub fn get_spotify() -> String {
None => "",
};
- format!(
+ let data = format!(
" {} {} - {} {}",
CONFIG.spotify.icon, artist, title, CONFIG.seperator
- )
+ );
+ ThreadsData::Spotify(data)
}
diff --git a/src/utils/time.rs b/src/utils/time.rs
index fcfb691..191d91b 100644
--- a/src/utils/time.rs
+++ b/src/utils/time.rs
@@ -1,13 +1,16 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
use chrono::prelude::*;
-pub fn get_time() -> String {
+pub fn get_time() -> ThreadsData {
let now = Local::now();
- format!(
+ let data = format!(
" {} {} {}",
CONFIG.time.icon,
now.format(&CONFIG.time.format),
CONFIG.seperator
- )
+ );
+
+ ThreadsData::Time(data)
}
diff --git a/src/utils/uptime.rs b/src/utils/uptime.rs
index b5de930..140c4c4 100644
--- a/src/utils/uptime.rs
+++ b/src/utils/uptime.rs
@@ -1,12 +1,11 @@
use crate::config::CONFIG;
-use std::fs::File;
-use std::io::Read;
+use crate::types::ThreadsData;
+use std::fs::read_to_string;
-pub fn get_uptime() -> Result<String, std::io::Error> {
- let mut buf = String::new();
- match File::open("/proc/uptime") {
- Ok(mut file) => file.read_to_string(&mut buf)?,
- _ => return Ok("cant find uptime file!".to_string()),
+pub fn get_uptime() -> ThreadsData {
+ let buf = match read_to_string("/proc/uptime") {
+ Ok(data) => data,
+ _ => return ThreadsData::Uptime("cant find uptime file!".to_string()),
};
let buf: f32 = buf.split(' ').collect::<Vec<&str>>()[0].parse().unwrap();
@@ -21,5 +20,5 @@ pub fn get_uptime() -> Result<String, std::io::Error> {
format!("{} min", minutes)
};
let result = format!(" {} {} {}", CONFIG.uptime.icon, uptime, CONFIG.seperator);
- Ok(result)
+ ThreadsData::Uptime(result)
}
diff --git a/src/utils/volume.rs b/src/utils/volume.rs
index c020ce8..05c1dbe 100644
--- a/src/utils/volume.rs
+++ b/src/utils/volume.rs
@@ -1,8 +1,9 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
use alsa::mixer::{Mixer, SelemChannelId, SelemId};
// getting volume percentage
-pub fn get_volume() -> String {
+pub fn get_volume() -> ThreadsData {
let card = if CONFIG.volume.card == "PULSE" {
"pulse"
} else {
@@ -27,5 +28,6 @@ pub fn get_volume() -> String {
((raw_volume as f64 / range as f64) * 100.) as u64
};
- format!(" {} {}% {}", CONFIG.volume.icon, vol, CONFIG.seperator)
+ let data = format!(" {} {}% {}", CONFIG.volume.icon, vol, CONFIG.seperator);
+ ThreadsData::Sound(data)
}
diff --git a/src/utils/weather.rs b/src/utils/weather.rs
index 9dc053b..11e446f 100644
--- a/src/utils/weather.rs
+++ b/src/utils/weather.rs
@@ -1,7 +1,8 @@
use crate::config::CONFIG;
+use crate::types::ThreadsData;
// will make a GET request from wttr.in
-pub fn get_weather() -> String {
+pub fn get_weather() -> ThreadsData {
let format = if CONFIG.weather.format.is_empty() {
String::from("%l:+%t")
} else {
@@ -18,5 +19,6 @@ pub fn get_weather() -> String {
Err(_) => err_string,
};
- format!(" {} {} {}", CONFIG.weather.icon, res, CONFIG.seperator)
+ let data = format!(" {} {} {}", CONFIG.weather.icon, res, CONFIG.seperator);
+ ThreadsData::Weather(data)
}
generated by cgit v1.2.3 (git 2.39.1) at 2025-09-11 13:49:36 +0000