#include "utils/WiFiUtils.h"
#include <vector>
#define TRIESONE 25 // количество попыток подключения к одной сети из несколких
#define TRIES 40    // количество попыток подключения сети если она одна

void routerConnect() {
    WiFi.setAutoConnect(false);
    WiFi.persistent(false);

    WiFi.mode(WIFI_STA);
    byte triesOne = TRIESONE;

    String _ssid = jsonReadStr(settingsFlashJson, "routerssid");
    String _password = jsonReadStr(settingsFlashJson, "routerpass");
    std::vector<String> _ssidList;
    std::vector<String> _passwordList;
    for (size_t i = 0; i < 3; i++)
    {
        String _stmp = selectFromMarkerToMarker(_ssid, ",", i);
        String _ptmp = selectFromMarkerToMarker(_password, ",", i);
        if (_stmp == "not found" && _ssidList.size() == 0)
        {
            triesOne = TRIES;
            _ssidList.push_back(_ssid);
            _passwordList.push_back(_password);
            break;
        }
        if (_stmp != "not found")
        {
            _ssidList.push_back(_stmp);
            _passwordList.push_back(_ptmp);
        }
    }

    if (_ssid == "" && _password == "") {
        WiFi.begin();
    } else {
        WiFi.begin(_ssidList[0].c_str(), _passwordList[0].c_str());
#ifdef ESP32
        WiFi.setTxPower(WIFI_POWER_19_5dBm);
#else
        WiFi.setOutputPower(20.5);
#endif

        if (_ssidList.size() > 1)
        {
            SerialPrint("i", "WIFI", "ssid list: " + _ssid);
            SerialPrint("i", "WIFI", "pass list: " + _password);
        }
    }
    for (size_t i = 0; i < _ssidList.size(); i++) {
        triesOne = TRIESONE;
        if (WiFi.status() == WL_CONNECTED)
            break;
        WiFi.begin(_ssidList[i].c_str(), _passwordList[i].c_str());
        SerialPrint("i", "WIFI", "ssid connect: " + _ssidList[i]);
        SerialPrint("i", "WIFI", "pass connect: " + _passwordList[i]);
        while (--triesOne && WiFi.status() != WL_CONNECTED) {
//            SerialPrint("i", "WIFI", ": " + String((int)WiFi.status()));
            if (WiFi.status() == WL_CONNECT_FAILED || WiFi.status() ==  WL_WRONG_PASSWORD)
            {
                SerialPrint("E", "WIFI", "password is not correct");
                triesOne = 1;
                jsonWriteInt(errorsHeapJson, "passer", 1);
                break;
            }
            Serial.print(".");
            delay(1000);
        }
        Serial.println("");
    }

    if (WiFi.status() != WL_CONNECTED) {
        Serial.println("");
        startAPMode();
    } else {
        Serial.println("");
        SerialPrint("i", "WIFI", "http://" + WiFi.localIP().toString());
        jsonWriteStr(settingsFlashJson, "ip", WiFi.localIP().toString());

        mqttInit();
    }
    SerialPrint("i", F("WIFI"), F("Network Init"));
}

bool startAPMode() {
    SerialPrint("i", "WIFI", "AP Mode");

    WiFi.disconnect();
    WiFi.mode(WIFI_AP);

    String _ssidAP = jsonReadStr(settingsFlashJson, "apssid");
    String _passwordAP = jsonReadStr(settingsFlashJson, "appass");

    WiFi.softAP(_ssidAP.c_str(), _passwordAP.c_str());
    IPAddress myIP = WiFi.softAPIP();

    SerialPrint("i", "WIFI", "AP IP: " + myIP.toString());
    jsonWriteStr(settingsFlashJson, "ip", myIP.toString());

    if (jsonReadInt(errorsHeapJson, "passer") != 1) {
        ts.add(
            WIFI_SCAN, 30 * 1000, [&](void*) {
                String sta_ssid = jsonReadStr(settingsFlashJson, "routerssid");

                SerialPrint("i", "WIFI", "scanning for " + sta_ssid);

                if (RouterFind(sta_ssid)) {
                    ts.remove(WIFI_SCAN);
                    WiFi.scanDelete();
                    routerConnect();
                }
            },
            nullptr, true);
    }
    return true;
}

boolean RouterFind(String ssid) {
    bool res = false;
    int n = WiFi.scanComplete();
    SerialPrint("i", "WIFI", "scan result: " + String(n, DEC));

    if (n == -2) {  //Сканирование не было запущено, запускаем
        SerialPrint("i", "WIFI", "start scanning");
        WiFi.scanNetworks(true, false);  // async, show_hidden
    }

    else if (n == -1) {  //Сканирование все еще выполняется
        SerialPrint("i", "WIFI", "scanning in progress");
    }

    else if (n == 0) {  //ни одна сеть не найдена
        SerialPrint("i", "WIFI", "no networks found");
        WiFi.scanNetworks(true, false);
    }
    else if (n > 0)
    {
        std::vector<String> _ssidList;
        for (size_t i = 0; i < 3; i++)
        {
            String _stmp = selectFromMarkerToMarker(ssid, ",", i);
            if (_stmp == "not found" && _ssidList.size() == 0)
            {
                _ssidList.push_back(ssid);
                break;
            }
            if (_stmp != "not found")
                _ssidList.push_back(_stmp);
        }

        for (int8_t i = 0; i < n; i++)
        {
            if (WiFi.SSID(i) == _ssidList[0] || WiFi.SSID(i) == _ssidList[1] ||
                WiFi.SSID(i) == _ssidList[2])
            {
                res = true;
            }
            // SerialPrint("i", "WIFI", (res ? "*" : "") + String(i, DEC) + ") " + WiFi.SSID(i));
            jsonWriteStr_(ssidListHeapJson, String(i), WiFi.SSID(i));

            // String(WiFi.RSSI(i)
        }
    }
    SerialPrint("i", "WIFI", ssidListHeapJson);
    WiFi.scanDelete();
    return res;
}

// boolean isNetworkActive() {
//     return WiFi.status() == WL_CONNECTED;
// }

uint8_t RSSIquality() {
    uint8_t res = 0;
    if (WiFi.status() == WL_CONNECTED) {
        int rssi = WiFi.RSSI();
        if (rssi >= -50) {
            res = 6;  //"Excellent";
        } else if (rssi < -50 && rssi >= -60) {
            res = 5;  //"Very good";
        } else if (rssi < -60 && rssi >= -70) {
            res = 4;  //"Good";
        } else if (rssi < -70 && rssi >= -80) {
            res = 3;  //"Low";
        } else if (rssi < -80 && rssi > -100) {
            res = 2;  //"Very low";
        } else if (rssi <= -100) {
            res = 1;  //"No signal";
        }
    }
    return res;
}