#include <Keypad.h>
#include <LiquidCrystal_I2C.h>
#include <Servo.h>

int servoPin = 11;
const int keypadRows = 4;
const int keypadColumns = 4;

const byte rowPins[keypadRows] = {9, 8, 7, 6};
const byte colPins[keypadColumns] = {5, 4, 3, 2};

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

Keypad keypad = Keypad(makeKeymap((char*)keys), rowPins, colPins, keypadRows, keypadColumns);
LiquidCrystal_I2C lcd(0x27, 20, 4);
Servo servo;

int servoUnlockPos = 0;
int servoLockPos = 90;

String password = "1234";
String input = "";

int attempts = 5; // Initialize attempts counter
bool lockoutMode = false; // Flag for lockout mode

void setup() {
  lcd.begin(20, 4);
  lcd.init();
  lcd.setCursor(0, 0);
  lcd.backlight();
  showMessage();
  servo.attach(servoPin);
  servo.write(servoLockPos); // Initialize servo to lock position
}

void showMessage() {
  lcd.setCursor(4, 0);
  lcd.print("WELCOME");
  delay(1000);

  lcd.setCursor(0, 2);
  String message = "HOMING SYSTEM";
  for (int i = 0; i < message.length(); i++) {
    lcd.print(message[i]);
    delay(100);
  }
  delay(500);
}

void loop() {
  if (lockoutMode) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("TOO MANY ATTEMPTS!");
    delay(1000);
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("WAIT FOR 30 SECONDS");
    delay(1000);
    for (int i = 30; i > 0; i--) {
     lcd.setCursor(0, 1);
      lcd.print("TIME REMAINING: " + String(i) + " SECONDS");
      delay(1000);
    }
    lockoutMode = false;
    attempts = 5; // Reset attempts counter
  }
lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Enter password");
  lcd.setCursor(0, 2);
  
  lcd.print("Attempts: " + String(attempts));
  input = inputPassword();

  if (input == password) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Unlocked!");
    servo.write(servoUnlockPos); // Move servo to unlock position
    delay(1000);
    servo.write(servoLockPos); // Move servo back to lock position
    attempts = 5; // Reset attempts counter
  } else {
    attempts--; // Decrement attempts counter
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("WRONG PASSWORD!");
    delay(1000);
    lcd.clear();
    lcd.setCursor(0, 2);
    lcd.print("Attempts remaining: " + String(attempts));
    delay(1000);
    if (attempts == 0) {
      lockoutMode = true;
    }
  }
}

 String inputPassword() {
  String input = "";
  while (input.length() < 4) {
    char key = keypad.getKey();
    if (key >= '0' && key <= '9') {
      lcd.print('*');
      input += key;
      Serial.print(key); 
      Serial.println(); 
    }
  }
  String password = String(input);
  Serial.print(password);
  return password;
}