#include "i2c_addresses.h"
#include "i2c.h"
#include "slave_drivers/slave_driver_uhk_module.h"
#include "bridge_protocol.h"
#include "main.h"
#include "peripherals/test_led.h"
#include "test_states.h"

uhk_module_state_t UhkModuleStates[UHK_MODULE_MAX_COUNT];
uhk_module_field_t currentUhkModuleField = UhkModuleField_SendKeystatesRequestCommand;
uhk_module_state_t uhkModuleExternalStates[UHK_MODULE_MAX_COUNT];
uint8_t txBuffer[2];

void InitUhkModules()
{
    for (uint8_t moduleId=0; moduleId<UHK_MODULE_MAX_COUNT; moduleId++) {
        uhk_module_state_t* uhkModuleState = UhkModuleStates + moduleId;
        uhkModuleState->isTestLedOn = true;
        uhkModuleState->ledPwmBrightness = 0x64;
    }
}

void UhkSlaveUhkModuleHandler(uint8_t uhkModuleId)
{
    uhk_module_state_t *uhkModuleInternalState = UhkModuleStates + uhkModuleId;
    //uhk_module_state_t *uhkModuleExternalState = uhkModuleExternalStates + uhkModuleId;

    switch (currentUhkModuleField) {
        case UhkModuleField_SendKeystatesRequestCommand:
            txBuffer[0] = BridgeCommand_GetKeyStates;
            I2cAsyncWrite(I2C_ADDRESS_LEFT_KEYBOARD_HALF, txBuffer, 1);
            currentUhkModuleField = UhkModuleField_ReceiveKeystates;
            break;
        case UhkModuleField_ReceiveKeystates:
            I2cAsyncRead(I2C_ADDRESS_LEFT_KEYBOARD_HALF, CurrentKeyStates[SLOT_ID_LEFT_KEYBOARD_HALF], LEFT_KEYBOARD_HALF_KEY_COUNT);
            currentUhkModuleField = UhkModuleField_SendPwmBrightnessCommand;
            break;
        case UhkModuleField_SendPwmBrightnessCommand:
            txBuffer[0] = BridgeCommand_SetLedPwmBrightness;
            txBuffer[1] = uhkModuleInternalState->ledPwmBrightness;
            I2cAsyncWrite(I2C_ADDRESS_LEFT_KEYBOARD_HALF, txBuffer, 2);
            currentUhkModuleField = UhkModuleField_SendTestLedCommand;
            break;
        case UhkModuleField_SendTestLedCommand:
            txBuffer[0] = BridgeCommand_SetTestLed;
            txBuffer[1] = uhkModuleInternalState->isTestLedOn;
            I2cAsyncWrite(I2C_ADDRESS_LEFT_KEYBOARD_HALF, txBuffer, 2);
            currentUhkModuleField = UhkModuleField_SendDisableKeyMatrixScanState;
            break;
        case UhkModuleField_SendDisableKeyMatrixScanState:
            txBuffer[0] = BridgeCommand_SetDisableKeyMatrixScanState;
            txBuffer[1] = TestStates.disableKeyMatrixScan;
            I2cAsyncWrite(I2C_ADDRESS_LEFT_KEYBOARD_HALF, txBuffer, 2);
            currentUhkModuleField = UhkModuleField_SendLedPwmBrightness;
            break;
        case UhkModuleField_SendLedPwmBrightness:
            txBuffer[0] = BridgeCommand_SetDisableKeyMatrixScanState;
            txBuffer[1] = TestStates.disableKeyMatrixScan;
            I2cAsyncWrite(I2C_ADDRESS_LEFT_KEYBOARD_HALF, txBuffer, 2);
            currentUhkModuleField = UhkModuleField_DisableLedSdb;
            break;
        case UhkModuleField_DisableLedSdb:
            txBuffer[0] = BridgeCommand_SetDisableLedSdb;
            txBuffer[1] = TestStates.disableLedSdb;
            I2cAsyncWrite(I2C_ADDRESS_LEFT_KEYBOARD_HALF, txBuffer, 2);
            currentUhkModuleField = UhkModuleField_SendKeystatesRequestCommand;
            break;
    }
}