from machine import Pin,I2C
from utime import sleep, sleep_ms
import network, time
import ujson
from umqtt.simple import MQTTClient
from ssd1306 import SSD1306_I2C

MQTT_CLIENT_ID = "clien353dennisi333"
MQTT_BROKER    = "broker.hivemq.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC_SUB = "lenguaje/xd"


ancho = 128
alto =  64

i2c=I2C(0, scl=Pin(22), sda=Pin(21))
oled = SSD1306_I2C(ancho, alto, i2c)


def conectaWifi (red, password):
      global miRed
      miRed = network.WLAN(network.STA_IF)     
      if not miRed.isconnected():              #Si no está conectado…
          miRed.active(True)                   #activa la interface
          miRed.connect(red, password)         #Intenta conectar con la red
          print('Conectando a la red', red +"…")
          timeout = time.time ()
          while not miRed.isconnected():           #Mientras no se conecte..
              if (time.ticks_diff (time.time (), timeout) > 10):
                  return False
      return True


if conectaWifi("Wokwi-GUEST", ""):

    print ("Conexión exitosa!")
    print('Datos de la red (IP/netmask/gw/DNS):', miRed.ifconfig())

    print("Conectando a  MQTT server... ",MQTT_BROKER,"...", end="")
    client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
    client.connect()

    def sub_cb(topic, msg):
        global variable
        dato_recibido = ujson.loads(msg.decode())
        print(f'Recibido:{dato}')
        print(dato_recibido["Traduccion"])
        variable = dato_recibido["Traduccion"]


    while True:
        oled.fill(0)    
        oled.text("Traducción",0,10)
        oled.show()
        
        client.set_callback(sub_cb)
        client.subscribe(MQTT_TOPIC_SUB)
        print ("esperando")

        if variable == "uno":
            oled.text("Uno - 1",0,30)
            oled.show()

        elif variable == "dos":
            oled.text("Dos - 2",0,30)
            oled.show()



        client.wait_msg()
        time.sleep(3)



else:
    print ("Imposible conectar")
    miRed.active (False)