#include <WiFi.h>
#include <PubSubClient.h>
#include <Wire.h>
#include <RTClib.h>
#include <ESP32Servo.h>
#include <DHT.h>

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

#define TOKEN "eCCZMhmpE8lfh81AOtWO" 
const char* mqtt_server = "mqtt.thingsboard.cloud";

const int dhtPin = 33;
const int trigPin = 5;
const int echoPin = 18;
const int servoPin = 14;
const int gasSensorPin = 34;

WiFiClient wifiClient;
PubSubClient client(wifiClient);

RTC_DS3231 rtc;
Servo feederServo;
DHT dht(dhtPin, DHT22);

const int servoOpenAngle = 90;
const int servoCloseAngle = 0;
const float containerHeight = 100.0;

long lastMsg = 0;
char attributes[1000];

void setup() {
  Serial.begin(9600);
  delay(10);

  // Connect to Wi-Fi
  Serial.print("Connecting to: ");
  Serial.println(ssid);
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print("...");
  }
  Serial.println("\nWi-Fi connected");
  Serial.print("IP address: ");
  Serial.println(WiFi.localIP());

  client.setServer(mqtt_server, 1883);

  // Initialize the RTC
  if (!rtc.begin()) {
    Serial.println("Couldn't find RTC");
    while (1);
  }

  if (rtc.lostPower()) {
    Serial.println("RTC lost power, setting time!");
    rtc.adjust(DateTime(F(__DATE__), F(__TIME__)));
  }

  // Initialize the sensors
  feederServo.attach(servoPin);
  feederServo.write(servoCloseAngle);
  pinMode(trigPin, OUTPUT);
  pinMode(echoPin, INPUT);
  dht.begin();
}

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

  long now = millis();
  if (now - lastMsg > 5000) {
    lastMsg = now;
    getData();
  }

  DateTime nowTime = rtc.now();
  int currentHour = nowTime.hour();

  if (currentHour == 6 || currentHour == 12 || currentHour == 16 || currentHour == 20) {
    feedChicken();
  }

  client.loop();
}

void getData() {
  DateTime nowTime = rtc.now();
  char timeString[9];
  snprintf(timeString, sizeof(timeString), "%02d:%02d:%02d", nowTime.hour(), nowTime.minute(), nowTime.second());

  float temperature = dht.readTemperature();
  float humidity = dht.readHumidity();

  float distance = measureDistance();
  float feedHeight = containerHeight - distance;

  int gasValue = analogRead(gasSensorPin);

  int servoAngle = feederServo.read();

  String payload = "{";
  payload += "\"Time\":\""; payload += timeString; payload += "\",";
  payload += "\"Temperature\":"; payload += temperature; payload += ",";
  payload += "\"Humidity\":"; payload += humidity; payload += ",";
  payload += "\"Distance\":"; payload += distance; payload += ",";
  payload += "\"FeedHeight\":"; payload += feedHeight; payload += ",";  // Send feed height
  payload += "\"GasValue\":"; payload += gasValue; payload += ",";
  payload += "\"ServoAngle\":"; payload += servoAngle;
  payload += "}";

  payload.toCharArray(attributes, 1000);
  client.publish("v1/devices/me/telemetry", attributes);
  Serial.println("Telemetry data sent:");
  Serial.println(attributes);
}

float measureDistance() {
  digitalWrite(trigPin, LOW);
  delayMicroseconds(2);

  digitalWrite(trigPin, HIGH);
  delayMicroseconds(10);
  digitalWrite(trigPin, LOW);

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

  Serial.print("Distance: ");
  Serial.print(distance);
  Serial.println(" cm");

  return distance;
}

void feedChicken() {
  for (int i = 0; i < 10; i++) {
    feederServo.write(servoOpenAngle);
    delay(5000);
    feederServo.write(servoCloseAngle);
    delay(300000);
  }

  Serial.println("Chicken fed!");
}

void reconnect() {
  while (!client.connected()) {
    Serial.print("Connecting to ThingsBoard...");
    if (client.connect("ESP32_Client", TOKEN, "")) {
      Serial.println("Connected");
    } else {
      Serial.print("Failed, rc=");
      Serial.print(client.state());
      Serial.println(" : retrying in 5 seconds");
      delay(5000);
    }
  }
}