fix(site): improve wasm init speed, various copy changes

Author: coadler

cmd/wasm/main_js.go

@@ -72,9 +72,32 @@ func newWush(cfg js.Value) map[string]any {
 		cfg.Get("onWebrtcCandidate"),
 	)
 
-	err = ov.PickDERPHome(ctx)
-	if err != nil {
-		panic(err)
+	// Try to get stored DERP home from localStorage
+	localStorage := js.Global().Get("localStorage")
+	storedDerpHome := localStorage.Call("getItem", "derpHome")
+	if !storedDerpHome.IsNull() {
+		// Parse stored DERP home and use it
+		derpID := uint16(js.Global().Get("parseInt").Invoke(storedDerpHome).Int())
+		if region := dm.Regions[int(derpID)]; region != nil {
+			ov.DerpRegionID = derpID
+			hlog("Using stored DERP home: %s", region.RegionName)
+		} else {
+			// If stored DERP home is invalid, pick a new one
+			err = ov.PickDERPHome(ctx)
+			if err != nil {
+				panic(err)
+			}
+			// Store the newly picked DERP home
+			localStorage.Call("setItem", "derpHome", fmt.Sprint(ov.DerpRegionID))
+		}
+	} else {
+		// No stored DERP home, pick a new one
+		err = ov.PickDERPHome(ctx)
+		if err != nil {
+			panic(err)
+		}
+		// Store the picked DERP home
+		localStorage.Call("setItem", "derpHome", fmt.Sprint(ov.DerpRegionID))
 	}
 
 	s, err := tsserver.NewServer(ctx, logger, ov, dm)
@@ -91,11 +114,13 @@ func newWush(cfg js.Value) map[string]any {
 		panic(err)
 	}
 
-	_, err = ts.Up(ctx)
-	if err != nil {
-		panic(err)
-	}
-	hlog("WireGuard is ready")
+	go func() {
+		_, err = ts.Up(ctx)
+		if err != nil {
+			panic(err)
+		}
+		hlog("WireGuard is ready")
+	}()
 
 	cpListener, err := ts.Listen("tcp", ":4444")
 	if err != nil {

cmd/wush/cp.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"net/http/httputil"
 	"net/netip"
+	"net/url"
 	"os"
 	"path/filepath"
 	"strings"
@@ -59,7 +60,13 @@ func initAuth(authFlag *string, ca *overlay.ClientAuth) serpent.MiddlewareFunc {
 				}
 			}
 
-			err := ca.Parse(strings.TrimSpace(*authFlag))
+			// If the user provided a URL, extract the auth key from the fragment.
+			authKey := *authFlag
+			if u, err := url.Parse(*authFlag); err == nil && u.Fragment != "" {
+				authKey = u.Fragment
+			}
+
+			err := ca.Parse(strings.TrimSpace(authKey))
 			if err != nil {
 				return fmt.Errorf("parse auth key: %w", err)
 			}

cmd/wush/serve.go

@@ -86,6 +86,8 @@ func serveCmd() *serpent.Command {
 				hlog("Your auth key is:")
 				fmt.Println("\t>", cliui.Code(r.ClientAuth().AuthKey()))
 				hlog("Use this key to authenticate other " + cliui.Code("wush") + " commands to this instance.")
+				hlog("Visit the following link to connect via the browser:")
+				fmt.Println("\t>", cliui.Code("https://wush.dev#"+r.ClientAuth().AuthKey()))
 			} else {
 				fmt.Println(cliui.Code(r.ClientAuth().AuthKey()))
 				hlog("The auth key has been printed to stdout")