add dl button to shorts results

This commit is contained in:
Simon 2023-10-31 16:59:38 +07:00
parent 79a002956b
commit 5406007315
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4

View File

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