#define BLYNK_TEMPLATE_ID "TMPL3jkW8j8L9"
#define BLYNK_TEMPLATE_NAME "smartwastemangementsystem3"
#define BLYNK_AUTH_TOKEN "p2owAzqmaAsfrrMSCuqykEEDXr4YSZQ6"
#include <WiFi.h>              // Library for WiFi
#include <PubSubClient.h>      // Library for MQTT
#include <LiquidCrystal_I2C.h> // Library for I2C LCD

#define TRIG_PIN 12            // HC-SR04 Trigger Pin
#define ECHO_PIN 13            // HC-SR04 Echo Pin
#define PIR_PIN 14             // PIR Sensor Pin
#define RED_LED_PIN 27         // Red LED Pin
#define YELLOW_LED_PIN 26      // Yellow LED Pin
#define GREEN_LED_PIN 25       // Green LED Pin
#define BLUE_LED_PIN 33        // Blue LED Pin

LiquidCrystal_I2C lcd(0x27, 16, 2); // Set the LCD address to 0x27 for a 16 chars and 2-line display

const char* ssid = "WOKWI-GUEST";        // Your WiFi SSID
const char* password = ""; // Your WiFi Password
const char* mqtt_server = "broker.hivemq.com"; // MQTT broker

WiFiClient espClient;
PubSubClient client(espClient);

void setup() {
  Serial.begin(115200);
  
  lcd.begin();
  lcd.backlight();
  lcd.setCursor(0, 0);
  lcd.print("Smart Waste");
  lcd.setCursor(0, 1);
  lcd.print("Management");

  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);
  pinMode(PIR_PIN, INPUT);
  pinMode(RED_LED_PIN, OUTPUT);
  pinMode(YELLOW_LED_PIN, OUTPUT);
  pinMode(GREEN_LED_PIN, OUTPUT);
  pinMode(BLUE_LED_PIN, OUTPUT);

  setup_wifi();
  client.setServer(mqtt_server, 1883);
}

void setup_wifi() {
  delay(10);
  Serial.println();
  Serial.print("Connecting to ");
  Serial.println(ssid);

  WiFi.begin(ssid, password);

  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }

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

void reconnect() {
  while (!client.connected()) {
    Serial.print("Attempting MQTT connection...");
    if (client.connect("ESP32Client")) {
      Serial.println("connected");
    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      delay(5000);
    }
  }
}

long readDistance() {
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);

  long duration = pulseIn(ECHO_PIN, HIGH);
  long distance = duration * 0.034 / 2;

  return distance;
}

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

  long distance = readDistance();
  bool motionDetected = digitalRead(PIR_PIN);

  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Distance: ");
  lcd.print(distance);
  lcd.print(" cm");

  if (motionDetected) {
    lcd.setCursor(0, 1);
    lcd.print("Motion Detected");
    digitalWrite(BLUE_LED_PIN, HIGH);
  } else {
    digitalWrite(BLUE_LED_PIN, LOW);
  }

  if (distance < 10) {
    digitalWrite(RED_LED_PIN, HIGH);
    digitalWrite(YELLOW_LED_PIN, LOW);
    digitalWrite(GREEN_LED_PIN, LOW);
    client.publish("waste/bin", "Full");
  } else if (distance < 20) {
    digitalWrite(RED_LED_PIN, LOW);
    digitalWrite(YELLOW_LED_PIN, HIGH);
    digitalWrite(GREEN_LED_PIN, LOW);
    client.publish("waste/bin", "Half Full");
  } else {
    digitalWrite(RED_LED_PIN, LOW);
    digitalWrite(YELLOW_LED_PIN, LOW);
    digitalWrite(GREEN_LED_PIN, HIGH);
    client.publish("waste/bin", "Empty");
  }

  delay(2000); // Update every 2 seconds
}