#include <WiFi.h>
#include <WiFiClientSecure.h>
#include <PubSubClient.h>
#include "DHTesp.h"

// WiFi credentials
const char* ssid = "Wokwi-GUEST";
const char* password = "";

// HiveMQ Cloud MQTT broker
const char* mqtt_server = "20c653d0ff6a404faa48d459d3c9ff08.s1.eu.hivemq.cloud";
const int mqtt_port = 8883;
const char* mqtt_user = "Arahim";
const char* mqtt_pass = "123456Abc";

const char* topic_sub = "aim/LedControl";
const char* topic_temp = "aim/readDHT_Temp";
const char* topic_humid = "aim/readDHT_Humid";

WiFiClientSecure espClient;
PubSubClient client(espClient);
DHTesp dht;

#define LED_PIN 19
#define DHT_PIN 33

unsigned long lastReadTime = 0;
const unsigned long interval = 2000;

void setup_wifi() {
    Serial.print("Connecting to ");
    Serial.println(ssid);
    WiFi.begin(ssid, password);
    while (WiFi.status() != WL_CONNECTED) {
        delay(500);
        Serial.print(".");
    }
    Serial.println("\nWiFi connected");
    Serial.println(WiFi.localIP());
}

void callback(char* topic, byte* payload, unsigned int length) {
    String msg = "";
    for (unsigned int i = 0; i < length; i++) {
        msg += (char)payload[i];
    }

    Serial.print("Received on ");
    Serial.print(topic);
    Serial.print(": ");
    Serial.println(msg);

    if (msg == "led1_1") digitalWrite(LED_PIN, HIGH);
    if (msg == "led1_0") digitalWrite(LED_PIN, LOW);
}

void reconnect() {
    while (!client.connected()) {
        Serial.print("Connecting to MQTT...");
        String clientId = "ESP32Client-" + String(random(0xffff), HEX);
        if (client.connect(clientId.c_str(), mqtt_user, mqtt_pass)) {
            Serial.println("connected");
            client.subscribe(topic_sub);
        } else {
            Serial.print("failed, rc=");
            Serial.print(client.state());
            Serial.println(" retrying in 5 seconds");
            delay(5000);
        }
    }
}

void setup() {
    Serial.begin(115200);
    pinMode(LED_PIN, OUTPUT);
    digitalWrite(LED_PIN, LOW);

    setup_wifi();

    espClient.setInsecure(); // For Wokwi testing (insecure TLS)

    client.setServer(mqtt_server, mqtt_port);
    client.setCallback(callback);

    dht.setup(DHT_PIN, DHTesp::DHT22);  // <-- DHT22 sensor setup
}

void loop() {
    if (!client.connected()) reconnect();
    client.loop();

    unsigned long now = millis();
    if (now - lastReadTime > interval) {
        lastReadTime = now;

        TempAndHumidity data = dht.getTempAndHumidity();
        if (!isnan(data.temperature) && !isnan(data.humidity)) {
            char tempStr[8], humidStr[8];
            dtostrf(data.temperature, 1, 2, tempStr);
            dtostrf(data.humidity, 1, 2, humidStr);

            client.publish(topic_temp, tempStr);
            client.publish(topic_humid, humidStr);

            Serial.print("Temp: ");
            Serial.print(tempStr);
            Serial.print(" °C | Humid: ");
            Serial.print(humidStr);
            Serial.println(" %");
        } else {
            Serial.println("Failed to read from DHT22");
        }
    }
}