#include <WiFi.h>
#include <PubSubClient.h>
#include <DHT.h>
#include <ESP32Servo.h>  // Replace Servo.h with ESP32Servo.h

#define DHTPIN 15
#define DHTTYPE DHT22

DHT dht(DHTPIN, DHTTYPE);
Servo lockServo;

// WiFi and MQTT credentials
const char* ssid = "Wokwi-GUEST";
const char* password = "";
const char* mqtt_server = "broker.hivemq.com";  // Public broker

WiFiClient espClient;
PubSubClient client(espClient);

// Define pins and initial values
int lockServoPin = 4;
int buttonPin = 5;
int redPin = 6;
int greenPin = 7;

float tempThreshold = 25.0;
bool doorLocked = true;

void setup() {
    Serial.begin(115200);
    dht.begin();
    lockServo.attach(lockServoPin);
    pinMode(buttonPin, INPUT_PULLUP);
    pinMode(redPin, OUTPUT);
    pinMode(greenPin, OUTPUT);

    // WiFi connection
    WiFi.begin(ssid, password);
    while (WiFi.status() != WL_CONNECTED) {
        delay(500);
        Serial.print(".");
    }
    Serial.println("WiFi connected");

    // MQTT connection
    client.setServer(mqtt_server, 1883);
    reconnect();
}

void reconnect() {
    while (!client.connected()) {
        Serial.print("Attempting MQTT connection...");
        if (client.connect("ESP32Client")) {
            Serial.println("connected");
            client.subscribe("home/door");
        } else {
            Serial.print("failed, rc=");
            Serial.print(client.state());
            delay(2000);
        }
    }
}
void loop() {
    if (!client.connected()) {
        reconnect();
    }
    client.loop();

    // Temperature control logic
    float temperature = dht.readTemperature();
    if (!isnan(temperature)) { // Check for valid reading
        String tempStr = String(temperature);
        client.publish("home/temperature", tempStr.c_str());
    }
    if (temperature > tempThreshold) {
        client.publish("home/fanStatus", "ON");
        delay(2000); // Optional delay to avoid too-frequent updates

    } else { 
        client.publish("home/fanStatus", "OFF");
        delay(2000); // Optional delay to avoid too-frequent updates

    }

    // Door lock control
    if (digitalRead(buttonPin) == LOW) {  // Simulate RFID
        doorLocked = !doorLocked;
        updateLockStatus();
        delay(1000);  // Debounce
    }
}

void updateLockStatus() {
    if (doorLocked) {
        lockServo.write(90); // Locked position
        digitalWrite(redPin, HIGH);
        digitalWrite(greenPin, LOW);
        client.publish("home/doorStatus", "Locked");
    } else {
        lockServo.write(0); // Unlocked position
        digitalWrite(redPin, LOW);
        digitalWrite(greenPin, HIGH);
        client.publish("home/doorStatus", "Unlocked");
    }
}