From 75f426f45687b560dec843956ef89b583484fb05 Mon Sep 17 00:00:00 2001
From: Michael C <michael@mchang.name>
Date: Thu, 30 Jun 2022 16:38:11 -0400
Subject: [PATCH] fix embeds not being detected correctly

- add awaiter for key element
- refresh ID with segments if videoID is invalid
---
 src/content.ts | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/content.ts b/src/content.ts
index e97d5309..2ed30119 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -100,7 +100,10 @@ const playerButtons: Record<string, {button: HTMLButtonElement, image: HTMLImage
 // Direct Links after the config is loaded
 utils.wait(() => Config.config !== null, 1000, 1).then(() => videoIDChange(getYouTubeVideoID(document)));
 // wait for hover preview to appear, and refresh attachments if ever found
-window.addEventListener("DOMContentLoaded", () => utils.waitForElement(".ytp-inline-preview-ui").then(() => refreshVideoAttachments()));
+window.addEventListener("DOMContentLoaded", () => {
+    utils.waitForElement(".ytp-inline-preview-ui").then(() => refreshVideoAttachments())
+    utils.waitForElement("[data-sessionlink='feature=player-title']").then(() => videoIDChange(getYouTubeVideoID(document)))
+});
 addPageListeners();
 addHotkeyListener();
 
@@ -206,6 +209,9 @@ function messageListener(request: Message, sender: unknown, sendResponse: (respo
             submitSponsorTimes();
             break;
         case "refreshSegments":
+            // update video on refresh if videoID invalid
+            if (!sponsorVideoID) videoIDChange(getYouTubeVideoID(document));
+            // fetch segments
             sponsorsLookup(false).then(() => sendResponse({
                 found: sponsorDataFound,
                 status: lastResponseStatus,