HA configuration

lib/HomeAssistantMqttHandler/include/HomeAssistantMqttHandler.h

@@ -3,14 +3,51 @@
 
 #include "AmsMqttHandler.h"
 #include "HomeAssistantStatic.h"
+#include "AmsConfiguration.h"
 
 class HomeAssistantMqttHandler : public AmsMqttHandler {
 public:
-    HomeAssistantMqttHandler(MQTTClient* mqtt, char* buf, const char* clientId, const char* topic, HwTools* hw) : AmsMqttHandler(mqtt, buf) {
+    HomeAssistantMqttHandler(MQTTClient* mqtt, char* buf, const char* clientId, const char* topic, uint8_t boardType, HomeAssistantConfig config, HwTools* hw) : AmsMqttHandler(mqtt, buf) {
         this->clientId = clientId;
         this->topic = String(topic);
+        this->config = config;
         this->hw = hw;
         l1Init = l2Init = l2eInit = l3Init = l3eInit = l4Init = l4eInit = rtInit = rteInit = pInit = sInit = false;
+
+        if(strlen(config.discoveryNameTag) > 0) {
+            snprintf_P(buf, 128, PSTR("AMS reader (%s)"), config.discoveryNameTag);
+            deviceName = String(buf);
+            snprintf_P(buf, 128, PSTR(" (%s)"), config.discoveryNameTag);
+            sensorNamePostFix = String(buf);
+        } else {
+            deviceName = "AMS reader";
+            sensorNamePostFix = "";
+        }
+        deviceModel = boardTypeToString(boardType);
+        manufacturer = boardManufacturerToString(boardType);
+
+        #if defined(ESP8266)
+            String hostname = WiFi.hostname();
+        #elif defined(ESP32)
+            String hostname = WiFi.getHostname();
+        #endif
+
+        deviceUid = hostname; // Maybe configurable in the future?
+
+        if(strlen(config.discoveryHostname) > 0) {
+            snprintf_P(buf, 128, PSTR("http://%s.local/"), config.discoveryHostname);
+            String deviceUrl = String(buf);
+        } else {
+            snprintf_P(buf, 128, PSTR("http://%s.local/"), hostname);
+            String deviceUrl = String(buf);
+        }
+
+        if(strlen(config.discoveryTopic) > 0) {
+            discoveryTopic = String(config.discoveryTopic);
+            if(!discoveryTopic.endsWith("/")) discoveryTopic += "/";
+        } else {
+            discoveryTopic = "homeassistant/sensor/";
+        }
     };
     bool publish(AmsData* data, AmsData* previousState, EnergyAccounting* ea, EntsoeApi* eapi);
     bool publishTemperatures(AmsConfiguration*, HwTools*);
@@ -33,15 +70,15 @@ public:
     void publishSystemSensors();
 
 private:
-    String haTopic = "homeassistant/sensor/";
+    HomeAssistantConfig config;
+    String deviceName;
+    String deviceModel;
+    String deviceUid;
+    String manufacturer;
+    String deviceUrl;
 
-    String haName = "AMS reader";
-    #if defined(ESP32)
-        String haModel = "ESP32";
-    #elif defined(ESP8266)
-        String haModel = "ESP8266";
-    #endif
-    String haManuf = "AmsToMqttBridge";
+    String discoveryTopic;
+    String sensorNamePostFix;
 
     bool l1Init, l2Init, l2eInit, l3Init, l3eInit, l4Init, l4eInit, rtInit, rteInit, pInit, sInit;
     bool tInit[32] = {false};
@@ -50,5 +87,75 @@ private:
     String clientId;
     String topic;
     HwTools* hw;
+
+    String boardTypeToString(uint8_t b) {
+        switch(b) {
+            case 5:
+                #if defined(ESP8266)
+                    return "Pow-K";
+                #elif defined(ESP32)
+                    return "Pow-K+";
+                #endif
+            case 7:
+                #if defined(ESP8266)
+                    return "Pow-U";
+                #elif defined(ESP32)
+                    return "Pow-U+";
+                #endif
+            case 6:
+                return "Pow-P1";
+            case 51:
+                return "S2 mini";
+            case 50:
+                return "ESP32-S2";
+            case 201:
+                return "LOLIN D32";
+            case 202:
+                return "HUZZAH32";
+            case 203:
+                return "DevKitC";
+            case 200:
+                return "ESP32";
+            case 2:
+                return "HAN Reader 2.0 by Max Spencer";
+            case 0:
+                return "Custom hardware by Roar Fredriksen";
+            case 1:
+                return "Kamstrup module by Egil Opsahl";
+            case 3:
+                return "Pow-K";
+            case 4:
+                return "Pow-U";
+            case 101:
+                return "D1 mini";
+            case 100:
+                return "ESP8266";
+            case 70:
+                return "ESP32-C3";
+            case 71:
+                return "ESP32-C3-DevKitM-1";
+        }
+        #if defined(ESP8266)
+            return "ESP8266";
+        #elif defined(ESP32)
+            return "ESP32";
+        #endif
+    };
+
+    String boardManufacturerToString(uint8_t b) {
+        if(b >= 3 && b <= 7)
+            return "amsleser.no";
+        if(b < 50)
+            return "Custom";
+        switch(b) {
+            case 51:
+            case 101:
+            case 201:
+                return "Wemos";
+            case 202:
+                return "Adafruit";
+        }
+        return "Espressif";
+    };
 };
 #endif