#include "dispatch.h"
#include "pico/stdlib.h"
#include "wire.h"
#include "timer_queue.h"
#include "net.h"
#include "igmp.h"
#include "debug_log.h"
#include "hardware/sync.h"

static timer_queue timers;

static void igmp_reannounce() {
    auto& ns = net_get_state();
    igmp::send_all_reports(ns.mac, ns.ip);
    dispatch_schedule_ms(60000, igmp_reannounce);
}

void dispatch_init() {
    net_init();
    dispatch_schedule_ms(60000, igmp_reannounce);
    dlog("dispatch_init complete");
}

timer_handle dispatch_schedule_ms(uint32_t ms, std::function<void()> fn) {
    auto h = timers.schedule_ms(ms, std::move(fn));
    if (!h) dlogf("timer alloc failed: %lu ms", static_cast<unsigned long>(ms));
    return h;
}

bool dispatch_cancel_timer(timer_handle h) {
    return timers.cancel(h);
}

[[noreturn]] void dispatch_run(std::span<const handler_entry> handlers) {
    std::array<handler_fn, 128> handler_map{};
    for (auto& entry : handlers) {
        handler_map[entry.type_id] = entry.handle;
    }

    static std::array<uint8_t, 4096> tx_buf;

    auto dispatch_msg = [&](const DecodedMessage& msg, send_fn send) {
        if (msg.type_id < 0 || !handler_map[msg.type_id]) {
            dlogf("dispatch: unknown type_id %d", msg.type_id);
            return;
        }
        responder resp{msg.message_id, std::move(send)};
        handler_map[msg.type_id](resp, msg.payload);
    };

    net_set_handler([&](std::span<const uint8_t> payload,
                        send_fn send) {
        auto msg = try_decode(payload.data(), payload.size());
        if (!msg) return;
        dispatch_msg(*msg, std::move(send));
    });

    while (true) {
        uint32_t save = save_and_disable_interrupts();

        dlog_if_slow("timers", 1000, [&]{ timers.run(); });
        dlog_if_slow("net_poll", 1000, [&]{ net_poll(std::span{tx_buf}); });

        __wfi();
        restore_interrupts(save);
    }
}