tubearchivist
/
browser-extension
mirror of https://github.com/tubearchivist/browser-extension.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

implement extract video id on hover

This commit is contained in:
Simon 2023-11-09 18:21:43 +07:00
parent 4f54e1f863
commit f8d69f5883
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
extension/script.js
View File

@ -326,12 +326,9 @@ function getTitleContainers() {
return elements;
}
function buildVideoButton(titleContainer) {
function getVideoId(titleContainer) {
let href = getNearestLink(titleContainer);
if (!href) return;
const dlButton = document.createElement('a');
dlButton.classList.add('ta-button');
dlButton.href = '#';
let videoId;
if (href.startsWith('/watch?v')) {
@ -340,11 +337,16 @@ function buildVideoButton(titleContainer) {
} else if (href.startsWith('/shorts/')) {
videoId = href.split('/')[2];
}
return videoId;
}
function buildVideoButton(titleContainer) {
let videoId = getVideoId(titleContainer);
if (!videoId) return;
dlButton.setAttribute('data-id', videoId);
dlButton.setAttribute('data-type', 'video');
dlButton.title = `TA download video: ${titleContainer.innerText} [${videoId}]`;
const dlButton = document.createElement('a');
dlButton.classList.add('ta-button');
dlButton.href = '#';
Object.assign(dlButton.style, {
display: 'flex',
@ -435,7 +437,12 @@ function checkVideoExists(taButton) {
console.error(e);
}
let videoId = taButton.dataset.id;
if (!taButton.parentElement) return;
let videoId = getVideoId(taButton.parentElement);
taButton.setAttribute('data-id', videoId);
taButton.setAttribute('data-type', 'video');
taButton.title = `TA download video: ${taButton.parentElement.innerText} [${videoId}]`;
let message = { type: 'videoExists', videoId };
let sending = sendMessage(message);
sending.then(handleResponse, handleError);