#include "blackhat_app_i.h"
#include "blackhat_uart.h"

struct BlackhatUart {
    BlackhatApp* app;
    FuriThread* rx_thread;
    FuriStreamBuffer* rx_stream;
    uint8_t rx_buf[RX_BUF_SIZE + 1];
    void (*handle_rx_data_cb)(uint8_t* buf, size_t len, void* context);
    FuriHalSerialHandle* serial_handle;
};

typedef enum {
    WorkerEvtStop = (1 << 0),
    WorkerEvtRxDone = (1 << 1),
} WorkerEvtFlags;

void blackhat_uart_set_handle_rx_data_cb(
    BlackhatUart* uart,
    void (*handle_rx_data_cb)(uint8_t* buf, size_t len, void* context)
)
{
    furi_assert(uart);
    uart->handle_rx_data_cb = handle_rx_data_cb;
}

#define WORKER_ALL_RX_EVENTS (WorkerEvtStop | WorkerEvtRxDone)

void blackhat_uart_on_irq_cb(
    FuriHalSerialHandle* handle, FuriHalSerialRxEvent event, void* context
)
{
    BlackhatUart* uart = (BlackhatUart*)context;

    if (event == FuriHalSerialRxEventData) {
        uint8_t data = furi_hal_serial_async_rx(handle);
        furi_stream_buffer_send(uart->rx_stream, &data, 1, 0);
        furi_thread_flags_set(
            furi_thread_get_id(uart->rx_thread), WorkerEvtRxDone
        );
    }
}

static int32_t uart_worker(void* context)
{
    BlackhatUart* uart = (void*)context;

    while (1) {
        uint32_t events = furi_thread_flags_wait(
            WORKER_ALL_RX_EVENTS, FuriFlagWaitAny, FuriWaitForever
        );
        furi_check((events & FuriFlagError) == 0);
        if (events & WorkerEvtStop) break;
        if (events & WorkerEvtRxDone) {
            size_t len = furi_stream_buffer_receive(
                uart->rx_stream, uart->rx_buf, RX_BUF_SIZE, 0
            );

            if (len > 0) {
                if (uart->handle_rx_data_cb) {
                    uart->handle_rx_data_cb(uart->rx_buf, len, uart->app);
                } else {
                    uart->rx_buf[len] = '\0';
                }
            }
        }
    }

    furi_stream_buffer_free(uart->rx_stream);

    return 0;
}

void blackhat_uart_tx(BlackhatUart* uart, char* data, size_t len)
{
    furi_hal_serial_tx(uart->serial_handle, (uint8_t*)data, len);
}

BlackhatUart* blackhat_uart_init(BlackhatApp* app)
{
    BlackhatUart* uart = malloc(sizeof(BlackhatUart));
    uart->app = app;
    // Init all rx stream and thread early to avoid crashes
    uart->rx_stream = furi_stream_buffer_alloc(RX_BUF_SIZE, 1);
    uart->rx_thread = furi_thread_alloc();
    furi_thread_set_name(uart->rx_thread, "BlackhatUartRxThread");
    furi_thread_set_stack_size(uart->rx_thread, 1024);
    furi_thread_set_context(uart->rx_thread, uart);
    furi_thread_set_callback(uart->rx_thread, uart_worker);

    furi_thread_start(uart->rx_thread);

    uart->serial_handle = furi_hal_serial_control_acquire(UART_CH);
    furi_check(uart->serial_handle);
    furi_hal_serial_init(uart->serial_handle, 115200);
    furi_hal_serial_async_rx_start(
        uart->serial_handle, blackhat_uart_on_irq_cb, uart, false
    );

    return uart;
}

void blackhat_uart_free(BlackhatUart* uart)
{
    furi_assert(uart);

    furi_hal_serial_async_rx_stop(uart->serial_handle);
    furi_hal_serial_deinit(uart->serial_handle);
    furi_hal_serial_control_release(uart->serial_handle);

    furi_thread_flags_set(furi_thread_get_id(uart->rx_thread), WorkerEvtStop);
    furi_thread_join(uart->rx_thread);
    furi_thread_free(uart->rx_thread);

    free(uart);
}