import network
import networkconfig
import time

wlan_sta = network.WLAN(network.STA_IF)


def do_connect(ntwrk_ssid, netwrk_pass):
    sta_if = network.WLAN(network.STA_IF)
    sta_if.active(True)
    if not sta_if.isconnected():
        print('Trying to connect to %s...' % ntwrk_ssid)
        sta_if.connect(ntwrk_ssid, netwrk_pass)
        retry = 0
        while not sta_if.isconnected() and retry < 100:
            time.sleep(0.1)
            retry += 1
            print('.', end='')
        if sta_if.isconnected():
            print('\nConnected. Network config: ', sta_if.ifconfig())
            return True
        else:
            print('\nFailed. Not Connected to: ' + ntwrk_ssid)
            return False


def check_connection():
    global wlan_sta
    # First check if there already is any connection:
    if wlan_sta.isconnected():
        return True
    try:
        # ESP connecting to WiFi takes time, wait a bit and try again:
        time.sleep(3)
        if not wlan_sta.isconnected():
            # inside passwd file, there is a list of WiFi networks (CSV format)
            f = open("passwd.dat")
            data = f.readlines()
            f.close()
            # Search WiFis in range
            ssids_found = wlan_sta.scan()

            # matching...
            for ipass in data:
                ssid_list = ipass.strip("\n").split(";")

                for i in ssids_found:
                    if ssid_list[0] in i[0]:
                        print("OK. WiFi found.")
                        if do_connect(ssid_list[0], ssid_list[1]):
                            return True

            if not wlan_sta.isconnected():
                if networkconfig.start():
                    return True
        else:
            return True

    except OSError:
        # start web server for connection manager:
        if networkconfig.start():
            return True

    return False


if check_connection():

    # Main Code is here
    print("ESP OK")
    # to import your code;
    # import sample_mqtt.py

else:
    print("There is something wrong.")