#define ADC_IN A5
#define ADC_IN2 A4
#define ADC_REF 3.3
#define Buzzer 4
#include <LiquidCrystal_I2C.h>
LiquidCrystal_I2C lcd(0x27,20,4);

#include <WiFi.h>
#include <WiFiClient.h>
#include <PubSubClient.h>

int interval = 5000;
long previousMillis = 0;
int IntervalR = 3000;    // Interval at which to toggle LED3 (milliseconds)
long previousMillisR = 0;  
bool ledStateR = LOW;

const int PIRinput = 17;  
int pirState = LOW;

const char* ssid = "Wokwi-GUEST";
const char* password = "";

const char* hostname = "broker.hivemq.com";

WiFiClient espClient;
PubSubClient client(espClient);
const char* espClientName = "esp32Client_10251626"; // yourStudentID must be unique
int PORTNUM = 1883;

void setup_wifi()
{
  WiFi.begin(ssid, password);
  while(WiFi.status() != WL_CONNECTED )
  {
    delay(1000);
    Serial.print(".");
  }
  Serial.println();
  Serial.print("Connected to ");
  Serial.println(ssid);
  Serial.print("Given IP by the router to ESP32 is ");
  Serial.println(WiFi.localIP());
}

void connectMQTT()
{
  while(!client.connected() )
  {
    Serial.println("Connecting to MQTT ...");
    if (client.connect(espClientName) ) //, mqttUser, mqttPassword) )
    {
      Serial.println("Connected");
      MQTTSubscribe();
    }
    else
    {
      Serial.print("Failed with state ");
      Serial.print(client.state() );
      delay(2000);
    }
  }
}

void callback(String topic, byte* payload, unsigned int length)
{
  String messageTemp;
  Serial.print("Message received in topic: ");
  Serial.print(topic);
  Serial.print(" length is: ");
  Serial.println(length);

  Serial.print("Data received from broker: ");
  for (int i = 0; i<length; i++)
  {
    Serial.print( (char)payload[i] );
    messageTemp += (char)payload[i];
  }

}

void MQTTSubscribe() 
{
 client.subscribe("JCE/IoT/1");
 client.subscribe("JCE/IoT/2");
 client.subscribe("JCE/IoT/3");
 client.subscribe("JCE/IoT/4");
 client.subscribe("JCE/IoT/5");
 client.subscribe("JCE/IoT/6");
 client.subscribe("JCE/IoT/7");
}

void setup_MQTT() 
{
  client.setServer(hostname, PORTNUM);
  client.setCallback(callback); 
}


void setup()
{
 lcd.init();
 lcd.backlight();
 lcd.setCursor(1,0);
 lcd.print("Hello, Smart Human");
 delay(5000);

 pinMode(PIRinput, INPUT);
 pinMode(23, OUTPUT); 
 pinMode(18, OUTPUT);
 pinMode(19, OUTPUT);
 pinMode(26, OUTPUT);

 Serial.begin(115200);
 pinMode(Buzzer, OUTPUT);

 setup_wifi();
 setup_MQTT();
}

void loop()
{
  int adcRes = analogRead( ADC_IN );
  delay(10);
  

  //when someone is detected from a distance, it will show whether the bin is full or not
  if (analogRead(ADC_IN) <= 1365) 
  {
    digitalWrite(23, HIGH); 
    digitalWrite(18, LOW);
    digitalWrite(19, HIGH);

    lcd.clear();
    lcd.setCursor(0,1);
    lcd.print("Hungry!Need rubbish!");
    delay(1000);

    String payload = String(adcRes);
    delay(500);
    client.publish("JCE/IoT/1", payload.c_str());
    client.publish("JCE/IoT/2", "Off");

  }
  else
  {
    digitalWrite(23, HIGH);
    digitalWrite(18, HIGH);
    digitalWrite(19, HIGH);
  }

  if (analogRead(ADC_IN) <= 2730)
  if (analogRead(ADC_IN) > 1365)
  {
    digitalWrite(23, HIGH); 
    digitalWrite(18, HIGH);
    digitalWrite(19, LOW);

    lcd.clear();
    lcd.setCursor(0,1);
    lcd.print("I'm still hungry!:)");
    delay(1000);

    String payload = String(adcRes);
    delay(500);
    client.publish("JCE/IoT/1", payload.c_str());
    client.publish("JCE/IoT/2", "Off");
    
  }
  else
  {
    digitalWrite(23, HIGH);
    digitalWrite(18, HIGH);
    digitalWrite(19, HIGH);
  }

  if (analogRead(ADC_IN) > 2730)
  if (analogRead(ADC_IN) <= 4095)
  {
    digitalWrite(23, LOW); 
    digitalWrite(18, HIGH);
    digitalWrite(19, HIGH);
  
    lcd.clear();
    lcd.setCursor(0,1);
    lcd.print("Full! Please clear!");
    delay(1000);
    digitalWrite(Buzzer, HIGH);
    delay(300);
    digitalWrite(Buzzer, LOW);

    String payload = String(adcRes);
    delay(500);
    client.publish("JCE/IoT/1", payload.c_str());
    client.publish("JCE/IoT/2", "On");
  }
  else
  {
    digitalWrite(23, HIGH);
    digitalWrite(18, HIGH);
    digitalWrite(19, HIGH);
  }


  //To show when the rubbish thrown is hot or not
  int adcRes2 = analogRead( ADC_IN2 );
  if (analogRead(ADC_IN2) <= 2047) //represents temperature between 0 to 50 degrees Celcius
  {
    digitalWrite(26, HIGH); 
    lcd.clear();
    String payload2 = String(adcRes2);
    client.publish("JCE/IoT/4", payload2.c_str());
    client.publish("JCE/IoT/3","------");
    client.publish("JCE/IoT/5", "Off");
    
  }
  else
  {
      digitalWrite(26, LOW);
  }
  if (analogRead(ADC_IN2) > 2047)
  if (analogRead(ADC_IN2) <= 4095)//50 to 100
    {
      digitalWrite(26, LOW);

      lcd.clear();
      lcd.setCursor(2,1);
      lcd.print("Bin is too hot!!");
      lcd.setCursor(7,2);
      lcd.print("Ah!!!!");
      delay(500);
      String payload2 = String(adcRes2);
      client.publish("JCE/IoT/3","Bin is too hot!");
      client.publish("JCE/IoT/4", payload2.c_str());
      client.publish("JCE/IoT/5", "On");

      digitalWrite(Buzzer, HIGH);
      delay(3000);
      digitalWrite(Buzzer, LOW);
    }
  else
    {
    digitalWrite(26, HIGH);
    }
  digitalWrite(26, HIGH);

  //motion sensor
  const int IP = digitalRead(17);
  delay(100);
  long currentMillis = millis();
  if (currentMillis - previousMillis >= interval)
  {
    previousMillis = currentMillis;
    if (IP == 1)  // If motion is detected by the PIR sensor
    {
      if (pirState == LOW)  // Update pirState to indicate motion detection
        {
          Serial.println("Bin is opening!");
          lcd.clear();
          lcd.setCursor(3,3);
          lcd.print("Bin is opening");
          String payload = String(IP);
  	      client.publish("JCE/IoT/6", "On");
          client.publish("JCE/IoT/7", "Off");
          delay(5000);
          pirState = HIGH;
        }   
      else 
        {
          if (pirState == HIGH)  // Update pirState to indicate motion detection
            {
              Serial.println("Bin is closing...");
              lcd.clear();
              lcd.setCursor(3,3);
              lcd.print("Bin is closing");
              pirState = LOW;
              delay(3000);
              Serial.println("Bin is closed.");
              lcd.clear();
              lcd.setCursor(2,3);
              lcd.print("Bin is closed...");
              delay(3000);
              String payload = String(IP);
  	          client.publish("JCE/IoT/7", "On");
              client.publish("JCE/IoT/6", "Off");
              delay(3000);
              digitalWrite(Buzzer, HIGH); //buzzer make sound
              digitalWrite(Buzzer, LOW);
            }

        }
      }
  }

  if ( !client.connected() )
  {
    connectMQTT();
  }
  client.loop();
}