#include "fsl_i2c.h"
#include "slave_scheduler.h"
#include "slot.h"
#include "main.h"
#include "slave_drivers/slave_driver_led_driver.h"
#include "slave_drivers/slave_driver_uhk_module.h"
#include "i2c.h"
#include "i2c_addresses.h"
#include "test_states.h"

uint8_t currentBridgeSlaveId = 0;

slave_driver_initializer_t slaveDriverInitializers[] = {
    UhkModuleSlaveDriver_Init,
    LedSlaveDriver_Init,
};

uhk_slave_t slaves[] = {
    { .updater = UhkModuleSlaveDriver_Update, .moduleId = 0 },
    { .updater = LedSlaveDriver_Update,       .moduleId = 0 },
    { .updater = LedSlaveDriver_Update,       .moduleId = 1 },
};

static void bridgeProtocolCallback(I2C_Type *base, i2c_master_handle_t *handle, status_t status, void *userData)
{
    if (TestStates.disableI2c) {
        return;
    }
    uhk_slave_t *bridgeSlave = slaves + currentBridgeSlaveId;

    bridgeSlave->updater(bridgeSlave->moduleId);
    currentBridgeSlaveId++;

    if (currentBridgeSlaveId >= (sizeof(slaves) / sizeof(uhk_slave_t))) {
        currentBridgeSlaveId = 0;
    }
}

static void initSlaveDrivers()
{
    for (uint8_t i=0; i<sizeof(slaveDriverInitializers) / sizeof(slave_driver_initializer_t); i++) {
        slaveDriverInitializers[i]();
    }
}

void InitSlaveScheduler()
{
    initSlaveDrivers();
    I2C_MasterTransferCreateHandle(I2C_MAIN_BUS_BASEADDR, &I2cMasterHandle, bridgeProtocolCallback, NULL);

    // Kickstart the scheduler by triggering the first callback.
    slaves[0].updater(slaves[0].moduleId);
}