// --- Підключення необхідних бібліотек ---
#include <Arduino.h>          // Стандартна бібліотека Arduino (для millis() тощо)
#include <Keypad.h>           // Для роботи з матричною клавіатурою 4x4
#include <EEPROM.h>           // Для збереження пароля в енергонезалежній пам'яті Arduino
#include <Wire.h>             // Необхідна для комунікації по шині I2C
#include <LiquidCrystal_I2C.h> // Спеціальна бібліотека для роботи з LCD дисплеями через I2C адаптер

// --- Ініціалізація об'єкта дисплея ---
// Вказуємо I2C адресу дисплея (0x27 - найпоширеніша, але може бути 0x3F - перевіряй сканером!)
// Також вказуємо розміри дисплея: 16 символів у ширину, 2 рядки у висоту.
LiquidCrystal_I2C lcd(0x27, 16, 2);

// --- Визначення констант (піни та налаштування) ---
#define lockPin 10    // Цифровий пін D10 буде керувати сигнальним входом MOSFET модуля (відкрити/закрити замок)
#define buzzerPin 13  // Цифровий пін D13 (де часто вбудований світлодіод на Nano) буде підключений до зумера
#define EEPROM_ADDRESS 0  // Адреса в пам'яті EEPROM, з якої починається збереження нашого пароля

// --- Налаштування клавіатури 4x4 ---
const byte ROWS = 4; // Кількість рядів
const byte COLS = 4; // Кількість стовпців
// Матриця символів, що відповідає кнопкам на клавіатурі
const char keys[ROWS][COLS] = {
  {'1', '2', '3', 'A'},
  {'4', '5', '6', 'B'},
  {'7', '8', '9', 'C'},
  {'*', '0', '#', 'D'}
};
// Піни Arduino, до яких фізично підключені виводи РЯДІВ клавіатури
const byte rowPins[ROWS] = {6, 7, 8, 9};    // Підключені до D6, D7, D8, D9
// Піни Arduino, до яких фізично підключені виводи СТОВПЦІВ клавіатури
const byte colPins[COLS] = {A0, A1, A2, A3}; // Підключені до A0, A1, A2, A3 (використовуються як цифрові входи)
// Створення об'єкта keypad за допомогою бібліотеки Keypad
Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);

// --- Глобальні змінні (доступні у всіх функціях) ---
String password;              // Змінна для зберігання поточного діючого пароля (тип String)
const String secretCombination = "A123";  // Секретний код для активації режиму зміни пароля
bool passwordEntry = false;   // Прапорець (true/false), що показує, чи ми зараз в процесі введення пароля
bool changePasswordMode = false; // Прапорець, що показує, чи ми зараз в процесі зміни пароля
String input = "";            // Змінна для накопичення символів, що вводяться користувачем
unsigned long lastActivityTime = 0; // Змінна для зберігання часу останнього натискання кнопки (в мілісекундах)

// --- Функція setup() - виконується один раз при увімкненні або перезавантаженні Arduino ---
void setup() {
  pinMode(lockPin, OUTPUT);    // Налаштовуємо пін замка (D10) як ВИХІД
  pinMode(buzzerPin, OUTPUT);  // Налаштовуємо пін зумера (D13) як ВИХІД
  digitalWrite(lockPin, LOW);  // Одразу встановлюємо низький рівень на піні замка (замок закритий)
  lcd.init();                  // Ініціалізуємо LCD дисплей
  lcd.backlight();             // Вмикаємо підсвітку дисплея
  lcd.print("Welcome!");       // Виводимо вітальне повідомлення
  loadPassword();              // Викликаємо функцію для завантаження пароля з EEPROM
  lastActivityTime = millis(); // Запам'ятовуємо поточний час як час останньої активності
}

// --- Функція loop() - основний цикл програми, виконується нескінченно ---
void loop() {
  // Перевірка на бездіяльність для вимкнення підсвітки
  // Якщо з моменту останнього натискання пройшло 20000 мілісекунд (20 секунд)...
  if (millis() - lastActivityTime >= 20000) {
    lcd.noBacklight();  // ...вимикаємо підсвітку дисплея
  }

  char key = keypad.getKey(); // Читаємо, яка кнопка натиснута зараз (якщо жодна, поверне NO_KEY)
  if (key) { // Якщо якась кнопка була натиснута...
    // ...одразу вмикаємо підсвітку (на випадок, якщо вона була вимкнена)
    lcd.backlight();
    // ...оновлюємо час останньої активності
    lastActivityTime = millis();
    // ...видаємо короткий звуковий сигнал зумером (частота 1000 Гц, тривалість 100 мс)
    tone(buzzerPin, 1000, 100);

    // --- Логіка обробки натиснутої клавіші ---

    // Сценарій 1: Ми НЕ в режимі введення пароля і НЕ в режимі зміни пароля (очікуємо '*')
    if (!passwordEntry && !changePasswordMode) {
      if (key == '*') {  // Якщо натиснули '*'...
        passwordEntry = true; // ...входимо в режим введення пароля
        input = "";           // ...очищуємо буфер введення
        lcd.clear();          // ...очищуємо екран
        lcd.print("Enter password:"); // ...просимо ввести пароль
        lcd.setCursor(0, 1);  // ...переводимо курсор на другий рядок для відображення вводу
      }
    }
    // Сценарій 2: Ми ВЖЕ в режимі введення пароля АБО в режимі його зміни
    else {
      if (key == '#') {  // Якщо натиснули '#' (завершення введення)...
        // --- Обробка після натискання '#' ---
        if (changePasswordMode) { // Якщо ми були в режимі ЗМІНИ пароля...
          if (input.length() > 0) { // ...і новий пароль не порожній...
            password = input; // ...зберігаємо введений рядок як новий пароль
            savePassword();   // ...записуємо новий пароль в EEPROM
            lcd.clear();
            lcd.print("Password changed"); // ...повідомляємо про успіх
            tone(buzzerPin, 2000, 500); // ...видаємо довгий підтверджуючий сигнал
            delay(2000);      // ...чекаємо 2 секунди
            changePasswordMode = false; // ...виходимо з режиму зміни пароля
            passwordEntry = false; // (також виходимо з режиму введення)
            input = "";           // ...очищуємо буфер
            lcd.clear();
            lcd.print("Welcome!"); // ...повертаємо вітальне повідомлення
          } else { // Якщо новий пароль порожній - нічого не робимо, просто виходимо
             changePasswordMode = false;
             passwordEntry = false;
             input = "";
             lcd.clear();
             lcd.print("Cancelled");
             delay(1000);
             lcd.clear();
             lcd.print("Welcome!");
          }
        } else {  // Якщо ми були в режимі ВВЕДЕННЯ пароля (не зміни)...
          // --- Перевірка введеного пароля ---
          if (input == password) {  // Якщо введене співпадає зі збереженим паролем...
            // *** ДОСТУП ДОЗВОЛЕНО ***
            lcd.clear();
            lcd.print("Password correct");
            tone(buzzerPin, 2000, 500); // ...переможний сигнал
            digitalWrite(lockPin, HIGH); // ...ВІДКРИВАЄМО ЗАМОК (подаємо HIGH на MOSFET)
            delay(3000);                 // ...тримаємо замок відкритим 3 секунди
            digitalWrite(lockPin, LOW);  // ...ЗАКРИВАЄМО ЗАМОК (знімаємо сигнал з MOSFET)
            lcd.clear();
            lcd.print("Locked");        // ...повідомляємо, що замок закрито
            delay(1500);                // ...невелика пауза
            lcd.clear();
            lcd.print("Welcome!");      // ...повертаємо вітання
          } else if (input == secretCombination) { // Якщо введене співпадає з кодом зміни пароля...
            // *** АКТИВАЦІЯ РЕЖИМУ ЗМІНИ ПАРОЛЯ ***
            changePasswordMode = true; // ...встановлюємо прапорець режиму зміни
            passwordEntry = false;     // ...(виходимо з режиму простого введення)
            input = "";                // ...очищуємо буфер
            lcd.clear();
            lcd.print("Enter new pass:"); // ...просимо ввести новий пароль
            lcd.setCursor(0, 1);
          } else {  // Якщо введене не співпало ні з паролем, ні з кодом зміни...
            // *** ДОСТУП ЗАБОРОНЕНО ***
            lcd.clear();
            lcd.print("Wrong password");
            // Видаємо три коротких тривожних сигнали
            for (int i = 0; i < 3; i++) {
              tone(buzzerPin, 500, 200);
              delay(300);
            }
            delay(2000); // ...чекаємо 2 секунди
            lcd.clear();
            lcd.print("Welcome!"); // ...повертаємо вітання
          }
          // Після обробки '#' в режимі введення, скидаємо прапорець та буфер
          passwordEntry = false;
          input = "";
        }
      } else { // Якщо натиснута НЕ '#' (а цифра, буква або '*')
        // --- Додавання символу до буфера введення ---
        if (input.length() < 16) { // Обмеження довжини пароля (щоб не переповнити буфер)
            input += key; // Додаємо натиснутий символ до рядка 'input'
            lcd.print('*'); // !!! Замість символу виводимо на екран зірочку '*' для секретності !!!
        }
      }
    }
  }
}

// --- Допоміжна функція для завантаження пароля з EEPROM ---
void loadPassword() {
  char buffer[17]; // Створюємо буфер для читання (розмір пароля + 1 для '\0')
  EEPROM.get(EEPROM_ADDRESS, buffer); // Читаємо дані з EEPROM, починаючи з адреси EEPROM_ADDRESS, в буфер
  buffer[16] = '\0'; // Гарантуємо, що рядок закінчується нуль-символом
  password = String(buffer); // Перетворюємо прочитаний масив символів на об'єкт String
  // Перевірка на "перший запуск": якщо пароль порожній або містить "сміття"
  if (password.length() == 0 || password.length() > 16) {
    password = "1234"; // ...встановлюємо пароль за замовчуванням "1234"
    savePassword();    // ...і одразу зберігаємо його в EEPROM
  }
}

// --- Допоміжна функція для збереження пароля в EEPROM ---
void savePassword() {
  char buffer[17]; // Створюємо буфер для запису
  // Копіюємо поточний пароль (змінна 'password' типу String) в масив символів 'buffer'
  password.toCharArray(buffer, sizeof(buffer));
  // Записуємо вміст буфера 'buffer' в EEPROM, починаючи з адреси EEPROM_ADDRESS
  EEPROM.put(EEPROM_ADDRESS, buffer);
  // Примітка: EEPROM має обмежений ресурс циклів запису (~100,000).
  // Ця функція викликається тільки при реальній зміні пароля, що є правильним підходом.
}