#include <Wire.h>                                                   //Import the required libraries
#include "DHTesp.h"
#if defined(ESP32)
#include <WiFiMulti.h>
WiFiMulti wifiMulti;
#define DEVICE "ESP32"
#elif defined(ESP8266)
#include <ESP8266WiFiMulti.h>
ESP8266WiFiMulti wifiMulti;
#define DEVICE "ESP8266"
#endif
#include <PubSubClient.h>
#define TZ_INFO "ICT-7"                                                                                                 //InfluxDB v2 timezone

#include <InfluxDbClient.h>
#include <InfluxDbCloud.h>

#define WIFI_SSID "Wokwi-GUEST"                                                                                        //Network Name
#define WIFI_PASSWORD ""

const int DHT_PIN = 15;
const int LED_PIN = 12;

DHTesp dhtSensor;

String stMac;
char mac[50];
char clientId[50];

const char* mqtt_server = "broker.emqx.io";
const int mqtt_port = 1883;

WiFiClient espClient;
PubSubClient client(espClient);

// Create variable to hold mqtt messages
#define MSG_BUFFER_SIZE	(100)
char msg[MSG_BUFFER_SIZE];

void setup_wifi() {
  delay(10);
  WiFi.mode(WIFI_STA);                                              //Setup wifi connection
  wifiMulti.addAP(WIFI_SSID, WIFI_PASSWORD);

  Serial.print("Connecting to wifi");                               //Connect to WiFi
  while (wifiMulti.run() != WL_CONNECTED)
  {
    Serial.print(".");
    delay(100);
  }
  Serial.println();

  Serial.println("WiFi connected");
  Serial.println("IP address: ");
  Serial.println(WiFi.localIP());
  Serial.println(WiFi.macAddress());
  stMac = WiFi.macAddress();
  stMac.replace(":", "_");
  Serial.println(stMac);

  setup_mqtt();
  timeSync(TZ_INFO, "pool.ntp.org", "time.nis.gov");                 //Accurate time is necessary for certificate validation and writing in batches

}

void setup_mqtt() {
  client.setServer(mqtt_server, mqtt_port);
  client.setCallback(callback);
}

void TurnOnLight() {
  digitalWrite(LED_PIN, HIGH);
}

void TurnOffLight() {
  digitalWrite(LED_PIN, LOW);
}

void reconnect() {
  // Loop until we're reconnected
  while (!client.connected()) {
    Serial.print("Attempting MQTT connection...");
    long r = random(1000);
    sprintf(clientId, "clientId-%ld", r);
    if (client.connect(clientId)) {
      Serial.println("connected");
      // Once connected, publish an announcement...
      String message = String("DHT22-SENSOR,device=" + String(DEVICE) + " connected=1");
      message.toCharArray(msg, message.length() + 1);
      client.publish("DHT22-SENSOR/1", msg);
      // ... and resubscribe
      client.subscribe("DHT22-SENSOR/1");
    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      // Wait 5 seconds before retrying
      delay(5000);
      setup_mqtt();
    }
  }
}

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


void setup() {
  // put your setup code here, to run once:
  Serial.begin(115200);
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
  Serial.println("Hello, ESP32!");

  setup_wifi();

  pinMode(LED_PIN, OUTPUT);
}

void loop() {
  TempAndHumidity  data = dhtSensor.getTempAndHumidity();

  float t = data.temperature;
  float h = data.humidity;

  if (wifiMulti.run() != WL_CONNECTED)                               //Check WiFi connection and reconnect if needed
    Serial.println("Wifi connection lost");


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

  if (t > 50 || h > 50) {            // check if the input is HIGH
    TurnOnLight();  // turn LED ON
    Serial.println("Temperature or Humidity detected!");

  } else {
    TurnOffLight();
    Serial.println("Temperature or Humidity UnDetected!");
  }

  String message = String("DHT22-SENSOR,device=" + String(DEVICE) + " temperature=" + String(t, 2) + " " + " humidity=" + String(h, 1) + " ");
  message.toCharArray(msg, message.length() + 1);
  Serial.println(msg);
  client.publish("DHT22-SENSOR/1", msg);

  Serial.println("Wait 5s");
  delay(5000);
}