Xinqi Bao's Git

Remove units from numbers
[slstatus.git] / components / battery.c
index 5031608..ff06f40 100644 (file)
@@ -7,6 +7,26 @@
 
 #if defined(__linux__)
        #include <limits.h>
 
 #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)
 
        const char *
        battery_perc(const char *bat)
@@ -23,7 +43,7 @@
                        return NULL;
                }
 
                        return NULL;
                }
 
-               return bprintf("%d%%", perc);
+               return bprintf("%d", perc);
        }
 
        const char *
        }
 
        const char *
                        return NULL;
                }
 
                        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 (!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;
                        }
 
                                return NULL;
                        }
 
                struct apm_power_info apm_info;
 
                if (load_apm_power_info(&apm_info)) {
                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;
                }
 
                return NULL;