support multiple batteries

1 files changed, 8 insertions, 7 deletions

diff --git a/bin/fancy-prompt b/bin/fancy-prompt
index 148cb05..b99d1c3 100755
--- a/bin/fancy-prompt
+++ b/bin/fancy-prompt
@@ -1,4 +1,5 @@
 #!/usr/bin/env perl
+use 5.010;
 # keep these commented out in actual use, for speed (saves ~2ms)
 # use strict;
 # use warnings;
@@ -178,13 +179,13 @@ print $prompt_escape eq 'zsh' ? "%{\n%}" : "\n";
 print join(' ', $errstr, $promptstr, '');
 
 sub battery {
-    my $now = slurp('/sys/class/power_supply/BAT0/charge_now');
-    $now = slurp('/sys/class/power_supply/BAT0/energy_now')
-        if !defined($now);
-    my $full = slurp('/sys/class/power_supply/BAT0/charge_full');
-    $full = slurp('/sys/class/power_supply/BAT0/energy_full')
-        if !defined($full);
-    return if !defined($full);
+    my $now = 0;
+    my $full = 0;
+    for my $battery (glob('/sys/class/power_supply/*')) {
+        $now  += slurp("$battery/charge_now")  // slurp("$battery/energy_now");
+        $full += slurp("$battery/charge_full") // slurp("$battery/energy_full");
+    }
+    return unless $full;
     return int(100 * $now / $full);
 }