From 37130576e9340fba87cc1c984b3474fa5de174f0 Mon Sep 17 00:00:00 2001
From: Alexey Khit <alexey.khit@gmail.com>
Date: Sun, 10 Sep 2023 07:58:20 +0300
Subject: [PATCH] Add support webrtc go2rtc source with auth #539

---
 internal/webrtc/client.go | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/internal/webrtc/client.go b/internal/webrtc/client.go
index 39c8bcc8..a652d3ce 100644
--- a/internal/webrtc/client.go
+++ b/internal/webrtc/client.go
@@ -1,6 +1,7 @@
 package webrtc
 
 import (
+	"encoding/base64"
 	"errors"
 	"io"
 	"net/http"
@@ -62,7 +63,7 @@ func streamsHandler(rawURL string) (core.Producer, error) {
 // ex: ws://localhost:1984/api/ws?src=camera1
 func go2rtcClient(url string) (core.Producer, error) {
 	// 1. Connect to signalign server
-	conn, _, err := websocket.DefaultDialer.Dial(url, nil)
+	conn, _, err := Dial(url)
 	if err != nil {
 		return nil, err
 	}
@@ -212,3 +213,27 @@ func whepClient(url string) (core.Producer, error) {
 
 	return prod, nil
 }
+
+// Dial - websocket.Dial with Basic auth support
+func Dial(rawURL string) (*websocket.Conn, *http.Response, error) {
+	u, err := url.Parse(rawURL)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	if u.User == nil {
+		return websocket.DefaultDialer.Dial(rawURL, nil)
+	}
+
+	user := u.User.Username()
+	pass, _ := u.User.Password()
+	u.User = nil
+
+	header := http.Header{
+		"Authorization": []string{
+			"Basic " + base64.StdEncoding.EncodeToString([]byte(user+":"+pass)),
+		},
+	}
+
+	return websocket.DefaultDialer.Dial(u.String(), header)
+}