#include <Keypad.h>
#include <TM1637Display.h>
#include <ESP32Servo.h>
// TM1637 Display Pins
#define CLK 2
#define DIO 4
TM1637Display display(CLK, DIO);
#define SERVO_PIN 5

// Define password
const String PASSWORD = "1234";
String input = "";

// Servo setup
Servo lockServo;
bool isUnlocked = false;

// Keypad setup
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] = {32, 33, 25, 26}; 
byte colPins[COLS] = {27, 14, 12, 13}; 

Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);
// Password
String password = "";
String enteredCode = "";

void setup() {
  Serial.begin(115200);
  display.setBrightness(7);
  display.showNumberDec(0, true); // Show 0000

  // Attach servo and set to locked position
  lockServo.attach(SERVO_PIN);
  lockServo.write(0); // 0 degrees = locked
}

void loop() {
  char key = keypad.getKey();

  if (key) {
    if (key >= '0' && key <= '9') {
      if (enteredCode.length() < 4) {
        enteredCode += key;
        updateDisplay(enteredCode);
      }
    } else if (key == '#') {
      if (enteredCode == password) {
        Serial.println("Access Granted");
        unlockDoor();
      } else {
        Serial.println("Access Denied");
        displayError();
      }
      enteredCode = "";
    } else if (key == '*') {
      enteredCode = "";
      display.showNumberDec(0, true); // Reset display
    }
  }
}

void updateDisplay(String code) {
  int num = code.toInt();
  display.showNumberDec(num, true, 4, 0);
}

void unlockDoor() {
  lockServo.write(90); // Open position
  delay(5000); // Wait 5 seconds
  lockServo.write(0); // Lock again
  display.showNumberDec(0, true);
}

void displayError() {
  display.showNumberDec(0, true); // Reset to 0000
  // Optional: flash error or show "----" if supported
}