#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <DHT.h>
#include <ESP32Servo.h>
#include <WiFi.h>
#include <PubSubClient.h>

// ─── WiFi ───────────────────────────────────────────
const char* ssid     = "Wokwi-GUEST";
const char* password = "";

// ─── MQTT ───────────────────────────────────────────
const char* mqtt_server = "broker.hivemq.com";
const int   mqtt_port   = 1883;
const char* mqtt_client = "esp32-bodega-001";

const char* topic_temp  = "bodega/temperatura";
const char* topic_hum   = "bodega/humedad";
const char* topic_servo = "bodega/servo/cmd";  // 👈 topic para recibir comandos

WiFiClient   espClient;
PubSubClient client(espClient);

// ─── OLED ───────────────────────────────────────────
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET    -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// ─── DHT22 ──────────────────────────────────────────
#define DHTPIN   18
#define DHTTYPE  DHT22
DHT dht(DHTPIN, DHTTYPE);

// ─── SERVO ──────────────────────────────────────────
Servo miServo;
#define SERVO_PIN 13
bool servoOn = false;  // estado del servo

// ─── Callback MQTT (recibe comandos) ────────────────
void callback(char* topic, byte* payload, unsigned int length) {
  String mensaje = "";
  for (int i = 0; i < length; i++) {
    mensaje += (char)payload[i];
  }
  Serial.println("Mensaje recibido: " + mensaje);

  if (String(topic) == "bodega/servo/cmd") {
    if (mensaje == "ON") {
      miServo.write(90);
      servoOn = true;
      Serial.println("Servo → ON");
    } else if (mensaje == "OFF") {
      miServo.write(0);
      servoOn = false;
      Serial.println("Servo → OFF");
    }
  }
}

// ────────────────────────────────────────────────────
void conectarWiFi() {
  Serial.print("Conectando a WiFi");
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("\nWiFi OK. IP: " + WiFi.localIP().toString());
}

void reconectarMQTT() {
  while (!client.connected()) {
    Serial.print("Conectando a HiveMQ...");
    if (client.connect(mqtt_client)) {
      Serial.println("conectado.");
      client.subscribe("bodega/servo/cmd");  // 👈 se suscribe al topic de comandos
    } else {
      Serial.print("Error rc=");
      Serial.print(client.state());
      Serial.println(" reintentando en 3s");
      delay(3000);
    }
  }
}

// ─── Setup ──────────────────────────────────────────
void setup() {
  Serial.begin(115200);

  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 failed"));
    for(;;);
  }
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("Iniciando...");
  display.display();

  dht.begin();

  miServo.attach(SERVO_PIN);
  miServo.write(0);

  conectarWiFi();
  client.setServer(mqtt_server, mqtt_port);
  client.setCallback(callback);  // 👈 registra el callback

  display.clearDisplay();
  display.setCursor(0, 0);
  display.println("WiFi + MQTT OK");
  display.display();
  delay(1000);
}

// ─── Loop ───────────────────────────────────────────
void loop() {
  if (!client.connected()) reconectarMQTT();
  client.loop();

  float h = dht.readHumidity();
  float t = dht.readTemperature();

  if (isnan(h) || isnan(t)) {
    Serial.println(F("Error leyendo DHT"));
    return;
  }

  char bufTemp[8], bufHum[8];
  dtostrf(t, 4, 2, bufTemp);
  dtostrf(h, 4, 2, bufHum);

  client.publish(topic_temp, bufTemp);
  client.publish(topic_hum,  bufHum);
  Serial.printf("→ Temp: %s C | Hum: %s %%\n", bufTemp, bufHum);

  // ── OLED ──
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.print("Temp: "); display.print(t); display.println(" C");
  display.setCursor(0, 12);
  display.print("Hum:  "); display.print(h); display.println(" %");
  display.setCursor(0, 24);
  display.println(servoOn ? "Servo: ON" : "Servo: OFF");
  display.setCursor(0, 36);
  display.println("MQTT: HiveMQ OK");
  display.display();

  delay(2000);
}