#include "Defines.h"
#include "DisplayHelper.h"
#include "Settings.h"
#include "LEDRunner.h"
#include "InputHelper.h"
#include "Enums.h"
#include "WrapHelper.h"

class EyeTrainerMain
{
public:
  EyeTrainerMain() { }

  void setup() 
  { 
      pinMode(LED_BUILTIN, OUTPUT);
      m_LEDRunner.setup();
      m_DisplayHelper.setup();
      m_InputHelper.setup();
      UpdateModeDisplay();
  }

  void loop()
  {
    bool inputEvent = m_InputHelper.readInputs(millis());
    
    if(inputEvent)
    {
      ProcessChangeMode();
      int upDownButtonsShortClickState = m_InputHelper.GetUpDownButtonsShortClicked();
      int upDownButtonsHoldState = m_InputHelper.GetUpDownButtonsHold();
      switch(m_deviceMode)
      {
        case DeviceMode::RUN:                  ProcessRun(upDownButtonsShortClickState, upDownButtonsHoldState);             break;
        case DeviceMode::SETTINGS_MODE:        ProcessSettingsRunMode(upDownButtonsShortClickState, upDownButtonsHoldState); break;
        case DeviceMode::SETTINGS_TIME_SWITCH: ProcessSwitchTime(upDownButtonsShortClickState, upDownButtonsHoldState);      break;
        case DeviceMode::SETTINGS_BRIGHTNESS:  ProcessBrightness(upDownButtonsShortClickState, upDownButtonsHoldState);      break;
        case DeviceMode::SETTINGS_TIME_MODE:   ProcessTimeMode(upDownButtonsShortClickState, upDownButtonsHoldState);        break;
        case DeviceMode::TEST:                 ProcessTest(upDownButtonsShortClickState, upDownButtonsHoldState);            break;
      }
    }
    unsigned long currentTime = millis();
    unsigned long deltaTime = currentTime - m_PrevTime;
    m_PrevTime = currentTime;
    m_LEDRunner.update(deltaTime);
  }

  void runTest()
  {
    m_LEDRunner.runTest();
    m_DisplayHelper.runTest();
  }

private:
  enum DeviceMode m_deviceMode  = DeviceMode::RUN;

  LEDRunner     m_LEDRunner     = LEDRunner();
  DisplayHelper m_DisplayHelper = DisplayHelper();
  InputHelper   m_InputHelper   = InputHelper();
  unsigned long m_PrevTime      = 0;

  void UpdateModeDisplay()
  {
    m_DisplayHelper.UpdateModeDisplay(m_LEDRunner.getRun(), m_deviceMode);
  }

  void ProcessChangeMode()
  {
    bool mainButtonShortClicked = m_InputHelper.getButtonMain().IsNowShortClicked();
    if(!mainButtonShortClicked)
    return;

    m_DisplayHelper.setDrawTitle(true);
    m_deviceMode = static_cast<DeviceMode>(wrapInt(static_cast<int>(m_deviceMode) + 1, 6)); 
    UpdateModeDisplay();
    if(m_deviceMode != DeviceMode::RUN)
    {
      m_LEDRunner.setRun(false);
    }
  }

  void ProcessRun(int upDownButtons, int upDownButtonsHold)
  {
    if(Settings::getInstance().TimeModeState == TimeMode::MANUAL && upDownButtons != BTN_DOWN_UP_NO_SHORT_CLICKED)
    {
      m_LEDRunner.manualUpdate(Settings::getInstance().RunModeState, upDownButtons);
    }
    else
    {
      if(upDownButtons == BTN_UP_SHORT_CLICKED)
      {
        m_DisplayHelper.setDrawTitle(false);
        m_LEDRunner.setRun(true);
        UpdateModeDisplay();
      }
      if(upDownButtons == BTN_DOWN_SHORT_CLICKED)
      {
        m_DisplayHelper.setDrawTitle(false);
        m_LEDRunner.setRun(false);
        UpdateModeDisplay();
      }
    }
  }

  void ProcessSettingsRunMode(int upDownButtons, int upDownButtonsHold)
  {
    if(upDownButtons != BTN_DOWN_UP_NO_SHORT_CLICKED)
    {
      m_DisplayHelper.setDrawTitle(false);
      Settings::getInstance().switchRunMode(upDownButtons);
      UpdateModeDisplay();
    }
  }
    
  void ProcessSwitchTime(int upDownButtons, int upDownButtonsHold)
  {
    if(upDownButtons != BTN_DOWN_UP_NO_SHORT_CLICKED)
    {
      m_DisplayHelper.setDrawTitle(false);
      Settings::getInstance().setSwitchTime(Settings::getInstance().SwitchTime + upDownButtons * 50);
      UpdateModeDisplay();
    }
    if(upDownButtonsHold != 0)
    {
      m_DisplayHelper.setDrawTitle(false);
      Settings::getInstance().setSwitchTime(Settings::getInstance().SwitchTime + upDownButtonsHold * 50);
      UpdateModeDisplay();
    }
  }
    
  void ProcessBrightness(int upDownButtons, int upDownButtonsHold)
  {
    if(upDownButtons != BTN_DOWN_UP_NO_SHORT_CLICKED)
    {
      m_LEDRunner.setAllHigh();
      m_DisplayHelper.setDrawTitle(false);
      Settings::getInstance().setBrightness(Settings::getInstance().Brightness + upDownButtons);
      m_LEDRunner.updateBrightness(Settings::getInstance().Brightness);
      UpdateModeDisplay();
    }

    if(upDownButtonsHold != 0)
    {
      m_LEDRunner.setAllHigh();
      m_DisplayHelper.setDrawTitle(false);
      Settings::getInstance().setBrightness(Settings::getInstance().Brightness + upDownButtonsHold);
      m_LEDRunner.updateBrightness(Settings::getInstance().Brightness);
      UpdateModeDisplay();
    }
  }

  void ProcessTimeMode(int upDownButtons, int upDownButtonsHold)
  {
    if(upDownButtons != BTN_DOWN_UP_NO_SHORT_CLICKED)
    {
      m_DisplayHelper.setDrawTitle(false);
      Settings::getInstance().switchTimeMode(upDownButtons);
      UpdateModeDisplay();
    }
  }

  void ProcessTest(int upDownButtons, int upDownButtonsHold)
  {
    if(upDownButtons != BTN_DOWN_UP_NO_SHORT_CLICKED)
    {
      runTest();
    }
  }
};