// Copyright 2024 Stefan Kerkmann (@KarlK90)
// Copyright 2021 QMK
// SPDX-License-Identifier: GPL-2.0-or-later

#include "uart.h"
#include "gpio.h"
#include "chibios_config.h"
#include <hal.h>

#ifndef UART_DRIVER
#    define UART_DRIVER SD1
#endif

#ifndef UART_TX_PIN
#    define UART_TX_PIN A9
#endif

#ifndef UART_TX_PAL_MODE
#    ifdef USE_GPIOV1
#        define UART_TX_PAL_MODE PAL_MODE_ALTERNATE_PUSHPULL
#    else
#        define UART_TX_PAL_MODE 7
#    endif
#endif

#ifndef UART_RX_PIN
#    define UART_RX_PIN A10
#endif

#ifndef UART_RX_PAL_MODE
#    ifdef USE_GPIOV1
#        define UART_RX_PAL_MODE PAL_MODE_INPUT
#    else
#        define UART_RX_PAL_MODE 7
#    endif
#endif

#ifndef UART_CTS_PIN
#    define UART_CTS_PIN A11
#endif

#ifndef UART_CTS_PAL_MODE
#    ifdef USE_GPIOV1
#        define UART_CTS_PAL_MODE PAL_MODE_INPUT
#    else
#        define UART_CTS_PAL_MODE 7
#    endif
#endif

#ifndef UART_RTS_PIN
#    define UART_RTS_PIN A12
#endif

#ifndef UART_RTS_PAL_MODE
#    ifdef USE_GPIOV1
#        define UART_RTS_PAL_MODE PAL_MODE_ALTERNATE_PUSHPULL
#    else
#        define UART_RTS_PAL_MODE 7
#    endif
#endif

#ifndef UART_CR1
#    define UART_CR1 0
#endif

#ifndef UART_CR2
#    define UART_CR2 0
#endif

#ifndef UART_CR3
#    define UART_CR3 0
#endif

#ifndef UART_WRDLEN
#    define UART_WRDLEN 3
#endif

#ifndef UART_STPBIT
#    define UART_STPBIT 0
#endif

#ifndef UART_PARITY
#    define UART_PARITY 0
#endif

#ifndef UART_ATFLCT
#    define UART_ATFLCT 0
#endif

#if defined(MCU_KINETIS)
static SerialConfig serialConfig = {SERIAL_DEFAULT_BITRATE};
#elif defined(WB32F3G71xx) || defined(WB32FQ95xx)
static SerialConfig serialConfig = {
    SERIAL_DEFAULT_BITRATE, UART_WRDLEN, UART_STPBIT, UART_PARITY, UART_ATFLCT,
};
#else
static SerialConfig serialConfig = {
    SERIAL_DEFAULT_BITRATE,
    UART_CR1,
    UART_CR2,
    UART_CR3,
};
#endif

void uart_init(uint32_t baud) {
    static bool is_initialised = false;

    if (is_initialised) {
        return;
    }
    is_initialised = true;

#if defined(MCU_KINETIS)
    serialConfig.sc_speed = baud;
#else
    serialConfig.speed = baud;
#endif

#if defined(USE_GPIOV1)
    palSetLineMode(UART_TX_PIN, UART_TX_PAL_MODE);
    palSetLineMode(UART_RX_PIN, UART_RX_PAL_MODE);
#else
    palSetLineMode(UART_TX_PIN, PAL_MODE_ALTERNATE(UART_TX_PAL_MODE) | PAL_OUTPUT_TYPE_PUSHPULL | PAL_OUTPUT_SPEED_HIGHEST);
    palSetLineMode(UART_RX_PIN, PAL_MODE_ALTERNATE(UART_RX_PAL_MODE) | PAL_OUTPUT_TYPE_PUSHPULL | PAL_OUTPUT_SPEED_HIGHEST);
#endif

    sdStart(&UART_DRIVER, &serialConfig);
}

void uart_write(uint8_t data) {
    sdPut(&UART_DRIVER, data);
}

uint8_t uart_read(void) {
    return (uint8_t)sdGet(&UART_DRIVER);
}

void uart_transmit(const uint8_t *data, uint16_t length) {
    sdWrite(&UART_DRIVER, data, length);
}

void uart_receive(uint8_t *data, uint16_t length) {
    sdRead(&UART_DRIVER, data, length);
}

bool uart_available(void) {
    return !sdGetWouldBlock(&UART_DRIVER);
}