/********* Pleasedontcode.com **********

	Pleasedontcode thanks you for automatic code generation! Enjoy your code!

	- Terms and Conditions:
	You have a non-exclusive, revocable, worldwide, royalty-free license
	for personal and commercial use. Attribution is optional; modifications
	are allowed, but you're responsible for code maintenance. We're not
	liable for any loss or damage. For full terms,
	please visit pleasedontcode.com/termsandconditions.

	- Project: **Title:** Lock System
	- Source Code NOT compiled for: Arduino Nano ESP32
	- Source Code created on: 2024-10-05 14:23:35

********* Pleasedontcode.com **********/

/****** SYSTEM REQUIREMENTS *****/
/****** SYSTEM REQUIREMENT 1 *****/
	/* The servo motor installed in the door will lock or */
	/* unlock based on the correct PIN code received by */
	/* the button. If the door is unlocked and the PIN is */
	/* correct, it will lock. If the door is locked and */
	/* the PIN is correct, it will unlock. */
/****** SYSTEM REQUIREMENT 2 *****/
	/* The webserver provides a web page about lock or */
	/* unlock state of the door. */
/****** END SYSTEM REQUIREMENTS *****/


/********* User code review feedback **********
#### Feedback 1 ####
- add a physical button to enable introduction of pin on keypad
********* User code review feedback **********/

/****** DEFINITION OF LIBRARIES *****/
#include 	// Library for controlling servo motors
#include 	// Library for keypad input
#include        // Library for WiFi connectivity
#include   // Library for web server functionality
#include "ButtonHandler.h" // Include the button handler header

/****** FUNCTION PROTOTYPES *****/
void setup(void);
void loop(void);
void handleRoot(); // Function to handle root URL

/***** DEFINITION OF DIGITAL INPUT PINS *****/
const uint8_t keypad_Keypad3x4_R1_PIN_D4 = 4;
const uint8_t keypad_Keypad3x4_R2_PIN_D5 = 5;
const uint8_t keypad_Keypad3x4_R3_PIN_D6 = 6;
const uint8_t keypad_Keypad3x4_R4_PIN_D7 = 7;
const uint8_t buttonPin = 2; // Define the pin for the button

/***** DEFINITION OF DIGITAL OUTPUT PINS *****/
const uint8_t keypad_Keypad3x4_C1_PIN_D8 = 8;
const uint8_t keypad_Keypad3x4_C2_PIN_D9 = 9;
const uint8_t keypad_Keypad3x4_C3_PIN_D10 = 10;

/***** DEFINITION OF KEYPAD VARIABLES *****/
const uint8_t KEYPAD3X4_ROWS = 4; // four rows
const uint8_t KEYPAD3X4_COLS = 3; // three columns
char Keys_Keypad3x4[KEYPAD3X4_ROWS][KEYPAD3X4_COLS] = {
    {'1','2','3'},
    {'4','5','6'},
    {'7','8','9'},
    {'*','0','#'},
};

Keypad keypad = Keypad(makeKeymap(Keys_Keypad3x4), keypad_Keypad3x4_R1_PIN_D4, keypad_Keypad3x4_C1_PIN_D8, KEYPAD3X4_ROWS, KEYPAD3X4_COLS);
Servo doorServo; // Servo object for door lock
WebServer server(80); // Create a web server on port 80
ButtonHandler pinButton(buttonPin); // Create an instance of ButtonHandler

// Define the correct PIN code
const String correctPIN = "1234"; // Example PIN code
String enteredPIN = ""; // Variable to store entered PIN
bool isLocked = true; // Initial state of the door
bool inputEnabled = false; // State to track if input is enabled

void setup(void)
{
    // Initialize the servo motor
    doorServo.attach(9); // Attach the servo to pin 9
    doorServo.write(isLocked ? 0 : 90); // Lock the door initially

    // Setup keypad pins
    pinMode(keypad_Keypad3x4_R1_PIN_D4, INPUT_PULLUP);
    pinMode(keypad_Keypad3x4_R2_PIN_D5, INPUT_PULLUP);
    pinMode(keypad_Keypad3x4_R3_PIN_D6, INPUT_PULLUP);
    pinMode(keypad_Keypad3x4_R4_PIN_D7, INPUT_PULLUP);
    
    // Initialize the button
    pinButton.begin();

    // Setup WiFi connection
    WiFi.begin("yourSSID", "yourPASSWORD"); // Replace with your WiFi credentials
    while (WiFi.status() != WL_CONNECTED) {
        delay(1000);
    }
    
    // Start the web server
    server.on("/", handleRoot); // Handle root URL
    server.begin(); // Start the server
}

void loop(void)
{
    // Check if the button is pressed
    if (pinButton.isPressed()) {
        inputEnabled = !inputEnabled; // Toggle input mode
        delay(300); // Debounce delay
    }

    // Check for keypad input only if input is enabled
    if (inputEnabled) {
        char key = keypad.getKey();
        if (key) {
            if (key == '#') { // If '#' is pressed, check the entered PIN
                if (enteredPIN == correctPIN) {
                    // Toggle lock state
                    isLocked = !isLocked;
                    doorServo.write(isLocked ? 0 : 90); // Lock or unlock the door
                }
                enteredPIN = ""; // Reset entered PIN
            } else if (key == '*') { // If '*' is pressed, reset the entered PIN
                enteredPIN = "";
            } else {
                enteredPIN += key; // Append the key to the entered PIN
            }
        }
    }

    // Handle web server requests
    server.handleClient();
}

void handleRoot() {
    // Serve the HTML content
    String html = "
Door Lock System
";
    html += "
The door is currently " + String(isLocked ? "Locked" : "Unlocked") + "
";
    html += "";
    server.send(200, "text/html", html); // Send the HTML response
}