from machine import Pin, I2C, SPI
from max7219 import Max7219
import network
import time
from umqtt.simple import MQTTClient

MQTT_CLIENT_ID = "mqtt-explorer-6d7c6a71"
MQTT_BROKER    = "test.mosquitto.org"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "NikitaWea"

spi = SPI(1)
ss = Pin(15, Pin.OUT)
display = Max7219(32, 8, spi, ss)

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

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

def display_text(text):
    display.fill(0)
    if text == 'kryg':
        display.text(' o', 1, 0, 1)
    elif text == 'kvadr':
        display.text('.', 1, 0, 1)
        display.text('.', -1, 0, -1)
        display.text('.', 1, -2, 1)
        display.text('.', -1, -2, -1)
    elif text == 'treg':
        display.text('.', 3, 1, 1)
        display.text('.', 1, 1, 1)
        display.text('.', 3, -1, 1)
        display.text('.', 3, -3, 1)
        display.text('.', 3, -5, 1)
        display.text('.', 1, -3, 1)
        display.text('.', 1, -1, 1)
        display.text('.', -1, 1, 1)
        display.text('.', -1, -1, 1)
        display.text('.', -3, 1, 1)
        display.text('.', -2, 0, 1)
        display.text('.', -3, 2, 1)
        display.text('.', 2, -4, 1)
        
    display.show()
    time.sleep(5)
    display.fill(0)
    

# Обработка входящих сообщений
def sub_callback(topic, msg):
    print("Message received:", msg)
    display_text(msg.decode('utf-8'))


# Настройка MQTT клиента
client = MQTTClient("esp32_client", MQTT_BROKER)
client.set_callback(sub_callback)
client.connect()
client.subscribe(MQTT_TOPIC)

print("MQTT client connected and subscribed to topic:", MQTT_TOPIC)

while True:
    client.check_msg()  # Проверка сообщений
    time.sleep(1)  # Задержка для снижения нагрузки на процессор