From 7b40dc44c26e92e18bba3a06cbe94386b7c6caea Mon Sep 17 00:00:00 2001
From: Simon <simobilleter@gmail.com>
Date: Mon, 13 May 2024 21:53:45 +0200
Subject: [PATCH] fix checkVideoExists, check siblings in getNearestLink

---
 extension/script.js | 32 ++++++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/extension/script.js b/extension/script.js
index 8aebfd9..d0337f2 100644
--- a/extension/script.js
+++ b/extension/script.js
@@ -389,6 +389,24 @@ function buildVideoButton(titleContainer) {
 }
 
 function getNearestLink(element) {
+  // Check siblings
+  let sibling = element;
+  while (sibling) {
+    sibling = sibling.previousElementSibling;
+    if (sibling && sibling.tagName === 'A' && sibling.getAttribute('href') !== '#') {
+      return sibling.getAttribute('href');
+    }
+  }
+
+  sibling = element;
+  while (sibling) {
+    sibling = sibling.nextElementSibling;
+    if (sibling && sibling.tagName === 'A' && sibling.getAttribute('href') !== '#') {
+      return sibling.getAttribute('href');
+    }
+  }
+
+  // Check parent elements
   for (let i = 0; i < 5 && element && element !== document; i++) {
     if (element.tagName === 'A' && element.getAttribute('href') !== '#') {
       return element.getAttribute('href');
@@ -453,12 +471,14 @@ function checkVideoExists(taButton) {
     console.error(e);
   }
 
-  let aElem = taButton?.parentElement?.querySelector('a');
-  let videoId = getVideoId(aElem);;
-  if (aElem) {
-    taButton.setAttribute('data-id', videoId);
-    taButton.setAttribute('data-type', 'video');
-    taButton.title = `TA download video: ${taButton.parentElement.innerText} [${videoId}]`;
+  let videoId = taButton.dataset.id;
+  if (!videoId) {
+    videoId = getVideoId(taButton);
+    if (videoId) {
+      taButton.setAttribute('data-id', videoId);
+      taButton.setAttribute('data-type', 'video');
+      taButton.title = `TA download video: ${taButton.parentElement.innerText} [${videoId}]`;
+    }
   }
 
   let message = { type: 'videoExists', videoId };