From 77729789e36102b8579587ae71b6c33be5f44f44 Mon Sep 17 00:00:00 2001
From: redphx <96280+redphx@users.noreply.github.com>
Date: Sun, 7 Jul 2024 11:20:43 +0700
Subject: [PATCH] Fix problems in the Guide menu #436 #438

---
 src/modules/ui/guide-menu.ts | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/src/modules/ui/guide-menu.ts b/src/modules/ui/guide-menu.ts
index ebe4baa..989d21e 100644
--- a/src/modules/ui/guide-menu.ts
+++ b/src/modules/ui/guide-menu.ts
@@ -105,6 +105,12 @@ export class GuideMenu {
         buttons.push(GuideMenu.#BUTTONS.streamSetting);
         AppInterface && buttons.push(GuideMenu.#BUTTONS.appSettings);
 
+        // Reload stream
+        buttons.push(GuideMenu.#BUTTONS.reloadStream);
+
+        // Back to home
+        buttons.push(GuideMenu.#BUTTONS.backToHome);
+
         const $buttons = GuideMenu.#renderButtons(buttons);
         $btnQuit.insertAdjacentElement('afterend', $buttons);
 
@@ -117,11 +123,13 @@ export class GuideMenu {
         const where = (e as any).where as GuideMenuTab;
 
         if (where === GuideMenuTab.HOME) {
-            const $root = document.querySelector('#gamepass-dialog-root div[role=dialog]') as HTMLElement;
-            if (STATES.isPlaying) {
-                GuideMenu.#injectHomePlaying($root);
-            } else {
-                GuideMenu.#injectHome($root);
+            const $root = document.querySelector('#gamepass-dialog-root div[role=dialog] div[role=tabpanel] div[class*=HomeLandingPage]') as HTMLElement;
+            if ($root) {
+                if (STATES.isPlaying) {
+                    GuideMenu.#injectHomePlaying($root);
+                } else {
+                    GuideMenu.#injectHome($root);
+                }
             }
         }
     }