# Módulos para trabajar con OLED y el sensor
from machine import Pin, ADC, SoftI2C
# Crear instancias de la pantalla OLED
from ssd1306 import SSD1306_I2C
from time import sleep
# Conexión con wifi, manejar instancias de un cliente MQTT
import network
from umqtt.simple import MQTTClient

# Constantes para conexión con el cliente MQTT
MQTT_CLIENT_ID = ""
MQTT_BROKER = "broker.hivemq.com"
MQTT_USER = ""
MQTT_PASSWORD = ""
MQTT_TOPIC = "jisg/gas"
MQTT_PORT = 1883

# Asignación de pines para la OLED 
i2c = SoftI2C(scl=Pin(22), sda=Pin(21))

# Instanciamos el objeto que controlará la OLED
oled = SSD1306_I2C(128, 64, i2c)

# Variables para los leds
led_verde = Pin(12, Pin.OUT)
led_rojo = Pin(13, Pin.OUT)

# Definición del sensor con el ADC (Convertidor ANALÓGICO-DIGITAL) 
sensor = ADC(Pin(32))

# Ajustes para el sensor
sensor.width(ADC.WIDTH_10BIT)  # Regula la precisión del sensor
sensor.atten(ADC.ATTN_11DB)  # Trabaja sobre 3.3V

def conectar_wifi():
    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.1)
    print("¡Conectado!")

def conectar_broker():
    print("Conectando al broker: ", MQTT_BROKER)
    cliente = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD, port=MQTT_PORT)
    cliente.connect()
    return cliente

# Antes de iniciar el ciclo
conectar_wifi()
cliente = conectar_broker()

# Ciclo infinito
while True:
    # Leer el sensor
    lectura = sensor.read()

    # Limpiar pantalla
    oled.fill(0)
    if lectura > 200:
        oled.text("Alerta CO Alto!", 0, 0)
        led_rojo.value(1)
        led_verde.value(0)
    else:
        led_verde.value(1)
        led_rojo.value(0)
        oled.text("CO Normal", 0, 0)
    oled.show()
    
    # Publicar en el tópico MQTT
    cliente.publish(MQTT_TOPIC, str(lectura))
    print("Publicando en {}. En el tópico {}, el valor {}".format(MQTT_BROKER, MQTT_TOPIC, lectura))
    sleep(10)