import tm1637
from machine import Pin, SPI
from utime import sleep
import network
import ujson
from umqtt.simple import MQTTClient

MQTT_CLIENT_ID = "broker.hivemq.com-111"
MQTT_BROKER    = "broker.hivemq.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "alferov"

def sub_cb(topic, msg):
    global n
    print(topic, msg)
    if topic == b"alferov" and b'/number ' in msg:
        n = int(msg.split()[1])

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="")
  sleep(0.1)
print(" Connected!")

print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(sub_cb)
client.connect()
client.subscribe(MQTT_TOPIC)

print("Connected!")

dis = tm1637.TM1637(clk=Pin(25), dio=Pin(26))

n = 0

while True:
    client.wait_msg()
    client.check_msg()
    f = n
    while n >= 0:
        if f > n:
            for i in range(10, -1, -1):
                if n < 10:
                    dis.show("  "+str(n)+str(i))
                elif n < 100:
                    dis.show(" "+str(n)+str(i))
                else:
                    dis.show(str(n)+str(i))
                sleep(0.1)
        else:
            if n < 10:
                dis.show("  "+str(n)+"0")
            elif n < 100:
                dis.show(" "+str(n)+"0")
            else:
                dis.show(str(n)+"0")
            sleep(0.1)
        n = n - 1