#include <Keypad.h>
#include <Wire.h>
#include <LiquidCrystal_I2C.h>
#include <ESP32Servo.h> // Knižnica na ovládanie serva

// Inicializácia LCD displeja
LiquidCrystal_I2C lcd(0x27, 16, 2); // adresa LCD pre 16 znakov a 2 riadky

Servo servoMotor;       // Objekt pre servo motor
const int servoPin = 27; // Pin pre servo motor

String pad;
const byte numRows = 4;
const byte numCols = 4;
String password = "4321";
char keypressed;
bool changeMode = false; // Režim na zmenu hesla
bool enterCurrentPassword = false; // Indikátor na zadanie aktuálneho hesla pre zmenu
String newPassword;      // Dočasný reťazec pre nové heslo

char keymap[numRows][numCols] =
{
  {'1', '2', '3', 'A'},
  {'4', '5', '6', ' '},
  {'7', '8', '9', ' '},
  {'*', '0', '#', ' '}
};

// Priradenie pinov pre riadky a stĺpce klávesnice
byte rowPins[numRows] = {23, 19, 18, 5}; // Zmenené na vhodné piny pre ESP32
byte colPins[numCols] = {4, 16, 17, 15}; // Zmenené na vhodné piny pre ESP32

Keypad myKeypad = Keypad(makeKeymap(keymap), rowPins, colPins, numRows, numCols); // mapping keypad

void setup() {
  // Inicializácia LCD
  lcd.init();
  lcd.backlight();
  lcd.setCursor(0, 0);
  lcd.print("Keypad");
  lcd.setCursor(0, 1);
  lcd.print("Test");
  delay(1000);
  lcd.clear();
  
  // Úvodná správa
  lcd.setCursor(0, 0);
  lcd.print("Heslo:");
  delay(1000);
  lcd.clear();

  // Inicializácia serva
  servoMotor.attach(servoPin); // Pripojenie serva na pin 27
  servoMotor.write(0);          // Servo začne v pozícii 0 stupňov
  servoMotor.detach();          // Odstránenie signálu na začiatku, aby sa nepohlo
}

void loop() {
  readKeypad();

  if (keypressed == '#') { // Potvrdenie vstupu
    if (changeMode) {
      // Ak sme v režime zmeny hesla, potvrdiť nové heslo
      password = pad; // Nastaviť nové heslo
      lcd.setCursor(0, 1);
      lcd.print("Heslo zmenene");
      changeMode = false; // Ukončiť režim zmeny hesla
      delay(2000);
      pad = ""; // Resetovať vstup
      lcd.clear();
    } else if (enterCurrentPassword) {
      // Skontrolovať aktuálne heslo pred zmenou
      if (pad == password) {
        lcd.setCursor(0, 1);
        lcd.print("nove heslo");
        changeMode = true; // Aktivovať režim zmeny hesla
        enterCurrentPassword = false;
        pad = ""; // Resetovať vstup pre nové heslo
        delay(1000);
        lcd.clear();
      } else {
        lcd.setCursor(0, 1);
        lcd.print("Zle heslo");
        delay(2000);
        enterCurrentPassword = false; // Zrušiť pokus o zmenu hesla
        pad = ""; // Resetovať vstup
        lcd.clear();
      }
    } else {
      // Overenie bežného hesla pre prístup
      if (pad == password) {
        lcd.setCursor(0, 1);
        lcd.print("Otvorene");
        servoMotor.attach(servoPin); // Pripojiť servo
        servoMotor.write(90);        // Otočiť servo o 90 stupňov
        delay(1000);                 // Krátke zdržanie
        servoMotor.detach();         // Odstrániť signál, aby servo ostalo na mieste
      } else {
        lcd.setCursor(0, 1);
        lcd.print("Zamietnute");
      }
      delay(2000);
      pad = ""; // Resetovanie vstupu
      lcd.clear();
    }
  }

  if (keypressed == '*') { // Mazanie posledného znaku
    if (pad.length() > 0) {
      pad.remove(pad.length() - 1); // Odstrániť posledný znak
    }
    lcd.clear();
  }

  if (keypressed == '0') { // Aktivácia režimu zmeny hesla
    lcd.setCursor(0, 1);
    lcd.print("zmena hesla");
    enterCurrentPassword = true; // Indikovať potrebu zadať aktuálne heslo pre zmenu
    pad = ""; // Resetovať vstup
    delay(1000);
    lcd.clear();
  }

  if (keypressed == 'A') { // Tlačidlo na zatvorenie serva (otočiť späť na 0 stupňov)
    servoMotor.attach(servoPin);  // Pripojiť servo
    servoMotor.write(0);          // Servo otočí späť na 0 stupňov
    delay(1000);                  // Krátke zdržanie
    servoMotor.detach();          // Odstrániť signál, aby servo ostalo na mieste
    lcd.setCursor(0, 1);
    lcd.print("Zavrete");
    delay(2000);
    lcd.clear();
  }

  // Vypísať aktuálne zadané heslo na riadok ako "Heslo:xxxx"
  lcd.setCursor(0, 0);
  lcd.print("Heslo:");    // Zobraziť text "Heslo:"
  lcd.print(pad);         // Zobraziť aktuálny obsah hesla
  delay(100);
}

void readKeypad() {
  keypressed = myKeypad.getKey(); // Detekcia stlačenia klávesnice
  
  // Pridať číslice len ak to nie sú špeciálne klávesy a ak už nie je zadané 4 znaky
  if (keypressed != '#' && keypressed != '*' && keypressed != '0' && keypressed != 'A' && keypressed != ' ' && keypressed != '\0') {
    if (pad.length() < 4) { // Ak ešte nie je zadaný 4-ciferný kód
      pad += keypressed; // Pridať kláves do reťazca
    }
  }
}