From f88f3dbd4993625a4bc8f5a23593e09c83d5b6cf Mon Sep 17 00:00:00 2001
From: Fuwn <contact@fuwn.me>
Date: Sun, 26 Nov 2023 18:51:04 -0800
Subject: feat(stream): user agent flag

---
 src/bashly.yml        |  2 ++
 src/stream_command.sh | 20 ++++++++++++++++++--
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/bashly.yml b/src/bashly.yml
index f29aac3..01fda35 100644
--- a/src/bashly.yml
+++ b/src/bashly.yml
@@ -67,6 +67,8 @@ commands:
       - long: --browser
         arg: browser
         default: firefox
+      - long: --user-agent
+        arg: user-agent
       - long: --fix
       - long: --download
       - long: --username
diff --git a/src/stream_command.sh b/src/stream_command.sh
index 1333222..351ddd8 100644
--- a/src/stream_command.sh
+++ b/src/stream_command.sh
@@ -1,11 +1,27 @@
+user_agent="${args['--user-agent']}"
+
+if [[ -n "${user_agent}" ]]; then
+	user_agent="--user-agent=${user_agent}"
+fi
+
 if [[ -n "${args['--download']}" ]]; then
 	START=$(date +%s)
 
-	yt-dlp --format 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best' --cookies-from-browser "${args[--browser]}" --all-subs --embed-subs "${args[uri]}"
+	yt-dlp \
+		--format 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best' \
+		--cookies-from-browser "${args[--browser]}" \
+		--all-subs \
+		--embed-subs \
+		"${user_agent}" \
+		"${args[uri]}"
 
 	printf "\ntook %s seconds\n" $(($(date +%s || true) - START))
 
 	return
 fi
 
-mpv --ytdl-raw-options-append=format='bestvideo[ext=mp4]+bestaudio[ext=m4a]/best' --ytdl-raw-options-append=cookies-from-browser="${args[--browser]}" "${args[uri]}"
+mpv \
+	--ytdl-raw-options-append=format='bestvideo[ext=mp4]+bestaudio[ext=m4a]/best' \
+	--ytdl-raw-options-append=cookies-from-browser="${args[--browser]}" \
+	"${user_agent}" \
+	"${args[uri]}"
-- 
cgit v1.2.3