#include <WiFi.h>
#include <PubSubClient.h>
#include "DHTesp.h"
//#include <ArduinoJson.h> // Biblioteca para manipulação de JSON

// Substitua pelas suas informações de rede
const char* ssid = "Wokwi-GUEST";
const char* password = "";

// Substitua pelas suas informações do broker MQTT 
const char* mqttServer = "broker.hivemq.com";
const int mqttPort = 1883;
const char* mqttUser = "";
const char* mqttPassword = "";

// Configuração do sensor DHT22, Potencoopmetro, 
const int DHT_PIN = 15;
const int potentiometerPin = 34;
const int motorPumpPin = 5;
int waterBottle = 0;

int pumpState = LOW;


DHTesp dhtSensor;

WiFiClient espClient;
PubSubClient client(espClient);


/*void callback(char* topic, byte* payload, unsigned int length) {
  if (strcmp(topic, "sensors/motor-pump") == 0) {
if (payload[0] == '1') {
digitalWrite(motorPumpPin, HIGH); // Turn on the motor pump (LED) Serial.println("Pump on");
waterBottle += 2;
} else if (payload[0] == '0') {
digitalWrite(motorPumpPin, LOW); // Turn off the motor pump (LED) Serial.println("pump off");
} 
}
}*/



void callback(char* topic, byte* payload, unsigned int lenght){
  if (strcmp(topic, "sensors/motor-pump") == 0){
    if (payload[0] == '1'){
      digitalWrite(motorPumpPin, HIGH);
      Serial.println(("Pump on"));
      waterBottle += 1;
          } else if(payload[0] == '0'){
            digitalWrite(motorPumpPin, LOW);
            Serial.println("Pump off");
          }
  }
}

void reconnect() {
  // Loop até que estejamos reconectados
  while (!client.connected()) {
    Serial.print("Tentando conectar ao MQTT Broker...");
    // Tente conectar
    if (client.connect("")) {
      Serial.println("Conectado ao MQTT Broker!");
    } else {
      Serial.print("Falha na conexão, rc=");
      Serial.print(client.state());
      Serial.println("Tentando novamente em 2 segundos");
      delay(2000);
    }
  }
}

void setup() {
  Serial.begin(9600);
  Serial.print("Conectando ao WiFi");
  WiFi.begin("Wokwi-GUEST", "", 6);
  while (WiFi.status() != WL_CONNECTED) {
    delay(100);
    Serial.print(".");
  }
  Serial.println(" Conectado!");
  client.setServer(mqttServer, mqttPort);
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
  pinMode(motorPumpPin, OUTPUT);
}
void loop() {
  if (!client.connected()) {
    reconnect();
  }
  client.loop();
  TempAndHumidity data = dhtSensor.getTempAndHumidity();
  int potentiometerValue = analogRead(potentiometerPin);

  // Print sensor data to serial monitor
  Serial.println("Temperature: " + String(data.temperature, 2) + "°C");
  Serial.println("Humidity: " + String(data.humidity, 1) + "%"); 
  Serial.println("Soil Moisture: " + String(potentiometerValue));

  // Publish sensor data to MQTT topics
  char temperatureTopic[] = "sensors/temperature";
  char humidityTopic[] = "sensors/humidity";
  char potentiometerTopic[] = "sensors/potentiometer";
  char waterTopic[] = "sensors/water";

  client.publish(temperatureTopic, String(data.temperature, 2).c_str());
  client.publish(humidityTopic, String(data.humidity, 1).c_str());
  client.publish(potentiometerTopic, String(potentiometerValue).c_str());

  // Control the pump based on the potentiometer value
  if (potentiometerValue < 2000) {
    digitalWrite(motorPumpPin, HIGH); // Turn on the pump
    Serial.println("Pump on due to low soil moisture");
    client.publish(waterTopic, String("1").c_str());
  } else if (potentiometerValue > 3000) {
    digitalWrite(motorPumpPin, LOW); // Turn off the pump
    Serial.println("Pump off due to sufficient soil moisture");
    client.publish(waterTopic, String("0").c_str());
  }

  delay(3000); // Publishing interval (in milliseconds)
}

/*
void loop() {
  if (!client.connected()) {
    reconnect();
  }
  client.loop();
TempAndHumidity data = dhtSensor.getTempAndHumidity();
  int potentiometerValue = analogRead(potentiometerPin);
  // Print sensor data to serial monitor
Serial.println("Temperature: " + String(data.temperature, 2) + "°C");
Serial.println("Humidity: " + String(data.humidity, 1) + "%"); Serial.println("Soil Moisture: " + String(potentiometerValue));

  // Publish sensor data to MQTT topics
  char temperatureTopic[] = "sensors/temperature";
  char humidityTopic[] = "sensors/humidity";
  char potentiometerTopic[] = "sensors/potentiometer";
  char waterTopic[] = "sensors/water";

  client.publish(temperatureTopic, String(data.temperature, 2).c_str());
  client.publish(humidityTopic, String(data.humidity, 1).c_str());
  client.publish(potentiometerTopic, String(potentiometerValue).c_str());
  if (waterBottle >= 2){
    client.publish(waterTopic, String("1").c_str());
  } else {
    client.publish(waterTopic, String("0").c_str());
  }

  delay(3000); // Intervalo de publicação (em milissegundos)
}
*/


/*void sendMessage(const char* message) {
  // Construa a mensagem JSON
  DynamicJsonDocument jsonDoc(128);
  jsonDoc["Status da sala"] = message;
  char jsonBuffer[128];
  serializeJson(jsonDoc, jsonBuffer);

  // Publique a mensagem no tópico MQTT
  if (client.publish("sala/status", jsonBuffer)) {
    Serial.println("Mensagem enviada com sucesso: " + String(message));
  } else {
    Serial.println("Falha ao enviar mensagem: " + String(message));
  }
}
*/