From 9b1ecf73ceb617ef9325de20e5d5e2e3077d4e30 Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Wed, 4 Jun 2025 17:14:14 +0200
Subject: [PATCH 1/5] add sensor for wifi signal strength

---
 library.json |  2 +-
 src/esp.h    | 20 ++++++++++++++++++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/library.json b/library.json
index e79e522..9885b79 100644
--- a/library.json
+++ b/library.json
@@ -1,7 +1,7 @@
 {
   "$schema": "https://raw.githubusercontent.com/platformio/platformio-core/develop/platformio/assets/schema/library.json",
   "name": "ha-mqtt",
-  "version": "1.6.0",
+  "version": "1.7.0",
   "description": "Home Assistant classes for integration with MQTT auto discovery",
   "repository": {
     "type": "git",
diff --git a/src/esp.h b/src/esp.h
index 4dd511b..2987e4f 100644
--- a/src/esp.h
+++ b/src/esp.h
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <ESP8266WiFi.h>
 #include "TaskScheduler.h"
 #include "ha.h"
 
@@ -9,6 +10,7 @@ namespace HaESP {
     struct {
         bool heapStats = false;
         bool restartInfo = false;
+        bool wifiInfo = false;
     } activeSensors;
 
     Task tHeap(5 * TASK_MINUTE, TASK_FOREVER, []() {
@@ -28,6 +30,10 @@ namespace HaESP {
         Sensor::mapSensors["restart_reason"]->updateState(ESP.getResetReason().c_str());
         }, &ts);
 
+    Task tWifi(3 * TASK_MINUTE, TASK_FOREVER, []() {
+        Sensor::mapSensors["wifi_signal_strength"]->updateState(to_string(WiFi.RSSI()).c_str());
+        }, &ts);
+
     template <class T>
     Builder<T>& heapStats(Builder<T>& builder) {
         auto heap = new Sensor{ "Heap fragmentation", "heap_fragmentation" };
@@ -60,6 +66,19 @@ namespace HaESP {
         return builder;
     }
 
+    template <class T>
+    Builder<T>& wifiInfo(Builder<T>& builder) {
+        builder.addDiagnostic(Builder<Sensor>::instance((
+            new Sensor{ "WiFi Signal (RSSI)", "wifi_signal_strength" }))
+            .withDeviceClass("signal_strength")
+            .withUnitMeasure("dBm")
+            .withPrecision(0)
+            .build()
+        );
+        activeSensors.wifiInfo = true;
+        return builder;
+    }
+
     Builder<Button>& restartButton() {
         return Builder<Button>::instance(new Button{"Restart", "restart",
             [](const char* msg) {
@@ -72,5 +91,6 @@ namespace HaESP {
     void enableSensors() {
         if (activeSensors.heapStats) tHeap.enable();
         if (activeSensors.restartInfo) tRestartInfo.enable();
+        if (activeSensors.wifiInfo) tWifi.enable();
     }
 }
\ No newline at end of file

From 391fa927ba5da639940042a1e87820e4156ffcfe Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Thu, 5 Jun 2025 19:34:06 +0200
Subject: [PATCH 2/5] add support for SensorStateClass (used for long term
 statistics) and use it as default for Temperature and Humidity sensors

---
 src/ha.h | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/ha.h b/src/ha.h
index 56d182e..7d436f3 100644
--- a/src/ha.h
+++ b/src/ha.h
@@ -289,6 +289,7 @@ namespace Ha {
     struct Sensor : Component, State {
         const char* unitMeasure = nullptr;
         unsigned int precision = 2;
+        const char* sensorStateClass = nullptr;
         inline static unordered_map<string, Sensor*> mapSensors;
 
         Sensor(const char* name, const char* id) : Component(id, name, "sensor"), State(this) {
@@ -301,6 +302,7 @@ namespace Ha {
             State::buildConfig(jsonDoc);
             if (unitMeasure) jsonDoc["unit_of_measurement"] = unitMeasure;
             if (isNumericSensor()) jsonDoc["suggested_display_precision"] = precision;
+            if (sensorStateClass) jsonDoc["state_class"] = sensorStateClass;
         }
 
     private:
@@ -313,6 +315,7 @@ namespace Ha {
         TemperatureSensor(const char* id, const char* name = "Temperature") : Sensor(name, id) {
             deviceClass = "temperature";
             unitMeasure = "°C";
+            sensorStateClass = "measurement";
         }
     };
 
@@ -320,6 +323,7 @@ namespace Ha {
         HumiditySensor(const char* id, const char* name = "Humidity") : Sensor(name, id) {
             deviceClass = "humidity";
             unitMeasure = "%";
+            sensorStateClass = "measurement";
         }
     };
 
@@ -407,6 +411,11 @@ namespace Ha {
             return *this;
         }
 
+        Builder& withSensorStateClass(const char* value) {
+            cmp->sensorStateClass = value;
+            return *this;
+        }
+
         Builder& withMin(unsigned int value) {
             cmp->min = value;
             return *this;

From 57160661ec3dc776e7b9cae5b87607029474488f Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Fri, 6 Jun 2025 13:43:32 +0200
Subject: [PATCH 3/5] use type safe SensorStateClass-es

---
 src/enums.h | 26 ++++++++++++++++++++++++++
 src/ha.h    | 11 ++++++-----
 2 files changed, 32 insertions(+), 5 deletions(-)
 create mode 100644 src/enums.h

diff --git a/src/enums.h b/src/enums.h
new file mode 100644
index 0000000..0dce396
--- /dev/null
+++ b/src/enums.h
@@ -0,0 +1,26 @@
+#pragma once
+
+class EnumClass {
+    const char* value = nullptr;
+public:
+    EnumClass() {}
+
+    EnumClass(const char* v) {
+        value = v;
+    }
+
+    operator const char*() {
+        return value;
+    }
+
+    operator bool() {
+        return value != nullptr;
+    }
+};
+
+namespace Ha {
+    struct SensorStateClass : EnumClass {
+        SensorStateClass() : EnumClass() {}
+        SensorStateClass(const char* value) : EnumClass(value) {}
+    } MEASUREMENT{ "measurement" }, MEASUREMENT_ANGLE{ "measurement_angle" }, TOTAL{ "total" }, TOTAL_INCREASING{ "total_increasing" };
+}
diff --git a/src/ha.h b/src/ha.h
index 7d436f3..4e702c5 100644
--- a/src/ha.h
+++ b/src/ha.h
@@ -2,6 +2,7 @@
 
 #include <ArduinoJson.h>
 #include "list.h"
+#include "enums.h"
 
 using namespace std;
 
@@ -289,7 +290,7 @@ namespace Ha {
     struct Sensor : Component, State {
         const char* unitMeasure = nullptr;
         unsigned int precision = 2;
-        const char* sensorStateClass = nullptr;
+        SensorStateClass sensorStateClass;
         inline static unordered_map<string, Sensor*> mapSensors;
 
         Sensor(const char* name, const char* id) : Component(id, name, "sensor"), State(this) {
@@ -302,7 +303,7 @@ namespace Ha {
             State::buildConfig(jsonDoc);
             if (unitMeasure) jsonDoc["unit_of_measurement"] = unitMeasure;
             if (isNumericSensor()) jsonDoc["suggested_display_precision"] = precision;
-            if (sensorStateClass) jsonDoc["state_class"] = sensorStateClass;
+            if (sensorStateClass) jsonDoc["state_class"] = static_cast<const char*>(sensorStateClass);
         }
 
     private:
@@ -315,7 +316,7 @@ namespace Ha {
         TemperatureSensor(const char* id, const char* name = "Temperature") : Sensor(name, id) {
             deviceClass = "temperature";
             unitMeasure = "°C";
-            sensorStateClass = "measurement";
+            sensorStateClass = MEASUREMENT;
         }
     };
 
@@ -323,7 +324,7 @@ namespace Ha {
         HumiditySensor(const char* id, const char* name = "Humidity") : Sensor(name, id) {
             deviceClass = "humidity";
             unitMeasure = "%";
-            sensorStateClass = "measurement";
+            sensorStateClass = MEASUREMENT;
         }
     };
 
@@ -411,7 +412,7 @@ namespace Ha {
             return *this;
         }
 
-        Builder& withSensorStateClass(const char* value) {
+        Builder& withSensorStateClass(SensorStateClass value) {
             cmp->sensorStateClass = value;
             return *this;
         }

From 464b4f1645875dfd256cb48976154beb6c8fb960 Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Sat, 7 Jun 2025 08:29:23 +0200
Subject: [PATCH 4/5] add assignment operator for when EnumClass is used in a
 Builder

---
 src/enums.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/enums.h b/src/enums.h
index 0dce396..025fd31 100644
--- a/src/enums.h
+++ b/src/enums.h
@@ -9,6 +9,10 @@ public:
         value = v;
     }
 
+    void operator=(const char* v) {
+        value = v;
+    }
+
     operator const char*() {
         return value;
     }

From df53af65fa24ffe9b3b94299fbe58bb63feec5d2 Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Sat, 7 Jun 2025 09:14:32 +0200
Subject: [PATCH 5/5] use long term statistics as default for battery sensors

---
 src/ha.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/ha.h b/src/ha.h
index 4e702c5..1580459 100644
--- a/src/ha.h
+++ b/src/ha.h
@@ -348,6 +348,7 @@ namespace Ha {
             this->valueTemplate = valueTemplate;
             deviceClass = "battery";
             unitMeasure = "%";
+            sensorStateClass = MEASUREMENT;
         }
     };