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

// Keypad settings
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);

// LCD settings
LiquidCrystal_I2C lcd(0x27, 20, 4);

// Servo settings
Servo front;  // front door
Servo back;  // back door
int servoPinFront = 11;  // front door servo pin
int servoPinBack = 10;  // back door servo pin
int servoUnlockPos = 0;
int servoLockPos = 90;

// Ultrasonic sensor settings
int echoFront = 4;
int trigFront = 5;
int echoBack = 2;
int trigBack = 3;
long distanceFront;
long distanceBack;

// Buzzer settings
int buzzerFront = 12;
int buzzerBack = 13;

// Password settings
String password = "1234";
String input = "";

// Attempts counter
int attempts = 5;
bool lockoutMode = false;

void setup() {
  Serial.begin(9600);

  lcd.begin(20, 4);
  lcd.init();
  lcd.setCursor(0, 0);
  lcd.backlight();

  front.attach(servoPinFront);
  back.attach(servoPinBack);
  front.write(servoLockPos);  // Initialize front door servo to lock position
  back.write(servoLockPos);  // Initialize back door servo to lock position

  pinMode(echoFront, INPUT);
  pinMode(trigFront, OUTPUT);
  pinMode(echoBack, INPUT);
  pinMode(trigBack, OUTPUT);

  pinMode(buzzerFront, OUTPUT);
  pinMode(buzzerBack, OUTPUT);
}

void loop() {
  if (lockoutMode) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("LOCKOUT MODE");
    lcd.setCursor(0, 1);
    lcd.print("Wait 30 seconds");
    delay(30000);  // Wait for 30 seconds in lockout mode
    attempts = 5;
    lockoutMode = false;
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Home Automation");
    return;
  }

  distanceFront = getDistance(trigFront, echoFront);
  distanceBack = getDistance(trigBack, echoBack);

  if (distanceFront <= 15 && distanceFront > 0) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Intruder at front:");
    lcd.setCursor(1, 1);
    lcd.print(distanceFront);
    lcd.print(" cm");
    lcd.setCursor(1, 2);
    lcd.print("Enter password:");
    input = inputPassword();
    if (input == password) {
      lcd.clear();
      lcd.setCursor(0, 0);
      lcd.print("Unlocked!");
      lcd.setCursor(1, 1);
      lcd.print("Door opened.");
      front.write(servoUnlockPos);  // Move front door servo to unlock position
      tone(buzzerFront, 300, 250);  // Short beep
      delay(1000);
      front.write(servoLockPos);  // Move front door servo back to lock position
      lcd.clear();
      lcd.setCursor(0, 0);
      lcd.print("Home Automation");
    } else {
      attempts--;  // Decrement attempts counter
      lcd.clear();
      lcd.setCursor(0, 0);
      lcd.print("WRONG PASSWORD!");
      tone(buzzerFront, 500, 1000);  // Long beep
      delay(1000);
      lcd.clear();
      lcd.setCursor(0, 2);
      lcd.print("Attempts remaining: " + String(attempts));
      delay(1000);
      if (attempts == 0) {
        lockoutMode = true;
      }
    }
  } else if (distanceBack <= 15 && distanceBack > 0) {
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Intruder at back:");
    lcd.setCursor(1, 1);
    lcd.print(distanceBack);
    lcd.print(" cm");
    lcd.setCursor(1, 2);
    lcd.print("Enter password:");
    input = inputPassword();
    if (input == password) {
      lcd.clear();
      lcd.setCursor(0, 0);
      lcd.print("Unlocked!");
      lcd.setCursor(1, 1);
      lcd.print("Door opened.");
      back.write(servoUnlockPos);  // Move back door servo to unlock position
      tone(buzzerBack, 300, 250);  // Short beep
      delay(1000);
      back.write(servoLockPos);  // Move back door servo back to lock position
      lcd.clear();
      lcd.setCursor(0, 0);
      lcd.print("Home Automation");
    } else {
      attempts--;  // Decrement attempts counter
      lcd.clear();
      lcd.setCursor(0, 0);
      lcd.print("WRONG PASSWORD!");
      tone(buzzerBack, 500, 1000);  // Long beep
      delay(1000);
      lcd.clear();
      lcd.setCursor(0, 2);
      lcd.print("Attempts remaining: " + String(attempts));
      delay(1000);
      if (attempts == 0) {
        lockoutMode = true;
      }
    }
  }
}

String inputPassword() {
  String enteredPassword = "";
  char key;
  while (true) {
    key = keypad.getKey();
    if (key) {
      if (key == '#') {
        return enteredPassword;  // End input when '#' is pressed
      } else if (key == '*') {
        enteredPassword = "";  // Clear input when '*' is pressed
        lcd.setCursor(0, 3);
        lcd.print("Password Cleared");
        delay(1000);
        lcd.setCursor(0, 2);
        lcd.print("Enter password:   ");
      } else {
        enteredPassword += key;  // Append key to password
        lcd.setCursor(0, 3);
        lcd.print("Password: ");
        lcd.print(enteredPassword);
      }
    }
  }
}

long getDistance(int trigPin, int echoPin) {
  digitalWrite(trigPin, LOW);
  delayMicroseconds(2);
  digitalWrite(trigPin, HIGH);
  delayMicroseconds(10);
  digitalWrite(trigPin, LOW);
  
  long duration = pulseIn(echoPin, HIGH);
  long distance = (duration / 2) * 0.0344;  // Convert duration to distance in cm
  return distance;
}