//#include <ESP8266WiFi.h>
#include <WiFi.h>
#include <PubSubClient.h>
#include "DHTesp.h"

const int DHT_PIN = 23;

const int TRIG_PIN = 4;
const int ECHO_PIN = 5;

DHTesp dhtSensor;

String my_payload;
String my_topic;

const char* ssid = "Wokwi-GUEST";
const char* password = "";
const char* mqtt_server = "broker.hivemq.com";

WiFiClient WOKWI_client;
PubSubClient client(WOKWI_client);

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(".");
  }

  Serial.println("");
  Serial.println("WiFi connected");
  Serial.println("IP address: ");
  Serial.println(WiFi.localIP());
}

void reconnect() {
  while (!client.connected()) {
    Serial.print("Attempting MQTT connection...");
   
    if (client.connect("WOKWI_client")) {
      Serial.println("connected");
      client.subscribe("WSP32_ASSINA");


    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      // Wait 5 seconds before retrying
      delay(5000);
    }
  }
}


void callback(char* topic, byte* payload, unsigned int length) {
  Serial.print("Message arrived [");
  Serial.print(topic);
  Serial.print("] ");

// FUNÇÃO PARA CONCATENAR MSG
String conc_payload;
  for (int i = 0; i < length; i++) {
    conc_payload += ((char)payload[i]);
  }

  my_topic = topic;
  my_payload = conc_payload;

}


void setup() {
  pinMode(2, OUTPUT);  
  pinMode(15, OUTPUT);
  pinMode(19, OUTPUT);
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);

  Serial.begin(115200);
  setup_wifi();
  client.setServer(mqtt_server, 1883);
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
}

void Conectado_Wifi(){
  if (WiFi.status())
    digitalWrite(2, HIGH);
  else
    digitalWrite(2, LOW);
}

void Conectado_HIVEMQ(){
  if (client.connected())
    digitalWrite(15, HIGH);
  else
    digitalWrite(15, LOW);
}

long medirDistancia() {
  // Gera o pulso ultrassônico
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);

  long duracao = pulseIn(ECHO_PIN, HIGH);
  
  long distancia = duracao * 0.034 / 2;
  return distancia;
}

void recebe_dados(){
  if (my_payload == "A")
    digitalWrite(19, HIGH);
  if(my_payload =="a")
    digitalWrite(19, LOW);
}

void Publicartemp_hum() {
  TempAndHumidity data = dhtSensor.getTempAndHumidity();
  long distancia = medirDistancia();

  client.publish("TOPICO_WOKWI_PROVA_TEMP", String(data.temperature).c_str());
  client.publish("TOPICO_WOKWI_PROVA_HUM", String(data.humidity).c_str());
  client.publish("TOPICO_WOKWI_PROVA_PROX", String(distancia).c_str());

  Serial.println("Temp: " + String(data.temperature, 2) + "°C");
  Serial.println("Humidity: " + String(data.humidity, 1) + "%");
  Serial.println("Proximity: " + String(distancia) + " cm");
  Serial.println("---");
  delay(2000);
}

void loop() {
    reconnect();
    Conectado_Wifi();
    Conectado_HIVEMQ();
    Publicartemp_hum();
    client.loop();
    recebe_dados();
}