Allow setting arbitrary headers for frame socket

socket/framesocket.go

@@ -29,8 +29,8 @@ type FrameSocket struct {
 	log    waLog.Logger
 	lock   sync.Mutex
 
-	URL    string
-	Origin string
+	URL         string
+	HTTPHeaders http.Header
 
 	Frames       chan []byte
 	OnDisconnect func(remote bool)
@@ -52,8 +52,8 @@ func NewFrameSocket(log waLog.Logger, proxy Proxy) *FrameSocket {
 		Header: WAConnHeader,
 		Frames: make(chan []byte),
 
-		URL:    URL,
-		Origin: Origin,
+		URL:         URL,
+		HTTPHeaders: http.Header{"Origin": {Origin}},
 
 		Proxy: proxy,
 	}
@@ -108,9 +108,8 @@ func (fs *FrameSocket) Connect() error {
 		Proxy: fs.Proxy,
 	}
 
-	headers := http.Header{"Origin": []string{fs.Origin}}
 	fs.log.Debugf("Dialing %s", fs.URL)
-	conn, _, err := dialer.Dial(fs.URL, headers)
+	conn, _, err := dialer.Dial(fs.URL, fs.HTTPHeaders)
 	if err != nil {
 		cancel()
 		return fmt.Errorf("couldn't dial whatsapp web websocket: %w", err)