#pragma once
#include "../Figments/Input.h"

class Bounce {
public:
    void attach(int pin, PinMode buttonPinMode) {
        m_pin = pin;
        pinMode(m_pin, buttonPinMode);
        Log.info("Attaching a button to %d", pin);
    }

    void update() {
        int readResult = digitalRead(m_pin);
        if (m_state == Ready) {
            if (readResult == HIGH) {
                m_state = Started;
                m_downStart = millis();
                //Log.info("Button %d is started!", m_pin);
            }
        } else if (m_state == Started && millis() - m_downStart >= m_interval) {
            if (readResult == HIGH) {
                m_state = Confirmed;
                //Log.info("Button %d is CONFIRMED!", m_pin);
            } else {
                m_state = Ready;
                //Log.info("Button %d bounced back to ready!", m_pin);
            }
        } else if (m_state == Confirmed || m_state == Held) {
            if (readResult == LOW) {
                //Log.info("Button %d is released", m_pin);
                m_state = Released;
            } else if (m_state == Confirmed) {
                m_state = Held;
                //Log.info("Button %d is being held down!", m_pin);
            }
        } else if (m_state == Released) {
            //Log.info("Button %d is ready!", m_pin);
            m_state = Ready;
        }
    }

    void interval(uint8_t v) {
        m_interval = v;
    }

    bool fell() const {
        return m_state == Confirmed;
    }

    bool rose() const {
        return m_state == Released;
    }

    bool held() const {
        return m_state == Held;
    }

private:
    enum State {
        Ready,
        Started,
        Confirmed,
        Held,
        Released
    };

    State m_state = Ready;
    unsigned int m_pin = 0;
    unsigned int m_downStart = 0;
    unsigned int m_interval = 10;
};

class Buttons: public InputSource {
public:
    Buttons() : InputSource("Buttons") {}
    void onStart() override;
    InputEvent read() override;

    enum Chord {
        None = 0,
        Circle = 1,
        Triangle = 2,
        Cross = 4,
        CircleTriangle = Circle | Triangle,
        CircleCross = Circle | Cross,
        TriangleCross = Triangle | Cross,
        CircleTriangleCross = Circle | Triangle | Cross
    };

private:
    Bounce m_buttons[3];
    Chord m_buttonMap[3] = {Circle, Triangle, Cross};
    bool m_wasChord[3] = {false, false, false};
};