feat: support optional custom User Agent per-channel (#20, PR: #21)

* feat: support optional custom User Agent per-channel * fix(ci): run workflow when PR is opened
duncanleo · Dec 17, 2023 · 71530c6 · 71530c6
1 parent 2deb25d
commit 71530c6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -2,7 +2,7 @@ name: Test Build
 
 on:
   pull_request:
-    types: [synchronize]
+    types: [synchronize, opened]
 
 jobs:
   build:

diff --git a/config/channels.go b/config/channels.go
@@ -17,6 +17,9 @@ type Channel struct {
 	URL              string       `json:"url"`
 	ProxyConfig      *ProxyConfig `json:"proxy"`
 	DisableTranscode bool         `json:"disableTranscode"`
+
+	// UserAgent is a custom UA string that will be used by FFMPEG to make requests to the stream URL.
+	UserAgent *string `json:"userAgent,omitempty"`
 }
 
 var (

diff --git a/routes/stream.go b/routes/stream.go
@@ -39,6 +39,14 @@ func Stream(c *gin.Context) {
 		)
 	}
 
+	if channel.UserAgent != nil {
+		ffmpegArgs = append(
+			ffmpegArgs,
+			"-headers",
+			fmt.Sprintf("User-Agent: %s", *channel.UserAgent),
+		)
+	}
+
 	switch config.Cfg.GetEncoderProfile() {
 	case config.EncoderProfileVAAPI:
 		ffmpegArgs = append(
-Original file line number
+Diff line change
@@ Expand Up / @@ -2,7 +2,7 @@ name: Test Build @@
     on:
       pull_request:
-        types: [synchronize]
+        types: [synchronize, opened]
     jobs:
       build:
@@ Expand Down @@