From 14a038bbdaf444adead2aba0fdf9c6564624850f Mon Sep 17 00:00:00 2001
From: Li Haoyuan <1513624626@qq.com>
Date: Wed, 10 Jul 2024 13:40:32 +0800
Subject: [PATCH] feat: Add battery prompt for Windows

---
 internal/p10k.zsh | 53 ++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 52 insertions(+), 1 deletion(-)

diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index d2d261b2..fd030c5a 100644
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -1332,7 +1332,6 @@ _p9k_prompt_fvm_init() {
 }
 
 ################################################################
-# Segment that displays the battery status in levels and colors
 prompt_battery() {
   [[ $_p9k_os == (Linux|Android) ]] && _p9k_prompt_battery_set_args
   (( $#_p9k__battery_args )) && _p9k_prompt_segment "${_p9k__battery_args[@]}"
@@ -1344,6 +1343,16 @@ _p9k_prompt_battery_init() {
     _p9k__async_segments_compute+='_p9k_worker_invoke battery _p9k_prompt_battery_compute'
     return
   fi
+
+  if [[ $_p9k_os == Windows ]]; then
+    if command -v wmic > /dev/null 2>&1; then # Test "wimc" command
+      _p9k__async_segments_compute+='_p9k_worker_invoke battery _p9k_prompt_battery_compute'
+      return
+    else
+      typeset -g "_p9k__segment_cond_${_p9k__prompt_side}[_p9k__segment_index]"='${:-}'
+    fi
+  fi
+
   if [[ $_p9k_os != (Linux|Android) ||
         -z /sys/class/power_supply/(CMB*|BAT*|*battery)/(energy_full|charge_full|charge_counter)(#qN) ]]; then
     typeset -g "_p9k__segment_cond_${_p9k__prompt_side}[_p9k__segment_index]"='${:-}'
@@ -1456,6 +1465,48 @@ _p9k_prompt_battery_set_args() {
       fi
     ;;
 
+    Windows)
+      # See Win32_Battery class members at https://learn.microsoft.com/en-us/windows/win32/cimwin32prov/win32-battery
+      battery_data=$(wmic path Win32_Battery get EstimatedChargeRemaining,BatteryStatus,EstimatedRunTime /format:list)
+      bat_percent=$(echo "$battery_data" | grep "EstimatedChargeRemaining" | awk -F'=' '{print $2}')
+      bat_status=$(echo "$battery_data" | grep "BatteryStatus" | awk -F'=' '{print $2}')
+      bat_remain_minutes=$(echo "$battery_data" | grep "EstimatedRunTime" | awk -F'=' '{print $2}')
+
+      # Missing critical parameters
+      if [[ -z "$bat_percent" || -z "$bat_status" || -z "$bat_remain_minutes" ]]; then
+          return 0
+      fi
+
+      case $bat_status in 
+        '1'|'4'|'5')
+            # When the battery is charging, system will show that bat_remain_minutes is 71582788 (min)
+            (( bat_remain_minutes < 71582788 )) && remain=$((bat_remain_minutes/60)):${(l#2##0#)$((bat_remain_minutes%60))} || remain=''
+            if (( bat_percent < _POWERLEVEL9K_BATTERY_LOW_THRESHOLD )); then
+                state=LOW
+            else
+                state=DISCONNECTED
+            fi
+        ;;
+        '2'|'6'|'7'|'8'|'9'|'11')
+            if (( bat_percent == 100 )); then
+                remain=''
+                state=CHARGED
+            else
+                (( bat_remain_minutes < 71582788 )) && remain=$((bat_remain_minutes/60)):${(l#2##0#)$((bat_remain_minutes%60))} || remain=''
+                state=CHARGING
+            fi
+        ;;
+        '3')
+            (( bat_remain_minutes < 71582788 )) && remain=$((bat_remain_minutes/60)):${(l#2##0#)$((bat_remain_minutes%60))} || remain=''
+            state=CHARGED
+        ;;
+        *)
+            state=CHARGED
+            remain=''
+        ;;
+      esac
+    ;;
+
     *)
       return 0
     ;;