From ef6b0415290df29837c80ef44a0844823b07c262 Mon Sep 17 00:00:00 2001
From: Phat Nguyen <phat.nt@hotmail.com>
Date: Tue, 4 Jun 2024 18:01:55 +0700
Subject: [PATCH] Fix: ota not callback on first powerup perform

---
 examples/OneOpenAir/OneOpenAir.ino | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/OneOpenAir/OneOpenAir.ino b/examples/OneOpenAir/OneOpenAir.ino
index e323b9a..d794795 100644
--- a/examples/OneOpenAir/OneOpenAir.ino
+++ b/examples/OneOpenAir/OneOpenAir.ino
@@ -219,6 +219,7 @@ void setup() {
         #ifdef ESP8266
           // ota not supported
         #else
+          otaHandler.setHandlerCallback(otaHandlerCallback);
           otaHandler.updateFirmwareIfOutdated(ag->deviceId());
 
           /** Update first OTA */
@@ -485,6 +486,7 @@ static bool sgp41Init(void) {
 }
 
 static void otaHandlerCallback(OtaState state, String mesasge) {
+  Serial.println("OTA message: " + mesasge);
   switch (state) {
   case OtaState::OTA_STATE_BEGIN:
     displayExecuteOta(state, fwNewVersion, 0);
@@ -562,7 +564,7 @@ static void displayExecuteOta(OtaState state, String msg, int processing) {
       while (i != 0) {
         i = i - 1;
         if (ag->isOne()) {
-          oledDisplay.showFirmwareUpdateSuccess(String(i));
+          oledDisplay.showFirmwareUpdateSuccess(i);
         } else {
           Serial.println("Rebooting... " + String(i));
         }