#Modulos para trabajar con la OLED y El sensor
from machine import Pin
from time import sleep
from servo import Servo

#Declarar mi objeto servo
motor = Servo(pin = 15)

#Conexion wifi, manejar instancias de un cliente mqtt
import network
from umqtt.simple import MQTTClient

#Constantes para conexion con el cliente MQTT
MQTT_CLIENT_ID = ""
MQTT_BROKER = "broker.hivemq.com"
MQTT_USER = ""
MQTT_PASSWORD = ""
MQTT_TOPIC = "utng/jamb/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!")

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

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.set_callback(mensaje_recibido)
    cliente.connect()
    cliente.subscribe(MQTT_TOPIC)
    print("Conectado al topico: ", MQTT_TOPIC)
    return cliente

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

#Ciclo infinito
while True:
    cliente.wait_msg()