mirror of
https://github.com/jeffvli/feishin.git
synced 2026-06-17 17:04:16 +02:00
316 lines
13 KiB
TypeScript
316 lines
13 KiB
TypeScript
import { SetActivity, StatusDisplayType } from '@xhayper/discord-rpc';
|
|
import isElectron from 'is-electron';
|
|
import { useCallback, useEffect, useRef, useState } from 'react';
|
|
|
|
import { controller } from '/@/renderer/api/controller';
|
|
import {
|
|
DiscordDisplayType,
|
|
DiscordLinkType,
|
|
useAppStore,
|
|
useDiscordSettings,
|
|
useGeneralSettings,
|
|
usePlayerStore,
|
|
useTimestampStoreBase,
|
|
} from '/@/renderer/store';
|
|
import { sentenceCase } from '/@/renderer/utils';
|
|
import { LogCategory, logFn } from '/@/renderer/utils/logger';
|
|
import { logMsg } from '/@/renderer/utils/logger-message';
|
|
import { QueueSong, ServerType } from '/@/shared/types/domain-types';
|
|
import { PlayerStatus } from '/@/shared/types/types';
|
|
|
|
const discordRpc = isElectron() ? window.api.discordRpc : null;
|
|
type ActivityState = [QueueSong | undefined, number, PlayerStatus];
|
|
|
|
const MAX_FIELD_LENGTH = 127;
|
|
|
|
const truncate = (field: string) =>
|
|
field.length <= MAX_FIELD_LENGTH ? field : field.substring(0, MAX_FIELD_LENGTH - 1) + '…';
|
|
|
|
export const useDiscordRpc = () => {
|
|
const discordSettings = useDiscordSettings();
|
|
const generalSettings = useGeneralSettings();
|
|
const privateMode = useAppStore((state) => state.privateMode);
|
|
const [lastUniqueId, setlastUniqueId] = useState('');
|
|
|
|
const previousEnabledRef = useRef<boolean>(discordSettings.enabled);
|
|
|
|
const setActivity = useCallback(
|
|
async (current: ActivityState, previous: ActivityState) => {
|
|
if (
|
|
!current[0] || // No track
|
|
current[1] === 0 || // Start of track
|
|
(current[2] === 'paused' && !discordSettings.showPaused) // Track paused with show paused setting disabled
|
|
) {
|
|
let reason: string;
|
|
if (!current[0]) {
|
|
reason = 'no_track';
|
|
} else if (current[1] === 0) {
|
|
reason = 'start_of_track';
|
|
} else {
|
|
reason = 'paused_with_show_paused_disabled';
|
|
}
|
|
|
|
logFn.debug(logMsg[LogCategory.EXTERNAL].discordRpcActivityCleared, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
reason,
|
|
status: current[2],
|
|
},
|
|
});
|
|
return discordRpc?.clearActivity();
|
|
}
|
|
|
|
// Handle change detection
|
|
const song = current[0];
|
|
const trackChanged = lastUniqueId !== song._uniqueId;
|
|
|
|
/*
|
|
1. If the song has just started, update status
|
|
2. If we jump more then 1.2 seconds from last state, update status to match
|
|
3. If the current song id is completely different, update status
|
|
4. If the player state changed, update status
|
|
*/
|
|
if (
|
|
previous[1] === 0 ||
|
|
Math.abs(current[1] - previous[1]) > 1.2 ||
|
|
trackChanged ||
|
|
current[2] !== previous[2]
|
|
) {
|
|
if (trackChanged) {
|
|
logFn.debug(logMsg[LogCategory.EXTERNAL].discordRpcTrackChanged, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
artistName: song.artists?.[0]?.name,
|
|
songId: song._uniqueId,
|
|
songName: song.name,
|
|
},
|
|
});
|
|
setlastUniqueId(song._uniqueId);
|
|
}
|
|
|
|
let reason: string;
|
|
if (previous[1] === 0) {
|
|
reason = 'song_started';
|
|
} else if (Math.abs(current[1] - previous[1]) > 1.2) {
|
|
reason = 'time_jump';
|
|
} else if (trackChanged) {
|
|
reason = 'track_changed';
|
|
} else {
|
|
reason = 'player_state_changed';
|
|
}
|
|
|
|
logFn.debug(logMsg[LogCategory.EXTERNAL].discordRpcActivityUpdate, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
currentStatus: current[2],
|
|
currentTime: current[1],
|
|
previousStatus: previous[2],
|
|
previousTime: previous[1],
|
|
reason,
|
|
trackChanged,
|
|
},
|
|
});
|
|
|
|
const start = Math.round(Date.now() - current[1] * 1000);
|
|
const end = Math.round(start + song.duration);
|
|
|
|
const artists = song?.artists.map((artist) => artist.name).join(', ');
|
|
|
|
const statusDisplayMap = {
|
|
[DiscordDisplayType.ARTIST_NAME]: StatusDisplayType.STATE,
|
|
[DiscordDisplayType.FEISHIN]: StatusDisplayType.NAME,
|
|
[DiscordDisplayType.SONG_NAME]: StatusDisplayType.DETAILS,
|
|
};
|
|
|
|
const activity: SetActivity = {
|
|
details: truncate((song?.name && song.name.padEnd(2, ' ')) || 'Idle'),
|
|
instance: false,
|
|
largeImageKey: undefined,
|
|
largeImageText: truncate(
|
|
(song?.album && song.album.padEnd(2, ' ')) || 'Unknown album',
|
|
),
|
|
smallImageKey: undefined,
|
|
smallImageText: sentenceCase(current[2]),
|
|
state: truncate((artists && artists.padEnd(2, ' ')) || 'Unknown artist'),
|
|
statusDisplayType: statusDisplayMap[discordSettings.displayType],
|
|
// I would love to use the actual type as opposed to hardcoding to 2,
|
|
// but manually installing the discord-types package appears to break things
|
|
type: discordSettings.showAsListening ? 2 : 0,
|
|
};
|
|
|
|
if (
|
|
(discordSettings.linkType == DiscordLinkType.LAST_FM ||
|
|
discordSettings.linkType == DiscordLinkType.MBZ_LAST_FM) &&
|
|
song?.artistName
|
|
) {
|
|
activity.stateUrl =
|
|
'https://www.last.fm/music/' + encodeURIComponent(song.artists[0].name);
|
|
activity.detailsUrl =
|
|
'https://www.last.fm/music/' +
|
|
encodeURIComponent(song.albumArtists[0].name) +
|
|
'/' +
|
|
encodeURIComponent(song.album || '_') +
|
|
'/' +
|
|
encodeURIComponent(song.name);
|
|
}
|
|
|
|
if (
|
|
discordSettings.linkType == DiscordLinkType.MBZ ||
|
|
discordSettings.linkType == DiscordLinkType.MBZ_LAST_FM
|
|
) {
|
|
if (song?.mbzTrackId) {
|
|
activity.detailsUrl = 'https://musicbrainz.org/track/' + song.mbzTrackId;
|
|
} else if (song?.mbzRecordingId) {
|
|
activity.detailsUrl =
|
|
'https://musicbrainz.org/recording/' + song.mbzRecordingId;
|
|
}
|
|
}
|
|
|
|
if (current[2] === PlayerStatus.PLAYING) {
|
|
if (start && end) {
|
|
activity.startTimestamp = start;
|
|
activity.endTimestamp = end;
|
|
}
|
|
|
|
activity.smallImageKey = 'playing';
|
|
} else {
|
|
activity.smallImageKey = 'paused';
|
|
}
|
|
|
|
if (discordSettings.showServerImage && song) {
|
|
if (song._serverType === ServerType.JELLYFIN && song.imageUrl) {
|
|
activity.largeImageKey = song.imageUrl;
|
|
} else if (song._serverType === ServerType.NAVIDROME) {
|
|
try {
|
|
const info = await controller.getAlbumInfo({
|
|
apiClientProps: { serverId: song._serverId },
|
|
query: { id: song.albumId },
|
|
});
|
|
|
|
if (info.imageUrl) {
|
|
activity.largeImageKey = info.imageUrl;
|
|
}
|
|
} catch {
|
|
/* empty */
|
|
}
|
|
}
|
|
}
|
|
|
|
if (
|
|
activity.largeImageKey === undefined &&
|
|
generalSettings.lastfmApiKey &&
|
|
song?.album &&
|
|
song?.albumArtists.length
|
|
) {
|
|
const albumInfo = await fetch(
|
|
`https://ws.audioscrobbler.com/2.0/?method=album.getinfo&api_key=${generalSettings.lastfmApiKey}&artist=${encodeURIComponent(song.albumArtists[0].name)}&album=${encodeURIComponent(song.album)}&format=json`,
|
|
);
|
|
|
|
const albumInfoJson = await albumInfo.json();
|
|
|
|
if (albumInfoJson.album?.image?.[3]['#text']) {
|
|
activity.largeImageKey = albumInfoJson.album.image[3]['#text'];
|
|
}
|
|
}
|
|
|
|
// Fall back to default icon if not set
|
|
if (!activity.largeImageKey) {
|
|
activity.largeImageKey = 'icon';
|
|
}
|
|
|
|
// Initialize if needed
|
|
const isConnected = await discordRpc?.isConnected();
|
|
if (!isConnected) {
|
|
logFn.debug(logMsg[LogCategory.EXTERNAL].discordRpcInitialized, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
clientId: discordSettings.clientId,
|
|
},
|
|
});
|
|
await discordRpc?.initialize(discordSettings.clientId);
|
|
}
|
|
|
|
logFn.debug(logMsg[LogCategory.EXTERNAL].discordRpcSetActivity, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
albumName: song.album,
|
|
artistName: song.artists?.[0]?.name,
|
|
displayType: discordSettings.displayType,
|
|
hasLargeImage: !!activity.largeImageKey,
|
|
hasTimestamps: !!(activity.startTimestamp && activity.endTimestamp),
|
|
showAsListening: discordSettings.showAsListening,
|
|
songName: song.name,
|
|
status: current[2],
|
|
},
|
|
});
|
|
discordRpc?.setActivity(activity);
|
|
} else {
|
|
logFn.debug(logMsg[LogCategory.EXTERNAL].discordRpcUpdateSkipped, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
currentStatus: current[2],
|
|
currentTime: current[1],
|
|
previousStatus: previous[2],
|
|
previousTime: previous[1],
|
|
timeDiff: Math.abs(current[1] - previous[1]),
|
|
trackChanged,
|
|
},
|
|
});
|
|
}
|
|
},
|
|
[
|
|
discordSettings.showAsListening,
|
|
discordSettings.showServerImage,
|
|
discordSettings.showPaused,
|
|
generalSettings.lastfmApiKey,
|
|
discordSettings.clientId,
|
|
discordSettings.displayType,
|
|
discordSettings.linkType,
|
|
lastUniqueId,
|
|
],
|
|
);
|
|
|
|
// Quit Discord RPC if it was enabled and is now disabled
|
|
useEffect(() => {
|
|
if ((!discordSettings.enabled || privateMode) && Boolean(previousEnabledRef.current)) {
|
|
logFn.info(logMsg[LogCategory.EXTERNAL].discordRpcQuit, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
enabled: discordSettings.enabled,
|
|
privateMode,
|
|
},
|
|
});
|
|
|
|
previousEnabledRef.current = false;
|
|
|
|
return discordRpc?.quit();
|
|
}
|
|
}, [discordSettings.clientId, privateMode, discordSettings.enabled]);
|
|
|
|
useEffect(() => {
|
|
if (!discordSettings.enabled || privateMode) {
|
|
return;
|
|
}
|
|
|
|
logFn.info(logMsg[LogCategory.EXTERNAL].discordRpcEnabled, {
|
|
category: LogCategory.EXTERNAL,
|
|
meta: {
|
|
clientId: discordSettings.clientId,
|
|
subscribed: true,
|
|
},
|
|
});
|
|
|
|
const unsubSongChange = usePlayerStore.subscribe((state): ActivityState => {
|
|
const currentSong = state.getCurrentSong();
|
|
const currentTime = useTimestampStoreBase.getState().timestamp;
|
|
const status = state.player.status;
|
|
|
|
return [currentSong, currentTime, status];
|
|
}, setActivity);
|
|
|
|
return () => {
|
|
unsubSongChange();
|
|
};
|
|
}, [discordSettings.clientId, discordSettings.enabled, privateMode, setActivity]);
|
|
};
|