#include "DHTesp.h"
#include <WiFi.h>
#include "PubSubClient.h"
#include <LiquidCrystal_I2C.h>
//#include "MQ135.h"

const char* ssid = "Wokwi-GUEST";
const char* password = "";
const char* mqttServer = "test.mosquitto.org";
const int port = 1883;

const int LED_PIN_1 = 15;
const int LED_PIN_2 = 2;
const int LED_PIN_3 = 4;
const int LED_PIN_4 = 5;
const int DHT22_PIN = 18;
const int BUZZER_PIN = 19;
const int mq135_simulator = 32; 


DHTesp dhtSensor;
WiFiClient espClient;
PubSubClient client(espClient);
LiquidCrystal_I2C LCD = LiquidCrystal_I2C(0x27, 20, 4);


// decorations while connecting to wifi.
void spinner() 
{
  static int8_t counter = 0;
  const char* glyphs = "\xa1\xa5\xdb";
  LCD.setCursor(15, 1);
  LCD.print(glyphs[counter++]);
  if (counter == strlen(glyphs)) {
    counter = 0;
  }
}
// connect to wifi. 
void wifiConnect() 
{
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    spinner();
    Serial.print(".");
  }
  Serial.println(" Connected");
}
// activate BUZZER_PIN on sin value
// void activate_BUZZER_PIN()
// {
//   for(int x = 0; x < 180; x++)
//   {
//     float sinVal = sin(x*(3.1412/180));
//     int toneVal = 2000 + int(sinVal * 1000);
//     tone(BUZZER_PIN, toneVal);
//     delay(2);
//   }
// }
// deactivate BUZZER_PIN.
// initialize LCD
void LCD_init()
{
  LCD.init();
  LCD.backlight();
  LCD.setCursor(0, 0);
  LCD.print("Connecting to ");
  LCD.setCursor(0, 1);
  LCD.print("WiFi ");
}
// print local IP on LCD after connected.
void LCD_show_localIP()
{
  LCD.clear();
  LCD.setCursor(0, 0);
  LCD.print("Online");
  LCD.setCursor(0, 1);
  LCD.print("Local IP: ");
  LCD.print(WiFi.localIP());
  LCD.setCursor(0, 2);
  LCD.print("Getting data in ");
  LCD.print("3");
  delay(1000);
  LCD.setCursor(16, 2);
  LCD.print("2");
  delay(1000);
  LCD.setCursor(16, 2);
  LCD.print("1");
  delay(1000);
  LCD.clear();
}
// Print data on LCD
void LCD_handle(int temperature, int humidity, int mq135_value)
{
  LCD.setCursor(0, 0);
  LCD.print("Current data: ");
  LCD.setCursor(0, 1);
  LCD.print("Temperature: ");
  LCD.print(String(temperature) + "oC");
  LCD.setCursor(0, 2);
  LCD.print("Humidity: ");
  LCD.print(String(humidity) + "%");
  LCD.setCursor(0, 3);
  LCD.print("MQ135: ");
  LCD.print(String(mq135_value) + "PPM");
}

void Turn_On_All_Led()
{
  digitalWrite(LED_PIN_1, HIGH);
  digitalWrite(LED_PIN_2, HIGH);
  digitalWrite(LED_PIN_3, HIGH);
  digitalWrite(LED_PIN_4, HIGH);
}

void Turn_Off_All_Led()
{
  digitalWrite(LED_PIN_1, LOW);
  digitalWrite(LED_PIN_2, LOW);
  digitalWrite(LED_PIN_3, LOW);
  digitalWrite(LED_PIN_4, LOW);
}

void Turn_On_Warning_Led()
{
  digitalWrite(LED_PIN_1, LOW);
  digitalWrite(LED_PIN_2, LOW);
  digitalWrite(LED_PIN_3, HIGH);
  digitalWrite(LED_PIN_4, HIGH);
}

float Read_ppm_data()
{
  int value = analogRead(mq135_simulator);

  int to_5V_value = map(value, 0, 4095, 0, 1023);

  float to_ppm = to_5V_value * 1200 / 1023;

  return to_ppm;
}

void Warning(float temp, float humidity, float ppm)
{
  if (temp > 40 && temp <= 70 || ppm > 800 && ppm <= 1000) //cảnh báo cháy
  {
    Turn_On_Warning_Led();
    tone(BUZZER_PIN, 262, 250);
  }

  else if (temp > 70 || ppm > 1000 && ppm <= 1200) //phát sinh cháy
  {
    Turn_On_All_Led();
    tone(BUZZER_PIN, 523);
  }

  else
  {
    Turn_Off_All_Led();
    noTone(BUZZER_PIN);
  }

  Serial.println("Temp: " + String(temp, 2) + "°C");
  Serial.println("Humidity: " + String(humidity, 1) + "%");
  Serial.println("PPM In Air: " + String(ppm, 2) + "ppm");
  Serial.println("--------");
}

void mqttReconnect()
{
  while (!client.connected())
  {
    Serial.println("Attempting MQTT reconnection...");
    if (client.connect("21127147_21127365_21127644"))
    {
      Serial.println("Connected to your device!");
    }
    else
    {
      Serial.println("Try again in 5 seconds...");
      delay(5000);
    }
  }
}

void setup() {
  // put your setup code here, to run once:
  Serial.begin(115200);
  // pinMode(DHT22_PIN, INPUT);
  dhtSensor.setup(DHT22_PIN, DHTesp::DHT22);
  pinMode(LED_PIN_1, OUTPUT);
  pinMode(LED_PIN_2, OUTPUT);
  pinMode(LED_PIN_3, OUTPUT);
  pinMode(LED_PIN_4, OUTPUT);
  pinMode(BUZZER_PIN, OUTPUT);
  pinMode(mq135_simulator, INPUT);

  LCD_init();
  WiFi.begin(ssid, password);
  wifiConnect();
  LCD_show_localIP();

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

  client.setServer(mqttServer, port);

}


void loop() {
  // put your main code here, to run repeatedly:
  if (!client.connected())
  {
    mqttReconnect();
  }
  client.loop();

  TempAndHumidity data = dhtSensor.getTempAndHumidity();
  float ppm = Read_ppm_data();
  LCD_handle(data.temperature, data.humidity, ppm);

  //Warning(data.temperature, data.humidity, ppm);

  float temperature = data.temperature;
  float humidity = data.humidity;

  char buffer[50];
  sprintf(buffer, "{\"temperature\": %f, \"humidity\": %f, \"ppm\": %f}", temperature, humidity, ppm);

  client.publish("21127147_21127365_21127644/Data", buffer);
  delay(5000);
}