import network
import time
from machine import Timer
import random
from simple import MQTTClient
from display import display_number

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(1)
print(" Connected!")

# Timer Object
tim = Timer(-1)

# MQTT Server Parameters
MQTT_CLIENT_ID = "micropython-Esp32-PK"
MQTT_BROKER    = "broker.rithask.me"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "haptomapping"

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

# Callback funcion for Topic Subscription
def sub_cb(topic, msg):
  print(str(topic,'UTF-8'))
  msg = str(msg, 'UTF-8')
  print(str(msg,'UTF-8'))
  if msg != "":
    display_number(msg)

# Callback funcion for Publish Data
def pub_topic(tim,topic:str,msg=""):
  msg = ""
  client.publish(topic, msg)

# Timer for Periodic Data Publish
tim.init(period=5000, mode=Timer.PERIODIC,callback=lambda pub: pub_topic(tim,MQTT_TOPIC))

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

while True:
  try:
    client.check_msg()  # Non-blocking check for incoming messages
    time.sleep(1)  # Small sleep to avoid overwhelming the loop
  except OSError as e:
    print("Connection lost, reconnecting...")
    client.connect()
    client.subscribe(MQTT_TOPIC)
    time.sleep(1)