#include "furi_hal_console.h"
#include "common_defines.h"
#include "furi_string.h"

#include <stdbool.h>
#include <esp_log.h>
#include <memory.h>

#define TAG "FuriHalConsole"

#ifdef HEAP_PRINT_DEBUG
#define CONSOLE_BAUDRATE 1843200
#else
#define CONSOLE_BAUDRATE 230400
#endif

typedef struct {
    bool alive;
    FuriHalConsoleTxCallback tx_callback;
    void* tx_callback_context;
} FuriHalConsole;

FuriHalConsole furi_hal_console = {
    .alive = false,
    .tx_callback = NULL,
    .tx_callback_context = NULL,
};

void furi_hal_console_init() {
//    furi_hal_uart_init(FuriHalUartIdUSART1, CONSOLE_BAUDRATE);
    furi_hal_console.alive = true;
}

void furi_hal_console_enable() {
//    furi_hal_uart_set_irq_cb(FuriHalUartIdUSART1, NULL, NULL);
//    while(!LL_USART_IsActiveFlag_TC(USART1))
//        ;
//    furi_hal_uart_set_br(FuriHalUartIdUSART1, CONSOLE_BAUDRATE);
    furi_hal_console.alive = true;
}

void furi_hal_console_disable() {
//    while(!LL_USART_IsActiveFlag_TC(USART1))
//        ;
    furi_hal_console.alive = false;
}

void furi_hal_console_set_tx_callback(FuriHalConsoleTxCallback callback, void* context) {
    FURI_CRITICAL_ENTER();
    furi_hal_console.tx_callback = callback;
    furi_hal_console.tx_callback_context = context;
    FURI_CRITICAL_EXIT();
}

void furi_hal_console_tx(const uint8_t* buffer, size_t buffer_size) {
    if(!furi_hal_console.alive) return;

    FURI_CRITICAL_ENTER();
    // Transmit data

    if(furi_hal_console.tx_callback) {
        furi_hal_console.tx_callback(buffer, buffer_size, furi_hal_console.tx_callback_context);
    }

    char safe_buffer[buffer_size + 1];
    memcpy(safe_buffer, buffer, buffer_size);
    safe_buffer[buffer_size] = 0;

    ESP_LOGI(TAG, "%s", safe_buffer);
//    furi_hal_uart_tx(FuriHalUartIdUSART1, (uint8_t*)buffer, buffer_size);
////     Wait for TC flag to be raised for last char
//    while(!LL_USART_IsActiveFlag_TC(USART1))
//        ;
    FURI_CRITICAL_EXIT();
}

void furi_hal_console_tx_with_new_line(const uint8_t* buffer, size_t buffer_size) {
    if(!furi_hal_console.alive) return;

    FURI_CRITICAL_ENTER();

    char safe_buffer[buffer_size + 1];
    memcpy(safe_buffer, buffer, buffer_size);
    safe_buffer[buffer_size] = 0;
    ESP_LOGI(TAG, "%s", safe_buffer);

    // Transmit data
//    furi_hal_uart_tx(FuriHalUartIdUSART1, (uint8_t*)buffer, buffer_size);
    // Transmit new line symbols
//    furi_hal_uart_tx(FuriHalUartIdUSART1, (uint8_t*)"\r\n", 2);
    // Wait for TC flag to be raised for last char
//    while(!LL_USART_IsActiveFlag_TC(USART1))
//        ;
    FURI_CRITICAL_EXIT();
}

void furi_hal_console_printf(const char format[], ...) {
    FuriString* string;
    va_list args;
    va_start(args, format);
    string = furi_string_alloc_vprintf(format, args);
    va_end(args);
    furi_hal_console_tx((const uint8_t*)furi_string_get_cstr(string), furi_string_size(string));
    furi_string_free(string);
}

void furi_hal_console_puts(const char* data) {
    furi_hal_console_tx((const uint8_t*)data, strlen(data));
}