feishin/src/renderer/api/subsonic/subsonic-controller.ts

import md5 from 'md5';
import { z } from 'zod';
import { ssApiClient } from '/@/renderer/api/subsonic/subsonic-api';
import { ssNormalize } from '/@/renderer/api/subsonic/subsonic-normalize';
import { ssType } from '/@/renderer/api/subsonic/subsonic-types';
import {
    ArtistInfoArgs,
    AuthenticationResponse,
    FavoriteArgs,
    FavoriteResponse,
    LibraryItem,
    MusicFolderListArgs,
    MusicFolderListResponse,
    SetRatingArgs,
    RatingResponse,
    ScrobbleArgs,
    ScrobbleResponse,
    SongListResponse,
    TopSongListArgs,
    SearchArgs,
    SearchResponse,
    RandomSongListResponse,
    RandomSongListArgs,
    ServerInfo,
    ServerInfoArgs,
    StructuredLyricsArgs,
    StructuredLyric,
} from '/@/renderer/api/types';
import { randomString } from '/@/renderer/utils';

const authenticate = async (
    url: string,
    body: {
        legacy?: boolean;
        password: string;
        username: string;
    },
): Promise<AuthenticationResponse> => {
    let credential: string;
    let credentialParams: {
        p?: string;
        s?: string;
        t?: string;
        u: string;
    };

    const cleanServerUrl = url.replace(/\/$/, '');

    if (body.legacy) {
        credential = `u=${body.username}&p=${body.password}`;
        credentialParams = {
            p: body.password,
            u: body.username,
        };
    } else {
        const salt = randomString(12);
        const hash = md5(body.password + salt);
        credential = `u=${body.username}&s=${salt}&t=${hash}`;
        credentialParams = {
            s: salt,
            t: hash,
            u: body.username,
        };
    }

    await ssApiClient({ server: null, url: cleanServerUrl }).authenticate({
        query: {
            c: 'Feishin',
            f: 'json',
            v: '1.13.0',
            ...credentialParams,
        },
    });

    return {
        credential,
        userId: null,
        username: body.username,
    };
};

const getMusicFolderList = async (args: MusicFolderListArgs): Promise<MusicFolderListResponse> => {
    const { apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).getMusicFolderList({});

    if (res.status !== 200) {
        throw new Error('Failed to get music folder list');
    }

    return {
        items: res.body.musicFolders.musicFolder,
        startIndex: 0,
        totalRecordCount: res.body.musicFolders.musicFolder.length,
    };
};

// export const getAlbumArtistDetail = async (
//   args: AlbumArtistDetailArgs,
// ): Promise<SSAlbumArtistDetail> => {
//   const { server, signal, query } = args;
//   const defaultParams = getDefaultParams(server);

//   const searchParams: SSAlbumArtistDetailParams = {
//     id: query.id,
//     ...defaultParams,
//   };

//   const data = await api
//     .get('/getArtist.view', {
//       prefixUrl: server?.url,
//       searchParams,
//       signal,
//     })
//     .json<SSAlbumArtistDetailResponse>();

//   return data.artist;
// };

// const getAlbumArtistList = async (args: AlbumArtistListArgs): Promise<SSAlbumArtistList> => {
//   const { signal, server, query } = args;
//   const defaultParams = getDefaultParams(server);

//   const searchParams: SSAlbumArtistListParams = {
//     musicFolderId: query.musicFolderId,
//     ...defaultParams,
//   };

//   const data = await api
//     .get('rest/getArtists.view', {
//       prefixUrl: server?.url,
//       searchParams,
//       signal,
//     })
//     .json<SSAlbumArtistListResponse>();

//   const artists = (data.artists?.index || []).flatMap((index: SSArtistIndex) => index.artist);

//   return {
//     items: artists,
//     startIndex: query.startIndex,
//     totalRecordCount: null,
//   };
// };

// const getGenreList = async (args: GenreListArgs): Promise<SSGenreList> => {
//   const { server, signal } = args;
//   const defaultParams = getDefaultParams(server);

//   const data = await api
//     .get('rest/getGenres.view', {
//       prefixUrl: server?.url,
//       searchParams: defaultParams,
//       signal,
//     })
//     .json<SSGenreListResponse>();

//   return data.genres.genre;
// };

// const getAlbumDetail = async (args: AlbumDetailArgs): Promise<SSAlbumDetail> => {
//   const { server, query, signal } = args;
//   const defaultParams = getDefaultParams(server);

//   const searchParams = {
//     id: query.id,
//     ...defaultParams,
//   };

//   const data = await api
//     .get('rest/getAlbum.view', {
//       prefixUrl: server?.url,
//       searchParams: parseSearchParams(searchParams),
//       signal,
//     })
//     .json<SSAlbumDetailResponse>();

//   const { song: songs, ...dataWithoutSong } = data.album;
//   return { ...dataWithoutSong, songs };
// };

// const getAlbumList = async (args: AlbumListArgs): Promise<SSAlbumList> => {
//   const { server, query, signal } = args;
//   const defaultParams = getDefaultParams(server);

//   const searchParams = {
//     ...defaultParams,
//   };
//   const data = await api
//     .get('rest/getAlbumList2.view', {
//       prefixUrl: server?.url,
//       searchParams: parseSearchParams(searchParams),
//       signal,
//     })
//     .json<SSAlbumListResponse>();

//   return {
//     items: data.albumList2.album,
//     startIndex: query.startIndex,
//     totalRecordCount: null,
//   };
// };

const createFavorite = async (args: FavoriteArgs): Promise<FavoriteResponse> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).createFavorite({
        query: {
            albumId: query.type === LibraryItem.ALBUM ? query.id : undefined,
            artistId: query.type === LibraryItem.ALBUM_ARTIST ? query.id : undefined,
            id: query.type === LibraryItem.SONG ? query.id : undefined,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to create favorite');
    }

    return null;
};

const removeFavorite = async (args: FavoriteArgs): Promise<FavoriteResponse> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).removeFavorite({
        query: {
            albumId: query.type === LibraryItem.ALBUM ? query.id : undefined,
            artistId: query.type === LibraryItem.ALBUM_ARTIST ? query.id : undefined,
            id: query.type === LibraryItem.SONG ? query.id : undefined,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to delete favorite');
    }

    return null;
};

const setRating = async (args: SetRatingArgs): Promise<RatingResponse> => {
    const { query, apiClientProps } = args;

    const itemIds = query.item.map((item) => item.id);

    for (const id of itemIds) {
        await ssApiClient(apiClientProps).setRating({
            query: {
                id,
                rating: query.rating,
            },
        });
    }

    return null;
};

const getTopSongList = async (args: TopSongListArgs): Promise<SongListResponse> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).getTopSongsList({
        query: {
            artist: query.artist,
            count: query.limit,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to get top songs');
    }

    return {
        items:
            res.body.topSongs?.song?.map((song) =>
                ssNormalize.song(song, apiClientProps.server, ''),
            ) || [],
        startIndex: 0,
        totalRecordCount: res.body.topSongs?.song?.length || 0,
    };
};

const getArtistInfo = async (
    args: ArtistInfoArgs,
): Promise<z.infer<typeof ssType._response.artistInfo>> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).getArtistInfo({
        query: {
            count: query.limit,
            id: query.artistId,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to get artist info');
    }

    return res.body;
};

const scrobble = async (args: ScrobbleArgs): Promise<ScrobbleResponse> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).scrobble({
        query: {
            id: query.id,
            submission: query.submission,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to scrobble');
    }

    return null;
};

const search3 = async (args: SearchArgs): Promise<SearchResponse> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).search3({
        query: {
            albumCount: query.albumLimit,
            albumOffset: query.albumStartIndex,
            artistCount: query.albumArtistLimit,
            artistOffset: query.albumArtistStartIndex,
            query: query.query,
            songCount: query.songLimit,
            songOffset: query.songStartIndex,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to search');
    }

    return {
        albumArtists: res.body.searchResult3?.artist?.map((artist) =>
            ssNormalize.albumArtist(artist, apiClientProps.server),
        ),
        albums: res.body.searchResult3?.album?.map((album) =>
            ssNormalize.album(album, apiClientProps.server),
        ),
        songs: res.body.searchResult3?.song?.map((song) =>
            ssNormalize.song(song, apiClientProps.server, ''),
        ),
    };
};

const getRandomSongList = async (args: RandomSongListArgs): Promise<RandomSongListResponse> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).getRandomSongList({
        query: {
            fromYear: query.minYear,
            genre: query.genre,
            musicFolderId: query.musicFolderId,
            size: query.limit,
            toYear: query.maxYear,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to get random songs');
    }

    return {
        items: res.body.randomSongs?.song?.map((song) =>
            ssNormalize.song(song, apiClientProps.server, ''),
        ),
        startIndex: 0,
        totalRecordCount: res.body.randomSongs?.song?.length || 0,
    };
};

const getServerInfo = async (args: ServerInfoArgs): Promise<ServerInfo> => {
    const { apiClientProps } = args;

    const ping = await ssApiClient(apiClientProps).ping();

    if (ping.status !== 200) {
        throw new Error('Failed to ping server');
    }

    if (!ping.body.openSubsonic || !ping.body.serverVersion) {
        return { version: ping.body.version };
    }

    const res = await ssApiClient(apiClientProps).getServerInfo();

    if (res.status !== 200) {
        throw new Error('Failed to get server extensions');
    }

    const features: Record<string, number[]> = {};
    for (const extension of res.body.openSubsonicExtensions) {
        features[extension.name] = extension.versions;
    }

    return { features, id: apiClientProps.server?.id, version: ping.body.serverVersion };
};

export const getStructuredLyrics = async (
    args: StructuredLyricsArgs,
): Promise<StructuredLyric[]> => {
    const { query, apiClientProps } = args;

    const res = await ssApiClient(apiClientProps).getStructuredLyrics({
        query: {
            id: query.songId,
        },
    });

    if (res.status !== 200) {
        throw new Error('Failed to get structured lyrics');
    }

    const lyrics = res.body.lyricsList?.structuredLyrics;

    if (!lyrics) {
        return [];
    }

    return lyrics.map((lyric) => {
        const baseLyric = {
            artist: lyric.displayArtist || '',
            lang: lyric.lang,
            name: lyric.displayTitle || '',
            remote: false,
            source: apiClientProps.server?.name || 'music server',
        };

        if (lyric.synced) {
            return {
                ...baseLyric,
                lyrics: lyric.line.map((line) => [line.start!, line.value]),
                synced: true,
            };
        }
        return {
            ...baseLyric,
            lyrics: lyric.line.map((line) => [line.value]).join('\n'),
            synced: false,
        };
    });
};

export const ssController = {
    authenticate,
    createFavorite,
    getArtistInfo,
    getMusicFolderList,
    getRandomSongList,
    getServerInfo,
    getStructuredLyrics,
    getTopSongList,
    removeFavorite,
    scrobble,
    search3,
    setRating,
};