#include <DHTesp.h>
#include <PubSubClient.h>
#include <WiFi.h>
//#include <WiFiClient.h>
//#include <BlynkSimpleEsp32.h>

// DHT sensors connect pin d15
const int DHT_PIN = 15;
const int relayPin = 13;
DHTesp dht;

const char* ssid = "Wokwi-GUEST" ;
const char* password = "";
const char* mqttServer = "b81JwtRu9Bp47vytyhZd-1oxa7N2vq4" ;

WiFiClient espClient;
PubSubClient client(espClient);
unsigned long lastMsg = 0;
float temp = 0;
float hum = 0;

void setup_wifi() {
    delay(10);
    Serial.println();
    Serial.print("Connecting to ");
    Serial.println(ssid);
    WiFi.mode(WIFI_STA);
    WiFi.begin(ssid, password);

    while (WiFi.status() != WL_CONNECTED) {
        delay(500);
        Serial.println(".");
    }

      randomSeed(micros());
      Serial.println("");
      Serial.print("Connected to ");
      Serial.println(ssid);
      Serial.print("IP address: ");
      Serial.println(WiFi.localIP());
}

void callback(char* topic, byte* payload, unsigned int length) {
    Serial. print("Message arrived [");
    Serial.print(topic);
    Serial.print("] ");
    for (int i = 0; i < length; i++) {
        Serial.print((char)payload[i]);
    }
    Serial.println();
}

void reconnect() {
    while (!client.connected()) {
      Serial.print("Attempting MOTT connection...");
      String clientId = "ESP32Client-";
      clientId += String(random(0xffff), HEX);

      if (client.connect(clientId.c_str())) {
          Serial.println("Connected to MOTT broker");
          client.publish("/ThinkIOT/Publish", "Welcome" );
          client.subscribe("/ThinkIOT/Subscribe");
      }   else {
            Serial.print("Failed to connect to MOTT broker, rc=");
            Serial.print(client.state());
            Serial.println(" try again in 5 seconds");
            delay(5000);
      }
    }
}

void setup() {
    pinMode(relayPin, OUTPUT);
    Serial.begin(115200);
    setup_wifi();
    client.setServer(mqttServer, 1883);
    client.setCallback(callback);
    dht.setup(DHT_PIN, DHTesp::DHT22);
}

void loop() {
    if (WiFi.status() != WL_CONNECTED) {
        Serial.println("WiFi disconnected, reconnecting...");
        setup_wifi();
    }

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

    unsigned long now = millis();
    if (now = lastMsg > 2000) {
        lastMsg = now;
        TempAndHumidity data = dht.getTempAndHumidity();
        String temp_str = String(data.temperature, 2);
        client.publish("/ThinkI0T/temp", temp_str.c_str());
        String hum_str = String(data.humidity, 1);
        client.publish("/ThinkIOT/hum", hum_str.c_str());
        Serial.print("Temperature: ");
        Serial.println(temp_str);
        Serial.print("Humidity: ");
        Serial.println(hum_str);
    }

    if (!client.connected()) {
        digitalWrite(relayPin, HIGH); // Relay ON when disconnected
    } else {
        digitalWrite(relayPin, LOW); // Relay OFF when connected
    }
}