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,
} from './types';

export const splitPaginatedQuery = (key: any) => {
    const { startIndex, limit, ...query } = key;

    return {
        pagination: {
            limit,
            startIndex,
        },
        query,
    };
};

export type QueryPagination = {
    limit?: number;
    startIndex?: number;
};

export const queryKeys: Record<
    string,
    Record<string, (...props: any) => 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?: Omit<AlbumArtistListQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && pagination)
                return [serverId, 'albumArtists', 'list', query, pagination] as const;
            if (query) return [serverId, 'albumArtists', 'list', query] 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?: Omit<AlbumListQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && pagination)
                return [serverId, 'albums', 'list', query, pagination] as const;
            if (query) return [serverId, 'albums', 'list', query] as const;
            return [serverId, 'albums', 'list'] 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?: Omit<ArtistListQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && pagination)
                return [serverId, 'artists', 'list', query, pagination] as const;
            if (query) return [serverId, 'artists', 'list', query] as const;
            return [serverId, 'artists', 'list'] as const;
        },
        root: (serverId: string) => [serverId, 'artists'] as const,
    },
    genres: {
        list: (serverId: string) => [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?: Omit<PlaylistDetailQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && pagination)
                return [serverId, 'playlists', id, 'detail', query, pagination] as const;
            if (query) return [serverId, 'playlists', id, 'detail', query] as const;
            if (id) return [serverId, 'playlists', id, 'detail'] as const;
            return [serverId, 'playlists', 'detail'] as const;
        },
        detailSongList: (
            serverId: string,
            id: string,
            query?: Omit<PlaylistSongListQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && pagination)
                return [serverId, 'playlists', id, 'detailSongList', query, pagination] as const;
            if (query && id) return [serverId, 'playlists', id, 'detailSongList', query] as const;
            if (id) return [serverId, 'playlists', id, 'detailSongList'] as const;
            return [serverId, 'playlists', 'detailSongList'] as const;
        },
        list: (serverId: string, query?: PlaylistListQuery) => {
            if (query) return [serverId, 'playlists', 'list', query] as const;
            return [serverId, 'playlists', 'list'] as const;
        },
        root: (serverId: string) => [serverId, 'playlists'] as const,
        songList: (
            serverId: string,
            id?: string,
            query?: Omit<PlaylistSongListQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && id && pagination)
                return [serverId, 'playlists', id, 'songList', query, pagination] as const;
            if (query && id) return [serverId, 'playlists', id, 'songList', query] 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?: Omit<SongListQuery, 'startIndex' | 'limit'>,
            pagination?: QueryPagination,
        ) => {
            if (query && pagination) return [serverId, 'songs', 'list', query, pagination] as const;
            if (query) return [serverId, 'songs', 'list', query] as const;
            return [serverId, 'songs', 'list'] as const;
        },
        lyrics: (serverId: string, query?: LyricsQuery) => {
            if (query) return [serverId, 'song', 'lyrics', 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,
    },
    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,
    },
};