#include <Wire.h>
#include <LiquidCrystal_I2C.h>

#define relayPin 10
#define buzzerPin 11
#define setButtonPin 2
#define minuteButtonPin 3
#define secondButtonPin 4
#define startButtonPin 5

LiquidCrystal_I2C lcd(0x27, 16, 2);

int timerMinutes = 0;
int timerSeconds = 0;
bool setupMode = false;
bool timerRunning = false;

void setup() {
  pinMode(relayPin, OUTPUT);
  pinMode(buzzerPin, OUTPUT);
  pinMode(setButtonPin, INPUT_PULLUP);
  pinMode(minuteButtonPin, INPUT_PULLUP);
  pinMode(secondButtonPin, INPUT_PULLUP);
  pinMode(startButtonPin, INPUT_PULLUP);

  lcd.begin(16, 2);
  lcd.clear();
  // Splash Screen
  lcd.setCursor(2, 0);
  lcd.print("IDY CREATIONS");
  lcd.setCursor(1, 1);
  lcd.print("Curing Station");
  delay(2000);
  lcd.clear();

  lcd.setCursor(2, 0);
  lcd.print("Press Setup");
}

void loop() {
  checkButtons();

  if (setupMode) {
    displaySetupMode();
  } else if (timerRunning) {
    displayTimer();
  }
}

void checkButtons() {
  if (digitalRead(setButtonPin) == LOW) {
    // Set button is pressed
    while (digitalRead(setButtonPin) == LOW) {
      // Wait for the button to be released
      delay(10);
    }
    // Delay to avoid recognizing the next press immediately
    delay(200);

    if (setupMode) {
      // If in setup mode, reset timer to 0:0
      timerMinutes = 0;
      timerSeconds = 0;
    } else {
      // If not in setup mode, enter setup mode
      setupMode = true;
    }
  }
}


enum SetupState {
  DISPLAY_SETUP,
  WAIT_FOR_BUTTON_RELEASE,
};

SetupState setupState = DISPLAY_SETUP;

void displaySetupMode() {
  lcd.clear();
  lcd.setCursor(4, 0);
  lcd.print("Set Timer");
  lcd.setCursor(6, 1);
  lcd.print(String(timerMinutes) + ":" + String(timerSeconds));

  unsigned long lastButtonCheckTime = 0;
  const unsigned long buttonCheckInterval = 100;

  while (setupMode) {
    if (millis() - lastButtonCheckTime >= buttonCheckInterval) {
      // Update timerMinutes and timerSeconds when minute or second buttons are pressed
      if (digitalRead(minuteButtonPin) == LOW) {
        timerMinutes++;
        lcd.clear();
        lcd.setCursor(4, 0);
        lcd.print("Set Timer");
        lcd.setCursor(6, 1);
        lcd.print(String(timerMinutes) + ":" + String(timerSeconds));
        delay(200); // Debounce
        setupState = WAIT_FOR_BUTTON_RELEASE;
      }

      if (digitalRead(secondButtonPin) == LOW) {
        timerSeconds += 10;
        
        // Ensure seconds do not exceed 50
        if (timerSeconds > 50) {
          timerSeconds = 0;
        }
        lcd.clear();
        lcd.setCursor(4, 0);
        lcd.print("Set Timer");
        lcd.setCursor(6, 1);
        lcd.print(String(timerMinutes) + ":" + String(timerSeconds));
        delay(200); // Debounce
        setupState = WAIT_FOR_BUTTON_RELEASE;
      }

      if (digitalRead(startButtonPin) == LOW) {
        // Start button pressed, exit setup mode and start timer
        setupMode = false;
        timerRunning = true;
        digitalWrite(relayPin, HIGH); // Turn on the relay
        delay(200); // Debounce

        digitalWrite(buzzerPin, HIGH);
        delay(100); // Buzzer sound for .1 seconds
        digitalWrite(buzzerPin, LOW);
      }

      if (digitalRead(setButtonPin) == LOW) {
        // Setup button pressed, reset timer to 0:0
        timerMinutes = 0;
        timerSeconds = 0;
        lcd.clear();
        lcd.setCursor(4, 0);
        lcd.print("Set Timer");
        lcd.setCursor(6, 1);
        lcd.print(String(timerMinutes) + ":" + String(timerSeconds));
        setupMode = true;
        delay(200); // Debounce
        setupState = WAIT_FOR_BUTTON_RELEASE;
      }

      lastButtonCheckTime = millis();
    }
  }
}


void displayTimer() {
  lcd.clear();
  lcd.setCursor(1, 0);
  lcd.print("Time Remaining");
  lcd.setCursor(6, 1);
  lcd.print(timerMinutes);
  lcd.print(":");
  if (timerSeconds % 60 < 10) {
    lcd.print("0");
  }
  lcd.print(timerSeconds % 60);
  
  // Update the timer every second
  delay(1000);

  if (--timerSeconds < 0) {
    timerSeconds = 59;
    if (--timerMinutes < 0) {
      // Timer done, turn off relay, activate buzzer, and reset
      digitalWrite(relayPin, LOW);
      digitalWrite(buzzerPin, HIGH);
      delay(2000); // Buzzer sound for 2 seconds
      digitalWrite(buzzerPin, LOW);
      timerRunning = false;
      timerMinutes = 0;
      timerSeconds = 0;
      setupMode = true; // Set up mode after the timer is done
    }
  }
}