From 21c73660b9d75275c3fb8af26051040c000d1151 Mon Sep 17 00:00:00 2001
From: Florian <florian.hess@azon.ai>
Date: Wed, 18 Sep 2024 13:03:41 +0200
Subject: [PATCH] Refactor set_programming_mode

---
 components/tc_bus/tc_bus.cpp | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/components/tc_bus/tc_bus.cpp b/components/tc_bus/tc_bus.cpp
index 34a1fb9f..68fddb2a 100644
--- a/components/tc_bus/tc_bus.cpp
+++ b/components/tc_bus/tc_bus.cpp
@@ -337,6 +337,7 @@ namespace esphome
                 }
             } else if (cmd_data.type == COMMAND_TYPE_PROGRAMMING_MODE) {
                 ESP_LOGD(TAG, "Programming Mode: %s", YESNO(cmd_data.address == 1));
+                this->programming_mode_ = cmd_data.address == 1;
             }
 
             // Publish Command to Last Bus Command Sensor
@@ -511,15 +512,7 @@ namespace esphome
         void TCBusComponent::set_programming_mode(bool enabled)
         {
             this->programming_mode_ = enabled;
-
-            if(enabled)
-            {
-                send_command_generate(COMMAND_TYPE_PROGRAMMING_MODE, 1, 0);
-            }
-            else
-            {
-                send_command_generate(COMMAND_TYPE_PROGRAMMING_MODE, 0, 0);
-            }
+            send_command_generate(COMMAND_TYPE_PROGRAMMING_MODE, enabled ? 1 : 0, 0);
         }
 
     }  // namespace tc_bus