ujs refactor and add toggle webgl feature

1 files changed, 26 insertions, 7 deletions

diff --git a/sh/ujs b/sh/ujs
index c1a3e1f..76d85db 100755
--- a/sh/ujs
+++ b/sh/ujs
@@ -1,11 +1,30 @@
 #!/bin/sh
 
-if grep -q '^//user_pref("privacy.resistFingerprinting", false);$' "$HOME/.mozilla/firefox/xxxxxxxx.fly/user-overrides.js"; then
-	sed -i 's#^//\(user_pref("privacy.resistFingerprinting", false);\)$#\1#' "$HOME/.mozilla/firefox/xxxxxxxx.fly/user-overrides.js"
-	notify-send -u critical 'RFP disabled'
-else
-	sed -i 's#^\(user_pref("privacy.resistFingerprinting", false);\)$#//\1#' "$HOME/.mozilla/firefox/xxxxxxxx.fly/user-overrides.js"
-	notify-send 'RFP enabled'
-fi
+file="$HOME/.mozilla/firefox/xxxxxxxx.fly/user-overrides.js"
+while getopts rw opt; do
+	case $opt in
+		r)
+			if grep -q '^//user_pref("privacy.resistFingerprinting", false);$' "$file"; then
+				# if want to abstract into a function, this maybe useful: sed -i "s#^//\\($1\\)\$#\\1#" "$file"
+				sed -i 's#^//\(user_pref("privacy.resistFingerprinting", false);\)$#\1#' "$file"
+				notify-send -u critical 'RFP disabled'
+			else
+				sed -i 's#^\(user_pref("privacy.resistFingerprinting", false);\)$#//\1#' "$file"
+				notify-send 'RFP enabled'
+			fi
+			;;
+		w)
+			if grep -q '^//user_pref("webgl.disabled", false);$' "$file"; then
+				sed -i 's#^//\(user_pref("webgl.disabled", false);\)$#\1#' "$file"
+				# note webgl is enabled by uncomment, but RFP is disabled by uncomment
+				notify-send -u critical 'webgl enabled'
+			else
+				sed -i 's#^\(user_pref("webgl.disabled", false);\)$#//\1#' "$file"
+				notify-send 'webgl disabled'
+			fi
+			;;
+		\?)exit 1;;
+	esac
+done
 upd -j
 o "$BROWSER"