#include "spiInterrupts.h"

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

#include <stddef.h>

user_handler_t spi1UserHandler = nullptr;
user_args_t spi1UserArgs = nullptr;

user_handler_t spi2UserHandler = nullptr;
user_args_t spi2UserArgs = nullptr;

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

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

/**
 * @brief  This function handles SPIx interrupt request.
 * @param  None
 * @retval None
 */
void spi::spi1IrqHandler(void* spiHandle)
{
    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;
    }
}

/* 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();
}