#include <cstdio>
#include <cstring>

#include "lr1121_malnus.hpp"

static void applyPaPreset(lr1121::Config &cfg, bool hp_mode)
{
    cfg.pa_sel = hp_mode ? 0x01 : 0x00;
    cfg.tx_dbm = hp_mode ? 14 : 10;
}

static bool parseArgs(int argc, char **argv, bool &verbose, bool &do_reset, bool &hp_mode)
{
    for (int i = 1; i < argc; ++i) {
        if (std::strcmp(argv[i], "-v") == 0) {
            verbose = true;
            continue;
        }
        if (std::strcmp(argv[i], "--reset") == 0) {
            do_reset = true;
            continue;
        }
        if (std::strcmp(argv[i], "--hp") == 0) {
            hp_mode = true;
            continue;
        }
        if (std::strcmp(argv[i], "--lp") == 0) {
            hp_mode = false;
            continue;
        }
        std::fprintf(stderr, "Unknown option: %s\nUsage: sudo ./lora_rx [-v] [--reset] [--lp|--hp]\n", argv[i]);
        return false;
    }
    return true;
}

int main(int argc, char **argv)
{
    lr1121::Config cfg;
    cfg.freq_hz = lr1121::FREQ_433;
    cfg.busy_gpio = 24;
    cfg.reset_gpio = 25;
    cfg.dio9_gpio = 4;
    cfg.sf = 0x07;
    cfg.bw = 0x04;
    cfg.cr = 0x01;
    constexpr uint32_t rx_timeout_ms = 1000;
    bool do_reset = false;
    bool hp_mode = false;

    if (!parseArgs(argc, argv, cfg.verbose, do_reset, hp_mode)) return 2;
    applyPaPreset(cfg, hp_mode);
    std::printf("RX %u Hz, tmo=%ums, gpio(busy=%u reset=%u dio9=%u)%s\n",
                cfg.freq_hz, rx_timeout_ms, cfg.busy_gpio, cfg.reset_gpio, cfg.dio9_gpio, cfg.verbose ? " [v]" : "");

    lr1121::Radio radio;
    if (!radio.begin(cfg)) {
        std::fprintf(stderr, "ERROR: radio init failed\n");
        return 1;
    }
    if (do_reset) {
        std::puts("Applying soft settings reset...");
        if (!radio.softResetSettings()) {
            std::fprintf(stderr, "ERROR: soft settings reset failed\n");
            return 1;
        }
    }
    std::puts("Radio OK - listening...");

    uint8_t buf[256];
    int pkt = 0;
    uint32_t timeout_total = 0;
    uint32_t crc_total = 0;

    for (;;) {
        lr1121::RxInfo info{};
        const int r = radio.receive(buf, static_cast<uint8_t>(sizeof(buf) - 1), rx_timeout_ms, &info);

        if (r > 0) {
            buf[r] = '\0';
            std::printf("[%04d] %3d B rssi=%4d snr=%3d '%s'\n",
                        ++pkt, r, info.rssi_dbm, info.snr_db, buf);
        } else if (r == -1) {
            ++timeout_total;
            if ((timeout_total % 10u) == 0u) std::printf("timeout x%u\n", timeout_total);
        } else if (r == -2) {
            ++crc_total;
            std::printf("crc error x%u\n", crc_total);
        } else {
            std::printf("rx code=%d\n", r);
        }
    }
}