wifi/espat, rtl8720dn, wifinina: move towards standard common interface for wifi adaptors

Signed-off-by: deadprogram <ron@hybridgroup.com>

espat/wifidriver.go

@@ -0,0 +1,19 @@
+package espat
+
+import (
+	"time"
+
+	"tinygo.org/x/drivers"
+)
+
+func (d *Device) ConnectToAccessPoint(ssid, pass string, timeout time.Duration) error {
+	if len(ssid) == 0 {
+		return drivers.ErrWiFiMissingSSID
+	}
+
+	return d.ConnectToAP(ssid, pass, 10)
+}
+
+func (d *Device) Disconnect() error {
+	return d.DisconnectFromAP()
+}

examples/espat/espconsole/main.go

@@ -114,7 +114,9 @@ func connectToAP() {
 	println("Connecting to wifi network '" + ssid + "'")
 
 	adaptor.SetWifiMode(espat.WifiModeClient)
-	adaptor.ConnectToAP(ssid, pass, 10)
+	if err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second); err != nil {
+		failMessage(err.Error())
+	}
 
 	println("Connected.")
 	ip, err := adaptor.GetClientIP()

examples/espat/esphub/main.go

@@ -100,7 +100,9 @@ func connectToAP() {
 	println("Connecting to wifi network '" + ssid + "'")
 
 	adaptor.SetWifiMode(espat.WifiModeClient)
-	adaptor.ConnectToAP(ssid, pass, 10)
+	if err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second); err != nil {
+		failMessage(err.Error())
+	}
 
 	println("Connected.")
 	ip, err := adaptor.GetClientIP()

examples/espat/espstation/main.go

@@ -89,7 +89,9 @@ func connectToAP() {
 	println("Connecting to wifi network '" + ssid + "'")
 
 	adaptor.SetWifiMode(espat.WifiModeClient)
-	adaptor.ConnectToAP(ssid, pass, 10)
+	if err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second); err != nil {
+		failMessage(err.Error())
+	}
 
 	println("Connected.")
 	ip, err := adaptor.GetClientIP()

examples/espat/mqttclient/main.go

@@ -109,7 +109,9 @@ func connectToAP() {
 	println("Connecting to wifi network '" + ssid + "'")
 
 	adaptor.SetWifiMode(espat.WifiModeClient)
-	adaptor.ConnectToAP(ssid, pass, 10)
+	if err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second); err != nil {
+		failMessage(err.Error())
+	}
 
 	println("Connected.")
 	ip, err := adaptor.GetClientIP()

examples/espat/mqttsub/main.go

@@ -130,7 +130,9 @@ func connectToAP() {
 	println("Connecting to wifi network '" + ssid + "'")
 
 	adaptor.SetWifiMode(espat.WifiModeClient)
-	adaptor.ConnectToAP(ssid, pass, 10)
+	if err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second); err != nil {
+		failMessage(err.Error())
+	}
 
 	println("Connected.")
 	ip, err := adaptor.GetClientIP()

examples/espat/tcpclient/main.go

@@ -92,7 +92,9 @@ func connectToAP() {
 	println("Connecting to wifi network '" + ssid + "'")
 
 	adaptor.SetWifiMode(espat.WifiModeClient)
-	adaptor.ConnectToAP(ssid, pass, 10)
+	if err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second); err != nil {
+		failMessage(err.Error())
+	}
 
 	println("Connected.")
 	ip, err := adaptor.GetClientIP()

examples/rtl8720dn/mqttclient/main.go

@@ -65,7 +65,7 @@ func run() error {
 	}
 	net.UseDriver(rtl)
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/mqttsub/main.go

@@ -74,7 +74,7 @@ func run() error {
 	}
 	net.UseDriver(rtl)
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/ntpclient/main.go

@@ -47,7 +47,7 @@ func run() error {
 	}
 	net.UseDriver(rtl)
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/tcpclient/main.go

@@ -47,7 +47,7 @@ func run() error {
 	}
 	net.UseDriver(rtl)
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/tlsclient/main.go

@@ -76,7 +76,7 @@ func run() error {
 	net.UseDriver(rtl)
 	http.SetBuf(buf[:])
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/udpstation/main.go

@@ -43,7 +43,7 @@ func run() error {
 	net.UseDriver(rtl)
 	http.SetBuf(buf[:])
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/webclient-tinyterm/main.go

@@ -74,7 +74,7 @@ func run() error {
 	http.SetBuf(buf[:])
 
 	fmt.Fprintf(terminal, "ConnectToAP()\r\n")
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/webclient/main.go

@@ -43,7 +43,7 @@ func run() error {
 	net.UseDriver(rtl)
 	http.SetBuf(buf[:])
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/rtl8720dn/webserver/main.go

@@ -43,7 +43,7 @@ func run() error {
 	}
 	http.UseDriver(rtl)
 
-	err = rtl.ConnectToAP(ssid, password)
+	err = rtl.ConnectToAccessPoint(ssid, password, 10*time.Second)
 	if err != nil {
 		return err
 	}

examples/wifinina/connect/main.go

@@ -128,19 +128,15 @@ func waitSerial() {
 
 // connect to access point
 func connectToAP() {
-	if len(ssid) == 0 || len(pass) == 0 {
-		for {
-			println("Connection failed: Either ssid or password not set")
-			time.Sleep(10 * time.Second)
-		}
-	}
 	time.Sleep(2 * time.Second)
 	println("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		println("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
+		for {
+			println(err)
+			time.Sleep(1 * time.Second)
+		}
 	}
+
 	println("Connected.")
 }

examples/wifinina/http-get/main.go

@@ -132,22 +132,18 @@ func waitSerial() {
 
 // connect to access point
 func connectToAP() {
-	if len(ssid) == 0 || len(pass) == 0 {
+	time.Sleep(2 * time.Second)
+	println("Connecting to " + ssid)
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
 		for {
-			println("Connection failed: Either ssid or password not set")
-			time.Sleep(10 * time.Second)
+			println(err)
+			time.Sleep(1 * time.Second)
 		}
 	}
-	time.Sleep(2 * time.Second)
-	message("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		message("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
-	}
-	message("Connected.")
-	time.Sleep(2 * time.Second)
+
+	println("Connected.")
+
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {
 		message(err.Error())

examples/wifinina/mqttclient/main.go

@@ -104,14 +104,16 @@ func main() {
 func connectToAP() {
 	time.Sleep(2 * time.Second)
 	println("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		println("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
+		for {
+			println(err)
+			time.Sleep(1 * time.Second)
+		}
 	}
+
 	println("Connected.")
-	time.Sleep(2 * time.Second)
+
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {
 		println(err.Error())

examples/wifinina/mqttsub/main.go

@@ -116,13 +116,16 @@ func publishing() {
 func connectToAP() {
 	time.Sleep(2 * time.Second)
 	println("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		println("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
+		for {
+			println(err)
+			time.Sleep(1 * time.Second)
+		}
 	}
+
 	println("Connected.")
+
 	time.Sleep(2 * time.Second)
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {

examples/wifinina/ntpclient/main.go

@@ -151,21 +151,18 @@ func clearBuffer() {
 
 // connect to access point
 func connectToAP() {
-	if len(ssid) == 0 || len(pass) == 0 {
+	time.Sleep(2 * time.Second)
+	println("Connecting to " + ssid)
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
 		for {
-			println("Connection failed: Either ssid or password not set")
-			time.Sleep(10 * time.Second)
+			println(err)
+			time.Sleep(1 * time.Second)
 		}
 	}
-	time.Sleep(2 * time.Second)
-	message("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		message("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
-	}
-	message("Connected.")
+
+	println("Connected.")
+
 	time.Sleep(2 * time.Second)
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {

examples/wifinina/tcpclient/main.go

@@ -113,14 +113,17 @@ func sendBatch() {
 // connect to access point
 func connectToAP() {
 	time.Sleep(2 * time.Second)
-	message("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		message("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
+	println("Connecting to " + ssid)
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
+		for {
+			println(err)
+			time.Sleep(1 * time.Second)
+		}
 	}
-	message("Connected.")
+
+	println("Connected.")
+
 	time.Sleep(2 * time.Second)
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {

examples/wifinina/tlsclient/main.go

@@ -122,21 +122,18 @@ func makeHTTPSRequest() {
 
 // connect to access point
 func connectToAP() {
-	if len(ssid) == 0 || len(pass) == 0 {
+	time.Sleep(2 * time.Second)
+	println("Connecting to " + ssid)
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
 		for {
-			println("Connection failed: Either ssid or password not set")
-			time.Sleep(10 * time.Second)
+			println(err)
+			time.Sleep(1 * time.Second)
 		}
 	}
-	time.Sleep(2 * time.Second)
-	message("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		message("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
-	}
-	message("Connected.")
+
+	println("Connected.")
+
 	time.Sleep(2 * time.Second)
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {

examples/wifinina/udpstation/main.go

@@ -76,15 +76,17 @@ func main() {
 // connect to access point
 func connectToAP() {
 	time.Sleep(2 * time.Second)
-	message("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		message("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
+	println("Connecting to " + ssid)
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
+		for {
+			println(err)
+			time.Sleep(1 * time.Second)
+		}
 	}
-	message("Connected.")
-	time.Sleep(2 * time.Second)
+
+	println("Connected.")
+
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {
 		message(err.Error())

examples/wifinina/webclient/main.go

@@ -124,22 +124,18 @@ func makeHTTPRequest() {
 
 // connect to access point
 func connectToAP() {
-	if len(ssid) == 0 || len(pass) == 0 {
+	time.Sleep(2 * time.Second)
+	println("Connecting to " + ssid)
+	err := adaptor.ConnectToAccessPoint(ssid, pass, 10*time.Second)
+	if err != nil { // error connecting to AP
 		for {
-			println("Connection failed: Either ssid or password not set")
-			time.Sleep(10 * time.Second)
+			println(err)
+			time.Sleep(1 * time.Second)
 		}
 	}
-	time.Sleep(2 * time.Second)
-	message("Connecting to " + ssid)
-	adaptor.SetPassphrase(ssid, pass)
-	for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
-		message("Connection status: " + st.String())
-		time.Sleep(1 * time.Second)
-		st, _ = adaptor.GetConnectionStatus()
-	}
-	message("Connected.")
-	time.Sleep(2 * time.Second)
+
+	println("Connected.")
+
 	ip, _, _, err := adaptor.GetIP()
 	for ; err != nil; ip, _, _, err = adaptor.GetIP() {
 		message(err.Error())

rtl8720dn/wifidriver.go

@@ -0,0 +1,20 @@
+package rtl8720dn
+
+import (
+	"time"
+
+	"tinygo.org/x/drivers"
+)
+
+func (r *RTL8720DN) ConnectToAccessPoint(ssid, pass string, timeout time.Duration) error {
+	if len(ssid) == 0 {
+		return drivers.ErrWiFiMissingSSID
+	}
+
+	return r.ConnectToAP(ssid, pass)
+}
+
+func (r *RTL8720DN) Disconnect() error {
+	_, err := r.Rpc_wifi_disconnect()
+	return err
+}

wifi.go

@@ -0,0 +1,16 @@
+package drivers
+
+import (
+	"errors"
+	"time"
+)
+
+var (
+	ErrWiFiMissingSSID    = errors.New("missing SSID")
+	ErrWiFiConnectTimeout = errors.New("WiFi connect timeout")
+)
+
+type WiFiDriver interface {
+	ConnectToAccessPoint(ssid, pass string, timeout time.Duration) error
+	Disconnect() error
+}

wifinina/wifidriver.go

@@ -0,0 +1,26 @@
+package wifinina
+
+import (
+	"time"
+
+	"tinygo.org/x/drivers"
+)
+
+func (d *Device) ConnectToAccessPoint(ssid, pass string, timeout time.Duration) error {
+	if len(ssid) == 0 {
+		return drivers.ErrWiFiMissingSSID
+	}
+
+	start := time.Now()
+	d.SetPassphrase(ssid, pass)
+
+	for time.Since(start) < timeout {
+		st, _ := d.GetConnectionStatus()
+		if st == StatusConnected {
+			return nil
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+
+	return drivers.ErrWiFiConnectTimeout
+}