about summary refs log tree commit diff stats
path: root/src/pages/background/background.js
diff options
context:
space:
mode:
Diffstat (limited to 'src/pages/background/background.js')
-rw-r--r--src/pages/background/background.js10
1 files changed, 5 insertions, 5 deletions
diff --git a/src/pages/background/background.js b/src/pages/background/background.js
index 3c6efd36..8e56f62d 100644
--- a/src/pages/background/background.js
+++ b/src/pages/background/background.js
@@ -86,10 +86,8 @@ browser.webRequest.onBeforeRequest.addListener(
       initiator = new URL(details.originUrl);
     else if (details.initiator)
       initiator = new URL(details.initiator);
-
-    let newUrl;
-
-    if (!newUrl) newUrl = youtubeHelper.redirect(url, details, initiator)
+    
+    let newUrl = youtubeHelper.redirect(url, details, initiator)
     if (youtubeMusicHelper.isYoutubeMusic(url, initiator)) newUrl = youtubeMusicHelper.redirect(url, details.type)
 
     if (!newUrl) newUrl = twitterHelper.redirect(url, initiator);
@@ -149,7 +147,9 @@ browser.tabs.onRemoved.addListener(
 browser.webRequest.onHeadersReceived.addListener(
   async e => {
     await wholeInit();
-    return twitterHelper.removeXFrameOptions(e);
+    let response = twitterHelper.removeXFrameOptions(e)
+    if (!response) youtubeHelper.removeXFrameOptions(e)
+    return response;
   },
   { urls: ["<all_urls>"], },
   ["blocking", "responseHeaders"]