#include "board.h"
#include "mw.h"

#include "drivers/serial_common.h"
#include "serial_common.h"

#include "telemetry_frsky.h"
#include "telemetry_hott.h"
#include "telemetry_msp.h"

#include "telemetry_common.h"

static bool isTelemetryConfigurationValid = false; // flag used to avoid repeated configuration checks

bool isTelemetryProviderFrSky(void)
{
    return masterConfig.telemetry_provider == TELEMETRY_PROVIDER_FRSKY;
}

bool isTelemetryProviderHoTT(void)
{
    return masterConfig.telemetry_provider == TELEMETRY_PROVIDER_HOTT;
}

bool isTelemetryProviderMSP(void)
{
    return masterConfig.telemetry_provider == TELEMETRY_PROVIDER_MSP;
}
bool canUseTelemetryWithCurrentConfiguration(void)
{
    if (!feature(FEATURE_TELEMETRY)) {
        return false;
    }

    if (!feature(FEATURE_SOFTSERIAL)) {
        if (masterConfig.telemetry_port == TELEMETRY_PORT_SOFTSERIAL_1 || masterConfig.telemetry_port == TELEMETRY_PORT_SOFTSERIAL_2) {
            // softserial feature must be enabled to use telemetry on softserial ports
            return false;
        }
    }

    if (isTelemetryProviderHoTT()) {
        if (masterConfig.telemetry_port == TELEMETRY_PORT_UART) {
            // HoTT requires a serial port that supports RX/TX mode swapping
            return false;
        }
    }

    return true;
}

void initTelemetry(serialPorts_t *serialPorts)
{
    // Force telemetry to uart when softserial disabled
    if (!feature(FEATURE_SOFTSERIAL))
        masterConfig.telemetry_port = TELEMETRY_PORT_UART;

#ifdef FY90Q
    // FY90Q does not support softserial
    masterConfig.telemetry_port = TELEMETRY_PORT_UART;
    serialPorts->telemport = serialPorts->mainport;
#endif

    isTelemetryConfigurationValid = canUseTelemetryWithCurrentConfiguration();

#ifndef FY90Q
    if (masterConfig.telemetry_port == TELEMETRY_PORT_SOFTSERIAL_1)
        serialPorts->telemport = &(softSerialPorts[0].port);
    else if (masterConfig.telemetry_port == TELEMETRY_PORT_SOFTSERIAL_2)
        serialPorts->telemport = &(softSerialPorts[1].port);
    else
        serialPorts->telemport = serialPorts->mainport;
#endif

    checkTelemetryState();
}

static bool telemetryEnabled = false;

bool determineNewTelemetryEnabledState(void)
{
    bool enabled = true;

    if (masterConfig.telemetry_port == TELEMETRY_PORT_UART) {
        if (!masterConfig.telemetry_switch)
            enabled = f.ARMED;
        else
            enabled = rcOptions[BOXTELEMETRY];
    }

    return enabled;
}

bool shouldChangeTelemetryStateNow(bool newState)
{
    return newState != telemetryEnabled;
}

static void configureTelemetryPort(void)
{
    if (isTelemetryProviderFrSky()) {
        configureFrSkyTelemetryPort();
    }

    if (isTelemetryProviderHoTT()) {
        configureHoTTTelemetryPort();
    }
}

void freeTelemetryPort(void)
{
    if (isTelemetryProviderFrSky()) {
        freeFrSkyTelemetryPort();
    }

    if (isTelemetryProviderHoTT()) {
        freeHoTTTelemetryPort();
    }
}

void checkTelemetryState(void)
{
    if (!isTelemetryConfigurationValid) {
        return;
    }

    bool newEnabledState = determineNewTelemetryEnabledState();

    if (!shouldChangeTelemetryStateNow(newEnabledState)) {
        return;
    }

    if (newEnabledState)
        configureTelemetryPort();
    else
        freeTelemetryPort();

    telemetryEnabled = newEnabledState;
}

void handleTelemetry(void)
{
    if (!isTelemetryConfigurationValid || !determineNewTelemetryEnabledState())
        return;

    if (isTelemetryProviderFrSky()) {
        handleFrSkyTelemetry();
    }

    if (isTelemetryProviderHoTT()) {
        handleHoTTTelemetry();
    }
    if (isTelemetryProviderMSP()) {
        handleMSPTelemetry();
    }
}