// ==== LED and Button pin definitions ====
const int redLED = 2;      // Red LED pin
const int yellowLED = 4;   // Yellow LED pin
const int greenLED = 6;    // Green LED pin
const int blueLED = 8;     // Blue LED pin
const int whiteLED = 10;   // White LED pin (new)

const int redButton = 3;      // Red button pin
const int yellowButton = 5;   // Yellow button pin
const int greenButton = 7;    // Green button pin
const int blueButton = 9;     // Blue button pin
const int whiteButton = 11;   // White button pin (new)

const int buzzer = 12;     // Buzzer pin

// ==== Game data ====
int sequence[100];      // Store the game sequence (LED colors)
int playerInput[100];   // Store what the player presses
int level = 0;          // Current length of the sequence
bool gameOver = false;  // Flag to check if the game is over

void setup() {
  // Set LED pins as output
  pinMode(redLED, OUTPUT);
  pinMode(yellowLED, OUTPUT);
  pinMode(greenLED, OUTPUT);
  pinMode(blueLED, OUTPUT);
  pinMode(whiteLED, OUTPUT);

  // Set button pins as input with pull-up resistors
  pinMode(redButton, INPUT_PULLUP);
  pinMode(yellowButton, INPUT_PULLUP);
  pinMode(greenButton, INPUT_PULLUP);
  pinMode(blueButton, INPUT_PULLUP);
  pinMode(whiteButton, INPUT_PULLUP);

  pinMode(buzzer, OUTPUT);

  Serial.begin(9600);  // For debugging if needed

  // Start the game
  startGame();
}

void loop() {
  if (!gameOver) {
    // Show the sequence to the player
    playSequence();

    // Wait for player's input
    if (!getPlayerInput()) {
      // If wrong, end the game
      endGame();
    } else {
      delay(500); // Small pause before next round
    }
  } else {
    // Restart the game after short delay
    delay(1500);
    startGame();
  }
}

// Reset and start a new game
void startGame() {
  level = 0;       // Reset sequence length
  gameOver = false;

  // Flash all LEDs and play a tone to show game start
  allLEDsOn();
  tone(buzzer, 1000, 500);
  delay(500);
  allLEDsOff();

  // Make random numbers different each time
  randomSeed(analogRead(A0));
  delay(500);
}

// Add 2 new random colors to the sequence and replay all
void playSequence() {
  sequence[level] = random(1, 6);     // First new color (1–5)
  sequence[level + 1] = random(1, 6); // Second new color
  level += 2; // Increase sequence length by 2

  // Show the sequence step by step
  for (int i = 0; i < level; i++) {
    lightUpLED(sequence[i]);
    delay(300);
  }
}

// Read the player's input and check if correct
bool getPlayerInput() {
  for (int i = 0; i < level; i++) {
    bool inputReceived = false;

    // Wait until a button is pressed
    while (!inputReceived) {
      if (digitalRead(redButton) == LOW) {
        playerInput[i] = 1;
        lightUpLED(1);
        inputReceived = true;
      } else if (digitalRead(yellowButton) == LOW) {
        playerInput[i] = 2;
        lightUpLED(2);
        inputReceived = true;
      } else if (digitalRead(greenButton) == LOW) {
        playerInput[i] = 3;
        lightUpLED(3);
        inputReceived = true;
      } else if (digitalRead(blueButton) == LOW) {
        playerInput[i] = 4;
        lightUpLED(4);
        inputReceived = true;
      } else if (digitalRead(whiteButton) == LOW) {
        playerInput[i] = 5;
        lightUpLED(5);
        inputReceived = true;
      }
    }

    // Check if player's input matches the sequence
    if (playerInput[i] != sequence[i]) {
      return false;  // Wrong input
    }
    delay(200); // Small pause before next input
  }
  return true; // All correct
}

// Show "game over" animation
void endGame() {
  gameOver = true;

  for (int i = 0; i < 5; i++) {
    allLEDsOn();
    tone(buzzer, 1000, 200);
    delay(200);
    allLEDsOff();
    delay(200);
  }
}

// Turn on one LED and play its tone
void lightUpLED(int color) {
  switch (color) {
    case 1: digitalWrite(redLED, HIGH);    tone(buzzer, 500, 300); break;
    case 2: digitalWrite(yellowLED, HIGH); tone(buzzer, 600, 300); break;
    case 3: digitalWrite(greenLED, HIGH);  tone(buzzer, 700, 300); break;
    case 4: digitalWrite(blueLED, HIGH);   tone(buzzer, 800, 300); break;
    case 5: digitalWrite(whiteLED, HIGH);  tone(buzzer, 900, 300); break;
  }
  delay(300);

  noTone(buzzer); // Stop the sound
  allLEDsOff();   // Turn off all LEDs
}

// Helper: turn on all LEDs
void allLEDsOn() {
  digitalWrite(redLED, HIGH);
  digitalWrite(yellowLED, HIGH);
  digitalWrite(greenLED, HIGH);
  digitalWrite(blueLED, HIGH);
  digitalWrite(whiteLED, HIGH);
}

// Helper: turn off all LEDs
void allLEDsOff() {
  digitalWrite(redLED, LOW);
  digitalWrite(yellowLED, LOW);
  digitalWrite(greenLED, LOW);
  digitalWrite(blueLED, LOW);
  digitalWrite(whiteLED, LOW);
}