From 6b92cecef43b824c0dbae69312fcf669a5058853 Mon Sep 17 00:00:00 2001
From: Mustafa Salih <37256744+MustafaSalih1993@users.noreply.github.com>
Date: Thu, 29 Apr 2021 21:59:59 +0300
Subject: getting battery percentage programmatically (#50)

* getting battery percentage programmatically

* cargo.lock edit
---
 src/utils/battery.rs | 39 +++++++++++++++++----------------------
 1 file changed, 17 insertions(+), 22 deletions(-)

(limited to 'src/utils/battery.rs')

diff --git a/src/utils/battery.rs b/src/utils/battery.rs
index 041965a..d3688e7 100644
--- a/src/utils/battery.rs
+++ b/src/utils/battery.rs
@@ -1,37 +1,32 @@
 use crate::config::CONFIG;
 use crate::types::ThreadsData;
-use std::fs::read_to_string;
+use battery::Manager;
+
+// TODO: better error handeling
 
 // getting battery percentage
 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
-    );
-    let battery_charge_now_file = format!(
-        "/sys/class/power_supply/{}/charge_now",
-        CONFIG.battery.source
-    );
-
-    let buf = match read_to_string(battery_full_cap_file) {
-        Ok(file) => file,
-        Err(_) => return error,
+    let battery_manager = if let Ok(manager) = Manager::new() {
+        manager
+    } else {
+        return ThreadsData::Battery(String::from("Cannot Create Battery Manager!"));
     };
-    let full_design = buf.trim().parse::<u32>().unwrap();
 
-    let buf = match read_to_string(&battery_charge_now_file) {
-        Ok(data) => data,
-        _ => return error,
+    let mut batteries = if let Ok(batteries) = battery_manager.batteries() {
+        batteries
+    } else {
+        return ThreadsData::Battery(String::from("Cannot Get Battery!"));
     };
 
-    let charge_now = buf.trim().parse::<u32>().unwrap();
+    let percentage = if let Some(battery) = batteries.next() {
+        f32::from(battery.unwrap().state_of_charge()) * 100.0
+    } else {
+        return ThreadsData::Battery(String::from("Cannot Read Battery!"));
+    };
 
-    let battery_percentage = (charge_now as f32 / full_design as f32) * 100.0;
     let result = format!(
         "  {}  {:.0}%  {}",
-        CONFIG.battery.icon, battery_percentage, CONFIG.seperator
+        CONFIG.battery.icon, percentage, CONFIG.seperator
     );
     ThreadsData::Battery(result)
 }
-- 
cgit v1.2.3