From 935e7f365fa33243da7436bc7831020d7f5239a6 Mon Sep 17 00:00:00 2001
From: Phat Nguyen <phat.nt@hotmail.com>
Date: Mon, 10 Jun 2024 22:04:23 +0700
Subject: [PATCH] Fix display message is not correct.

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

diff --git a/examples/OneOpenAir/OneOpenAir.ino b/examples/OneOpenAir/OneOpenAir.ino
index d3ec231..7cbf2e7 100644
--- a/examples/OneOpenAir/OneOpenAir.ino
+++ b/examples/OneOpenAir/OneOpenAir.ino
@@ -221,8 +221,11 @@ void setup() {
         if (apiClient.isFetchConfigureFailed()) {
           if (ag->isOne()) {
             if (apiClient.isNotAvailableOnDashboard()) {
+              stateMachine.displaySetAddToDashBoard();
               stateMachine.displayHandle(
                   AgStateMachineWiFiOkServerOkSensorConfigFailed);
+            } else {
+              stateMachine.displayClearAddToDashBoard();
             }
           }
           stateMachine.handleLeds(
@@ -945,7 +948,6 @@ static void appLedHandler(void) {
     if (wifiConnector.isConnected() == false) {
       state = AgStateMachineWiFiLost;
     } else if (apiClient.isFetchConfigureFailed()) {
-      stateMachine.displaySetAddToDashBoard();
       state = AgStateMachineSensorConfigFailed;
     } else if (apiClient.isPostToServerFailed()) {
       state = AgStateMachineServerLost;
@@ -965,6 +967,11 @@ static void appDispHandler(void) {
         state = AgStateMachineWiFiLost;
       } else if (apiClient.isFetchConfigureFailed()) {
         state = AgStateMachineSensorConfigFailed;
+        if (apiClient.isNotAvailableOnDashboard()) {
+          stateMachine.displaySetAddToDashBoard();
+        } else {
+          stateMachine.displayClearAddToDashBoard();
+        }
       } else if (apiClient.isPostToServerFailed()) {
         state = AgStateMachineServerLost;
       }