#include "pomodoEnum.hpp"
#include "LedControl.h"
#include "binary.h"
#include "ReadEncoderUtils.cpp"
#include "DisplayManager.hpp"
#include "PomodoCompanion.cpp"

#define ENCODER_CLK 21
#define ENCODER_DT  20
#define ENCODER_SW  19

#define BUTTON_PIN 5

const int SPEAKER_PIN = 21;

DisplayManager dm(12, 11, 10, 9, 8, 7);
PomodoCompanion pc(25, 5, 4, 0);
ReadEncoderUtils re(ENCODER_CLK, ENCODER_DT, ENCODER_SW);

int counter = 0;

int lastClk = HIGH;


volatile int encoderState;
volatile int previousEncoderState;
volatile boolean displayRefreshNeeded = true;

void encoderISR() {
  if (digitalRead(ENCODER_CLK) == LOW) {
    encoderState = digitalRead(ENCODER_DT);
    
    // if (encoderState != previousEncoderState) {
      pc.updateValue(re.readEncoder());
      displayRefreshNeeded = true;
    // }
    
    previousEncoderState = encoderState;
  }
}

long int modeLastChanged = 0;
long int lastRotaryChanged = 0;

void handlePushButton() {
  if (re.readPushButton() && millis() - modeLastChanged > 300) {
    modeLastChanged = millis();
    Pomodo state = pc.setNextState();
    if (state != FOCUS_TIME || state != BREAK_TIME)
      displayRefreshNeeded = true;
  }
}
void setup() {
  Serial.begin(115200);
  pinMode(ENCODER_CLK, INPUT);
  pinMode(ENCODER_DT, INPUT);
  pinMode(ENCODER_SW, INPUT_PULLUP);
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  attachInterrupt(digitalPinToInterrupt(ENCODER_CLK), encoderISR, CHANGE);
  attachInterrupt(digitalPinToInterrupt(ENCODER_SW), [] () {handlePushButton();}, FALLING);
  dm.DisplayText("Hi! Let's start a new session \x03");
  delay(1500);
  dm.clear();
  // workTime = GetFocusTime();
}

// [X] - Create function getting/setting FocusTime/BreakTime/NbRounds
// [X] - Start Timer
// [ ] - Buzz sound
// [X] - Button to go next
// [ ] - Button to restart

LiquidCrystal* lc = dm.getLiquidCrystal();

const unsigned long interval = 1000;
unsigned long previousMillis = 0;
int remainingMinutes;
int remainingSeconds;
unsigned long currentMillis;

bool isFirstPart = true;

void sessionRecapDisplay() {
  lc->clear();
  if (isFirstPart) {
    lc->print("Focus ");
    lc->print(pc.getFocusTime());
    lc->print(" min");
    lc->setCursor(0, 2);
    lc->print("Break ");
    lc->print(pc.getBreakTime());
    lc->print(" min");
    isFirstPart = !isFirstPart;
    delay(1000);
  } else {
    lc->print("Rounds ");
    lc->print(pc.getNbRounds());
    lc->print("x");
    lc->setCursor(0, 2);
    lc->print("Total ");
    lc->print((pc.getFocusTime() + pc.getBreakTime()) * pc.getNbRounds());
    lc->print(" min");
    isFirstPart = !isFirstPart;
    delay(1000);
  }
}

void StateDisplayManager() {
  Pomodo state = pc.getPomodoState();
  switch(state) {
    case SET_FOCUS_TIME: 
      lc->clear();
      lc->print("Let's setup ^.^");
      lc->setCursor(0,2);
      lc->print("Focus Time: ");
      lc->print(pc.getFocusTime());
      displayRefreshNeeded = false;
      break;
    case SET_BREAK_TIME:
      lc->clear();
      lc->print("Time to drink !");
      lc->setCursor(0,2);
      lc->print("BreakTime: ");
      lc->print(pc.getBreakTime());
      displayRefreshNeeded = false;
      break;
    case NB_POMODO_ROUND:
      lc->clear();
      lc->print("Nearly there D:");
      lc->setCursor(0,2);
      lc->print("Nb Rounds: ");
      lc->print(pc.getNbRounds());
      displayRefreshNeeded = false;
      break;
    case SESSION_RECAP: 
      sessionRecapDisplay();
      break;
    case SUMMARY:
      lc->clear();
      lc->print("FINISHED OMG GG");
      displayRefreshNeeded = false;
      break;
  }
}
bool nextTimer = true;
bool needUpdateRound = false;

int lastState = HIGH;
void handleNextBtn() {
  int value = digitalRead(BUTTON_PIN);
  if (lastState != value) {
    if (value == LOW) {
      Pomodo state = pc.setNextState();
      if (state == FOCUS_TIME || state == BREAK_TIME) {
        nextTimer = true;
      } else {
        displayRefreshNeeded = true;
      }
    }
  }
}

void TimeHelper(
    int baseMinutes, 
    int baseSeconds, 
    char* label, 
    bool countAsRound = false
) {
  if (nextTimer) {
    tone(6, 262, 250);
    previousMillis = millis();
    remainingMinutes = baseMinutes;
    remainingSeconds = baseSeconds;

    lc->clear();
    lc->print("   ");
		lc->print(label);
		lc->print(" Time  ");

    nextTimer = false;
    needUpdateRound = true;
	} else if (!nextTimer && (currentMillis - previousMillis >= interval)) {
		previousMillis = currentMillis;

    if (remainingMinutes > 0 || remainingSeconds > 0) {
      if (remainingSeconds == 0) {
        remainingMinutes--;
        remainingSeconds = 59;
      } else {
        remainingSeconds--;
      }

			lc->setCursor(0,2);
			lc->print("                ");
      lc->setCursor(5, 2);
      lc->print(remainingMinutes);
      lc->print(" : ");
      lc->print(remainingSeconds);
    } else {
      if (countAsRound && needUpdateRound ) {
        pc.setCurrentRound(pc.getCurrentRound() + 1);
        
        needUpdateRound = false;
      }

        lc->clear();
        lc->print("End ");
				lc->print(label);
				lc->print(" ");
        lc->print(pc.getCurrentRound());
        lc->print("! :D");
        lc->setCursor(0, 2);

        if (pc.isSessionComplete()) {
          lc->print("Btn to finish!");
          tone(6, 262, 400);
        } else {
          lc->print("Btn for ");
					lc->print(countAsRound ? "Break" : "Focus");
					lc->print("!");
          tone(6, 262, 400);
        }

      handleNextBtn();
		}
	}
}

void timerManagerDisplay() {
  Pomodo state = pc.getPomodoState();
  if (state == FOCUS_TIME) {
    TimeHelper(pc.getFocusTime(), 0, "Focus", true);
  } else if (state == BREAK_TIME) {
    TimeHelper(pc.getBreakTime(), 0, "Break");
  }
}

void loop() {
  currentMillis = millis();

  if (displayRefreshNeeded) {
    StateDisplayManager();
    // delay(300);
  } 

  timerManagerDisplay();
}


/*
// void focusTimer() {
//   if (nextTimer) {
//     // remainingMinutes = pc.getWorkTime();
//     previousMillis = millis();
//     remainingMinutes = 0;
//     remainingSeconds = 3;

//     lc->clear();
//     lc->print("   Focus Time  ");
//     lc->setCursor(5, 2);

//     nextTimer = false;
//     needUpdateRound = true;
//   } else if (!nextTimer && (currentMillis - previousMillis >= interval)) {
//     previousMillis = currentMillis;

//     if (remainingMinutes > 0 || remainingSeconds > 0) {
//       if (remainingSeconds == 0) {
//         remainingMinutes--;
//         remainingSeconds = 59;
//       } else {
//         remainingSeconds--;
//       }
//       lc->setCursor(5, 2);
//       lc->print(remainingMinutes);
//       lc->print(" : ");
//       lc->print(remainingSeconds);
//     } else {
//       if (needUpdateRound) {
//         pc.setCurrentRound(pc.getCurrentRound() + 1);
//         needUpdateRound = false;
//       }

//         lc->clear();
//         lc->print("End focus ");
//         lc->print(pc.getCurrentRound());
//         lc->print("! :D");
//         lc->setCursor(0, 2);

//         if (pc.isSessionComplete()) {
//           lc->print("Btn to finish!");
//         } else {
//           lc->print("Btn for Break!");
//         }

//       handleNextBtn();
//     }
//   }
// }
*/