DonCon2040

#ifndef UTILS_MENU_H_
#define UTILS_MENU_H_

#include "utils/InputState.h"
#include "utils/SettingsStore.h"

#include <map>
#include <memory>
#include <stack>
#include <string>
#include <vector>

namespace Doncon::Utils {

class Menu {
  public:
    enum class Page : uint8_t {
        Main,

        DeviceMode,
        Drum,
        Led,
        Reset,
        Bootsel,

        DrumDebounceDelay,
        DrumTriggerThresholds,
        DrumDoubleTrigger,

        DrumTriggerThresholdKaLeft,
        DrumTriggerThresholdDonLeft,
        DrumTriggerThresholdDonRight,
        DrumTriggerThresholdKaRight,

        DrumDoubleTriggerThresholds,

        DrumDoubleTriggerThresholdKaLeft,
        DrumDoubleTriggerThresholdDonLeft,
        DrumDoubleTriggerThresholdDonRight,
        DrumDoubleTriggerThresholdKaRight,

        LedBrightness,
        LedEnablePlayerColor,

        BootselMsg,
    };

    struct State {
        Page page;
        uint16_t selected_value;
        uint16_t original_value;
    };

    struct Descriptor {
        enum class Type : uint8_t {
            Menu,
            Selection,
            Value,
            Toggle,
            RebootInfo,
        };

        enum class Action : uint8_t {
            None,
            GotoParent,

            GotoPageDeviceMode,
            GotoPageDrum,
            GotoPageLed,
            GotoPageReset,
            GotoPageBootsel,

            GotoPageDrumDebounceDelay,
            GotoPageDrumDoubleTrigger,
            GotoPageDrumTriggerThresholds,
            GotoPageDrumDoubleTriggerThresholds,

            GotoPageDrumTriggerThresholdKaLeft,
            GotoPageDrumTriggerThresholdDonLeft,
            GotoPageDrumTriggerThresholdDonRight,
            GotoPageDrumTriggerThresholdKaRight,

            GotoPageDrumDoubleTriggerThresholdKaLeft,
            GotoPageDrumDoubleTriggerThresholdDonLeft,
            GotoPageDrumDoubleTriggerThresholdDonRight,
            GotoPageDrumDoubleTriggerThresholdKaRight,

            GotoPageLedBrightness,
            GotoPageLedEnablePlayerColor,

            SetUsbMode,

            SetDrumDebounceDelay,

            SetDoubleTriggerOff,
            SetDoubleTriggerAlways,

            SetDrumTriggerThresholdKaLeft,
            SetDrumTriggerThresholdDonLeft,
            SetDrumTriggerThresholdDonRight,
            SetDrumTriggerThresholdKaRight,

            SetDrumDoubleTriggerThresholdKaLeft,
            SetDrumDoubleTriggerThresholdDonLeft,
            SetDrumDoubleTriggerThresholdDonRight,
            SetDrumDoubleTriggerThresholdKaRight,

            SetLedBrightness,
            SetLedEnablePlayerColor,

            DoReset,
            DoRebootToBootsel,
        };

        Type type;
        std::string name;
        std::vector<std::pair<std::string, Action>> items;
        uint16_t max_value;
    };

    const static std::map<Page, const Descriptor> descriptors;

  private:
    class Buttons {
      public:
        enum class Id : uint8_t { Up, Down, Left, Right, Confirm, Back };

      private:
        struct State {
            enum class Repeat : uint8_t {
                Idle,
                RepeatDelay,
                Repeat,
                FastRepeat,
            };
            Repeat repeat;
            uint32_t pressed_since;
            uint32_t last_repeat;
            bool pressed;
        };

        std::map<Id, State> m_states;

      public:
        Buttons();

        void update(const InputState::Controller &state);
        [[nodiscard]] bool getPressed(Id id) const;
    };

    std::shared_ptr<SettingsStore> m_store;
    Buttons m_buttons;
    bool m_active{false};

    std::stack<State> m_state_stack{{{.page = Page::Main, .selected_value = 0, .original_value = 0}}};

    uint16_t getCurrentValue(Page page);
    void gotoPage(Page page);
    void gotoParent(bool do_restore);

    void performAction(Descriptor::Action action, uint16_t value);

  public:
    Menu(std::shared_ptr<SettingsStore> settings_store);

    void activate();
    void update(const InputState::Controller &controller_state);
    [[nodiscard]] bool active() const;
    [[nodiscard]] State getState() const;
};
} // namespace Doncon::Utils

#endif // UTILS_MENU_H_