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

const int DHT_PIN = 15;
DHTesp dht;
const char* ssid = "Wokwi-GUEST"; 
const char* password = "";
const char* mqtt_server = "test.mosquitto.org";
WiFiClient espClient;
PubSubClient client(espClient);
unsigned long lastMsg = 0;
float temp = 0;
float hum = 0;

const int LED_GREEN_PIN = 2; 

const int LED_RED_PIN = 5; 

const int LED_YELLOW_PIN = 4; 

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.print(".");
}
randomSeed(micros());
Serial.println("");
Serial.println("WiFi connected");
Serial.println("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]);
}
}

void reconnect() {
while (!client.connected()) {
Serial.print("Attempting MQTT connection...");
String clientId = "ESP32Client-";
clientId += String(random(0xffff), HEX);
if (client.connect(clientId.c_str())) {
Serial.println("Connected");
client.publish("/ThinkIOT/Publish", "Welcome");
client.subscribe("/ThinkIOT/Subscribe");
} else {
Serial.print("failed, rc=");
Serial.print(client.state());
Serial.println(" try again in 5 seconds");
delay(5000);
}
}
}

void setup() {
pinMode(2, OUTPUT);
Serial.begin(115200);
setup_wifi();
client.setServer(mqtt_server, 1883);
client.setCallback(callback);
dht.setup(DHT_PIN, DHTesp::DHT22);
pinMode(LED_GREEN_PIN, OUTPUT);
pinMode(LED_RED_PIN, OUTPUT);
pinMode(LED_YELLOW_PIN, OUTPUT);
}

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

client.loop();
unsigned long now = millis();
if (now - lastMsg > 2000) { 
lastMsg = now;
TempAndHumidity data = dht.getTempAndHumidity();
String temp = String(data.temperature, 2);
client.publish("/ThinkIOT/temputs", temp.c_str()); 

String hum = String(data.humidity, 1);
client.publish("/ThinkIOT/humuts", hum.c_str()); 

Serial.print("Temperature: ");
Serial.println(temp);
Serial.print("Humidity: ");
Serial.println(hum);

int status_green = digitalRead(LED_GREEN_PIN);
int status_red = digitalRead(LED_RED_PIN);
int status_yellow = digitalRead(LED_YELLOW_PIN);

if (data.temperature < 24) {
digitalWrite(LED_YELLOW_PIN, LOW); 

} else {
digitalWrite(LED_YELLOW_PIN, HIGH); 

}

if (data.temperature > 0) {
digitalWrite(LED_GREEN_PIN, HIGH); 

} else {
digitalWrite(LED_GREEN_PIN, LOW); 

}

if (data.temperature == 0) {
digitalWrite(LED_RED_PIN, HIGH); 

delay(1000); 
digitalWrite(LED_RED_PIN, LOW); 
} else {
digitalWrite(LED_RED_PIN, LOW); 

}
}
}