diff --git a/plugins/discord/back.js b/plugins/discord/back.js index 0ebcc624..2fb8cc79 100644 --- a/plugins/discord/back.js +++ b/plugins/discord/back.js @@ -24,7 +24,10 @@ module.exports = (win, {activityTimoutEnabled, activityTimoutTime}) => { details: songInfo.title, state: songInfo.artist, largeImageKey: "logo", - largeImageText: songInfo.views + " - " + songInfo.likes, + largeImageText: [ + songInfo.uploadDate, + songInfo.views.toString().replace(/\B(?=(\d{3})+(?!\d))/g, ",") + " views" + ].join(' || '), }; if (songInfo.isPaused) { @@ -50,8 +53,7 @@ module.exports = (win, {activityTimoutEnabled, activityTimoutTime}) => { }); // Startup the rpc client - rpc - .login({ + rpc.login({ clientId, }) .catch(console.error); diff --git a/preload.js b/preload.js index c3d51d16..94f2a72a 100644 --- a/preload.js +++ b/preload.js @@ -29,6 +29,10 @@ document.addEventListener("DOMContentLoaded", () => { }); }); + // inject song-info provider + const songInfoProviderPath = path.join(__dirname, "providers", "song-info-front.js") + fileExists(songInfoProviderPath, require(songInfoProviderPath)); + // Add action for reloading global.reload = () => remote.getCurrentWindow().webContents.loadURL(config.get("url")); diff --git a/providers/song-info-front.js b/providers/song-info-front.js new file mode 100644 index 00000000..42c3afb2 --- /dev/null +++ b/providers/song-info-front.js @@ -0,0 +1,22 @@ +const { ipcRenderer } = require("electron"); + +const injectListener = () => { + var oldXHR = window.XMLHttpRequest; + function newXHR() { + var realXHR = new oldXHR(); + realXHR.addEventListener("readystatechange", () => { + if(realXHR.readyState==4 && realXHR.status==200){ + if (realXHR.responseURL.includes('/player')){ + // if the request is the contains the song info send the response to ipcMain + ipcRenderer.send( + "song-info-request", + realXHR.responseText + ); + } + } + }, false); + return realXHR; + } + window.XMLHttpRequest = newXHR; +} +module.exports = injectListener; diff --git a/providers/song-info.js b/providers/song-info.js index 4ddf9cab..5bea89b8 100644 --- a/providers/song-info.js +++ b/providers/song-info.js @@ -1,73 +1,20 @@ +const { ipcMain } = require("electron"); const { nativeImage } = require("electron"); const fetch = require("node-fetch"); -// This selects the song title -const titleSelector = ".title.style-scope.ytmusic-player-bar"; - -// This selects the song image -const imageSelector = - "#layout > ytmusic-player-bar > div.middle-controls.style-scope.ytmusic-player-bar > img"; - -// This selects the song subinfo, this includes artist, views, likes -const subInfoSelector = - "#layout > ytmusic-player-bar > div.middle-controls.style-scope.ytmusic-player-bar > div.content-info-wrapper.style-scope.ytmusic-player-bar > span"; - -// This selects the progress bar, used for songlength and current progress +// This selects the progress bar, used for current progress const progressSelector = "#progress-bar"; -// Grab the title using the selector -const getTitle = (win) => { - return win.webContents - .executeJavaScript( - "document.querySelector('" + titleSelector + "').innerText" - ) - .catch((error) => { - console.log(error); - }); -}; - -// Grab the image src using the selector -const getImageSrc = (win) => { - return win.webContents - .executeJavaScript("document.querySelector('" + imageSelector + "').src") - .catch((error) => { - console.log(error); - }); -}; - -// Grab the subinfo using the selector -const getSubInfo = async (win) => { - // Get innerText of subinfo element - const subInfoString = await win.webContents.executeJavaScript( - 'document.querySelector("' + subInfoSelector + '").innerText' - ); - - // Split and clean the string - const splittedSubInfo = subInfoString.replaceAll("\n", "").split(" • "); - - // Make sure we always return 3 elements in the aray - const subInfo = []; - for (let i = 0; i < 3; i++) { - // Fill array with empty string if not defined - subInfo.push(splittedSubInfo[i] || ""); - } - - return subInfo; -}; // Grab the progress using the selector const getProgress = async (win) => { - // Get max value of the progressbar element - const songDuration = await win.webContents.executeJavaScript( - 'document.querySelector("' + progressSelector + '").max' - ); // Get current value of the progressbar element const elapsedSeconds = await win.webContents.executeJavaScript( 'document.querySelector("' + progressSelector + '").value' ); - return { songDuration, elapsedSeconds }; + return elapsedSeconds; }; // Grab the native image using the src @@ -77,6 +24,7 @@ const getImage = async (src) => { return nativeImage.createFromBuffer(buffer); }; +// To find the paused status, we check if the title contains `-` const getPausedStatus = async (win) => { const title = await win.webContents.executeJavaScript("document.title"); return !title.includes("-"); @@ -86,8 +34,8 @@ const getPausedStatus = async (win) => { const songInfo = { title: "", artist: "", - views: "", - likes: "", + views: 0, + uploadDate: "", imageSrc: "", image: null, isPaused: true, @@ -95,6 +43,17 @@ const songInfo = { elapsedSeconds: 0, }; +const handleData = async (_event, responseText) => { + let data = JSON.parse(responseText); + songInfo.title = data?.videoDetails?.title; + songInfo.artist = data?.videoDetails?.author; + songInfo.views = data?.videoDetails?.viewCount; + songInfo.imageSrc = data?.videoDetails?.thumbnail?.thumbnails?.pop()?.url; + songInfo.songDuration = data?.videoDetails?.lengthSeconds; + songInfo.image = await getImage(songInfo.imageSrc); + songInfo.uploadDate = data?.microformat?.microformatDataRenderer?.uploadDate; +} + const registerProvider = (win) => { // This variable will be filled with the callbacks once they register const callbacks = []; @@ -105,32 +64,21 @@ const registerProvider = (win) => { }; win.on("page-title-updated", async () => { - // Save the old title temporarily - const oldTitle = songInfo.title; // Get and set the new data - songInfo.title = await getTitle(win); songInfo.isPaused = await getPausedStatus(win); - const { songDuration, elapsedSeconds } = await getProgress(win); - songInfo.songDuration = songDuration; + const elapsedSeconds = await getProgress(win); songInfo.elapsedSeconds = elapsedSeconds; - // If title changed then we do need to update other info - if (oldTitle !== songInfo.title) { - const subInfo = await getSubInfo(win); - songInfo.artist = subInfo[0]; - songInfo.views = subInfo[1]; - songInfo.likes = subInfo[2]; - songInfo.imageSrc = await getImageSrc(win); - songInfo.image = await getImage(songInfo.imageSrc); - } - // Trigger the callbacks callbacks.forEach((c) => { c(songInfo); }); }); + // This will be called when the song-info-front finds a new request with song data + ipcMain.on('song-info-request', handleData); + return registerCallback; };