#include <Wire.h> // Include the Wire library for I2C
#include <LiquidCrystal_I2C.h> // Include the LiquidCrystal I2C library

// Define LED pins
const int ledPins[] = {2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13}; // Include 12 LED pins
const int numLEDs = sizeof(ledPins) / sizeof(ledPins[0]);

// Define buzzer pin
const int buzzerPin = A1; // Change to pin 14 for the buzzer

// Define I2C LCD address and dimensions
const int lcdColumns = 16;
const int lcdRows = 2;
const int lcdAddress = 0x27; // Adjust the address according to your module
LiquidCrystal_I2C lcd(lcdAddress, lcdColumns, lcdRows);

// Define pushbutton pin
const int buttonPin = A0;

// Define notes for "We Are the Champions"
#define NOTE_B0 31
#define NOTE_C1 33
#define NOTE_CS1 35
#define NOTE_D1 37
#define NOTE_DS1 39
#define NOTE_E1 41
#define NOTE_F1 44
#define NOTE_FS1 46
#define NOTE_G1 49
#define NOTE_GS1 52
#define NOTE_A1 55
#define NOTE_AS1 58
#define NOTE_B1 62
#define NOTE_C2 65
#define NOTE_CS2 69
#define NOTE_D2 73
#define NOTE_DS2 78
#define NOTE_E2 82
#define NOTE_F2 87
#define NOTE_FS2 93
#define NOTE_G2 98
#define NOTE_GS2 104
#define NOTE_A2 110
#define NOTE_AS2 117
#define NOTE_B2 123
#define NOTE_C3 131
#define NOTE_CS3 139
#define NOTE_D3 147
#define NOTE_DS3 156
#define NOTE_E3 165
#define NOTE_F3 175
#define NOTE_FS3 185
#define NOTE_G3 196
#define NOTE_GS3 208
#define NOTE_A3 220
#define NOTE_AS3 233
#define NOTE_B3 247
#define NOTE_C4 262
#define NOTE_CS4 277
#define NOTE_D4 294
#define NOTE_DS4 311
#define NOTE_E4 330
#define NOTE_F4 349
#define NOTE_FS4 370
#define NOTE_G4 392
#define NOTE_GS4 415
#define NOTE_A4 440
#define NOTE_AS4 466
#define NOTE_B4 494
#define NOTE_C5 523
#define NOTE_CS5 554
#define NOTE_D5 587
#define NOTE_DS5 622
#define NOTE_E5 659
#define NOTE_F5 698
#define NOTE_FS5 740
#define NOTE_G5 784
#define NOTE_GS5 831
#define NOTE_A5 880
#define NOTE_AS5 932
#define NOTE_B5 988
#define NOTE_C6 1047
#define NOTE_CS6 1109
#define NOTE_D6 1175
#define NOTE_DS6 1245
#define NOTE_E6 1319
#define NOTE_F6 1397
#define NOTE_FS6 1480
#define NOTE_G6 1568
#define NOTE_GS6 1661
#define NOTE_A6 1760
#define NOTE_AS6 1865
#define NOTE_B6 1976
#define NOTE_C7 2093
#define NOTE_CS7 2217
#define NOTE_D7 2349
#define NOTE_DS7 2489
#define NOTE_E7 2637
#define NOTE_F7 2794
#define NOTE_FS7 2960
#define NOTE_G7 3136
#define NOTE_GS7 3322
#define NOTE_A7 3520
#define NOTE_AS7 3729
#define NOTE_B7 3951
#define NOTE_C8 4186
#define NOTE_CS8 4435
#define NOTE_D8 4699
#define NOTE_DS8 4978

// Notes and durations for "We Are the Champions"
int melody[] = {
  NOTE_G4, NOTE_G4, NOTE_A4, NOTE_G4, NOTE_C5, NOTE_B4,
  NOTE_G4, NOTE_G4, NOTE_A4, NOTE_G4, NOTE_D5, NOTE_C5,
  NOTE_G4, NOTE_G4, NOTE_G5, NOTE_E5, NOTE_C5, NOTE_B4, NOTE_A4,
  NOTE_F5, NOTE_F5, NOTE_E5, NOTE_C5, NOTE_D5, NOTE_C5
};

int noteDurations[] = {
  4, 8, 4, 4, 4, 2,
  4, 8, 4, 4, 4, 2,
  4, 8, 4, 4, 4, 8, 4,
  4, 8, 4, 4, 4, 2
};

// Closing melody notes and durations
int closingMelody[] = {
  NOTE_C5, NOTE_G4, NOTE_E4, NOTE_A4, NOTE_B4,
  NOTE_A4, NOTE_GS4, NOTE_A4, NOTE_G4
};

int closingNoteDurations[] = {
  4, 8, 4, 4, 4, 4, 4, 4, 2
};

void setup() {
  // Initialize LEDs
  for (int i = 0; i < numLEDs; i++) {
    pinMode(ledPins[i], OUTPUT);
  }

  // Initialize buzzer
  pinMode(buzzerPin, OUTPUT);

  // Initialize I2C LCD
  lcd.init();
  lcd.backlight();

  // Initialize pushbutton
  pinMode(buttonPin, INPUT);
  
  // Seed random number generator
  randomSeed(analogRead(0));
}

void loop() {
  // If the button is pressed, start the game
  if (digitalRead(buttonPin) == HIGH) {
    int randomNumber = random(numLEDs); // Generate a random LED index
    displayMessage("___Berputarrr___");
    spinLEDsWithMelody(); // Spin LEDs with melody
    delay(2000); // Spin for 2 seconds
    digitalWrite(ledPins[randomNumber], HIGH); // Turn on the LED where it stopped
    displayMessage("Masalah Nomor " + String(randomNumber + 1)); // Display result on LCD
    playClosingMelody(); // Play the closing melody immediately
    delay(3000); // Wait for 3 seconds before resetting
    clearLEDs(); // Turn off all LEDs
    displayMessage(""); // Clear LCD
  }
}

// Function to spin LEDs with melody
void spinLEDsWithMelody() {
  int notesCount = sizeof(melody) / sizeof(melody[0]);
  int noteIndex = 0;

  unsigned long startMillis = millis();
  unsigned long elapsedMillis = 0;
  int ledIndex = 0;

  while (elapsedMillis < 2000) { // Spin and play melody for 2 seconds
    unsigned long currentMillis = millis();
    elapsedMillis = currentMillis - startMillis;

    if (elapsedMillis > noteDurations[noteIndex] * 100) {
      noteIndex++;
      if (noteIndex >= notesCount) {
        noteIndex = 0;
      }
      tone(buzzerPin, melody[noteIndex], noteDurations[noteIndex] * 100);
    }

    digitalWrite(ledPins[ledIndex], HIGH);
    delay(100);
    digitalWrite(ledPins[ledIndex], LOW);
    ledIndex++;
    if (ledIndex >= numLEDs) {
      ledIndex = 0;
    }
  }
  noTone(buzzerPin); // Stop the tone
}

// Function to play the closing melody
void playClosingMelody() {
  int notesCount = sizeof(closingMelody) / sizeof(closingMelody[0]);
  for (int thisNote = 0; thisNote < notesCount; thisNote++) {
    int noteDuration = 1000 / closingNoteDurations[thisNote];
    tone(buzzerPin, closingMelody[thisNote], noteDuration);
    int pauseBetweenNotes = noteDuration * 1.30;
    delay(pauseBetweenNotes);
    noTone(buzzerPin);
  }
}

// Function to turn off all LEDs
void clearLEDs() {
  for (int i = 0; i < numLEDs; i++) {
    digitalWrite(ledPins[i], LOW);
  }
}

// Function to display message on LCD
void displayMessage(String message) {
  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print(message);
}