#include <WiFi.h>
#include <ThingSpeak.h>
#include <DHTesp.h>

char ssid[] = "Wokwi-GUEST";
char pass[] = "";
WiFiClient client;

long myChannelNumber = 2684781;
const char * myWriteAPIKey = "LQUWUY1SJ62WWR6Z";
int statusCode;

DHTesp gayathri;
TempAndHumidity data;
const int trigger = 13;
const int echo = 12;

void setup() {
  Serial.begin(115200);
  Serial.println("Hello, ESP32!");
  
  // Initialize DHT
  gayathri.setup(18, DHTesp::DHT22);
  
  // Initialize pins
  pinMode(2, OUTPUT);
  pinMode(4, OUTPUT);
  pinMode(17, OUTPUT);
  pinMode(trigger, OUTPUT);
  pinMode(echo, INPUT);

  // Connect to WiFi
  WiFi.mode(WIFI_STA);
  WiFi.begin(ssid, pass);
  
  // Wait for connection
  while (WiFi.status() != WL_CONNECTED) {
    Serial.print(".");
    delay(500);
  }
  Serial.println("Connected to WiFi");

  // Initialize ThingSpeak
  ThingSpeak.begin(client);
}

int finding_distance() {
  digitalWrite(trigger, HIGH);
  delay(10); // Adjust this delay as necessary
  digitalWrite(trigger, LOW);
  float a = pulseIn(echo, HIGH);
  return (a * 0.034 / 2);
}

void loop() {
  // Read temperature and humidity
  data = gayathri.getTempAndHumidity();
  int distance = finding_distance();

  if (data.temperature > 70 && data.humidity < 30 && distance < 200) {
    digitalWrite(2, HIGH);
    digitalWrite(4, LOW);
    digitalWrite(17, LOW);
    Serial.println("You are in danger");
  } else if (data.temperature >= 30 && data.temperature <= 70 &&
             data.humidity >= 30 && data.humidity <= 70 && distance <= 300 && distance >= 200) {
    digitalWrite(2, LOW);
    digitalWrite(4, HIGH);
    digitalWrite(17, LOW);
    Serial.println("You are unsafe");
  } else if (data.temperature < 30 && data.humidity > 70 && distance > 300) {
    digitalWrite(2, LOW);
    digitalWrite(4, LOW);
    digitalWrite(17, HIGH);
    Serial.println("You are safe");
  } else {
    digitalWrite(2, LOW);
    digitalWrite(4, LOW);
    digitalWrite(17, LOW);
  }

  Serial.print("Humidity: ");
  Serial.println(data.humidity);
  Serial.print("Temperature: ");
  Serial.println(data.temperature);
  Serial.print("Distance: ");
  Serial.println(distance);
  Serial.println("-----------------");

  // Update ThingSpeak
  ThingSpeak.setField(3, data.humidity);
  ThingSpeak.setField(2, data.temperature);
  ThingSpeak.setField(1, distance);

  statusCode = ThingSpeak.writeFields(myChannelNumber, myWriteAPIKey);

  if (statusCode == 200) {
    Serial.println("Channel update successful.");
  } else {
    Serial.println("Problem Writing data. HTTP error code: " + String(statusCode));
  }

  delay(10000); // Delay before the next loop iteration
}