/**
 * @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();
}