from machine import Pin, ADC, SoftI2C
from ssd1306 import SSD1306_I2C
from time import sleep
import network
from umqtt.simple import MQTTClient

# Cliente MQTT Constantes
MQTT_BROKER = "broker.hivemq.com"
MQTT_USER = ""
MQTT_PASSWORD = ""
MQTT_CLIENT_ID = "unique_client_id"  # Debes poner un ID único
MQTT_TOPIC = "utng/wdmr/gas"
MQTT_PORT = 1883

# Método para conectarnos a WIFI
def wifi_connect():
    print("Conectando...", end="")
    sta_if = network.WLAN(network.STA_IF)
    sta_if.active(True)
    sta_if.connect('Wokwi-GUEST', '')
    while not sta_if.isconnected():
        print(".", end="")
        sleep(0.3)
    print(" Conectada")

# Método para publicar a un tópico
def publish(sensor_value):
    client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, port=MQTT_PORT, user=MQTT_USER, password=MQTT_PASSWORD, keepalive=0)
    client.connect()
    print("[OK]")
    client.publish(MQTT_TOPIC, str(sensor_value))
    print("Se publicó el valor a %s en el tópico %s" % (MQTT_BROKER, MQTT_TOPIC))
    client.disconnect()

# ASIGNACIÓN DE PINES ESP32
i2c = SoftI2C(scl=Pin(21), sda=Pin(22))
oled_width = 128
oled_height = 64

oled = SSD1306_I2C(oled_width, oled_height, i2c)

# VARIABLES PARA LOS LEDS Y SENSOR
led_verde = Pin(2, Pin.OUT)
led_rojo = Pin(13, Pin.OUT)

sensor = ADC(Pin(32))
# AJUSTAR LA RESOLUCIÓN Y EL VOLTAJE
sensor.width(ADC.WIDTH_10BIT)  # Regular la precisión del sensor
sensor.atten(ADC.ATTN_11DB)  # Trabajando con el 3.3V

# Intentando conectar wifi
wifi_connect()

# Ciclo infinito
while True:
    lectura = sensor.read()
    # Publicamos el valor del sensor
    publish(lectura)
    # Limpiar pantalla
    oled.fill(0)

    if lectura > 200:
        led_rojo.value(1)
        oled.text('Alerta CO High', 0, 0)
        led_verde.value(0)
    else:
        led_verde.value(1)
        oled.text('CO normal', 0, 0)
        led_rojo.value(0)

    oled.show()
    sleep(1)