From 3cf929875613cd8664a3ccd41f0dbe7ea05d6245 Mon Sep 17 00:00:00 2001
From: Empire <empirephoenix@yahoo.de>
Date: Fri, 28 Feb 2025 22:07:20 +0100
Subject: [PATCH] add awake and charge indicator led functions

---
 rust/src/main.rs      |  5 +++++
 rust/src/plant_hal.rs | 10 ++++++++++
 2 files changed, 15 insertions(+)

diff --git a/rust/src/main.rs b/rust/src/main.rs
index 4bb6f2b..3f3b1ad 100644
--- a/rust/src/main.rs
+++ b/rust/src/main.rs
@@ -180,6 +180,8 @@ struct PlantStateMQTT<'a> {
     next_pump: &'a str,
 }
 
+
+
 fn safe_main() -> anyhow::Result<()> {
     // It is necessary to call this function once. Otherwise some patches to the runtime
     // implemented by esp-idf-sys might not link properly. See https://github.com/esp-rs/esp-idf-template/issues/71
@@ -256,7 +258,9 @@ fn safe_main() -> anyhow::Result<()> {
         log(log::LogMessage::YearInplausibleForceConfig, 0,0,"","");
     }
 
+    
     println!("cur is {}", cur);
+    board.update_charge_indicator();
 
     if board.get_restart_to_conf() {
         log(log::LogMessage::ConfigModeSoftwareOverride, 0,0,"","");
@@ -941,6 +945,7 @@ fn wait_infinity(wait_type: WaitType, reboot_now: Arc<AtomicBool>) -> ! {
             led_count += 1;
         };
         unsafe {
+            BOARD_ACCESS.lock().unwrap().update_charge_indicator();
             //do not trigger watchdog
             for i in 0..8 {
                 BOARD_ACCESS.lock().unwrap().fault(i, i < led_count);
diff --git a/rust/src/plant_hal.rs b/rust/src/plant_hal.rs
index 113e45e..d0c7bf3 100644
--- a/rust/src/plant_hal.rs
+++ b/rust/src/plant_hal.rs
@@ -216,7 +216,17 @@ pub struct BackupHeader{
 }
 
 impl PlantCtrlBoard<'_> {
+    pub fn update_charge_indicator(&mut self){
+        let is_charging = match self.battery_driver.average_current() {
+            OkStd(current) => current < 20,
+            Err(_) => false,
+        };
+        self.shift_register.decompose()[CHARGING].set_state(is_charging.into()).unwrap();
+    }
+
+
     pub fn deep_sleep(&mut self, duration_in_ms:u64) -> !{
+        self.shift_register.decompose()[AWAKE].set_low().unwrap();
         unsafe { 
             //if we dont do this here, we might just revert a newly flashed firmeware
             mark_app_valid();