#include "ch.h"
#include "hal.h"
#include "annepro2.h"
#include "annepro2_ble.h"
#include "spi_master.h"
#include "qmk_ap2_led.h"

static const SerialConfig ledUartConfig = {
  .speed = 115200,
};

static const SerialConfig bleUartConfig = {
  .speed = 115200,
};

static uint8_t ledMcuWakeup[11] = {
    0x7b, 0x10, 0x43, 0x10, 0x03, 0x00, 0x00, 0x7d, 0x02, 0x01, 0x02
};

static bool ledEnabled = false;

ble_capslock_t BLECapsLock = {._dummy = {0}, .caps_lock = false};

uint16_t annepro2LedMatrix[MATRIX_ROWS * MATRIX_COLS] = {
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,
  0,0,0,0,0,0,0,0,0,0,0,0,0,0,
};

void OVERRIDE keyboard_pre_init_kb(void) {
#if HAL_USE_SPI == TRUE
    spi_init();
#endif
}

void OVERRIDE keyboard_post_init_kb(void) {
    // Start LED UART
    sdStart(&SD0, &ledUartConfig);
    sdWrite(&SD0, ledMcuWakeup, 11);

    // wait to receive response from wakeup
    wait_ms(15);

    // loop to clear out receive buffer from shine wakeup
    while(!sdGetWouldBlock(&SD0))
        sdGet(&SD0);

    // Start BLE UART
    sdStart(&SD1, &bleUartConfig);
    annepro2_ble_startup();

    // Give the send uart thread some time to
    // send out the queue before we read back
    wait_ms(15);

    // loop to clear out receive buffer from ble wakeup
    while(!sdGetWouldBlock(&SD1))
        sdGet(&SD1);

    keyboard_post_init_user();
}

void OVERRIDE matrix_init_kb(void) {
    matrix_init_user();
}

void matrix_scan_kb() {
    // if there's stuff on the ble serial buffer
    // read it into the capslock struct
    while(!sdGetWouldBlock(&SD1)) {
        sdReadTimeout(&SD1, (uint8_t *) &BLECapsLock, sizeof(ble_capslock_t), 10);

        // if it's capslock from ble, darken led
        if (BLECapsLock.caps_lock) {
            annepro2LedClearMask(MATRIX_COLS * 2);
        } else {
            annepro2LedSetMask(MATRIX_COLS * 2);
        }
    }


    matrix_scan_user();
}

/*!
 * @returns false   processing for this keycode has been completed.
 */
bool OVERRIDE process_record_kb(uint16_t keycode, keyrecord_t *record) {
    if (record->event.pressed) {
        if (AP2_LED_ENABLED && AP2_LED_DYNAMIC_PROFILE) {
            annepro2LedForwardKeypress(record->event.key.row, record->event.key.col);
        }

        switch (keycode) {
            case KC_AP2_BT1:
                annepro2_ble_broadcast(0);
                return false;

            case KC_AP2_BT2:
                annepro2_ble_broadcast(1);
                return false;

            case KC_AP2_BT3:
                annepro2_ble_broadcast(2);
                return false;

            case KC_AP2_BT4:
                annepro2_ble_broadcast(3);
                return false;

            case KC_AP2_USB:
                annepro2_ble_disconnect();
                return false;

            case KC_AP2_BT_UNPAIR:
                annepro2_ble_unpair();
                return false;

            case KC_AP_LED_OFF:
                annepro2LedDisable();
                ledEnabled = false;
                break;

            case KC_AP_LED_ON:
                if (ledEnabled) {
                    annepro2LedNextProfile();
                } else {
                    ledEnabled = true;
                    annepro2LedEnable();
                }
                break;

            case KC_AP_LED_NEXT_PROFILE:
                annepro2LedNextProfile();
                break;

            case KC_AP_LED_PREV_PROFILE:
                annepro2LedPrevProfile();
                break;

            case KC_AP_LED_NEXT_INTENSITY:
                annepro2LedNextIntensity();
                return false;
            case KC_AP_LED_SPEED:
                annepro2LedNextAnimationSpeed();
                return false;
            default:
                break;
        }
    }
    return process_record_user(keycode, record);
}