import network

# Thông tin WiFi kết nối và WiFi phát
ssid_sta = 'Wokwi-GUEST'
password_sta = ''
ssid_ap = 'ESP32'
password_ap = '123456789'

# Chế độ kết nối WiFi (Station)
def testSTA():
    wlan = network.WLAN(network.STA_IF)  # Chọn chế độ STA
    wlan.active(True)                    # Kích hoạt WiFi STA
    print("Scan:", wlan.scan())         # Quét các mạng WiFi xung quanh
    print("Connected:", wlan.isconnected())  # Kiểm tra đã kết nối chưa
    wlan.connect(ssid_sta, password_sta)     # Bắt đầu kết nối
    while not wlan.isconnected():
        pass
    print("MAC:", wlan.config('mac'))       # In địa chỉ MAC
    print("Config:", wlan.ifconfig())       # In địa chỉ IP, subnet, gateway, DNS

# Chế độ phát WiFi (Access Point)
def testAP():
    ap = network.WLAN(network.AP_IF)        # Chọn chế độ AP
    ap.config(essid=ssid_ap, authmode=3, password=password_ap)  # Cấu hình SSID, mật khẩu
    ap.config(max_clients=5)                # Số thiết bị tối đa
    ap.active(True)                         # Bắt đầu phát WiFi
    print("Access Point started:", ssid_ap)

# Kết nối WiFi đơn giản
def testWifiConnect():
    wifi = network.WLAN(network.STA_IF)
    wifi.active(True)
    if not wifi.isconnected():
        print("Connecting to network...")
        wifi.connect(ssid_sta, password_sta)
        while not wifi.isconnected():
            pass
    print("Connected to:", wifi.config('essid'))
    print("Network config:", wifi.ifconfig())

# testAP()             # Gọi hàm này nếu bạn muốn ESP phát WiFi
# testSTA()            # Gọi hàm này nếu bạn muốn ESP quét & kết nối WiFi (có scan)
# testWifiConnect()    # Gọi hàm này nếu bạn chỉ cần kết nối đơn giản