src/charge_controller.rs

@@ -1,100 +1,189 @@
-use std::sync::{Arc, Mutex, MutexGuard};
+use std::{
+    sync::{
+        Arc, Mutex, MutexGuard,
+    },
+};
 
 use embedded_hal::digital::v2::{OutputPin, PinState};
 use embedded_svc::event_bus::{EventBus, Postbox};
-use esp_idf_hal::gpio::{Gpio7, Output};
+use esp_idf_hal::gpio::{Gpio6, Output};
 use esp_idf_svc::eventloop::{
     Background, EspBackgroundEventLoop, EspEventFetchData, EspEventLoop, EspEventPostData,
     EspSubscription, EspTypedEventDeserializer, EspTypedEventSerializer, EspTypedEventSource, User,
 };
 use esp_idf_sys::c_types;
-use log::{warn};
+use log::{info, warn, debug};
 use serde_json::json;
 
 use crate::voltage_detection::{VoltageDetectionWorker, VOLTAGE_EVENTLOOP};
 
-pub static mut CHARGE_STATE_EVENT_LOOP: Option<
+const WAITING_OFF_SECONDS: u8 = 60;
+
+pub static mut DC_OUT_STATE_EVENT_LOOP: Option<
     EspEventLoop<esp_idf_svc::eventloop::User<Background>>,
 > = None;
 
 #[derive(Debug, Clone, Copy, PartialEq)]
-pub enum ChargeStatus {
+pub enum DcOutStatus {
-    Charging,
+    WaitingOn(u8),
-    Charged,
+    On,
+    Off,
+    WaitingOff,
+    TurningOff(u8),
 }
 
 #[derive(Debug, Clone, Copy)]
-pub struct ChargeControllerState {
+pub struct DcOutControllerState {
-    pub status: ChargeStatus,
+    pub status: DcOutStatus,
     pub pin_state: PinState,
 }
 
-impl ChargeControllerState {
+impl DcOutControllerState {
     pub fn new() -> Self {
         Self {
-            status: ChargeStatus::Charging,
+            status: DcOutStatus::On,
             pin_state: PinState::Low,
         }
     }
 
+    fn handle_adapter_down(&mut self) {
+        info!("status: {:?}", self);
+        match self.status {
+            DcOutStatus::On => {
+                self.status = DcOutStatus::WaitingOff;
+            }
+            DcOutStatus::WaitingOn(_) => {
+                self.status = DcOutStatus::Off;
+            }
+            DcOutStatus::TurningOff(seconds) => {
+                let seconds = seconds - 1;
+                if seconds > 0 {
+                    self.status = DcOutStatus::TurningOff(seconds);
+                } else {
+                    self.status = DcOutStatus::Off;
+                }
+            }
+            _ => {}
+        };
+    }
+
+    fn turn_off(&mut self) {
+        match self.status {
+            DcOutStatus::On => {
+                self.status = DcOutStatus::TurningOff(WAITING_OFF_SECONDS);
+            }
+            DcOutStatus::WaitingOff => {
+                self.status = DcOutStatus::TurningOff(WAITING_OFF_SECONDS);
+            }
+            DcOutStatus::WaitingOn(_) => {
+                self.status = DcOutStatus::Off;
+            }
+            DcOutStatus::TurningOff(seconds) => {
+                let seconds = seconds - 1;
+                if seconds > 0 {
+                    self.status = DcOutStatus::TurningOff(seconds);
+                } else {
+                    self.status = DcOutStatus::Off;
+                }
+            }
+            _ => {}
+        };
+    }
+
+    fn turn_on(&mut self) {
+        match self.status {
+            DcOutStatus::WaitingOff => {
+                self.status = DcOutStatus::On;
+            }
+            DcOutStatus::Off => {
+                self.status = DcOutStatus::WaitingOn(WAITING_OFF_SECONDS);
+            }
+            DcOutStatus::WaitingOn(seconds) => {
+                let seconds = seconds - 1;
+                if seconds > 0 {
+                    self.status = DcOutStatus::WaitingOn(seconds);
+                } else {
+                    self.status = DcOutStatus::On;
+                }
+            }
+            DcOutStatus::TurningOff(seconds) => {
+                let seconds = seconds - 1;
+                if seconds > 0 {
+                    self.status = DcOutStatus::TurningOff(seconds);
+                } else {
+                    self.status = DcOutStatus::On;
+                }
+            }
+            _ => {}
+        };
+    }
+
     pub fn to_json(&self) -> String {
         let status = match self.status {
-            ChargeStatus::Charging => "Charging",
+            DcOutStatus::WaitingOn(_) => "WaitingOn",
-            ChargeStatus::Charged => "Charged",
+            DcOutStatus::On => "On",
+            DcOutStatus::Off => "Off",
+            DcOutStatus::WaitingOff => "WaitingOff",
+            DcOutStatus::TurningOff(_) => "TurningOff",
         };
         let pin_state = match self.pin_state {
             PinState::Low => "Low",
             PinState::High => "High",
         };
-        json!({ "status": status, "pin_state": pin_state}).to_string()
+        let seconds: i16 = match self.status {
+            DcOutStatus::WaitingOn(seconds) => seconds.into(),
+            DcOutStatus::TurningOff(seconds) => seconds.into(),
+            _ => -1,
+        };
+        json!({ "status": status, "pin_state": pin_state, "seconds": seconds }).to_string()
     }
 }
 
-impl EspTypedEventSource for ChargeControllerState {
+impl EspTypedEventSource for DcOutControllerState {
     fn source() -> *const c_types::c_char {
         b"Charge\0".as_ptr() as *const _
     }
 }
 
-impl EspTypedEventSerializer<ChargeControllerState> for ChargeControllerState {
+impl EspTypedEventSerializer<DcOutControllerState> for DcOutControllerState {
     fn serialize<R>(
-        event: &ChargeControllerState,
+        event: &DcOutControllerState,
         f: impl for<'a> FnOnce(&'a EspEventPostData) -> R,
     ) -> R {
         f(&unsafe { EspEventPostData::new(Self::source(), Self::event_id(), event) })
     }
 }
 
-impl EspTypedEventDeserializer<ChargeControllerState> for ChargeControllerState {
+impl EspTypedEventDeserializer<DcOutControllerState> for DcOutControllerState {
     fn deserialize<R>(
         data: &EspEventFetchData,
-        f: &mut impl for<'a> FnMut(&'a ChargeControllerState) -> R,
+        f: &mut impl for<'a> FnMut(&'a DcOutControllerState) -> R,
     ) -> R {
         f(unsafe { data.as_payload() })
     }
 }
 
-type ChargeCtlPin = Gpio7<Output>;
+type DcOutPin = Gpio6<Output>;
 
-pub struct ChargeController {
+pub struct DcOutController {
-    pub state: ChargeControllerState,
+    pub state: DcOutControllerState,
     voltage_subscription: Option<EspSubscription<User<Background>>>,
-    pin: Arc<Mutex<ChargeCtlPin>>,
+    pin: Arc<Mutex<DcOutPin>>,
 }
 
-impl ChargeController {
+impl DcOutController {
     pub fn new() -> anyhow::Result<Self> {
-        let pin = unsafe { Gpio7::<Output>::new() }
+        let pin = unsafe { Gpio6::<Output>::new() }
             .into_output()
             .map_err(|err| anyhow::anyhow!("Make Gpio6 Into output Failed. {}", err))?;
 
         match EspBackgroundEventLoop::new(&Default::default()) {
-            Ok(eventloop) => unsafe { CHARGE_STATE_EVENT_LOOP = Some(eventloop) },
+            Ok(eventloop) => unsafe { DC_OUT_STATE_EVENT_LOOP = Some(eventloop) },
             Err(err) => anyhow::bail!("Init Event Loop failed. {:?}", err),
         }
 
         anyhow::Ok(Self {
-            state: ChargeControllerState::new(),
+            state: DcOutControllerState::new(),
             voltage_subscription: None,
             pin: Arc::new(Mutex::new(pin)),
         })
@@ -107,39 +196,33 @@ impl ChargeController {
         if let Some(event_loop) = unsafe { VOLTAGE_EVENTLOOP.as_mut() } {
             let voltage_subscription = event_loop
                 .subscribe(move |obj: &VoltageDetectionWorker| {
-                    match state.status {
+                    if obj.adapter_voltage < 1000 {
-                        ChargeStatus::Charging => {
+                        if obj.battery_voltage < 1000 {
-                            if obj.battery_voltage < 12600 {
+                            state.turn_off();
-                                state.status = ChargeStatus::Charging;
                         } else {
-                                state.status = ChargeStatus::Charged;
+                            state.handle_adapter_down();
                         }
-                        }
-                        ChargeStatus::Charged => {
-                            if obj.battery_voltage < 10500 {
-                                state.status = ChargeStatus::Charging;
                     } else {
-                                state.status = ChargeStatus::Charged;
+                        state.turn_on();
-                            }
-                        }
                     }
 
                     match pin.lock() {
                         Ok(pin) => {
-                            if let Err(err) = Self::output_ctl(&mut state, pin) {
+                            if let Err(err) = Self::output_ctl(state, pin) {
                                 warn!("Put Control Pin State Failed. {}", err);
                             }
                         }
                         Err(_) => todo!(),
                     }
 
-                    if let Some(event_loop) = unsafe { CHARGE_STATE_EVENT_LOOP.as_mut() } {
+                    if let Some(event_loop) = unsafe { DC_OUT_STATE_EVENT_LOOP.as_mut() } {
                         if let Err(err) = event_loop.post(&state, None) {
                             warn!("Post DC Out Status Failed. {}", err);
                         }
                     } else {
-                        warn!("CHARGE_STATE_EVENT_LOOP is None");
+                        warn!("DC_OUT_STATE_EVENT_LOOP is None");
                     }
+                    debug!("status: {:?}", state);
                 })
                 .map_err(|err| anyhow::anyhow!("Subscribe Voltage Failed. {}", err))?;
             self.voltage_subscription = Some(voltage_subscription);
@@ -149,14 +232,14 @@ impl ChargeController {
         anyhow::Ok(())
     }
     fn output_ctl(
-        state: &mut ChargeControllerState,
+        mut state: DcOutControllerState,
-        mut pin: MutexGuard<ChargeCtlPin>,
+        mut pin: MutexGuard<DcOutPin>,
     ) -> anyhow::Result<()> {
-        if ChargeStatus::Charging == state.status {
+        if DcOutStatus::Off == state.status && state.pin_state == PinState::Low {
             pin.set_high()
                 .map_err(|err| anyhow::anyhow!("Set DC Output Control Pin High Failed. {}", err))?;
             state.pin_state = PinState::High;
-        } else if ChargeStatus::Charged == state.status {
+        } else if DcOutStatus::On == state.status && state.pin_state == PinState::High {
             pin.set_low()
                 .map_err(|err| anyhow::anyhow!("Set DC Output Control Pin Low Failed. {}", err))?;
             state.pin_state = PinState::Low;

src/dc_out_controller.rs

@@ -47,6 +47,7 @@ impl DcOutControllerState {
     }
 
     fn handle_adapter_down(&mut self) {
+        info!("status: {:?}", self);
         match self.status {
             DcOutStatus::On => {
                 self.status = DcOutStatus::WaitingOff;
@@ -207,7 +208,7 @@ impl DcOutController {
 
                     match pin.lock() {
                         Ok(pin) => {
-                            if let Err(err) = Self::output_ctl(&mut state, pin) {
+                            if let Err(err) = Self::output_ctl(state, pin) {
                                 warn!("Put Control Pin State Failed. {}", err);
                             }
                         }
@@ -221,6 +222,7 @@ impl DcOutController {
                     } else {
                         warn!("DC_OUT_STATE_EVENT_LOOP is None");
                     }
+                    debug!("status: {:?}", state);
                 })
                 .map_err(|err| anyhow::anyhow!("Subscribe Voltage Failed. {}", err))?;
             self.voltage_subscription = Some(voltage_subscription);
@@ -230,14 +232,14 @@ impl DcOutController {
         anyhow::Ok(())
     }
     fn output_ctl(
-        state: &mut DcOutControllerState,
+        mut state: DcOutControllerState,
         mut pin: MutexGuard<DcOutPin>,
     ) -> anyhow::Result<()> {
-        if DcOutStatus::Off == state.status {
+        if DcOutStatus::Off == state.status && state.pin_state == PinState::Low {
             pin.set_high()
                 .map_err(|err| anyhow::anyhow!("Set DC Output Control Pin High Failed. {}", err))?;
             state.pin_state = PinState::High;
-        } else if DcOutStatus::On == state.status {
+        } else if DcOutStatus::On == state.status && state.pin_state == PinState::High {
             pin.set_low()
                 .map_err(|err| anyhow::anyhow!("Set DC Output Control Pin Low Failed. {}", err))?;
             state.pin_state = PinState::Low;

src/main.rs

@@ -15,13 +15,12 @@ mod message_queue;
 mod time;
 mod voltage_detection;
 mod wifi;
-mod charge_controller;
 
 use crate::{
     beep::{ringtone, Beep},
     dc_out_controller::{DcOutController, DcOutControllerState, DC_OUT_STATE_EVENT_LOOP},
     message_queue::MqDto,
-    voltage_detection::{VoltageDetectionWorker, VOLTAGE_EVENTLOOP}, charge_controller::{CHARGE_STATE_EVENT_LOOP, ChargeControllerState, ChargeController},
+    voltage_detection::{VoltageDetectionWorker, VOLTAGE_EVENTLOOP},
 };
 use crate::{
     message_queue::MessageQueue, time::Time, voltage_detection::VoltageDetection, wifi::Internet,
@@ -57,7 +56,7 @@ fn main() {
         blink.play();
     });
 
-    let voltage_detection = VoltageDetection::new();
+    let mut voltage_detection = VoltageDetection::new();
 
     voltage_detection.unwrap()
         .watching()
@@ -78,12 +77,6 @@ fn main() {
         .watch()
         .expect("Can not watch for dc_out_controller");
 
-    let mut charge_controller = ChargeController::new().expect("Can not get ChargeController instance");
-    charge_controller
-        .watch()
-        .expect("Can not watch for charge_controller");
-
-
     sleep(Duration::from_millis(100));
 
     let mut _mq = MessageQueue::new();
@@ -98,7 +91,6 @@ fn main() {
 
     let _mq_tx_for_voltage = _mq.tx.clone();
     let _mq_tx_for_dc_out_state = _mq.tx.clone();
-    let _mq_tx_for_charge_state = _mq.tx.clone();
 
     let _voltage_subscription;
     if let Some(voltage_event_loop) = unsafe { VOLTAGE_EVENTLOOP.as_mut() } {
@@ -159,29 +151,6 @@ fn main() {
         panic!("DC_OUT_STATE_EVENT_LOOP is undefined!");
     }
 
-    let _charge_state_subscription;
-    if let Some(charge_state_event_loop) = unsafe { CHARGE_STATE_EVENT_LOOP.as_mut() } {
-        _charge_state_subscription = charge_state_event_loop
-            .subscribe(move |message: &ChargeControllerState| {
-                match _mq_tx_for_charge_state.lock() {
-                    Ok(tx) => {
-                        let result = tx.send(MqDto {
-                            topic: "charge_state".to_string(),
-                            message: message.to_json(),
-                        });
-
-                        if let Err(err) = result {
-                            warn!("send charge_state message failed. {}", err)
-                        }
-                    }
-                    Err(err) => warn!("send charge_state to mq message failed. {}", err),
-                }
-            })
-            .expect(" Listening Event Loop Failed");
-    } else {
-        panic!("CHARGE_STATE_EVENT_LOOP is undefined!");
-    }
-
     loop {
         sleep(Duration::from_millis(100));
     }