import sys
import os
import time
import ipaddress
import wifi
import socketpool
import ssl

import adafruit_requests

from secrets import secrets


ITERATIONS = 5
DELAY = 0


print("""-"""*49)
print(os.uname().machine)
print(os.uname().version)

# Nice scenario: connect only when enabled
print("""-"""*49)
for _ in range(ITERATIONS):
    wifi.radio.enabled = True
    print("Enabled?  ", wifi.radio.enabled)
    print("Connect...", wifi.radio.connect(secrets["ssid"], secrets["password"]))
    print("RSSI      ", wifi.radio.ap_info.rssi)
    print()
    wifi.radio.enabled = False
    time.sleep(DELAY)

# Same, with multiple sequential connects while already connected
print("""-"""*49)
for _ in range(ITERATIONS):
    wifi.radio.enabled = True
    print("Enabled?  ", wifi.radio.enabled)
    for _ in range(ITERATIONS):
        print("Connect...", wifi.radio.connect(secrets["ssid"], secrets["password"]))
    print("RSSI      ", wifi.radio.ap_info.rssi)
    print()
    wifi.radio.enabled = False
    time.sleep(DELAY)

# Stress test: try connect and GET while wifi started, then stopped
# it should recover each time wifi is re-enabled
# user code should enable before connect to avoid socket exceptions
pool = socketpool.SocketPool(wifi.radio)
requests = adafruit_requests.Session(pool, ssl.create_default_context())

for _ in range(ITERATIONS):
    print("""-"""*49)
    wifi.radio.enabled = True
    print("Enabled?  ", wifi.radio.enabled)
    print("Connect...", wifi.radio.connect(secrets["ssid"], secrets["password"]))

    try:
        r = requests.get("http://wifitest.adafruit.com/testwifi/index.html")
        print(r.text)
    except (BrokenPipeError) as e:
        sys.print_exception(e)
    try:
        r = requests.get("https://httpbin.org/get")
        print(r.json())
    except (OSError) as e:
        sys.print_exception(e)

    print("""-"""*49)
    wifi.radio.enabled = False
    print("Enabled?  ", wifi.radio.enabled)
    try:
        print("Connect...", wifi.radio.connect(secrets["ssid"], secrets["password"]))
    except (ConnectionError, RuntimeError) as e:
        sys.print_exception(e)

    try:
        r = requests.get("http://wifitest.adafruit.com/testwifi/index.html")
        print(r.text)
    except (RuntimeError, BrokenPipeError, OSError) as e:
        sys.print_exception(e)
    try:
        r = requests.get("https://httpbin.org/get")
        print(r.json())
    except (RuntimeError, OSError) as e:
        sys.print_exception(e)

    print()
    time.sleep(DELAY)

print("Done.")

ITERATIONS = 10
DELAY = 0

for _ in range(ITERATIONS):
    print("Enabled?  ", wifi.radio.enabled)
    try:
        for network in wifi.radio.start_scanning_networks():
            print("{0:02X}:{1:02X}:{2:02X}:{3:02X}:{4:02X}:{5:02X}".format(*network.bssid), end=" ")
            print("{:>2d}".format(network.channel), end=" ")
            print(network.rssi, network.authmode, network.country, network.ssid)
        wifi.radio.stop_scanning_networks()
    except (RuntimeError) as e:
        print("RuntimeError:", e)
    wifi.radio.enabled = not wifi.radio.enabled
    time.sleep(DELAY)

print("Done.")