#include <DHT.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <ESP32Servo.h> 
#include <ESP32Firebase.h> 
#include <WiFi.h>
// Thông số Firebase
#define REFERENCE_URL "https://final-iots-14d89-default-rtdb.asia-southeast1.firebasedatabase.app/"

// Thông số màn hình OLED
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET -1
// Thông số DHT22
#define DHTPIN 4
#define DHTTYPE DHT22
// Thông số HRSC4
#define TRIG_PIN 12
#define ECHO_PIN 14
// Thông số LED và Servo
#define RED_LED_PIN 2
#define GREEN_LED_PIN 15
#define SERVO_PIN 19
// Khởi tạo đối tượng
DHT dht(DHTPIN, DHTTYPE);
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);
Firebase firebase(REFERENCE_URL);
Servo myservo;

const char* WIFI_NAME = "Wokwi-GUEST";
const char* WIFI_PASSWORD = ""; 

// Hàm thiết lập
void setup() {
  Serial.begin(115200);
  // Khởi tạo các cảm biến và màn hình
  dht.begin();
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(WHITE);
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);
  pinMode(GREEN_LED_PIN, OUTPUT);
  pinMode(RED_LED_PIN, OUTPUT);
  myservo.attach(SERVO_PIN); myservo.write(0);
  WiFi.begin(WIFI_NAME, WIFI_PASSWORD);   
  while (WiFi.status() != WL_CONNECTED)    
  {
    delay(1000);
    Serial.println("Wifi not connected"); 
  }
  Serial.println("Wifi connected !");    
  Serial.println("Local IP: " + String(WiFi.localIP())); 
  WiFi.mode(WIFI_STA); 
}
// Chương trình chớp tắt led
void blink_led_red(){
  digitalWrite(RED_LED_PIN, HIGH);  delay(1000);
  digitalWrite(RED_LED_PIN, LOW);   delay(1000);
}

// Hàm đo khoảng cách bằng HRSC4
long measureDistance() {
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);
  long duration = pulseIn(ECHO_PIN, HIGH);
  return duration * 0.034 / 2;
}

// Hàm loop
void loop() {
  // Đọc dữ liệu từ cảm biến
  float humidity = dht.readHumidity();
  float temperature = dht.readTemperature();
  long distance = measureDistance();
  // Hiển thị dữ liệu lên OLED
  display.setCursor(0, 0);
  display.println("Ten: Nguyen Van Nhat");
  display.println("MSSV: 21161343");
  display.print("Nhiet do: ");
  display.print(temperature);
  display.println(" C");
  display.print("Do am: ");
  display.print(humidity);
  display.println(" %");
  display.print("Khoang cach: ");
  display.print(distance);
  display.println(" cm");
  display.display();
  // Hiện thị dữ liệu trên wokwi
  Serial.println("-------------------");
  Serial.print("Temperature: ");
  Serial.print(temperature);
  Serial.println(" °C");
  Serial.print("Humidity: ");
  Serial.print(humidity);
  Serial.println(" %");
  Serial.print("Distance: ");
  Serial.print(distance);
  Serial.println(" cm");
  // Điều khiển LED và Servo
  if (temperature > 50) {
    blink_led_red();
  } else {
    digitalWrite(RED_LED_PIN, LOW);
  }
  if (distance > 80) {
    myservo.write(90);
    digitalWrite(GREEN_LED_PIN, HIGH);
  } else {
    digitalWrite(GREEN_LED_PIN, LOW);
    myservo.write(0);
  }
  // Gửi dữ liệu lên Firebase
  firebase.setFloat("Temperature", temperature); 
  firebase.setFloat("Humidity", humidity); 
  firebase.setFloat("Distance", distance); 
}