feat: Implement PMU management functions to keep power management unit awake

Author: 02loveslollipop

tx_websocket_gateway/LoRaBoards.cpp

@@ -44,6 +44,26 @@ static void setPmuFlag()
 {
     pmuInterrupt = true;
 }
+
+static void keepPmuAwake()
+{
+    if (!PMU) {
+        return;
+    }
+
+    const auto chip = PMU->getChipModel();
+    if (chip == XPOWERS_AXP2101) {
+        auto *axp2101 = static_cast<XPowersAXP2101 *>(PMU);
+        axp2101->disableSleep();
+        axp2101->disableLongPressShutdown();
+        axp2101->setLongPressRestart();
+        axp2101->disableWatchdog();
+        axp2101->setLowBatShutdownThreshold(0);
+    } else if (chip == XPOWERS_AXP192) {
+        auto *axp192 = static_cast<XPowersAXP192 *>(PMU);
+        axp192->disablePowerKeyLongPressPowerOff();
+    }
+}
 #endif
 
 bool beginPower()
@@ -76,6 +96,7 @@ bool beginPower()
     }
 
     PMU->setChargingLedMode(XPOWERS_CHG_LED_CTRL_CHG);
+    keepPmuAwake();
 
     pinMode(PMU_IRQ, INPUT_PULLUP);
     attachInterrupt(PMU_IRQ, setPmuFlag, FALLING);
@@ -847,4 +868,4 @@ bool beginGPS()
     SerialGPS.write("$PCAS11,3*1E\r\n");
     return result;
 }
-#endif
+#endif

tx_websocket_gateway/tx_websocket_gateway.ino

@@ -1,5 +1,8 @@
 #include <Arduino.h>
 #include <WiFi.h>
+#if defined(ESP32)
+#include <esp_wifi.h>
+#endif
 #include <ArduinoWebsockets.h>
 #include <ArduinoJson.h>
 #include <SPI.h>
@@ -82,13 +85,20 @@ void loop() {
 
     wsClient.poll();
     publishStatus();
+#ifdef HAS_PMU
+    loopPMU();
+#endif
     delay(5);
 }
 
 // ----- Wi-Fi & WebSocket ---------------------------------------------
 void connectWiFi() {
     Serial.printf("[WiFi] Connecting to %s\n", WIFI_SSID);
     WiFi.mode(WIFI_STA);
+#if defined(ESP32)
+    WiFi.setSleep(false);
+    esp_wifi_set_ps(WIFI_PS_NONE);
+#endif
     WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
 
     const uint32_t start = millis();