import network
import time
import umqtt.simple as qtt

def connect_wifi(ssid = 'Wokwi-GUEST', pwd = ''):
    Wifi_connection = network.WLAN(network.STA_IF)
    Wifi_connection.active(True)
    Wifi_connection.connect(ssid, pwd)

    while not Wifi_connection.isconnected():
        pass

    print('Wifi is OK')
    return Wifi_connection

def IbraheemZain(topic, msg):
    print('Callback function is called')
    print(f'this {topic} send this : {msg}')

    if msg == b'command_1':
        print("command 1 under excution")
    elif msg == b'command_2':
        print('command 2 under excution')
    else:
        print('no such command')


def main():
    connect_wifi()
 
    clnt = qtt.MQTTClient("Ibraheem_Zain", "broker.hivemq.com")
    clnt.set_callback(IbraheemZain);
    clnt.connect()

    print('connected to broker')

    clnt.subscribe("Wokwi ESP32")
    print("Done subscription")
    while True:
        clnt.check_msg()

main()