From 870a7592517d7d182806a0a022fea127790769aa Mon Sep 17 00:00:00 2001
From: hensm <m+github@matt.tf>
Date: Wed, 13 Mar 2019 07:51:38 +0000
Subject: [PATCH] Redirect cast extension chrome-extension:// URLs to regular
 API URL

---
 ext/src/compat/youtube.js    | 50 ++++++++++++++++++++++++++++++++++++
 ext/src/lib/userAgents.ts    | 22 +++++++++++-----
 ext/src/main.ts              | 11 +++++++-
 ext/src/shim/contentSetup.ts |  5 +++-
 4 files changed, 80 insertions(+), 8 deletions(-)

diff --git a/ext/src/compat/youtube.js b/ext/src/compat/youtube.js
index 264226f..63b89d3 100644
--- a/ext/src/compat/youtube.js
+++ b/ext/src/compat/youtube.js
@@ -8,3 +8,53 @@ document.addEventListener("beforescriptexecute", function onBeforeScriptExecute
     scriptElement.src = browser.runtime.getURL("vendor/webcomponents-lite.min.js");
     document.head.prepend(scriptElement);
 });
+
+
+const EXT_SENDER_SCRIPT_URLS = [
+    "chrome-extension://pkedcjkdefgpdelpbcmbmeomcjbeemfm/cast_sender.js"
+  , "chrome-extension://enhhojjnijigcajfphajepfemndkmdlo/cast_sender.js"
+];
+
+const SENDER_SCRIPT_URL = "https://www.gstatic.com/cv/js/sender/v1/cast_sender.js";
+
+
+// Store reference to original function
+const _createElement = document.createElement;
+
+function createElement () {
+    // Call original function
+    const element = _createElement.apply(this, arguments);
+
+    /**
+     * If the new element being created is a <script> element,
+     * replace the src property setter to intercept the new value.
+     *
+     * If it matches Chrome's cast extension sender script URL,
+     * replace it with the standard API URL, the request for which
+     * is handled in the main script.
+     */
+    if (element.nodeName === "SCRIPT") {
+        const { get, set } = Reflect.getOwnPropertyDescriptor(
+                Reflect.getPrototypeOf(element.wrappedJSObject), "src");
+
+        Reflect.defineProperty(element.wrappedJSObject, "src", {
+            configurable: true
+          , enumerable: true          
+          , get
+
+          , set: exportFunction(function (value) {
+                if (EXT_SENDER_SCRIPT_URLS.includes(value)) {
+                    return set.call(this, SENDER_SCRIPT_URL);
+                }
+                return set.call(this, value);
+            }, window)
+        });
+    }
+
+    return element;
+}
+
+// Redefine page's document.createElement function
+exportFunction(createElement, document, {
+    defineAs: "createElement"
+});
diff --git a/ext/src/lib/userAgents.ts b/ext/src/lib/userAgents.ts
index d2873a4..53b4ba2 100644
--- a/ext/src/lib/userAgents.ts
+++ b/ext/src/lib/userAgents.ts
@@ -7,18 +7,28 @@ const PLATFORM_LINUX = "Mozilla/5.0 (X11; Linux x86_64";
 const UA_PREFIX = "Mozilla/5.0";
 
 const UA_CHROME = "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.67 Safari/537.36";
+const UA_CHROME_LEGACY = "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2883.87 Safari/537.36";
 const UA_SAFARI = "AppleWebKit/605.1.15 (KHTML, like Gecko) Version/12.0 Safari/605.1.15";
 
 
-export function getChromeUserAgent (platform: string): string {
-    let platformComponent: string;
+function getPlatformComponent (platform: string) {
     switch (platform) {
-        case "mac": platformComponent = PLATFORM_MAC; break;
-        case "win": platformComponent = PLATFORM_WIN; break;
-        case "linux": platformComponent = PLATFORM_LINUX; break;
+        case "mac": return PLATFORM_MAC; break;
+        case "win": return PLATFORM_WIN; break;
+        case "linux": return PLATFORM_LINUX; break;
     }
+}
 
-    return `${UA_PREFIX} (${platformComponent}) ${UA_CHROME}`;
+export function getChromeUserAgent (
+        platform: string
+      , legacy: boolean = false): string {
+
+    const platformComponent = getPlatformComponent(platform);
+    const browserComponent = legacy
+        ? UA_CHROME_LEGACY
+        : UA_CHROME;
+
+    return `${UA_PREFIX} (${platformComponent}) ${browserComponent}`;
 }
 
 export function getSafariUserAgent (platform: string): string {
diff --git a/ext/src/main.ts b/ext/src/main.ts
index cbf156d..a37c9cb 100755
--- a/ext/src/main.ts
+++ b/ext/src/main.ts
@@ -173,10 +173,19 @@ async function onBeforeSendHeaders (
         currentUAString = getChromeUserAgent(os);
     }
 
+    const host = details.requestHeaders.find(
+            (header: any) => header.name === "Host");
+
     // Find and rewrite the User-Agent header
     for (const header of details.requestHeaders) {
         if (header.name.toLowerCase() === "user-agent") {
-            header.value = currentUAString;
+            // TODO: Move this somewhere else
+            if (host.value === "www.youtube.com") {
+                header.value = getChromeUserAgent(os, true);
+            } else {
+                header.value = currentUAString;
+            }
+
             break;
         }
     }
diff --git a/ext/src/shim/contentSetup.ts b/ext/src/shim/contentSetup.ts
index f59905c..0c34bb1 100644
--- a/ext/src/shim/contentSetup.ts
+++ b/ext/src/shim/contentSetup.ts
@@ -1,3 +1,6 @@
 "use strict";
 
-(window.wrappedJSObject as any).chrome = cloneInto({}, window);
+const _window = (window.wrappedJSObject as any);
+
+_window.chrome = cloneInto({}, window);
+_window.navigator.presentation = cloneInto({}, window);