extension/script.js

@@ -312,13 +312,8 @@ function buildVideoButton(titleContainer) {
   dlButton.classList.add('ta-button');
   dlButton.href = '#';
 
-  let videoId;
-  if (href.startsWith('/watch?v')) {
-    let params = new URLSearchParams(href);
-    videoId = params.get('/watch?v');
-  } else if (href.startsWith('/shorts/')) {
-    videoId = href.split('/')[2];
-  }
+  let params = new URLSearchParams(href);
+  let videoId = params.get('/watch?v');
   if (!videoId) return;
 
   dlButton.setAttribute('data-id', videoId);
@@ -343,8 +338,8 @@ function buildVideoButton(titleContainer) {
   dlIcon.innerHTML = defaultIcon;
   Object.assign(dlIcon.style, {
     filter: 'invert()',
-    width: '15px',
-    height: '15px',
+    width: '18px',
+    height: '18px',
     padding: '7px 8px',
   });
 
@@ -370,7 +365,7 @@ function getNearestLink(element) {
 }
 
 function processTitle(titleContainer) {
-  if (titleContainer.hasListener) return;
+  if (titleContainer.hasListener) return
   Object.assign(titleContainer.style, {
     display: 'flex',
     gap: '15px',