#include <WiFi.h>

#include <PubSubClient.h>

#include <DHTesp.h>


const int DHT_PIN = 15;

const int redled = 4;

const int yellowled = 2;

const int blueled = 32;

const int trigPin = 13;

const int echoPin = 12;

long duration;

int distance;

DHTesp dht;

const char* ssid = "Wokwi-GUEST"; ///  wifi ssid

const char* password = "";

const char* mqtt_server = "test.mosquitto.org";// mosquitto server url


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

  }


  randomSeed(micros());


  Serial.println("");

  Serial.println("WiFi connected");

  Serial.println("IP address: ");

  Serial.println(WiFi.localIP());

}




void callback(char* topic, byte* message, unsigned int length) {

  Serial.print("Message arrived on topic: ");

  Serial.print(topic);

  Serial.print(". Message: ");

  String messageTemp;



  for (int i = 0; i < length; i++) {

    Serial.print((char)message[i]);

    messageTemp += (char)message[i];

  }

  Serial.println();


  // Feel free to add more if statements to control more GPIOs with MQTT


  // If a message is received on the topic esp32/output, you check if the message is either "on" or "off".

  // Changes the output state according to the message

  if (String(topic) == "esp32.4/red") {

    Serial.print("Changing output to ");

    if (messageTemp == "on") {

      Serial.println("on");

      digitalWrite(redled, HIGH);

    }

    else if (messageTemp == "off") {

      Serial.println("off");

      digitalWrite(redled, LOW);

    }

  }

  if (String(topic) == "esp32.4/yellow") {

    Serial.print("Changing output to ");

    if (messageTemp == "on") {

      Serial.println("on");

      digitalWrite(yellowled, HIGH);

    }

    else if (messageTemp == "off") {

      Serial.println("off");

      digitalWrite(yellowled, LOW);

    }

  }

  if (String(topic) == "esp32.4/green") {

    Serial.print("Changing output to ");

    if (messageTemp == "on") {

      Serial.println("on");

      digitalWrite(blueled, HIGH);

    }

    else if (messageTemp == "off") {

      Serial.println("off");

      digitalWrite(blueled, LOW);

    }

  }
}




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("/esp32/Publish", "Welcome");

      client.subscribe("esp32.4/red");

      client.subscribe("esp32.4/yellow");

      client.subscribe("esp32.4/green");

    } else {

      Serial.print("failed, rc=");

      Serial.print(client.state());

      Serial.println(" try again in 5 seconds");

      delay(5000);

    }
  }

}

void setup() {

  pinMode(yellowled, OUTPUT);

  pinMode(blueled, OUTPUT);

  pinMode(redled, OUTPUT);

  pinMode(trigPin, OUTPUT); // Sets the trigPin as an Output

  pinMode(echoPin, INPUT); // Sets the echoPin as an Input

  Serial.begin(115200);

  setup_wifi();

  client.setServer(mqtt_server, 1883);

  client.setCallback(callback);

  dht.setup(DHT_PIN, DHTesp::DHT22);

}

void loop() {

  if (!client.connected()) {

    reconnect();

  }

  client.loop();

  // Clears the trigPin
  digitalWrite(trigPin, LOW);
  delayMicroseconds(2);
  // Sets the trigPin on HIGH state for 10 micro seconds
  digitalWrite(trigPin, HIGH);
  delayMicroseconds(10);
  digitalWrite(trigPin, LOW);

  // Reads the echoPin, returns the sound wave travel time in microseconds
  duration = pulseIn(echoPin, HIGH);

  // Calculate the distance
  distance = (duration * 0.034 / 2) + 1;

  unsigned long now = millis();

  if (now - lastMsg > 2000) { //perintah publish data

    lastMsg = now;

    TempAndHumidity data = dht.getTempAndHumidity();


    String temp = String(data.temperature, 2);

    client.publish("esp32.4/temp", temp.c_str()); // publish temp topic /ThinkIOT/temp

    String hum = String(data.humidity, 1);

    client.publish("esp32.4/hum", hum.c_str());   // publish hum topic /ThinkIOT/hum

    String dis = String(distance);

    client.publish("esp32.4/dis", dis.c_str());   // publish hum topic /ThinkIOT/hum


    Serial.print("Temperature: ");

    Serial.println(temp);

    Serial.print("Humidity: ");

    Serial.println(hum);

    Serial.print("Distance: ");

    Serial.println(dis);


  }

}