Youtube Web Extension refactor (#2487)

MozillaReality · Dec 18, 2019 · c4770a3 · c4770a3
1 parent 8dd6bff
commit c4770a3
Show file tree

Hide file tree

Showing 4 changed files with 232 additions and 290 deletions.
diff --git a/app/src/main/assets/web_extensions/webcompat_youtube/background.js b/app/src/main/assets/web_extensions/webcompat_youtube/background.js
@@ -1,6 +1,7 @@
+const CUSTOM_USER_AGENT = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12) AppleWebKit/602.1.21 (KHTML, like Gecko) Version/9.2 Safari/602.1.21';
 const targetUrls = [
-	"https://*.youtube.com/*",
-	"https://*.youtube-nocookie.com/*"
+    "https://*.youtube.com/*",
+    "https://*.youtube-nocookie.com/*"
 ];
 
 /**
@@ -10,24 +11,44 @@ const targetUrls = [
  *    video pages intended for mobile phones, as linked from Google search results).
  */
 function redirectUrl(req) {
-	let redirect = false;
-	const url = new URL(req.url);
-	if (url.host.startsWith("m.")) {
-		url.host = url.host.replace("m.", "www.");
-		redirect = true;
-	}
-	if (!url.searchParams.get("disable_polymer")) {
-		url.searchParams.set("disable_polymer", "1");
-		redirect = true;
-	}
-	if (!redirect) {
-		return null;
-	}
-	return { redirectUrl: url.toString() };
+    let redirect = false;
+    const url = new URL(req.url);
+    if (url.host.startsWith("m.")) {
+        url.host = url.host.replace("m.", "www.");
+        redirect = true;
+    }
+    if (!url.searchParams.get("disable_polymer")) {
+        url.searchParams.set("disable_polymer", "1");
+        redirect = true;
+    }
+    if (!redirect) {
+        return null;
+    }
+    return { redirectUrl: url.toString() };
+}
+
+/**
+ * Override UA. This is required to get the equirectangular video formats from Youtube.
+ * Otherwise youtube uses custom 360 controls.
+ */
+function overrideUA(req) {
+    for (const header of req.requestHeaders) {
+        if (header.name.toLowerCase() === "user-agent") {
+            header.value = CUSTOM_USER_AGENT;
+        }
+    }
+    return { requestHeaders: req.requestHeaders };
 }
 
 browser.webRequest.onBeforeRequest.addListener(
-	redirectUrl,
-	{ urls: targetUrls, types: ["main_frame"]},
-	["blocking"]
+    redirectUrl,
+    { urls: targetUrls, types: ["main_frame"]},
+    ["blocking"]
 );
+
+browser.webRequest.onBeforeSendHeaders.addListener(
+    overrideUA,
+    { urls: targetUrls },
+    ["blocking", "requestHeaders"]
+  );
+
diff --git a/app/src/main/assets/web_extensions/webcompat_youtube/main.css b/app/src/main/assets/web_extensions/webcompat_youtube/main.css
@@ -4,3 +4,8 @@
 .ytp-generic-popup, ytp-generic-popup {
   display: none;
 }
+
+/* Improve readability a bit in FxR. */
+hr, html, i, iframe, img, ins, kbd, label, legend, li, menu, object, ol, p, pre, q, s, samp, small, span, strike, strong, sub {
+  font-size: 110%;
+}