Initial commit

2023-12-20 22:52:04 +01:00
commit 678b65c9f8
12 changed files with 558 additions and 0 deletions
@@ -0,0 +1,156 @@
+/**
+ * @file network.cpp
+ * @author Alexander Klein (alex@kleiax.de)
+ * @brief
+ * @version 0.1
+ * @date 2023-09-18
+ *
+ * @copyright Copyright (c) 2023
+ *
+ */
+
+#include "network.h"
+
+Network::Network(const char *ssid, const char *passphrase)
+{
+    if (!WiFi.mode(WIFI_STA))
+        std::cout << "Network::connectWiFi failed WiFi.mode" << std::endl;
+
+    this->init(ssid, passphrase);
+}
+
+Network::Network(const char *ssid, const char *passphrase, NetworkAddresses addresses)
+{
+    this->addresses = addresses;
+
+    if (!WiFi.mode(WIFI_STA))
+        std::cout << "Network::connectWiFi failed WiFi.mode" << std::endl;
+
+    if (!WiFi.config(this->addresses.localIP,
+                     this->addresses.gateway,
+                     this->addresses.subnet,
+                     this->addresses.dnsServer))
+    {
+        std::cout << "STA Failed to configure" << std::endl;
+    }
+
+    this->init(ssid, passphrase);
+}
+
+Network::~Network()
+{
+    if (this->mqttClient)
+        delete mqttClient;
+}
+
+bool Network::activateMqtt(const char *user, const char *passphrase)
+{
+    this->mqttUser = user;
+    this->mqttPassphrase = passphrase;
+
+    this->mqttClient = new PubSubClient(this->wifiClient);
+    this->mqttClient->setServer(this->addresses.mqttServer, this->addresses.mqttPort);
+    this->mqttClient->setSocketTimeout(1);
+
+    if (this->wifiConnected)
+        return this->connectMqtt();
+    return false;
+}
+
+bool Network::activateMqtt(const char *user, const char *passphrase, IPAddress server, uint16_t port)
+{
+    this->addresses.mqttServer = server;
+    this->addresses.mqttPort = port;
+    return this->activateMqtt(user, passphrase);
+}
+
+void Network::printIPs()
+{
+    std::cout << std::endl;
+    if (!this->wifiConnected)
+    {
+        std::cout << "WiFi is not connected." << std::endl;
+        return;
+    }
+    std::cout << "WiFi is connected to" << std::endl;
+    std::cout << "IP address: " << std::endl;
+    std::cout << WiFi.localIP().toString().c_str() << std::endl;
+    std::cout << "WiFi MAC Address: " << WiFi.macAddress().c_str() << std::endl
+              << std::endl;
+}
+
+void Network::runAsChild()
+{
+    if (!this->initSuccessful)
+        return;
+
+    this->checkWifi();
+
+    if (this->wifiConnected && this->mqttClient)
+        this->checkMqtt();
+}
+
+void Network::init(const char *ssid, const char *passphrase)
+{
+    // Connect to Wi-Fi network with SSID and password
+    std::cout << "Connecting to " << ssid << std::endl;
+    WiFi.begin(ssid, passphrase);
+    uint8_t timeout = Network::wifiConnectTimeout;
+    while (WiFi.status() != WL_CONNECTED)
+    {
+        delay(Network::wifiConnectLoopTime);
+        std::cout << "." << std::flush;
+        timeout--;
+        if (timeout == 0)
+        {
+            std::cout << std::endl;
+            std::cout << "WiFi NOT connected." << std::endl;
+            return;
+        }
+    }
+
+    this->wifiConnected = true;
+    this->printIPs();
+}
+
+void Network::checkWifi()
+{
+    if ((WiFi.status() != WL_CONNECTED) && (millis() - this->lastWifiReconnectAttempt >= Network::wifiReconnectDelay))
+    {
+        std::cout << "Reconnecting to WiFi..." << std::endl;
+        WiFi.disconnect();
+        this->wifiConnected = WiFi.reconnect();
+        this->lastWifiReconnectAttempt = millis();
+    }
+}
+
+void Network::checkMqtt()
+{
+    if (!this->mqttClient->connected() && millis() - this->lastMqttReconnectAttempt > Network::mqttReconnectDelay)
+    {
+        this->mqttConnected = this->connectMqtt();
+        this->lastMqttReconnectAttempt = millis();
+    }
+
+    if (this->mqttConnected)
+        this->mqttClient->loop();
+}
+
+bool Network::connectMqtt()
+{
+    String clientId = "ESP32CrazyHead-";
+    clientId += String(random(0xffff), HEX);
+
+    if (this->mqttUser)
+    {
+        if (this->mqttClient->connect(clientId.c_str(), this->mqttUser, this->mqttPassphrase))
+            this->mqttClient->publish("CrazyHead/Info", "Connected to Mqtt-Broker");
+    }
+    else
+    {
+        if (this->mqttClient->connect(clientId.c_str()))
+            this->mqttClient->publish("CrazyHead/Info", "Connected to Mqtt-Broker");
+    }
+
+    return this->mqttClient->connected();
+}