From b298f84271603bf9a0b31f30b7ad2314ccfc9d28 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patrick=20Ha=C3=9Fel?= <git@patrick-hassel.de>
Date: Mon, 15 Apr 2024 14:11:35 +0200
Subject: [PATCH] mqttDisconnect() before 'restart' or 'otaEnd'

---
 lib/patrix/console.cpp | 1 +
 lib/patrix/wifi.cpp    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/lib/patrix/console.cpp b/lib/patrix/console.cpp
index ed191ab..da70292 100644
--- a/lib/patrix/console.cpp
+++ b/lib/patrix/console.cpp
@@ -113,6 +113,7 @@ void _reboot() {
   info("Rebooting...");
   delay(500);
   yield();
+  mqttDisconnect();
   ESP.restart();
 }
 
diff --git a/lib/patrix/wifi.cpp b/lib/patrix/wifi.cpp
index 2daad48..3f9189b 100644
--- a/lib/patrix/wifi.cpp
+++ b/lib/patrix/wifi.cpp
@@ -93,6 +93,7 @@ void otaSetup() {
   });
   ArduinoOTA.onEnd([] {
     info("OTA Complete!");
+    mqttDisconnect();
   });
   ArduinoOTA.onError([](ota_error_t e) {
     const char *name;