connect to wsurl，add http.Header param

browser.go

@@ -9,6 +9,7 @@ package rod
 
 import (
 	"context"
+	"net/http"
 	"reflect"
 	"strings"
 	"sync"
@@ -145,7 +146,8 @@ func (b *Browser) NoDefaultDevice() *Browser {
 
 // Connect to the browser and start to control it.
 // If fails to connect, try to launch a local browser, if local browser not found try to download one.
-func (b *Browser) Connect() error {
+// If connect websocket not chromium，use header and set Sec-WebSocket-Key, for example websocket for transfer chromium.
+func (b *Browser) Connect(h http.Header) error {
 	if b.client == nil {
 		u := b.controlURL
 		if u == "" {
@@ -156,7 +158,7 @@ func (b *Browser) Connect() error {
 			}
 		}
 
-		c, err := cdp.StartWithURL(b.ctx, u, nil)
+		c, err := cdp.StartWithURL(b.ctx, u, h)
 		if err != nil {
 			return err
 		}

browser_test.go

@@ -434,7 +434,7 @@ func TestBrowserConnectFailure(t *testing.T) {
 
 	c := g.Context()
 	c.Cancel()
-	err := rod.New().Context(c).Connect()
+	err := rod.New().Context(c).Connect(nil)
 	if err == nil {
 		g.Fatal("expected an error on connect failure")
 	}
@@ -447,7 +447,7 @@ func TestBrowserPool(t *testing.T) {
 
 	b, err := pool.Get(func() (*rod.Browser, error) {
 		browser := rod.New()
-		return browser, browser.Connect()
+		return browser, browser.Connect(nil)
 	})
 	g.E(err)
 	pool.Put(b)

must.go

@@ -48,7 +48,7 @@ func (b *Browser) WithPanic(fail func(interface{})) *Browser {
 
 // MustConnect is similar to [Browser.Connect].
 func (b *Browser) MustConnect() *Browser {
-	b.e(b.Connect())
+	b.e(b.Connect(nil))
 	return b
 }