Merge 4725934df3 into ca15cc9c0b

extension/script.js

@@ -135,12 +135,14 @@ function ensureTALinks() {
 
   let titleContainerNodes = getTitleContainers();
   for (let titleContainer of titleContainerNodes) {
-    if (titleContainer.hasTA) continue;
+    let parent = getNearestH3(titleContainer);
+    if (!parent) continue;
+    if (parent.hasTA) continue;
     let videoButton = buildVideoButton(titleContainer);
     if (videoButton == null) continue;
-    processTitle(titleContainer);
+    processTitle(parent);
-    titleContainer.appendChild(videoButton);
+    parent.appendChild(videoButton);
-    titleContainer.hasTA = true;
+    parent.hasTA = true;
   }
 }
 ensureTALinks = throttled(ensureTALinks, 700);
@@ -329,6 +331,8 @@ function getTitleContainers() {
 }
 
 function getVideoId(titleContainer) {
+  if (!titleContainer) return undefined;
+
   let href = getNearestLink(titleContainer);
   if (!href) return;
 
@@ -394,6 +398,16 @@ function getNearestLink(element) {
   return null;
 }
 
+function getNearestH3(element) {
+  for (let i = 0; i < 5 && element && element !== document; i++) {
+    if (element.tagName === 'H3') {
+      return element;
+    }
+    element = element.parentNode;
+  }
+  return null;
+}
+
 function processTitle(titleContainer) {
   if (titleContainer.hasListener) return;
   Object.assign(titleContainer.style, {
@@ -439,9 +453,9 @@ function checkVideoExists(taButton) {
     console.error(e);
   }
 
-  let videoId = taButton.dataset.id;
+  let aElem = taButton?.parentElement?.querySelector('a');
-  if (taButton.parentElement) {
+  let videoId = getVideoId(aElem);;
-    videoId = getVideoId(taButton.parentElement);
+  if (aElem) {
     taButton.setAttribute('data-id', videoId);
     taButton.setAttribute('data-type', 'video');
     taButton.title = `TA download video: ${taButton.parentElement.innerText} [${videoId}]`;