import network
import time
from machine import Pin, PWM
import dht
from umqtt.simple import MQTTClient

MQTT_CLIENT_ID = "lazaro"
MQTT_BROKER    = "broker.mqttdashboard.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
alerta_topico1 = "UMIDADE1"
alerta_topico2 = "UMIDADE2"
normalazicao_topico1 = "UMIDADE1_NORMAL"
normalazicao_topico2 = "UMIDADE2_NORMAL"

sensor1 = dht.DHT22(Pin(15))
sensor2 = dht.DHT22(Pin(2))
servo1 = PWM(Pin(13), freq=50)
servo2 = PWM(Pin(12), freq=50)
slide_switch = Pin(14, Pin.IN, Pin.PULL_UP)  

def ajuste_servo1(angulo1):
    duty1 = int(angulo1)
    servo1.duty(duty1)

def ajuste_servo2(angulo2):
    duty2 = int(angulo2)
    servo2.duty(duty2)

def publicar_mensagem(client, topico, mensagem):
    try:
        client.publish(topico, mensagem)
        print(f"Mensagem publicada em {topico}: {mensagem}")
    except Exception as e:
        print(f"Erro ao publicar em {topico}: {e}")
        print("Tentando reconectar ao MQTT...")
        try:
            client.connect()
            print("Reconectado ao MQTT")
            client.publish(topico, mensagem)
            print(f"Mensagem publicada em {topico} após reconexão: {mensagem}")
        except Exception as e:
            print("Falha ao reconectar e publicar:", e)

print("Conectando ao WiFi", end="")
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(" Conectado!")

print("Conectando ao servidor MQTT... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)

try:
    client.connect()
    print("Conectado!")
    client.publish(alerta_topico1, "Teste de conexão MQTT")
    print("Mensagem de teste publicada em alerta_topico1")
except Exception as e:
    print("Falha ao conectar ao MQTT:", e)
    sys.exit()

# Loop principal
while True:
    try:
        if slide_switch.value() == 0:  
            print("Chave ativada, desativando envios e retornando servos para 0º")
            ajuste_servo1(0)
            ajuste_servo2(0)
            continue  

        sensor1.measure()
        temperatura1 = sensor1.temperature()
        umidade1 = sensor1.humidity()
        sensor2.measure()
        temperatura2 = sensor2.temperature()
        umidade2 = sensor2.humidity()

        print("Sensor 1 - Temperatura:", temperatura1, "Umidade:", umidade1)
        print("Sensor 2 - Temperatura:", temperatura2, "Umidade:", umidade2)

        if temperatura1 >= 60:
            print("Temperatura 1 acima do normal, abrindo comporta em 50º")
            ajuste_servo1(50)
        else:
            print("Temperatura 1 normal")

        if temperatura2 >= 60:
            print("Temperatura 2 acima do normal, abrindo comporta em 180º")
            ajuste_servo2(180)
        else:
            print("Temperatura 2 normal")

        publicar_mensagem(client, alerta_topico1, "UMIDADE1 BAIXA" if umidade1 < 20 else "UMIDADE1 NORMAL")
        publicar_mensagem(client, alerta_topico2, "UMIDADE2 BAIXA" if umidade2 < 20 else "UMIDADE2 NORMAL")

    except OSError as e:
        print("Erro ao medir sensor ou publicar MQTT:", e)

    time.sleep(10)