#include "fsfw-hal/stm32h7/spi/spiInterrupts.h"
#include "fsfw-hal/stm32h7/spi/spiCore.h"

#include "stm32h7xx_hal.h"
#include "stm32h7xx_hal_dma.h"
#include "stm32h7xx_hal_spi.h"

#include <stddef.h>

user_handler_t spi1UserHandler = &spi::spiIrqHandler;
user_args_t spi1UserArgs = nullptr;

user_handler_t spi2UserHandler = &spi::spiIrqHandler;
user_args_t spi2UserArgs = nullptr;

/**
 * @brief  This function handles DMA Rx interrupt request.
 * @param  None
 * @retval None
 */
void spi::dmaRxIrqHandler(void* dmaHandle) {
    if(dmaHandle == nullptr) {
        return;
    }
    HAL_DMA_IRQHandler((DMA_HandleTypeDef *) dmaHandle);
}

/**
 * @brief  This function handles DMA Rx interrupt request.
 * @param  None
 * @retval None
 */
void spi::dmaTxIrqHandler(void* dmaHandle) {
    if(dmaHandle == nullptr) {
        return;
    }
    HAL_DMA_IRQHandler((DMA_HandleTypeDef *) dmaHandle);
}

/**
 * @brief  This function handles SPIx interrupt request.
 * @param  None
 * @retval None
 */
void spi::spiIrqHandler(void* spiHandle) {
    if(spiHandle == nullptr) {
        return;
    }
    //auto currentSpiHandle = spi::getSpiHandle();
    HAL_SPI_IRQHandler((SPI_HandleTypeDef *) spiHandle);
}

void spi::assignSpiUserHandler(spi::SpiBus spiIdx, user_handler_t userHandler,
        user_args_t userArgs) {
    if(spiIdx == spi::SpiBus::SPI_1) {
        spi1UserHandler = userHandler;
        spi1UserArgs = userArgs;
    }
    else {
        spi2UserHandler = userHandler;
        spi2UserArgs = userArgs;
    }
}

void spi::getSpiUserHandler(spi::SpiBus spiBus, user_handler_t *userHandler,
        user_args_t *userArgs) {
    if(userHandler == nullptr or userArgs == nullptr) {
        return;
    }
    if(spiBus == spi::SpiBus::SPI_1) {
        *userArgs = spi1UserArgs;
        *userHandler = spi1UserHandler;
    }
    else {
        *userArgs = spi2UserArgs;
        *userHandler = spi2UserHandler;
    }
}

void spi::assignSpiUserArgs(spi::SpiBus spiBus, user_args_t userArgs) {
    if(spiBus == spi::SpiBus::SPI_1) {
        spi1UserArgs = userArgs;
    }
    else {
        spi2UserArgs = userArgs;
    }
}

/* Do not change these function names! They need to be exactly equal to the name of the functions
defined in the startup_stm32h743xx.s files! */

extern "C" void SPI1_IRQHandler() {
    if(spi1UserHandler != NULL) {
        spi1UserHandler(spi1UserArgs);
        return;
    }
    Default_Handler();
}

extern "C"  void SPI2_IRQHandler() {
    if(spi2UserHandler != nullptr) {
        spi2UserHandler(spi2UserArgs);
        return;
    }
    Default_Handler();
}