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

// OLED 128x64
#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 27
#define DHTTYPE DHT22
DHT dht(DHTPIN, DHTTYPE);

// Calidad de aire
#define AIR_AOUT 34
#define AIR_DOUT 35

// Botón táctil simulado
#define TOUCH_PIN 14

// WiFi y MQTT
const char* ssid = "Wokwi-GUEST";
const char* password = "";
const char* mqtt_server = "broker.hivemq.com";
WiFiClient espClient;
PubSubClient client(espClient);

// Bitmaps de ojos (4 gestos)
const unsigned char eyes_open[] PROGMEM = { /* Poner bitmap */ };
const unsigned char eyes_closed[] PROGMEM = { /* Poner bitmap */ };
const unsigned char eyes_left[] PROGMEM = { /* Poner bitmap */ };
const unsigned char eyes_right[] PROGMEM = { /* Poner bitmap */ };

bool mostrarDatos = false;

void setup() {
  Serial.begin(115200);
  dht.begin();

  pinMode(AIR_DOUT, INPUT);
  pinMode(TOUCH_PIN, INPUT);

  // OLED
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)){
    Serial.println("Error OLED");
    while(true);
  }
  display.clearDisplay();
  display.display();

  // WiFi
  WiFi.begin(ssid, password);
  while(WiFi.status() != WL_CONNECTED){
    delay(500);
    Serial.print(".");
  }
  Serial.println("\nWiFi Conectado");

  // MQTT
  client.setServer(mqtt_server, 1883);
}

void reconnect() {
  while (!client.connected()) {
    Serial.print("Conectando MQTT...");
    if (client.connect("ESP32_Wokwi")) {
      Serial.println("Conectado");
    } else {
      Serial.print("Falló, rc=");
      Serial.print(client.state());
      delay(2000);
    }
  }
}

void publicarDatos(float temp, float hum, int aout, int dout, int boton){
  if (!client.connected()) reconnect();
  client.loop();
  client.publish("casa/esp32/temperatura", String(temp).c_str());
  client.publish("casa/esp32/humedad", String(hum).c_str());
  client.publish("casa/esp32/aire_analog", String(aout).c_str());
  client.publish("casa/esp32/aire_digital", String(dout).c_str());
  client.publish("casa/esp32/boton", String(boton).c_str());
}

void mostrarAnimacion(float temp){
  display.clearDisplay();
  // Cambiar gesto según temperatura
  if(temp < 20){
    display.drawBitmap(0,0, eyes_left, 32,16, SSD1306_WHITE);
  } else if(temp < 25){
    display.drawBitmap(0,0, eyes_open, 32,16, SSD1306_WHITE);
  } else if(temp < 30){
    display.drawBitmap(0,0, eyes_right, 32,16, SSD1306_WHITE);
  } else {
    display.drawBitmap(0,0, eyes_closed, 32,16, SSD1306_WHITE);
  }
  display.display();
}

void mostrarDatosPantalla(float temp, float hum, int aout, int dout){
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  display.println("DATOS SENSOR");
  display.print("Temp: "); display.print(temp); display.println(" C");
  display.print("Hum: "); display.print(hum); display.println(" %");
  display.print("Aire: "); display.print(aout); display.print(" DOUT: "); display.println(dout);
  display.display();
}

void loop() {
  float temp = dht.readTemperature();
  float hum = dht.readHumidity();
  int aout = analogRead(AIR_AOUT);
  int dout = digitalRead(AIR_DOUT);
  int boton = digitalRead(TOUCH_PIN);

  mostrarDatos = boton; // Si presiona, mostrar pantalla de datos

  if(mostrarDatos){
    mostrarDatosPantalla(temp, hum, aout, dout);
  } else {
    mostrarAnimacion(temp);
  }

  publicarDatos(temp, hum, aout, dout, boton);
  delay(1000);
}