#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <ESP32Servo.h>
#include <HX711.h>
#include <WiFi.h>
#include <PubSubClient.h>
#include "RTClib.h"

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define TRIG_PIN 13
#define ECHO_PIN 12
#define BUZZER_PIN 5
#define BUTTON_PIN 14
#define SERVO_PIN 15
#define LOADCELL_DOUT_PIN 18
#define LOADCELL_SCK_PIN 19

const char* ssid = "YOUR_WIFI_SSID";
const char* password = "YOUR_WIFI_PASSWORD";
const char* mqttServer = "broker.netpie.io";
const char* clientID = "YOUR_CLIENT_ID";
const char* token = "YOUR_TOKEN";
const char* secret = "YOUR_SECRET";

WiFiClient espClient;
PubSubClient client(espClient);
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);
HX711 scale;
RTC_DS1307 rtc;
Servo feederServo;
bool isFeeding = false;

void setup() {
    Serial.begin(115200);
    pinMode(TRIG_PIN, OUTPUT);
    pinMode(ECHO_PIN, INPUT);
    pinMode(BUZZER_PIN, OUTPUT);
    pinMode(BUTTON_PIN, INPUT_PULLUP);
    feederServo.attach(SERVO_PIN);
    scale.begin(LOADCELL_DOUT_PIN, LOADCELL_SCK_PIN);

    if (!rtc.begin()) {
        Serial.println("RTC ไม่ทำงาน!");
    }
    rtc.adjust(DateTime(F(__DATE__), F(__TIME__)));

    if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
        Serial.println("SSD1306 ไม่พบ!");
        for (;;);
    }
    display.clearDisplay();

    connectWiFi();
    client.setServer(mqttServer, 1883);
    client.setCallback(mqttCallback);
}

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

    checkButton();
    measureFoodLevel();
    displayData();
}

void connectWiFi() {
    WiFi.begin(ssid, password);
    while (WiFi.status() != WL_CONNECTED) {
        delay(500);
        Serial.print(".");
    }
    Serial.println("WiFi Connected!");
}

void reconnectMQTT() {
    while (!client.connected()) {
        if (client.connect(clientID, token, secret)) {
            client.subscribe("feed/control");
        } else {
            delay(5000);
        }
    }
}

void mqttCallback(char* topic, byte* payload, unsigned int length) {
    String msg;
    for (int i = 0; i < length; i++) {
        msg += (char)payload[i];
    }
    if (msg == "FEED") {
        feedPet();
    }
}

void feedPet() {
    if (!isFeeding) {
        isFeeding = true;
        feederServo.write(90);
        delay(3000);
        feederServo.write(0);
        isFeeding = false;
        tone(BUZZER_PIN, 1000, 500);
    }
}

void checkButton() {
    if (digitalRead(BUTTON_PIN) == LOW) {
        feedPet();
    }
}

void measureFoodLevel() {
    digitalWrite(TRIG_PIN, LOW);
    delayMicroseconds(2);
    digitalWrite(TRIG_PIN, HIGH);
    delayMicroseconds(10);
    digitalWrite(TRIG_PIN, LOW);
    int duration = pulseIn(ECHO_PIN, HIGH);
    int distance = duration * 0.034 / 2;
    Serial.print("ระดับอาหาร: ");
    Serial.print(distance);
    Serial.println(" cm");
}

void displayData() {
    display.clearDisplay();
    display.setTextSize(1);
    display.setTextColor(WHITE);
    display.setCursor(0, 0);
    DateTime now = rtc.now();
    display.printf("%02d:%02d:%02d\n", now.hour(), now.minute(), now.second());
    display.display();
}