This commit is contained in:
Su-Yong
2023-10-02 21:00:53 +09:00
parent 042083b112
commit aed1bbc6d7

View File

@ -4,103 +4,105 @@ import is from 'electron-is';
import type { SongInfo } from '../../providers/song-info'; import type { SongInfo } from '../../providers/song-info';
export default () => { export default () => {
ipcRenderer.on('update-song-info', (_, extractedSongInfo: SongInfo) => setTimeout(async () => { const setLyrics = (lyricsContainer: Element, lyrics: string | null) => {
const tabList = document.querySelectorAll('tp-yt-paper-tab'); lyricsContainer.innerHTML = `
const tabs = { <div id="contents" class="style-scope ytmusic-section-list-renderer description ytmusic-description-shelf-renderer genius-lyrics">
upNext: tabList[0], ${lyrics?.replaceAll(/\r\n|\r|\n/g, '<br/>') ?? 'Could not retrieve lyrics from genius'}
lyrics: tabList[1], </div>
discover: tabList[2], <yt-formatted-string class="footer style-scope ytmusic-description-shelf-renderer" style="align-self: baseline">
}; </yt-formatted-string>
`;
// Check if disabled if (lyrics) {
if (!tabs.lyrics?.hasAttribute('disabled')) { const footer = lyricsContainer.querySelector('.footer');
return;
if (footer) {
footer.textContent = 'Source: Genius';
}
} }
};
let hasLyrics = true; let unregister: (() => void) | null = null;
const lyrics = await ipcRenderer.invoke( ipcRenderer.on('update-song-info', (_, extractedSongInfo: SongInfo) => {
'search-genius-lyrics', unregister?.();
extractedSongInfo,
) as string;
if (!lyrics) {
// Delete previous lyrics if tab is open and couldn't get new lyrics
checkLyricsContainer(() => {
hasLyrics = false;
setTabsOnclick(undefined);
});
return;
}
if (is.dev()) { setTimeout(async () => {
console.log('Fetched lyrics from Genius'); const tabList = document.querySelectorAll<HTMLElement>('tp-yt-paper-tab');
} const tabs = {
upNext: tabList[0],
lyrics: tabList[1],
discover: tabList[2],
};
enableLyricsTab(); // Check if disabled
if (!tabs.lyrics?.hasAttribute('disabled')) return;
setTabsOnclick(enableLyricsTab); const lyrics = await ipcRenderer.invoke(
'search-genius-lyrics',
extractedSongInfo,
) as string | null;
checkLyricsContainer(); if (!lyrics) {
// Delete previous lyrics if tab is open and couldn't get new lyrics
tabs.upNext.click();
return;
}
if (is.dev()) {
console.log('Fetched lyrics from Genius');
}
const tryToInjectLyric = (callback?: () => void) => {
const lyricsContainer = document.querySelector(
'[page-type="MUSIC_PAGE_TYPE_TRACK_LYRICS"] > ytmusic-message-renderer',
);
if (lyricsContainer) {
callback?.();
setLyrics(lyricsContainer, lyrics);
applyLyricsTabState();
}
};
const applyLyricsTabState = () => {
if (lyrics) {
tabs.lyrics.removeAttribute('disabled');
tabs.lyrics.removeAttribute('aria-disabled');
} else {
tabs.lyrics.setAttribute('disabled', '');
tabs.lyrics.setAttribute('aria-disabled', '');
}
};
const lyricsTabHandler = () => {
const tabContainer = document.querySelector('ytmusic-tab-renderer');
if (!tabContainer) return;
const lyricsTabHandler = () => {
const tabContainer = document.querySelector('ytmusic-tab-renderer');
if (tabContainer) {
const observer = new MutationObserver((_, observer) => { const observer = new MutationObserver((_, observer) => {
checkLyricsContainer(() => observer.disconnect()); tryToInjectLyric(() => observer.disconnect());
}); });
observer.observe(tabContainer, { observer.observe(tabContainer, {
attributes: true, attributes: true,
childList: true, childList: true,
subtree: true, subtree: true,
}); });
} };
};
tabs.lyrics.addEventListener('click', lyricsTabHandler); applyLyricsTabState();
function checkLyricsContainer(callback = () => { tabs.discover.addEventListener('click', applyLyricsTabState);
}) { tabs.lyrics.addEventListener('click', lyricsTabHandler);
const lyricsContainer = document.querySelector( tabs.upNext.addEventListener('click', applyLyricsTabState);
'[page-type="MUSIC_PAGE_TYPE_TRACK_LYRICS"] > ytmusic-message-renderer',
);
if (lyricsContainer) {
callback();
setLyrics(lyricsContainer);
}
}
function setLyrics(lyricsContainer: Element) { tryToInjectLyric();
lyricsContainer.innerHTML = `<div id="contents" class="style-scope ytmusic-section-list-renderer description ytmusic-description-shelf-renderer genius-lyrics">
${
hasLyrics
? lyrics.replaceAll(/\r\n|\r|\n/g, '<br/>')
: 'Could not retrieve lyrics from genius'
}
</div> unregister = () => {
<yt-formatted-string class="footer style-scope ytmusic-description-shelf-renderer" style="align-self: baseline"></yt-formatted-string>`; tabs.discover.removeEventListener('click', applyLyricsTabState);
if (hasLyrics) { tabs.lyrics.removeEventListener('click', lyricsTabHandler);
const footer = lyricsContainer.querySelector('.footer'); tabs.upNext.removeEventListener('click', applyLyricsTabState);
if (footer) { };
footer.textContent = 'Source: Genius'; }, 500);
enableLyricsTab(); });
}
}
}
const defaultHandler = () => {};
function setTabsOnclick(callback: EventListenerOrEventListenerObject | undefined) {
for (const tab of [tabs.upNext, tabs.discover]) {
if (tab) {
tab.addEventListener('click', callback ?? defaultHandler);
}
}
}
function enableLyricsTab() {
tabs.lyrics.removeAttribute('disabled');
tabs.lyrics.removeAttribute('aria-disabled');
}
}, 500));
}; };