#include <Arduino.h>
#include "settings.h"
#include "Log.h"
#include "Button.h"
#include "Cmd.h"
#include "Port.h"
#include "System.h"

// Only enable those buttons that are not disabled (99 or >115)
// 0 -> 39: GPIOs
// 100 -> 115: Port-expander
#if (NEXT_BUTTON >= 0 && NEXT_BUTTON <= 39)
    #define BUTTON_0_ENABLE
#elif (NEXT_BUTTON >= 100 && NEXT_BUTTON <= 115)
    #define EXPANDER_0_ENABLE
#endif
#if (PREVIOUS_BUTTON >= 0 && PREVIOUS_BUTTON <= 39)
    #define BUTTON_1_ENABLE
#elif (PREVIOUS_BUTTON >= 100 && PREVIOUS_BUTTON <= 115)
    #define EXPANDER_1_ENABLE
#endif
#if (PAUSEPLAY_BUTTON >= 0 && PAUSEPLAY_BUTTON <= 39)
    #define BUTTON_2_ENABLE
#elif (PAUSEPLAY_BUTTON >= 100 && PAUSEPLAY_BUTTON <= 115)
    #define EXPANDER_2_ENABLE
#endif
#ifdef USEROTARY_ENABLE
    #if (DREHENCODER_BUTTON >= 0 && DREHENCODER_BUTTON <= 39)
        #define BUTTON_3_ENABLE
    #elif (DREHENCODER_BUTTON >= 100 && DREHENCODER_BUTTON <= 115)
        #define EXPANDER_3_ENABLE
    #endif
#endif
#if (BUTTON_4 >= 0 && BUTTON_4 <= 39)
    #define BUTTON_4_ENABLE
#elif (BUTTON_4 >= 100 && BUTTON_4 <= 115)
    #define EXPANDER_4_ENABLE
#endif
#if (BUTTON_5 >= 0 && BUTTON_5 <= 39)
    #define BUTTON_5_ENABLE
#elif (BUTTON_5 >= 100 && BUTTON_5 <= 115)
    #define EXPANDER_5_ENABLE
#endif

t_button gButtons[7];         // next + prev + pplay + rotEnc + button4 + button5 + dummy-button
uint8_t gShutdownButton = 99; // Helper used for Neopixel: stores button-number of shutdown-button
#ifdef PORT_EXPANDER_ENABLE
    extern bool Port_AllowReadFromPortExpander;
#endif

static volatile SemaphoreHandle_t Button_TimerSemaphore;

hw_timer_t *Button_Timer = NULL;
static void IRAM_ATTR onTimer();
static void Button_DoButtonActions(void);

void Button_Init() {
    #if (WAKEUP_BUTTON <= 39)
        esp_sleep_enable_ext0_wakeup((gpio_num_t)WAKEUP_BUTTON, 0);
    #endif

    #ifdef NEOPIXEL_ENABLE // Try to find button that is used for shutdown via longpress-action (only necessary for Neopixel)
        #if defined(BUTTON_0_ENABLE) || defined(EXPANDER_0_ENABLE)
            #if (BUTTON_0_LONG == CMD_SLEEPMODE)
                gShutdownButton = 0;
            #endif
        #endif
        #if defined(BUTTON_1_ENABLE) || defined(EXPANDER_1_ENABLE)
            #if (BUTTON_1_LONG == CMD_SLEEPMODE)
                gShutdownButton = 1;
            #endif
        #endif
        #if defined(BUTTON_2_ENABLE) || defined(EXPANDER_2_ENABLE)
            #if (BUTTON_2_LONG == CMD_SLEEPMODE)
                gShutdownButton = 2;
            #endif
        #endif
        #if defined(BUTTON_3_ENABLE) || defined(EXPANDER_3_ENABLE)
            #if (BUTTON_3_LONG == CMD_SLEEPMODE)
                gShutdownButton = 3;
            #endif
        #endif
        #if defined(BUTTON_4_ENABLE) || defined(EXPANDER_4_ENABLE)
            #if (BUTTON_4_LONG == CMD_SLEEPMODE)
                gShutdownButton = 4;
            #endif
        #endif
        #if defined(BUTTON_5_ENABLE) || defined(EXPANDER_5_ENABLE)
            #if (BUTTON_5_LONG == CMD_SLEEPMODE)
                gShutdownButton = 5;
            #endif
        #endif
    #endif

    // Activate internal pullups for all enabled buttons
    #ifdef BUTTON_0_ENABLE
        pinMode(NEXT_BUTTON, INPUT_PULLUP);
    #endif
    #ifdef BUTTON_1_ENABLE
        pinMode(PREVIOUS_BUTTON, INPUT_PULLUP);
    #endif
    #ifdef BUTTON_2_ENABLE
        pinMode(PAUSEPLAY_BUTTON, INPUT_PULLUP);
    #endif
    #ifdef BUTTON_3_ENABLE
        pinMode(DREHENCODER_BUTTON, INPUT_PULLUP);
    #endif
    #ifdef BUTTON_4_ENABLE
        pinMode(BUTTON_4, INPUT_PULLUP);
    #endif
    #ifdef BUTTON_5_ENABLE
        pinMode(BUTTON_5, INPUT_PULLUP);
    #endif

    // Create 1000Hz-HW-Timer (currently only used for buttons)
    Button_TimerSemaphore = xSemaphoreCreateBinary();
    Button_Timer = timerBegin(0, 240, true); // Prescaler: CPU-clock in MHz
    timerAttachInterrupt(Button_Timer, &onTimer, true);
    timerAlarmWrite(Button_Timer, 10000, true); // 100 Hz
    timerAlarmEnable(Button_Timer);
}

// If timer-semaphore is set, read buttons (unless controls are locked)
void Button_Cyclic() {
    if (xSemaphoreTake(Button_TimerSemaphore, 0) == pdTRUE) {
        if (System_AreControlsLocked()) {
            return;
        }

        unsigned long currentTimestamp = millis();
        #ifdef PORT_EXPANDER_ENABLE
            Port_Cyclic();
        #endif

        // Buttons can be mixed between GPIO and port-expander.
        // But at the same time only one of them can be for example NEXT_BUTTON
        #if defined(BUTTON_0_ENABLE) || defined(EXPANDER_0_ENABLE)
                gButtons[0].currentState = Port_Read(NEXT_BUTTON);
        #endif
        #if defined(BUTTON_1_ENABLE) || defined(EXPANDER_1_ENABLE)
                gButtons[1].currentState = Port_Read(PREVIOUS_BUTTON);
        #endif
        #if defined(BUTTON_2_ENABLE) || defined(EXPANDER_2_ENABLE)
                gButtons[2].currentState = Port_Read(PAUSEPLAY_BUTTON);
        #endif
        #if defined(BUTTON_3_ENABLE) || defined(EXPANDER_3_ENABLE)
                gButtons[3].currentState = Port_Read(DREHENCODER_BUTTON);
        #endif
        #if defined(BUTTON_4_ENABLE) || defined(EXPANDER_4_ENABLE)
                gButtons[4].currentState = Port_Read(BUTTON_4);
        #endif
        #if defined(BUTTON_5_ENABLE) || defined(EXPANDER_5_ENABLE)
                gButtons[5].currentState = Port_Read(BUTTON_5);
        #endif

        // Iterate over all buttons in struct-array
        for (uint8_t i = 0; i < sizeof(gButtons) / sizeof(gButtons[0]); i++) {
            if (gButtons[i].currentState != gButtons[i].lastState && currentTimestamp - gButtons[i].lastPressedTimestamp > buttonDebounceInterval) {
                if (!gButtons[i].currentState) {
                    gButtons[i].isPressed = true;
                    gButtons[i].lastPressedTimestamp = currentTimestamp;
                } else {
                    gButtons[i].isReleased = true;
                    gButtons[i].lastReleasedTimestamp = currentTimestamp;
                }
            }
            gButtons[i].lastState = gButtons[i].currentState;
        }
    }
    Button_DoButtonActions();
}

// Do corresponding actions for all buttons
void Button_DoButtonActions(void) {
    if (gButtons[0].isPressed && gButtons[1].isPressed) {
        gButtons[0].isPressed = false;
        gButtons[1].isPressed = false;
        Cmd_Action(BUTTON_MULTI_01);
    } else if (gButtons[0].isPressed && gButtons[2].isPressed) {
        gButtons[0].isPressed = false;
        gButtons[2].isPressed = false;
        Cmd_Action(BUTTON_MULTI_02);
    } else if (gButtons[0].isPressed && gButtons[3].isPressed) {
        gButtons[0].isPressed = false;
        gButtons[3].isPressed = false;
        Cmd_Action(BUTTON_MULTI_03);
    } else if (gButtons[0].isPressed && gButtons[4].isPressed) {
        gButtons[0].isPressed = false;
        gButtons[4].isPressed = false;
        Cmd_Action(BUTTON_MULTI_04);
    } else if (gButtons[0].isPressed && gButtons[5].isPressed) {
        gButtons[0].isPressed = false;
        gButtons[5].isPressed = false;
        Cmd_Action(BUTTON_MULTI_05);
    } else if (gButtons[1].isPressed && gButtons[2].isPressed) {
        gButtons[1].isPressed = false;
        gButtons[2].isPressed = false;
        Cmd_Action(BUTTON_MULTI_12);
    } else if (gButtons[1].isPressed && gButtons[3].isPressed) {
        gButtons[1].isPressed = false;
        gButtons[3].isPressed = false;
        Cmd_Action(BUTTON_MULTI_13);
    } else if (gButtons[1].isPressed && gButtons[4].isPressed) {
        gButtons[1].isPressed = false;
        gButtons[4].isPressed = false;
        Cmd_Action(BUTTON_MULTI_14);
    } else if (gButtons[1].isPressed && gButtons[5].isPressed) {
        gButtons[1].isPressed = false;
        gButtons[5].isPressed = false;
        Cmd_Action(BUTTON_MULTI_15);
    } else if (gButtons[2].isPressed && gButtons[3].isPressed) {
        gButtons[2].isPressed = false;
        gButtons[3].isPressed = false;
        Cmd_Action(BUTTON_MULTI_23);
    } else if (gButtons[2].isPressed && gButtons[4].isPressed) {
        gButtons[2].isPressed = false;
        gButtons[4].isPressed = false;
        Cmd_Action(BUTTON_MULTI_24);
    } else if (gButtons[2].isPressed && gButtons[5].isPressed) {
        gButtons[2].isPressed = false;
        gButtons[5].isPressed = false;
        Cmd_Action(BUTTON_MULTI_25);
    } else if (gButtons[3].isPressed && gButtons[4].isPressed) {
        gButtons[3].isPressed = false;
        gButtons[4].isPressed = false;
        Cmd_Action(BUTTON_MULTI_34);
    } else if (gButtons[3].isPressed && gButtons[5].isPressed) {
        gButtons[3].isPressed = false;
        gButtons[5].isPressed = false;
        Cmd_Action(BUTTON_MULTI_35);
    } else if (gButtons[4].isPressed && gButtons[5].isPressed) {
        gButtons[4].isPressed = false;
        gButtons[5].isPressed = false;
        Cmd_Action(BUTTON_MULTI_45);
    } else {
        for (uint8_t i = 0; i < sizeof(gButtons) / sizeof(gButtons[0]); i++) {
            if (gButtons[i].isPressed) {
                if (gButtons[i].lastReleasedTimestamp > gButtons[i].lastPressedTimestamp) {
                    if (gButtons[i].lastReleasedTimestamp - gButtons[i].lastPressedTimestamp >= intervalToLongPress) {
                        switch (i) { // Long-press-actions
                            case 0:
                                Cmd_Action(BUTTON_0_LONG);
                                gButtons[i].isPressed = false;
                                break;

                            case 1:
                                Cmd_Action(BUTTON_1_LONG);
                                gButtons[i].isPressed = false;
                                break;

                            case 2:
                                Cmd_Action(BUTTON_2_LONG);
                                gButtons[i].isPressed = false;
                                break;

                            case 3:
                                Cmd_Action(BUTTON_3_LONG);
                                gButtons[i].isPressed = false;
                                break;

                            case 4:
                                Cmd_Action(BUTTON_4_LONG);
                                gButtons[i].isPressed = false;
                                break;

                            case 5:
                                Cmd_Action(BUTTON_5_LONG);
                                gButtons[i].isPressed = false;
                                break;
                        }
                    } else {
                        switch (i) { // Short-press-actions
                            case 0:
                                Cmd_Action(BUTTON_0_SHORT);
                                gButtons[i].isPressed = false;
                                break;

                            case 1:
                                Cmd_Action(BUTTON_1_SHORT);
                                gButtons[i].isPressed = false;
                                break;

                            case 2:
                                Cmd_Action(BUTTON_2_SHORT);
                                gButtons[i].isPressed = false;
                                break;

                            case 3:
                                Cmd_Action(BUTTON_3_SHORT);
                                gButtons[i].isPressed = false;
                                break;

                            case 4:
                                Cmd_Action(BUTTON_4_SHORT);
                                gButtons[i].isPressed = false;
                                break;

                            case 5:
                                Cmd_Action(BUTTON_5_SHORT);
                                gButtons[i].isPressed = false;
                                break;
                        }
                    }
                }
            }
        }
    }
}

void IRAM_ATTR onTimer() {
    xSemaphoreGiveFromISR(Button_TimerSemaphore, NULL);
}