From 6468db55650f3d236960eb54221c1bebda32b51a Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Tue, 8 Oct 2024 21:03:55 +0200
Subject: [PATCH] add restart reason sensor

---
 src/esp.h | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/esp.h b/src/esp.h
index 1a97c90..db65f76 100644
--- a/src/esp.h
+++ b/src/esp.h
@@ -12,6 +12,10 @@ namespace HaESP {
         Sensor::mapSensors["heap_max_free_block"]->updateState(to_string(ESP.getMaxFreeBlockSize()).c_str());
         }, &ts);
 
+	Task tRestartInfo(TASK_IMMEDIATE, TASK_ONCE, [](){
+		Sensor::mapSensors["restart_reason"]->updateState(ESP.getResetReason().c_str());
+	}, &ts);
+
     template <class T>
     Builder<T>& heapStats(Builder<T>& builder) {
         builder.addDiagnostic(Builder<Sensor>::instance(new Sensor{ "Heap fragmentation", "heap_fragmentation" })
@@ -33,4 +37,11 @@ namespace HaESP {
         tHeap.enable();
         return builder;
     }
+
+    template <class T>
+    Builder<T>& restartInfo(Builder<T>& builder) {
+        builder.addDiagnostic(Builder<Sensor>::instance((new Sensor{"Restart reason", "restart_reason"})).build());
+        tRestartInfo.enable();
+        return builder;
+    }
 }
\ No newline at end of file