fix: detect the upgrade btn using the icon (#2364)

th-ch · Aug 22, 2024 · c5c1914 · c5c1914
1 parent b12e2f6
commit c5c1914
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/renderer.ts b/src/renderer.ts
@@ -172,10 +172,19 @@ async function onApiLoaded() {
 
   // Remove upgrade button
   if (window.mainConfig.get('options.removeUpgradeButton')) {
+    const itemsSelector = 'ytmusic-guide-section-renderer #items';
+    let selector = 'ytmusic-guide-entry-renderer:last-child';
+
+    const upgradeBtnIcon = document.querySelector<SVGGElement>('iron-iconset-svg[name="yt-sys-icons"] #youtube_music_monochrome');
+    if (upgradeBtnIcon) {
+      const path = upgradeBtnIcon.firstChild as SVGPathElement;
+      const data = path.getAttribute('d')!.substring(0, 15);
+      selector = `ytmusic-guide-entry-renderer:has(> tp-yt-paper-item > yt-icon path[d^="${data}"])`;
+    }
+
     const styles = document.createElement('style');
-    styles.innerHTML = `ytmusic-guide-section-renderer #items ytmusic-guide-entry-renderer:last-child {
-      display: none;
-    }`;
+    styles.textContent = `${itemsSelector} ${selector} { display: none; }`;
+
     document.head.appendChild(styles);
   }