From bb60f51816fd29dfd7a7b6c78730f5a57d2e9687 Mon Sep 17 00:00:00 2001
From: Probably Raging <probablyraging@gmail.com>
Date: Fri, 10 Feb 2023 06:40:02 +1100
Subject: [PATCH] Reverted previous changes to background.js

---
 src/scripts/background.js | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/scripts/background.js b/src/scripts/background.js
index 436f4fc..131436b 100644
--- a/src/scripts/background.js
+++ b/src/scripts/background.js
@@ -37,9 +37,13 @@ function hideShorts() {
                 chrome.storage.sync.get(['toggleNavState'], function (result) {
                     if (result.toggleNavState === 'on') hideShortsNavButton();
                 });
-                // Home Feed
+                // Home feed
                 chrome.storage.sync.get(['toggleHomeFeedState'], function (result) {
-                    if (result.toggleHomeFeedState === 'on') hideShortsShelfHomeFeed(), hideShortsVideosHomeSubscriptionFeed();
+                    if (result.toggleHomeFeedState === 'on') hideShortsShelfHomeFeed(), hideShortsVideosHomeFeed()
+                });
+                // Subscriptions feed
+                chrome.storage.sync.get(['toggleSubscriptionFeedState'], function (result) {
+                    if (result.toggleSubscriptionFeedState === 'on') hideShortsVideosSubscriptionFeed();
                 });
                 // Channel tab
                 chrome.storage.sync.get(['toggleTabState'], function (result) {
@@ -70,16 +74,20 @@ function hideShorts() {
     }
 
     // Hide shorts video elements in the home/subscription feeds
-    function hideShortsVideosHomeSubscriptionFeed() {
+    function hideShortsVideosHomeFeed() {
         if (document.title.toLowerCase() == 'youtube') {
             const elements = document.querySelectorAll('[href^="/shorts/"]');
             elements.forEach(element => {
                 // Ignore shorts in the notification menu
                 if (element.parentNode.id === 'item' || element.parentNode.parentNode.parentNode.parentNode.parentNode.id === 'submenu') return;
                 const parent = element.parentNode;
-                parent.parentNode.removeChild(parent);
+                if (parent.hasAttribute('rich-grid-thumbnail')) parent.parentNode.parentNode.style.display = 'none';
             });
         }
+    }
+
+    // Hide shorts video elements in the home/subscription feeds
+    function hideShortsVideosSubscriptionFeed() {
         if (document.title.toLowerCase() == 'subscriptions - youtube') {
             const elements = document.querySelectorAll('[href^="/shorts/"]');
             elements.forEach(element => {