import network
import time
from machine import Pin
from umqtt.simple import MQTTClient

# MQTT Server Parameters
MQTT_CLIENT_ID = ""
MQTT_BROKER    = "test.mosquitto.org"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "wokwi/weather/relay"

# Define relay pin
relay_pin = 23

# Initialize relay pin
relay = Pin(relay_pin, Pin.OUT)

# Connect to WiFi 
print("Connecting to 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(" Connected!\n")

# Connect to MQTT Broker
print("Connecting to MQTT server... \n", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.connect()
print("Connected!")

# Callback funcion for Topic Subscription
def on_message(topic, msg):
    #print(str(topic,'UTF-8'))
    #print(str(msg,'UTF-8'))

    # Processar a mensagem
    try:
        if msg.decode() == "true":
            relay.on()
            print("Relé Ligado")

        elif msg.decode() == "false":
            relay.off()
            print("Relé Desligado")

    except ValueError:
        print("Erro ao acionar o relé")

# MQTT Subcribtion details
client.set_callback(on_message)
client.subscribe(MQTT_TOPIC)

while True:
  client.check_msg()