#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <Keypad.h>
#include <ESP32Servo.h>
#include <RTClib.h> // library for RTC (Real-Time Clock)

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET    -1

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

const byte ROWS = 4;
const byte COLS = 4;

char keys[ROWS][COLS] = {
  {'1','2','3','A'},
  {'4','5','6','B'},
  {'7','8','9','C'},
  {'*','0','#','D'}
};

byte rowPins[ROWS] = {14, 12, 19, 18};
byte colPins[COLS] = {5, 4, 2, 15};

Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);

const char *password = "2525";
char enteredPassword[5];
int passwordIndex = 0;

Servo myservo;
const int servoPin = 26;

RTC_DS3231 rtc; // Create an instance of the RTC class

bool accessGranted = false;
int wrongAttempts = 0;
const int buzzerPin = 16; // Pin buzzer

void setup() {
  Serial.begin(115200);
  pinMode(buzzerPin, OUTPUT); // Set pin buzzer sebagai output

  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  
  display.display();
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(20, 0);
  display.println("Smart Security");
  display.display();

  display.setCursor(25, 12);
  display.println("Laboratorium");
  display.display();

  display.setCursor(10, 28);
  display.println("Silahkan Masukkan");
  display.display();

  display.setCursor(5, 38);
  display.println("Barang dan Password:");
  display.display();

  myservo.attach(servoPin);

  if (!rtc.begin()) {
    Serial.println("Couldn't find RTC");
    while (1);
  }
  if (rtc.lostPower()) {
    Serial.println("RTC lost power, let's set the time!");
    rtc.adjust(DateTime(F(__DATE__), F(__TIME__)));
  }
}

void loop() {
  DateTime now = rtc.now(); // Get current time
  char key = keypad.getKey();
  
  if (!accessGranted && key && key != '#' && key != '*') {
    if (passwordIndex < 4) {
      enteredPassword[passwordIndex] = key;
      passwordIndex++;
      displayPasswordInput();
    }
  } else if (key == '#' && !accessGranted) {
    if (strcmp(enteredPassword, password) == 0) {
      displayAccessAccepted();
      unlockLocker();
      succesBuzzer();
      delay(100);
      deactivateBuzzer();
      accessGranted = true;
    } else {
      displayInvalidPassword();
      succesBuzzer();
      delay(250);
      deactivateBuzzer();

      wrongAttempts++;
      if (wrongAttempts >= 3) { // Jika 3 kali salah, aktifkan buzzer
        activateBuzzer();
        delay(3000); // Diamkan buzzer selama 5 detik
        deactivateBuzzer();
        wrongAttempts = 0; // Setel kembali jumlah percobaan yang salah ke 0
      }

      delay(2000);
      resetPasswordInputDisplay();
    }
  } else if (key == '*' && !accessGranted) { // Handle "*" key to clear entered password
    passwordIndex = 0;
    memset(enteredPassword, 0, sizeof(enteredPassword));
    resetPasswordInputDisplay();
  }
  
  if (accessGranted) {
    displayHomeEntry(now);
  }
}

void displayPasswordInput() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(20, 0);
  display.println("Smart Security");
  display.display();

  display.setCursor(25, 12);
  display.println("Laboratorium");
  display.display();

  display.setCursor(10, 28);
  display.println("Silahkan Masukkan");
  display.display();

  display.setCursor(5, 38);
  display.println("Barang dan Password:");
  display.display();
  for (int i = 0; i < passwordIndex; i++) {
    display.print("*");
  }
  display.display();
}

void resetPasswordInputDisplay() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(20, 0);
  display.println("Smart Security");
  display.display();

  display.setCursor(25, 12);
  display.println("Laboratorium");
  display.display();

  display.setCursor(10, 28);
  display.println("Silahkan Masukkan");
  display.display();

  display.setCursor(5, 38);
  display.println("Barang dan Password:");
  display.display();
}

void displayAccessAccepted() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(15, 20);
  display.println("Access Accepted");
  display.display();
}

void displayHomeEntry(DateTime now) {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(20, 0);
  display.println("Smart Security");

  display.setCursor(25, 10);
  display.println("Laboratorium");

  display.setCursor(0, 30);
  display.print("Date: ");
  display.print(now.day());
  display.print("/");
  display.print(now.month());
  display.print("/");
  display.println(now.year());
  
  display.setCursor(0, 40);
  display.print("Time: ");
  display.print(now.hour());
  display.print(":");
  if (now.minute() < 10) {
    display.print("0"); // Add leading zero if minute is less than 10
  }
  display.println(now.minute());

  display.display();
}

void displayInvalidPassword() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(15, 20);
  display.println("Invalid Password");
  display.display();
}

void unlockLocker() {
  myservo.write(90);
  delay(2000);
  myservo.write(0);

  // Tampilkan "Loker Terbuka" setelah membuka loker
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(23,30);
  display.println("Loker Terbuka");
  display.display();
  delay(1000);
  display.clearDisplay();
}

void activateBuzzer() {
  tone(buzzerPin, 200); // Aktifkan buzzer pada frekuensi 200 Hz
}

void deactivateBuzzer() {
  noTone(buzzerPin); // Matikan buzzer
}

void succesBuzzer(){
  tone(buzzerPin, 200);// Setelah memasukkan password
}