from machine import Pin
from machine import SoftI2C
import ssd1306
import time
import dht
import network
import ujson
from umqtt.simple import MQTTClient
import utime as time
from machine import RTC

import ssl


# read in DER formatted certs & user key
with open("cliente2_key.txt", "rb") as f:
    key_data = f.read()
with open("cliente2_pem.txt", "rb") as f:
    cert_data = f.read()
with open("root_ca.txt", "rb") as f:
    ca_data = f.read()

print(key_data)
print(cert_data)
print(ca_data)




#Parametrizar el servidor MQTT.
MQTT_CLIENT_ID = "cliente2"
#MQTT_BROKER = "66.29.142.107"
MQTT_BROKER = "mqttbrokereventgrid.eastus2-1.ts.eventgrid.azure.net"
MQTT_PORT = 8883                    #Puerto por defecto sin SSL
MQTT_USER = "cliente2"
MQTT_PASSWORD = "123"

# Tópicos MQTT
#MQTT_TOPIC_SUB = "variables/rele"
#MQTT_TOPIC_PUB = "variables/temphum"
MQTT_TOPIC_SUB = "telemetry/rele"
MQTT_TOPIC_PUB = "telemetry/temphum"

# & put them in a dict
ssl_params = dict()
ssl_params["certfile"] = cert_data
ssl_params["keyfile"] = key_data
#ssl_params["cadata"] = ca_data
ssl_params["server_hostname"] = MQTT_BROKER
ssl_params["cert_reqs"] = ssl.CERT_REQUIRED


# Configurar el pin GPIO 2 como salida
led = Pin(15, Pin.OUT)

# Configurar el pin donde está conectado el DHT22
dht_pin = Pin(23)  # Cambia el número de pin si es necesario
sensor = dht.DHT22(dht_pin)

# Configurar SoftI2C (SDA en GPIO21, SCL en GPIO22)
i2c = SoftI2C(scl=Pin(22), sda=Pin(21))

# Inicializar la pantalla OLED (128x64)
oled = ssd1306.SSD1306_I2C(128, 64, i2c)

# Limpiar la pantalla antes de comenzar
oled.fill(0)

# Mostrar texto en la pantalla OLED
oled.text("Hola, MicroPython!", 0, 0)  # (texto, x, y)

# Actualizar la pantalla para reflejar los cambios
oled.show()

# Esperar unos segundos
time.sleep(5)

# Limpiar la pantalla nuevamente
oled.fill(0)
oled.show()

# Conectar a la red Wi-Fi
def connect_wifi():
    sta_if = network.WLAN(network.STA_IF)
    sta_if.active(True)
    sta_if.connect("Wokwi-GUEST", '')
    while not sta_if.isconnected():
        print(".", end="")
        time.sleep(0.1)
    print(" Connected!")

msg_str="off"
# Función de callback para cuando recibimos un mensaje
def mqtt_callback(topic, msg):
    #Cuando recibo datos por MQTT los recibo como bytes b'Hola'
    global msg_str
    topic_str = topic.decode('utf-8')
    msg_str = msg.decode('utf-8')
    print(f'Recibido mensaje en {topic_str}: {msg_str}')
    if msg_str == 'on':
        # Encender el LED
        led.on()
    if msg_str == 'off':
        # Apagar el LED
        led.off()

# Configuración y conexión al broker MQTT
def connect_mqtt():

    #client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, port=MQTT_PORT, user=MQTT_USER, password=MQTT_PASSWORD)
    client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, port=MQTT_PORT, user=MQTT_USER, ssl = True, ssl_params = ssl_params)
    #client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, MQTT_PORT, keepalive=60, ssl_params=ssl_params)
    #client.set_callback(mqtt_callback)
    client.connect()
    print('Conectado al broker MQTT')
    return client

# Publicar un mensaje
def publish_message(client, message):
    client.publish(MQTT_TOPIC_PUB, message)

# Suscribirse a un tema
def subscribe_topic(client):
    client.subscribe(MQTT_TOPIC_SUB)
    print(f'Suscrito al tema {MQTT_TOPIC_SUB}')

#Bucle Principal.
prev_weather = ""
prev_msg_str = "X"
ciclo = 0

try:
    connect_wifi()
    client = connect_mqtt()
    subscribe_topic(client)
    while True:
        client.check_msg()      # Verifica si hay mensajes sin bloquear el loop

        sensor.measure()
        message = ujson.dumps({
            "temperatura": sensor.temperature(),
            "humedad": sensor.humidity(),
        })
        if message != prev_weather:             #Solo publica cuando el dato a cambiado
            publish_message(client, message)
            print(message)
            prev_weather = message
            oled.fill(0)
            oled.text("TEMP: "+ str(sensor.temperature())+ " oC", 0, 0)  # (texto, x, y)
            oled.text("HUM:  "+ str(sensor.humidity())+" %", 0, 20)  # (texto, x, y)
            oled.text("LED:  "+ str(msg_str), 0, 40)  # (texto, x, y)
            oled.show()

        if msg_str != prev_msg_str:             #Solo publica cuando el dato a cambiado
            prev_msg_str = msg_str
            oled.fill(0)
            oled.text("TEMP: "+ str(sensor.temperature())+ " oC", 0, 0)  # (texto, x, y)
            oled.text("HUM:  "+ str(sensor.humidity())+" %", 0, 20)  # (texto, x, y)
            oled.text("LED:  "+ str(msg_str), 0, 40)  # (texto, x, y)
            oled.show()

        ciclo += 1
        oled.fill_rect(60, 50, 30, 10, 0)
        oled.text("Ciclo:  "+ str(ciclo), 0, 50)  # (texto, x, y)
        oled.show()
        time.sleep(0.05)

except OSError as e:
    print('Error de conexión', e)