various code improvements and some boyscout rule fixes :)

This commit is contained in:
Rick van Lieshout 2023-07-31 13:43:32 +02:00
parent de8a5a1b07
commit 586f7b595b
10 changed files with 180 additions and 99 deletions

View File

@ -1,10 +1,14 @@
{ {
"cSpell.words": [ "cSpell.words": [
"Brainz",
"flac", "flac",
"geqnfr", "geqnfr",
"hifi", "hifi",
"listenbrainz",
"playpause", "playpause",
"rescrobbler", "rescrobbler",
"scrobble",
"scrobbling",
"Songwhip", "Songwhip",
"trackid", "trackid",
"tracklist", "tracklist",

View File

@ -1,4 +0,0 @@
export const statuses = {
playing: "playing",
paused: "paused",
};

View File

@ -1,12 +1,35 @@
import axios from "axios"; import axios from "axios";
import { settingsStore } from "../../scripts/settings"; import { ipcRenderer } from "electron";
import Store from "electron-store";
import { settings } from "../../constants/settings"; import { settings } from "../../constants/settings";
import { MediaStatus } from "../../models/mediaStatus"; import { MediaStatus } from "../../models/mediaStatus";
import Store from "electron-store"; import { settingsStore } from "../../scripts/settings";
import { Logger } from "../logger";
import { StoreData } from "./models/storeData";
const ListenBrainzStore = new Store({name: "listenbrainz"}); const ListenBrainzStore = new Store({ name: "listenbrainz" });
export const ListenBrainzConstants = {
oldData: "oldData",
};
export class ListenBrainz { export class ListenBrainz {
/**
* Create the object to store old information in the Store :)
* @param title
* @param artists
* @param duration
* @returns data passed along in an object + a "listenedAt" key with the current time
*/
private static constructStoreData(title: string, artists: string, duration: number): StoreData {
return {
listenedAt: Math.floor(new Date().getTime() / 1000),
title,
artists,
duration,
};
}
/** /**
* Call the ListenBrainz API and create playing now payload and scrobble old song * Call the ListenBrainz API and create playing now payload and scrobble old song
* @param title * @param title
@ -14,70 +37,96 @@ export class ListenBrainz {
* @param status * @param status
* @param duration * @param duration
*/ */
public static async scrobble(title: string, artists: string, status: string, duration: number): Promise<any> { public static async scrobble(
title: string,
artists: string,
status: string,
duration: number
): Promise<void> {
try { try {
if (status == MediaStatus.paused) { if (status === MediaStatus.paused) {
return false; return;
} else { } else {
// Fetches the OldData required for scrobbling and proceeds to construct a playing_now data payload for the Playing Now area // Fetches the oldData required for scrobbling and proceeds to construct a playing_now data payload for the Playing Now area
const OldData = ListenBrainzStore.get("OldData") as string[]; const oldData = ListenBrainzStore.get(ListenBrainzConstants.oldData) as StoreData;
const playing_data = { const playing_data = {
listen_type: "playing_now", listen_type: "playing_now",
payload: [ payload: [
{ {
track_metadata: { track_metadata: {
additional_info: { additional_info: {
media_player: "Tidal Hi-Fi", media_player: "Tidal Hi-Fi",
submission_client: "Tidal Hi-Fi", submission_client: "Tidal Hi-Fi",
music_service: "tidal.com", music_service: "tidal.com",
duration: duration, duration: duration,
}, },
artist_name: artists, artist_name: artists,
track_name: title, track_name: title,
} },
} },
] ],
}; };
await axios.post(`${settingsStore.get(settings.ListenBrainz.api)}/1/submit-listens`, playing_data, { await axios.post(
headers:{ `${settingsStore.get<string, string>(settings.ListenBrainz.api)}/1/submit-listens`,
"Content-Type": "application/json", playing_data,
"Authorization": `Token ${settingsStore.get(settings.ListenBrainz.token)}` {
} headers: {
}); "Content-Type": "application/json",
if (!OldData) { Authorization: `Token ${settingsStore.get<string, string>(
ListenBrainzStore.set("OldData", [Math.floor(new Date().getTime() / 1000), title, artists, duration]); settings.ListenBrainz.token
} else if (OldData[1] != title) { )}`,
},
}
);
if (!oldData) {
ListenBrainzStore.set(
ListenBrainzConstants.oldData,
this.constructStoreData(title, artists, duration)
);
} else {
if (oldData.title !== title) {
// This constructs the data required to scrobble the data after the song finishes // This constructs the data required to scrobble the data after the song finishes
const scrobble_data = { const scrobble_data = {
listen_type: "single", listen_type: "single",
payload: [ payload: [
{ {
listened_at: OldData[0], listened_at: oldData.listenedAt,
track_metadata: { track_metadata: {
additional_info: { additional_info: {
media_player: "Tidal Hi-Fi", media_player: "Tidal Hi-Fi",
submission_client: "Tidal Hi-Fi", submission_client: "Tidal Hi-Fi",
music_service: "listen.tidal.com", music_service: "listen.tidal.com",
duration: OldData[3], duration: oldData.duration,
}, },
artist_name: OldData[2], artist_name: oldData.artists,
track_name: OldData[1], track_name: oldData.title,
} },
} },
] ],
}; };
await axios.post(`${settingsStore.get(settings.ListenBrainz.api)}/1/submit-listens`, scrobble_data, { await axios.post(
headers:{ `${settingsStore.get<string, string>(settings.ListenBrainz.api)}/1/submit-listens`,
"Content-Type": "application/json", scrobble_data,
"Authorization": `Token ${settingsStore.get(settings.ListenBrainz.token)}` {
} headers: {
}); "Content-Type": "application/json",
ListenBrainzStore.set("OldData", [Math.floor(new Date().getTime() / 1000), title, artists, duration]); Authorization: `Token ${settingsStore.get<string, string>(
settings.ListenBrainz.token
)}`,
},
}
);
ListenBrainzStore.set(
ListenBrainzConstants.oldData,
this.constructStoreData(title, artists, duration)
);
}
} }
} }
} catch (error) { } catch (error) {
console.log(JSON.stringify(error)); const logger = new Logger(ipcRenderer);
logger.log(JSON.stringify(error));
} }
} }
} }

View File

@ -0,0 +1,9 @@
/**
* Data saved for ListenBrainz
*/
export interface StoreData {
listenedAt: number;
title: string;
artists: string;
duration: number;
}

View File

@ -26,8 +26,7 @@ export class Logger {
public log(content: string, object: object = {}) { public log(content: string, object: object = {}) {
if (this.ipcRenderer) { if (this.ipcRenderer) {
this.ipcRenderer.send(globalEvents.log, { content, object }); this.ipcRenderer.send(globalEvents.log, { content, object });
} else {
console.log(`${content} \n ${JSON.stringify(object, null, 2)}`);
} }
console.log(`${content} \n ${JSON.stringify(object, null, 2)}`);
} }
} }

View File

@ -58,7 +58,7 @@ function setFlags() {
} }
/** /**
* Update the menuBarVisbility according to the store value * Update the menuBarVisibility according to the store value
* *
*/ */
function syncMenuBarWithStore() { function syncMenuBarWithStore() {

View File

@ -211,9 +211,12 @@
<span class="switch__slider"></span> <span class="switch__slider"></span>
</label> </label>
</div> </div>
</div>
<div class="group">
<p class="group__title">ListenBrainz</p>
<div class="group__option"> <div class="group__option">
<div class="group__description"> <div class="group__description">
<h4>ListenBrainz</h4> <h4>Enable ListenBrainz</h4>
<p>Scrobble your listens directly to ListenBrainz.</p> <p>Scrobble your listens directly to ListenBrainz.</p>
</div> </div>
<label class="switch"> <label class="switch">

View File

@ -4,20 +4,25 @@ import fs from "fs";
import Player from "mpris-service"; import Player from "mpris-service";
import { globalEvents } from "./constants/globalEvents"; import { globalEvents } from "./constants/globalEvents";
import { settings } from "./constants/settings"; import { settings } from "./constants/settings";
import { statuses } from "./constants/statuses";
import { Songwhip } from "./features/songwhip/songwhip"; import { Songwhip } from "./features/songwhip/songwhip";
import { ListenBrainz, ListenBrainzStore } from "./features/listenbrainz/listenbrainz"; import {
ListenBrainz,
ListenBrainzConstants,
ListenBrainzStore,
} from "./features/listenbrainz/listenbrainz";
import { Options } from "./models/options"; import { Options } from "./models/options";
import { downloadFile } from "./scripts/download"; import { downloadFile } from "./scripts/download";
import { addHotkey } from "./scripts/hotkeys"; import { addHotkey } from "./scripts/hotkeys";
import { settingsStore } from "./scripts/settings"; import { settingsStore } from "./scripts/settings";
import { setTitle } from "./scripts/window-functions"; import { setTitle } from "./scripts/window-functions";
import { StoreData } from "./features/listenbrainz/models/storeData";
import { MediaStatus } from "./models/mediaStatus";
const notificationPath = `${app.getPath("userData")}/notification.jpg`; const notificationPath = `${app.getPath("userData")}/notification.jpg`;
const appName = "Tidal Hifi"; const appName = "Tidal Hifi";
let currentSong = ""; let currentSong = "";
let player: Player; let player: Player;
let currentPlayStatus = statuses.paused; let currentPlayStatus = MediaStatus.paused;
const elements = { const elements = {
play: '*[data-test="play"]', play: '*[data-test="play"]',
@ -106,7 +111,7 @@ const elements = {
window.location.href.includes("/playlist/") || window.location.href.includes("/playlist/") ||
window.location.href.includes("/mix/") window.location.href.includes("/mix/")
) { ) {
if (currentPlayStatus === statuses.playing) { if (currentPlayStatus === MediaStatus.playing) {
// find the currently playing element from the list (which might be in an album icon), traverse back up to the mediaItem (row) and select the album cell. // find the currently playing element from the list (which might be in an album icon), traverse back up to the mediaItem (row) and select the album cell.
// document.querySelector("[class^='isPlayingIcon'], [data-test-is-playing='true']").closest('[data-type="mediaItem"]').querySelector('[class^="album"]').textContent // document.querySelector("[class^='isPlayingIcon'], [data-test-is-playing='true']").closest('[data-type="mediaItem"]').querySelector('[class^="album"]').textContent
const row = window.document.querySelector(this.currentlyPlaying).closest(this.mediaItem); const row = window.document.querySelector(this.currentlyPlaying).closest(this.mediaItem);
@ -179,7 +184,7 @@ function addCustomCss() {
* make sure it returns a number, if not use the default * make sure it returns a number, if not use the default
*/ */
function getUpdateFrequency() { function getUpdateFrequency() {
const storeValue = settingsStore.get(settings.updateFrequency) as number; const storeValue = settingsStore.get<string, number>(settings.updateFrequency);
const defaultValue = 500; const defaultValue = 500;
if (!isNaN(storeValue)) { if (!isNaN(storeValue)) {
@ -280,7 +285,7 @@ function handleLogout() {
defaultId: 2, defaultId: 2,
}) })
.then((result: { response: number }) => { .then((result: { response: number }) => {
if (logoutOptions.indexOf("Yes, please") == result.response) { if (logoutOptions.indexOf("Yes, please") === result.response) {
for (let i = 0; i < window.localStorage.length; i++) { for (let i = 0; i < window.localStorage.length; i++) {
const key = window.localStorage.key(i); const key = window.localStorage.key(i);
if (key.startsWith("_TIDAL_activeSession")) { if (key.startsWith("_TIDAL_activeSession")) {
@ -322,6 +327,8 @@ function addIPCEventListeners() {
case globalEvents.pause: case globalEvents.pause:
elements.click("pause"); elements.click("pause");
break; break;
default:
break;
} }
}); });
}); });
@ -336,9 +343,9 @@ function getCurrentlyPlayingStatus() {
// if pause button is visible tidal is playing // if pause button is visible tidal is playing
if (pause) { if (pause) {
status = statuses.playing; status = MediaStatus.playing;
} else { } else {
status = statuses.paused; status = MediaStatus.paused;
} }
return status; return status;
} }
@ -363,27 +370,41 @@ function updateMediaInfo(options: Options, notify: boolean) {
if (settingsStore.get(settings.notifications) && notify) { if (settingsStore.get(settings.notifications) && notify) {
new Notification({ title: options.title, body: options.artists, icon: options.icon }).show(); new Notification({ title: options.title, body: options.artists, icon: options.icon }).show();
} }
if (player) { updateMpris(options);
player.metadata = { updateListenBrainz(options);
...player.metadata, }
...{ }
"xesam:title": options.title,
"xesam:artist": [options.artists], function updateMpris(options: Options) {
"xesam:album": options.album, if (player) {
"mpris:artUrl": options.image, player.metadata = {
"mpris:length": convertDuration(options.duration) * 1000 * 1000, ...player.metadata,
"mpris:trackid": "/org/mpris/MediaPlayer2/track/" + getTrackID(), ...{
}, "xesam:title": options.title,
}; "xesam:artist": [options.artists],
player.playbackStatus = options.status == statuses.paused ? "Paused" : "Playing"; "xesam:album": options.album,
} "mpris:artUrl": options.image,
if (settingsStore.get(settings.ListenBrainz.enabled)) { "mpris:length": convertDuration(options.duration) * 1000 * 1000,
const data = ListenBrainzStore.get("OldData") as string[]; "mpris:trackid": "/org/mpris/MediaPlayer2/track/" + getTrackID(),
if (data && data[1] !== options.title) { },
ListenBrainz.scrobble(options.title, options.artists, options.status, convertDuration(options.duration)); };
} else if (!data) { player.playbackStatus = options.status === MediaStatus.paused ? "Paused" : "Playing";
ListenBrainz.scrobble(options.title, options.artists, options.status, convertDuration(options.duration)); }
} }
function updateListenBrainz(options: Options) {
if (settingsStore.get(settings.ListenBrainz.enabled)) {
const oldData = ListenBrainzStore.get(ListenBrainzConstants.oldData) as StoreData;
if (
(!oldData && options.status === MediaStatus.playing) ||
(oldData && oldData.title !== options.title)
) {
ListenBrainz.scrobble(
options.title,
options.artists,
options.status,
convertDuration(options.duration)
);
} }
} }
} }

View File

@ -1,11 +1,11 @@
import { BrowserWindow, dialog } from "electron"; import { BrowserWindow, dialog } from "electron";
import express, { Response } from "express"; import express, { Response } from "express";
import fs from "fs"; import fs from "fs";
import { settings } from "../constants/settings";
import { MediaStatus } from "../models/mediaStatus";
import { globalEvents } from "./../constants/globalEvents"; import { globalEvents } from "./../constants/globalEvents";
import { statuses } from "./../constants/statuses";
import { mediaInfo } from "./mediaInfo"; import { mediaInfo } from "./mediaInfo";
import { settingsStore } from "./settings"; import { settingsStore } from "./settings";
import { settings } from "../constants/settings";
/** /**
* Function to enable tidal-hifi's express api * Function to enable tidal-hifi's express api
@ -44,7 +44,7 @@ export const startExpress = (mainWindow: BrowserWindow) => {
expressApp.get("/next", (req, res) => handleGlobalEvent(res, globalEvents.next)); expressApp.get("/next", (req, res) => handleGlobalEvent(res, globalEvents.next));
expressApp.get("/previous", (req, res) => handleGlobalEvent(res, globalEvents.previous)); expressApp.get("/previous", (req, res) => handleGlobalEvent(res, globalEvents.previous));
expressApp.get("/playpause", (req, res) => { expressApp.get("/playpause", (req, res) => {
if (mediaInfo.status == statuses.playing) { if (mediaInfo.status === MediaStatus.playing) {
handleGlobalEvent(res, globalEvents.pause); handleGlobalEvent(res, globalEvents.pause);
} else { } else {
handleGlobalEvent(res, globalEvents.play); handleGlobalEvent(res, globalEvents.play);

View File

@ -1,12 +1,12 @@
import { MediaInfo } from "../models/mediaInfo"; import { MediaInfo } from "../models/mediaInfo";
import { statuses } from "./../constants/statuses"; import { MediaStatus } from "../models/mediaStatus";
export const mediaInfo = { export const mediaInfo = {
title: "", title: "",
artists: "", artists: "",
album: "", album: "",
icon: "", icon: "",
status: statuses.paused, status: MediaStatus.paused as string,
url: "", url: "",
current: "", current: "",
duration: "", duration: "",