#include <ESPmDNS.h>
#include "network.h"

WIFIMANAGER Network::WifiManager;
AsyncWebServer Network::webServer(80);

Network::Network(std::string_view hostname, std::string_view apSsid) {
    this->hostname = std::string(hostname);
    this->apSsid = std::string(apSsid);
    setHostname(hostname);
    WifiManager.startBackgroundTask(apSsid.data(), "");
    WifiManager.fallbackToSoftAp(true);
    WifiManager.attachWebServer(&webServer);
    WifiManager.attachUI();
    webServer.on("/", HTTP_GET, [this](AsyncWebServerRequest *request) {
        request->send(200, "text/html", this->defaultHomepage.data());
    });
    webServer.begin();
}

Network::~Network() {
    webServer.end();
    WifiManager.detachUI();
    WifiManager.detachWebServer();
}

void Network::reconnect() {
    WifiManager.stopWifi(true);
    WifiManager.startBackgroundTask(apSsid.data(), "");
}

bool Network::isConnected() const {
    return WiFi.status() == WL_CONNECTED;
}

std::string Network::getHostname() const {
    return WiFi.getHostname();
}

const std::string Network::getMacAddress() const {
    std::string mac = WiFi.macAddress().c_str(); // format: "AA:BB:CC:DD:EE:FF"
    std::string hexMac = "0x";
    for (size_t i = 0; i < mac.size(); ++i) {
        if (mac[i] != ':') {
            hexMac += mac[i];
        }
    }
    return hexMac;
}

bool Network::setHostname(const std::string_view &hostname) {
    this->hostname = std::string(hostname);
    return WiFi.setHostname(this->hostname.c_str());
}

void Network::registerMDNS() {
    if (!MDNS.begin(this->hostname.c_str())) {
        Serial.println("Error setting up MDNS responder!");
    } else {
        Serial.printf("mDNS responder started with hostname: %s\n", this->hostname.c_str());
    }
}