from machine import Pin
from time import sleep
from servo import Servo
#Conexion wifi, manejar instancias de un cliente mqtt
import network
from umqtt.simple import MQTTClient

#Configuración del servo
motor = Servo(pin = 15)

#Constantes para conexion con el cliente MQTT
MQTT_BROKER = "broker.hivemq.com"
MQTT_TOPIC = "utng/wdmr/servo"
MQTT_PORT = 1883

def conectar_wifi():
    print("Conectando a 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 A WIFI!")

    
def mensaje_recibido(topic, msg):
    angulo = int(msg)
    motor.move(angulo)

def conectar_broker():
    print("Conectando al broker: ", MQTT_BROKER)
    cliente = MQTTClient("", 
                        MQTT_BROKER,
                        port=MQTT_PORT,
                        keepalive=60)
    cliente.set_callback(mensaje_recibido)
    cliente.connect()
    cliente.subscribe(MQTT_TOPIC)
    return cliente

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

#Ciclo infinito
while True:
    cliente.wait_msg()