#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <DHT.h>
#include <ESP32Servo.h>

#define OLED_RESET 4
Adafruit_SSD1306 display(OLED_RESET);

#define DHTPIN 25
#define DHTTYPE DHT22
DHT dht(DHTPIN, DHTTYPE);

#define TRIG_PIN 18
#define ECHO_PIN 5
#define SERVO_PIN 17

Servo servo;

void setup() {
  Serial.begin(9600);
  dht.begin();
  
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);
  
  servo.attach(SERVO_PIN);
  servo.write(90); // Inisialisasi servo ke posisi tengah
  
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  display.display();
  delay(2000);
  display.clearDisplay();
}

void loop() {
  float humidity = dht.readHumidity();
  float temperature = dht.readTemperature();

  if (isnan(humidity) || isnan(temperature)) {
    Serial.println(F("Failed to read from DHT sensor!"));
    return;
  }

  float distance = getDistance();

  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  
  // Menampilkan kondisi cuaca berdasarkan suhu dan kelembaban
  if (temperature >= 28 && temperature <= 40 && humidity >= 10 && humidity <= 30) {
    display.println("Cuaca: Cerah");
  } else if (temperature >= 22 && temperature <= 30 && humidity >= 35 && humidity <= 60) {
    display.println("Cuaca: Mendung");
  } else if (temperature >= 20 && temperature <= 24 && humidity >= 25 && humidity <= 40) {
    display.println("Cuaca: Gerimis");
  } else if (temperature >= 16 && temperature <= 21 && humidity >= 55 && humidity <= 80) {
    display.println("Cuaca: Hujan Deras");
  } else {
    display.println("Cuaca: Tidak Diketahui");
  }

  display.display();
  
  // Control servo based on distance
  if (distance < 20) {
    servo.write(0); // Menutup
  } else if (distance >= 30 && distance <= 40) {
    servo.write(90); // Setengah terbuka
  } else {
    servo.write(180); // Terbuka penuh
  }

  delay(2000); // Delay for stability
}

float getDistance() {
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);
  
  long duration = pulseIn(ECHO_PIN, HIGH);
  float distance = duration * 0.034 / 2; // Calculate distance in cm
  return distance;
}