import { QueryFunctionContext } from '@tanstack/react-query'; import { LyricSource } from './types'; import type { AlbumListQuery, SongListQuery, AlbumDetailQuery, AlbumArtistListQuery, ArtistListQuery, PlaylistListQuery, PlaylistDetailQuery, PlaylistSongListQuery, UserListQuery, AlbumArtistDetailQuery, TopSongListQuery, SearchQuery, SongDetailQuery, RandomSongListQuery, LyricsQuery, LyricSearchQuery, GenreListQuery, SimilarSongsQuery, } from './types'; export const splitPaginatedQuery = (key: any) => { const { startIndex, limit, ...filter } = key || {}; if (startIndex !== undefined || limit !== undefined) { return { filter, pagination: { limit, startIndex, }, }; } return { filter, pagination: undefined, }; }; export type QueryPagination = { limit?: number; startIndex?: number; }; export const queryKeys: Record< string, Record QueryFunctionContext['queryKey']> > = { albumArtists: { detail: (serverId: string, query?: AlbumArtistDetailQuery) => { if (query) return [serverId, 'albumArtists', 'detail', query] as const; return [serverId, 'albumArtists', 'detail'] as const; }, list: (serverId: string, query?: AlbumArtistListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination) { return [serverId, 'albumArtists', 'list', filter, pagination] as const; } if (query) { return [serverId, 'albumArtists', 'list', filter] as const; } return [serverId, 'albumArtists', 'list'] as const; }, root: (serverId: string) => [serverId, 'albumArtists'] as const, topSongs: (serverId: string, query?: TopSongListQuery) => { if (query) return [serverId, 'albumArtists', 'topSongs', query] as const; return [serverId, 'albumArtists', 'topSongs'] as const; }, }, albums: { detail: (serverId: string, query?: AlbumDetailQuery) => [serverId, 'albums', 'detail', query] as const, list: (serverId: string, query?: AlbumListQuery, artistId?: string) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination && artistId) { return [serverId, 'albums', 'list', artistId, filter, pagination] as const; } if (query && pagination) { return [serverId, 'albums', 'list', filter, pagination] as const; } if (query && artistId) { return [serverId, 'albums', 'list', artistId, filter] as const; } if (query) { return [serverId, 'albums', 'list', filter] as const; } return [serverId, 'albums', 'list'] as const; }, related: (serverId: string, id: string, query?: AlbumDetailQuery) => { if (query) { return [serverId, 'albums', id, 'related', query] as const; } return [serverId, 'albums', id, 'related'] as const; }, root: (serverId: string) => [serverId, 'albums'], serverRoot: (serverId: string) => [serverId, 'albums'], songs: (serverId: string, query: SongListQuery) => [serverId, 'albums', 'songs', query] as const, }, artists: { list: (serverId: string, query?: ArtistListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination) { return [serverId, 'artists', 'list', filter, pagination] as const; } if (query) { return [serverId, 'artists', 'list', filter] as const; } return [serverId, 'artists', 'list'] as const; }, root: (serverId: string) => [serverId, 'artists'] as const, }, genres: { list: (serverId: string, query?: GenreListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination) { return [serverId, 'genres', 'list', filter, pagination] as const; } if (query) { return [serverId, 'genres', 'list', filter] as const; } return [serverId, 'genres', 'list'] as const; }, root: (serverId: string) => [serverId, 'genres'] as const, }, musicFolders: { list: (serverId: string) => [serverId, 'musicFolders', 'list'] as const, }, playlists: { detail: (serverId: string, id?: string, query?: PlaylistDetailQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination) { return [serverId, 'playlists', id, 'detail', filter, pagination] as const; } if (query) { return [serverId, 'playlists', id, 'detail', filter] as const; } if (id) return [serverId, 'playlists', id, 'detail'] as const; return [serverId, 'playlists', 'detail'] as const; }, detailSongList: (serverId: string, id: string, query?: PlaylistSongListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && id && pagination) { return [serverId, 'playlists', id, 'detailSongList', filter, pagination] as const; } if (query && id) { return [serverId, 'playlists', id, 'detailSongList', filter] as const; } if (id) return [serverId, 'playlists', id, 'detailSongList'] as const; return [serverId, 'playlists', 'detailSongList'] as const; }, list: (serverId: string, query?: PlaylistListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination) { return [serverId, 'playlists', 'list', filter, pagination] as const; } if (query) { return [serverId, 'playlists', 'list', filter] as const; } return [serverId, 'playlists', 'list'] as const; }, root: (serverId: string) => [serverId, 'playlists'] as const, songList: (serverId: string, id?: string, query?: PlaylistSongListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && id && pagination) { return [serverId, 'playlists', id, 'songList', filter, pagination] as const; } if (query && id) { return [serverId, 'playlists', id, 'songList', filter] as const; } if (id) return [serverId, 'playlists', id, 'songList'] as const; return [serverId, 'playlists', 'songList'] as const; }, }, search: { list: (serverId: string, query?: SearchQuery) => { if (query) return [serverId, 'search', 'list', query] as const; return [serverId, 'search', 'list'] as const; }, root: (serverId: string) => [serverId, 'search'] as const, }, server: { root: (serverId: string) => [serverId] as const, }, songs: { detail: (serverId: string, query?: SongDetailQuery) => { if (query) return [serverId, 'songs', 'detail', query] as const; return [serverId, 'songs', 'detail'] as const; }, list: (serverId: string, query?: SongListQuery) => { const { pagination, filter } = splitPaginatedQuery(query); if (query && pagination) { return [serverId, 'songs', 'list', filter, pagination] as const; } if (query) { return [serverId, 'songs', 'list', filter] as const; } return [serverId, 'songs', 'list'] as const; }, lyrics: (serverId: string, query?: LyricsQuery) => { if (query) return [serverId, 'song', 'lyrics', 'select', query] as const; return [serverId, 'song', 'lyrics'] as const; }, lyricsByRemoteId: (searchQuery: { remoteSongId: string; remoteSource: LyricSource }) => { return ['song', 'lyrics', 'remote', searchQuery] as const; }, lyricsSearch: (query?: LyricSearchQuery) => { if (query) return ['lyrics', 'search', query] as const; return ['lyrics', 'search'] as const; }, randomSongList: (serverId: string, query?: RandomSongListQuery) => { if (query) return [serverId, 'songs', 'randomSongList', query] as const; return [serverId, 'songs', 'randomSongList'] as const; }, root: (serverId: string) => [serverId, 'songs'] as const, similar: (serverId: string, query?: SimilarSongsQuery) => { if (query) return [serverId, 'song', 'similar', query] as const; return [serverId, 'song', 'similar'] as const; }, }, users: { list: (serverId: string, query?: UserListQuery) => { if (query) return [serverId, 'users', 'list', query] as const; return [serverId, 'users', 'list'] as const; }, root: (serverId: string) => [serverId, 'users'] as const, }, };