#ifndef FSFW_HAL_STM32H7_SPI_MSPINIT_H_
#define FSFW_HAL_STM32H7_SPI_MSPINIT_H_

#include <cstdint>

#include "../definitions.h"
#include "../dma.h"
#include "spiDefinitions.h"
#include "stm32h7xx_hal_spi.h"

#ifdef __cplusplus
extern "C" {
#endif

using mspCb = void (*)(void);

/**
 * @brief   This file provides MSP implementation for DMA, IRQ and Polling mode for the
 *          SPI peripheral. This configuration is required for the SPI communication to work.
 */
namespace spi {

struct MspCfgBase {
  MspCfgBase() {}
  MspCfgBase(stm32h7::GpioCfg sck, stm32h7::GpioCfg mosi, stm32h7::GpioCfg miso,
             mspCb cleanupCb = nullptr, mspCb setupCb = nullptr)
      : sck(sck), mosi(mosi), miso(miso), cleanupCb(cleanupCb), setupCb(setupCb) {}

  virtual ~MspCfgBase() = default;

  stm32h7::GpioCfg sck;
  stm32h7::GpioCfg mosi;
  stm32h7::GpioCfg miso;

  mspCb cleanupCb = nullptr;
  mspCb setupCb = nullptr;
};

struct MspPollingConfigStruct : public MspCfgBase {
  MspPollingConfigStruct() : MspCfgBase(){};
  MspPollingConfigStruct(stm32h7::GpioCfg sck, stm32h7::GpioCfg mosi, stm32h7::GpioCfg miso,
                         mspCb cleanupCb = nullptr, mspCb setupCb = nullptr)
      : MspCfgBase(sck, mosi, miso, cleanupCb, setupCb) {}
};

/* A valid instance of this struct must be passed to the MSP initialization function as a void*
argument */
struct MspIrqConfigStruct : public MspPollingConfigStruct {
  MspIrqConfigStruct() : MspPollingConfigStruct(){};
  MspIrqConfigStruct(stm32h7::GpioCfg sck, stm32h7::GpioCfg mosi, stm32h7::GpioCfg miso,
                     mspCb cleanupCb = nullptr, mspCb setupCb = nullptr)
      : MspPollingConfigStruct(sck, mosi, miso, cleanupCb, setupCb) {}

  SpiBus spiBus = SpiBus::SPI_1;
  user_handler_t spiIrqHandler = nullptr;
  user_args_t spiUserArgs = nullptr;
  IRQn_Type spiIrqNumber = SPI1_IRQn;
  // Priorities for NVIC
  // Pre-Empt priority ranging from 0 to 15. If FreeRTOS calls are used, only 5-15 are allowed
  IrqPriorities preEmptPriority = IrqPriorities::LOWEST;
  IrqPriorities subpriority = IrqPriorities::LOWEST;
};

/* A valid instance of this struct must be passed to the MSP initialization function as a void*
argument */
struct MspDmaConfigStruct : public MspIrqConfigStruct {
  MspDmaConfigStruct() : MspIrqConfigStruct(){};
  MspDmaConfigStruct(stm32h7::GpioCfg sck, stm32h7::GpioCfg mosi, stm32h7::GpioCfg miso,
                     mspCb cleanupCb = nullptr, mspCb setupCb = nullptr)
      : MspIrqConfigStruct(sck, mosi, miso, cleanupCb, setupCb) {}
  void (*dmaClkEnableWrapper)(void) = nullptr;

  dma::DMAIndexes txDmaIndex = dma::DMAIndexes::DMA_1;
  dma::DMAIndexes rxDmaIndex = dma::DMAIndexes::DMA_1;
  dma::DMAStreams txDmaStream = dma::DMAStreams::STREAM_0;
  dma::DMAStreams rxDmaStream = dma::DMAStreams::STREAM_0;
  IRQn_Type txDmaIrqNumber = DMA1_Stream0_IRQn;
  IRQn_Type rxDmaIrqNumber = DMA1_Stream1_IRQn;
  // Priorities for NVIC
  IrqPriorities txPreEmptPriority = IrqPriorities::LOWEST;
  IrqPriorities rxPreEmptPriority = IrqPriorities::LOWEST;
  IrqPriorities txSubpriority = IrqPriorities::LOWEST;
  IrqPriorities rxSubpriority = IrqPriorities::LOWEST;
};

using msp_func_t = void (*)(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);

void getMspInitFunction(msp_func_t* init_func, MspCfgBase** args);
void getMspDeinitFunction(msp_func_t* deinit_func, MspCfgBase** args);

void halMspInitDma(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);
void halMspDeinitDma(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);

void halMspInitInterrupt(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);
void halMspDeinitInterrupt(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);

void halMspInitPolling(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);
void halMspDeinitPolling(SPI_HandleTypeDef* hspi, MspCfgBase* cfg);

/**
 * Assign MSP init functions. Important for SPI configuration
 * @param init_func
 * @param init_args
 * @param deinit_func
 * @param deinit_args
 */
void setSpiDmaMspFunctions(MspDmaConfigStruct* cfg, msp_func_t initFunc = &spi::halMspInitDma,
                           msp_func_t deinitFunc = &spi::halMspDeinitDma);
void setSpiIrqMspFunctions(MspIrqConfigStruct* cfg, msp_func_t initFunc = &spi::halMspInitInterrupt,
                           msp_func_t deinitFunc = &spi::halMspDeinitInterrupt);
void setSpiPollingMspFunctions(MspPollingConfigStruct* cfg,
                               msp_func_t initFunc = &spi::halMspInitPolling,
                               msp_func_t deinitFunc = &spi::halMspDeinitPolling);

void mspErrorHandler(const char* const function, const char* const message);

}  // namespace spi

#ifdef __cplusplus
}
#endif

#endif /* FSFW_HAL_STM32H7_SPI_MSPINIT_H_ */