#include <WiFi.h>
#include "configuration.h"
#include "station_utils.h"
#include "battery_utils.h"
#include "aprs_is_utils.h"
#include "syslog_utils.h"
#include "pins_config.h"
#include "wifi_utils.h"
#include "lora_utils.h"
#include "gps_utils.h"
#include "bme_utils.h"
#include "display.h"
#include "utils.h"

extern WiFiClient           espClient;
extern Configuration        Config;
extern String               versionDate;
extern bool                 statusAfterBoot;
extern String               firstLine;
extern String               secondLine;
extern String               thirdLine;
extern String               fourthLine;
extern String               fifthLine;
extern String               sixthLine;
extern String               seventhLine;
extern uint32_t             lastBeaconTx;
extern uint32_t             lastScreenOn;
extern bool                 beaconUpdate;
extern int                  stationMode;
extern String               iGateBeaconPacket;
extern String               iGateLoRaBeaconPacket;
extern std::vector<String>  lastHeardStation;
extern int                  rssi;
extern float                snr;
extern int                  freqError;
extern String               distance;
extern uint32_t             lastWiFiCheck;
extern bool                 WiFiConnect;
extern bool                 WiFiConnected;


namespace Utils {

    void processStatus() {
        String status = Config.callsign + ">APLRG1,WIDE1-1";
        if (stationMode==1 || stationMode==2 || (stationMode==5 && WiFi.status() == WL_CONNECTED)) {
            delay(1000);
            status += ",qAC:>https://github.com/richonguzman/LoRa_APRS_iGate " + versionDate;
            APRS_IS_Utils::upload(status);
            SYSLOG_Utils::log("APRSIS Tx", status,0,0,0);
        } else {
            delay(5000);
            status += ":>https://github.com/richonguzman/LoRa_APRS_iGate " + versionDate;
            if (stationMode==4) {
                LoRa_Utils::changeFreqTx();
            }
            LoRa_Utils::sendNewPacket("APRS", status);
            if (stationMode==4) {
                LoRa_Utils::changeFreqRx();
            }        
        }
        statusAfterBoot = false;
    }

    String getLocalIP() {
        if (!WiFiConnected) {
            return "IP : 192.168.4.1";
        } else {
            return "IP :  " + String(WiFi.localIP()[0]) + "." + String(WiFi.localIP()[1]) + "." + String(WiFi.localIP()[2]) + "." + String(WiFi.localIP()[3]);
        }        
    }

    void setupDisplay() {
        setup_display();
        #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2) || defined(HELTEC_V3) || defined(ESP32_DIY_LoRa) || defined(ESP32_DIY_1W_LoRa)
        digitalWrite(internalLedPin,HIGH);
        #endif
        Serial.println("\nStarting Station: " + Config.callsign + "   Version: " + versionDate);
        show_display(" LoRa APRS", "", "   ( iGATE & Digi )", "", "", "Richonguzman / CA2RXU", "      " + versionDate, 4000);
        #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2) || defined(HELTEC_V3) || defined(ESP32_DIY_LoRa) || defined(ESP32_DIY_1W_LoRa)
        digitalWrite(internalLedPin,LOW);
        #endif
        firstLine   = Config.callsign;
        seventhLine = "     listening...";
    }

    void activeStations() {
        fourthLine = "Stations (" + String(Config.rememberStationTime) + "min) = ";
        if (lastHeardStation.size() < 10) {
            fourthLine += " ";
        }
        fourthLine += String(lastHeardStation.size());            
    }

    void checkBeaconInterval() {
        uint32_t lastTx = millis() - lastBeaconTx;
        String beaconPacket, secondaryBeaconPacket;
        if (lastTx >= Config.beaconInterval*60*1000) {
            beaconUpdate = true;    
        }
        if (beaconUpdate) {
            display_toggle(true);
            Serial.println("---- Sending iGate Beacon ----");
            STATION_Utils::deleteNotHeard();
            activeStations();
            if (Config.bme.active) {
                String sensorData = BME_Utils::readDataSensor();
                beaconPacket = iGateBeaconPacket.substring(0,iGateBeaconPacket.indexOf(":=")+20) + "_" + sensorData + iGateBeaconPacket.substring(iGateBeaconPacket.indexOf(":=")+21) + " + WX";
                if (Config.igateSendsLoRaBeacons && stationMode!=1) {
                    secondaryBeaconPacket = iGateLoRaBeaconPacket + sensorData + Config.iGateComment + " + WX";
                }
            } else {
                beaconPacket = iGateBeaconPacket;
                if (Config.igateSendsLoRaBeacons && stationMode!=1) {
                    secondaryBeaconPacket = iGateLoRaBeaconPacket + Config.iGateComment;
                }
            }
            #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2)
            if (Config.sendBatteryVoltage) {
                beaconPacket += " (Batt=" + String(BATTERY_Utils::checkBattery(),2) + "V)";
            }
            #endif
            if (Config.externalVoltageMeasurement) { 
                beaconPacket += " (Ext V=" + String(BATTERY_Utils::checkExternalVoltage(),2) + "V)";
            }
            if (stationMode==1 || stationMode==2) {
                thirdLine = getLocalIP();
                if (!Config.bme.active) {
                    fifthLine = "";
                }
                sixthLine = "";
                show_display(firstLine, secondLine, thirdLine, fourthLine, fifthLine, sixthLine, "SENDING iGate BEACON", 1000);     
                #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2)
                if (Config.sendBatteryVoltage) { 
                    sixthLine = "     (Batt=" + String(BATTERY_Utils::checkBattery(),2) + "V)";
                }
                #endif
                if (Config.externalVoltageMeasurement) { 
                    sixthLine = "    (Ext V=" + String(BATTERY_Utils::checkExternalVoltage(),2) + "V)";
                }
                seventhLine = "     listening...";
                APRS_IS_Utils::upload(beaconPacket);
                if (Config.igateSendsLoRaBeacons && stationMode==2) { 
                    LoRa_Utils::sendNewPacket("APRS", secondaryBeaconPacket);
                }
                show_display(firstLine, secondLine, thirdLine, fourthLine, fifthLine, sixthLine, seventhLine, 0);
            } else if (stationMode==3 || stationMode==4) {
                String Rx = String(Config.loramodule.digirepeaterRxFreq);
                String Tx = String(Config.loramodule.digirepeaterTxFreq);
                if (stationMode==3) {
                    secondLine = "Rx:" + String(Tx.substring(0,3)) + "." + String(Tx.substring(3,6));
                } else {
                    secondLine = "Rx:" + String(Rx.substring(0,3)) + "." + String(Rx.substring(3,6));
                }
                secondLine += " Tx:" + String(Tx.substring(0,3)) + "." + String(Tx.substring(3,6));
                thirdLine = "<<   DigiRepeater  >>";
                fifthLine = "";
                sixthLine = "";
                show_display(firstLine, secondLine, thirdLine, fourthLine, fifthLine, sixthLine, "SENDING iGate BEACON", 0);
                #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2)
                if (Config.sendBatteryVoltage) { 
                    sixthLine = "     (Batt=" + String(BATTERY_Utils::checkBattery(),2) + "V)";
                }
                #endif
                if (Config.externalVoltageMeasurement) { 
                    sixthLine = "    (Ext V=" + String(BATTERY_Utils::checkExternalVoltage(),2) + "V)";
                }
                seventhLine = "     listening...";
                if (stationMode==4) {
                    LoRa_Utils::changeFreqTx();
                }
                LoRa_Utils::sendNewPacket("APRS", beaconPacket);
                if (stationMode==4) {
                    LoRa_Utils::changeFreqRx();
                }
            } else if (stationMode==5) {
                if (!Config.bme.active) {
                    fifthLine = "";
                }
                sixthLine = "";
                if (WiFi.status() == WL_CONNECTED && espClient.connected()) {
                    APRS_IS_Utils::checkStatus();
                    thirdLine = getLocalIP();
                    show_display(firstLine, secondLine, thirdLine, fourthLine, fifthLine, sixthLine, "SENDING iGate BEACON", 1000);        
                    #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2)
                    if (Config.sendBatteryVoltage) { 
                        sixthLine = "     (Batt=" + String(BATTERY_Utils::checkBattery(),2) + "V)";
                    }
                    #endif
                    if (Config.externalVoltageMeasurement) { 
                        sixthLine = "    (Ext V=" + String(BATTERY_Utils::checkExternalVoltage(),2) + "V)";
                    }
                    seventhLine = "     listening...";
                    APRS_IS_Utils::upload(beaconPacket);
                    if (Config.igateSendsLoRaBeacons) { 
                        LoRa_Utils::sendNewPacket("APRS", secondaryBeaconPacket);
                    }
                    show_display(firstLine, secondLine, thirdLine, fourthLine, fifthLine, sixthLine, seventhLine, 0);
                } else {
                    show_display(firstLine, secondLine, thirdLine, fourthLine, fifthLine, sixthLine, "SENDING iGate BEACON", 0);
                    #if defined(TTGO_T_LORA32_V2_1) || defined(HELTEC_V2)
                    if (Config.sendBatteryVoltage) { 
                        sixthLine = "     (Batt=" + String(BATTERY_Utils::checkBattery(),2) + "V)";
                    }
                    #endif
                    if (Config.externalVoltageMeasurement) { 
                        sixthLine = "    (Ext V=" + String(BATTERY_Utils::checkExternalVoltage(),2) + "V)";
                    }
                    seventhLine = "     listening...";
                    LoRa_Utils::sendNewPacket("APRS", beaconPacket);
                }
            }
            lastBeaconTx = millis();
            lastScreenOn = millis();
            beaconUpdate = false;
        }
        if (statusAfterBoot) {
            processStatus();
        }
    }

    void checkDisplayInterval() {
        uint32_t lastDisplayTime = millis() - lastScreenOn;
        if (!Config.display.alwaysOn) {
            if (lastDisplayTime >= Config.display.timeout*1000) {
                display_toggle(false);
            }
        }
    }

    void checkWiFiInterval() {
        uint32_t WiFiCheck = millis() - lastWiFiCheck;
        if (WiFi.status() != WL_CONNECTED && WiFiCheck >= 15*60*1000) {
        WiFiConnect = true;
        }
        if (WiFiConnect) {
        Serial.println("\nConnecting to WiFi ...");
        WIFI_Utils::startWiFi();
        lastWiFiCheck = millis();
        WiFiConnect = false;
        }
    }

    void validateDigiFreqs() {
        if (stationMode==4) {
            if (abs(Config.loramodule.digirepeaterTxFreq - Config.loramodule.digirepeaterRxFreq) < 125000) {
                Serial.println("Tx Freq less than 125kHz from Rx Freq ---> NOT VALID");
                show_display("Tx Freq is less than ", "125kHz from Rx Freq", "device will autofix", "and then reboot", 1000);
                Config.loramodule.digirepeaterTxFreq = Config.loramodule.digirepeaterRxFreq;        // Inform about that but then change the digirepeaterTxFreq to digirepeaterRxFreq and reset the device
                Config.writeFile();
                ESP.restart();
            }
        }
    }

    void typeOfPacket(String packet, String packetType) {
        String sender;
        if (stationMode==1 || stationMode==2 || (stationMode==5 && WiFi.status()==WL_CONNECTED)) {
            if (packetType == "LoRa-APRS") {
                fifthLine = "LoRa Rx ----> APRS-IS";
            } else if (packetType == "APRS-LoRa") {
                fifthLine = "APRS-IS ----> LoRa Tx";
            }
            sender = packet.substring(0,packet.indexOf(">"));
        } else {
            fifthLine = "LoRa Rx ----> LoRa Tx";
            sender = packet.substring(3,packet.indexOf(">"));
        }
        for (int i=sender.length();i<9;i++) {
            sender += " ";
        }
        if (packet.indexOf("::") >= 10) {
            if (packetType == "APRS-LoRa") {
                String addresseeAndMessage = packet.substring(packet.indexOf("::")+2);
                String addressee = addresseeAndMessage.substring(0, addresseeAndMessage.indexOf(":"));
                addressee.trim();
                sixthLine = sender + " > " + addressee;
            } else {
                sixthLine = sender + "> MESSAGE";
            }
            seventhLine = "RSSI:" + String(rssi) + "dBm SNR: " + String(snr) + "dBm";
        } else if (packet.indexOf(":>") >= 10) {
            sixthLine = sender + "> NEW STATUS";
            seventhLine = "RSSI:" + String(rssi) + "dBm SNR: " + String(snr) + "dBm";
        } else if (packet.indexOf(":!") >= 10 || packet.indexOf(":=") >= 10) {
            sixthLine = sender + "> GPS BEACON";
            GPS_Utils::getDistance(packet);
            seventhLine = "RSSI:" + String(rssi) + "dBm";
            if (rssi <= -100) {
                seventhLine += " ";
            } else {
                seventhLine += "  ";
            }
            if (distance.indexOf(".") == 1) {
                seventhLine += " ";
            }
            seventhLine += "D:" + distance + "km";
        } else if (packet.indexOf(":T#") >= 10 && packet.indexOf(":=/") == -1) {
            sixthLine = sender + "> TELEMETRY";
            seventhLine = "RSSI:" + String(rssi) + "dBm SNR: " + String(snr) + "dBm";
        } else if (packet.indexOf(":`") >= 10) {
            sixthLine = sender + ">  MIC-E";
            seventhLine = "RSSI:" + String(rssi) + "dBm SNR: " + String(snr) + "dBm";
        } else if (packet.indexOf(":;") >= 10) {
            sixthLine = sender + ">  OBJECT";
            seventhLine = "RSSI:" + String(rssi) + "dBm SNR: " + String(snr) + "dBm";
        } else {
            sixthLine = sender + "> ??????????";
            seventhLine = "RSSI:" + String(rssi) + "dBm SNR: " + String(snr) + "dBm";
        }
    }

}