X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/74c4f4ebdae8a12fc95840954dde574692234b01..ec5c35ec9f23254f09e734d0c1880559774dfa52:/components/battery.c?ds=inline

diff --git a/components/battery.c b/components/battery.c
index 5031608..ff06f40 100644
--- a/components/battery.c
+++ b/components/battery.c
@@ -7,6 +7,26 @@
 
 #if defined(__linux__)
 	#include <limits.h>
+	#include <unistd.h>
+
+	#define CHARGE_NOW    "/sys/class/power_supply/%s/charge_now"
+	#define ENERGY_NOW    "/sys/class/power_supply/%s/energy_now"
+	#define CURRENT_NOW   "/sys/class/power_supply/%s/current_now"
+	#define POWER_NOW     "/sys/class/power_supply/%s/power_now"
+
+	static const char *
+	pick(const char *bat, const char *f1, const char *f2, char *path, size_t length)
+	{
+		if (esnprintf(path, length, f1, bat) > 0 && access(path, R_OK) == 0) {
+			return f1;
+		}
+
+		if (esnprintf(path, length, f2, bat) > 0 && access(path, R_OK) == 0) {
+			return f2;
+		}
+
+		return NULL;
+	}
 
 	const char *
 	battery_perc(const char *bat)
@@ -23,7 +43,7 @@
 			return NULL;
 		}
 
-		return bprintf("%d%%", perc);
+		return bprintf("%d", perc);
 	}
 
 	const char *
@@ -72,21 +92,14 @@
 			return NULL;
 		}
 
+		if (pick(bat, CHARGE_NOW, ENERGY_NOW, path, sizeof (path)) == NULL ||
+		    pscanf(path, "%d", &charge_now) < 0) {
+			return NULL;
+		}
+
 		if (!strcmp(state, "Discharging")) {
-			if (esnprintf(path, sizeof(path),
-                                      "/sys/class/power_supply/%s/charge_now",
-			              bat) < 0) {
-				return NULL;
-			}
-			if (pscanf(path, "%d", &charge_now) != 1) {
-				return NULL;
-			}
-			if (esnprintf(path, sizeof(path),
-			              "/sys/class/power_supply/%s/current_now",
-			              bat) < 0) {
-				return NULL;
-			}
-			if (pscanf(path, "%d", &current_now) != 1) {
+			if (pick(bat, CURRENT_NOW, POWER_NOW, path, sizeof (path)) == NULL ||
+			    pscanf(path, "%d", &current_now) < 0) {
 				return NULL;
 			}
 
@@ -131,7 +144,7 @@
 		struct apm_power_info apm_info;
 
 		if (load_apm_power_info(&apm_info)) {
-			return bprintf("%d%%", apm_info.battery_life);
+			return bprintf("%d", apm_info.battery_life);
 		}
 
 		return NULL;