From afd58dafac77a83b242f1024b02af1291e4eae8a Mon Sep 17 00:00:00 2001
From: Viktor Liu <viktor@netbird.io>
Date: Mon, 1 Sep 2025 17:27:51 +0200
Subject: [PATCH] Remove unused imports

---
 encryption/route53.go                  |  2 ++
 shared/relay/client/client.go          | 12 +-----------
 shared/relay/client/dialers_generic.go | 19 +++++++++++++++++++
 shared/relay/client/dialers_js.go      | 13 +++++++++++++
 4 files changed, 35 insertions(+), 11 deletions(-)
 create mode 100644 shared/relay/client/dialers_generic.go
 create mode 100644 shared/relay/client/dialers_js.go

diff --git a/encryption/route53.go b/encryption/route53.go
index 3c81ab103..48c7a3a1b 100644
--- a/encryption/route53.go
+++ b/encryption/route53.go
@@ -1,3 +1,5 @@
+//go:build !js
+
 package encryption
 
 import (
diff --git a/shared/relay/client/client.go b/shared/relay/client/client.go
index 5dabc5742..57a98614d 100644
--- a/shared/relay/client/client.go
+++ b/shared/relay/client/client.go
@@ -9,11 +9,8 @@ import (
 
 	log "github.com/sirupsen/logrus"
 
-	"github.com/netbirdio/netbird/client/iface"
 	auth "github.com/netbirdio/netbird/shared/relay/auth/hmac"
 	"github.com/netbirdio/netbird/shared/relay/client/dialer"
-	"github.com/netbirdio/netbird/shared/relay/client/dialer/quic"
-	"github.com/netbirdio/netbird/shared/relay/client/dialer/ws"
 	"github.com/netbirdio/netbird/shared/relay/healthcheck"
 	"github.com/netbirdio/netbird/shared/relay/messages"
 )
@@ -296,14 +293,7 @@ func (c *Client) Close() error {
 }
 
 func (c *Client) connect(ctx context.Context) (*RelayAddr, error) {
-	// Force WebSocket for MTUs larger than default to avoid QUIC DATAGRAM frame size issues
-	var dialers []dialer.DialeFn
-	if c.mtu > 0 && c.mtu > iface.DefaultMTU {
-		c.log.Infof("MTU %d exceeds default (%d), forcing WebSocket transport to avoid DATAGRAM frame size issues", c.mtu, iface.DefaultMTU)
-		dialers = []dialer.DialeFn{ws.Dialer{}}
-	} else {
-		dialers = []dialer.DialeFn{quic.Dialer{}, ws.Dialer{}}
-	}
+	dialers := c.getDialers()
 
 	rd := dialer.NewRaceDial(c.log, dialer.DefaultConnectionTimeout, c.connectionURL, dialers...)
 	conn, err := rd.Dial()
diff --git a/shared/relay/client/dialers_generic.go b/shared/relay/client/dialers_generic.go
new file mode 100644
index 000000000..a8ed79961
--- /dev/null
+++ b/shared/relay/client/dialers_generic.go
@@ -0,0 +1,19 @@
+//go:build !js
+
+package client
+
+import (
+	"github.com/netbirdio/netbird/client/iface"
+	"github.com/netbirdio/netbird/shared/relay/client/dialer"
+	"github.com/netbirdio/netbird/shared/relay/client/dialer/quic"
+	"github.com/netbirdio/netbird/shared/relay/client/dialer/ws"
+)
+
+// getDialers returns the list of dialers to use for connecting to the relay server.
+func (c *Client) getDialers() []dialer.DialeFn {
+	if c.mtu > 0 && c.mtu > iface.DefaultMTU {
+		c.log.Infof("MTU %d exceeds default (%d), forcing WebSocket transport to avoid DATAGRAM frame size issues", c.mtu, iface.DefaultMTU)
+		return []dialer.DialeFn{ws.Dialer{}}
+	}
+	return []dialer.DialeFn{quic.Dialer{}, ws.Dialer{}}
+}
diff --git a/shared/relay/client/dialers_js.go b/shared/relay/client/dialers_js.go
new file mode 100644
index 000000000..6bd0e6696
--- /dev/null
+++ b/shared/relay/client/dialers_js.go
@@ -0,0 +1,13 @@
+//go:build js
+
+package client
+
+import (
+	"github.com/netbirdio/netbird/shared/relay/client/dialer"
+	"github.com/netbirdio/netbird/shared/relay/client/dialer/ws"
+)
+
+func (c *Client) getDialers() []dialer.DialeFn {
+	// JS/WASM build only uses WebSocket transport
+	return []dialer.DialeFn{ws.Dialer{}}
+}