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,
  RatingArgs,
  RatingResponse,
  ScrobbleArgs,
  ScrobbleResponse,
  SongListResponse,
  TopSongListArgs,
} 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({ 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 { signal, serverId } = args;

  if (!serverId) {
    throw new Error('No server id');
  }

  const res = await ssApiClient({ serverId, signal }).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 { serverId, query, signal } = args;

  if (!serverId) {
    throw new Error('No server id');
  }

  const res = await ssApiClient({ serverId, signal }).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 {
    id: query.id,
    type: query.type,
  };
};

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

  if (!serverId) {
    throw new Error('No server id');
  }

  const res = await ssApiClient({ serverId, signal }).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 {
    id: query.id,
    type: query.type,
  };
};

const setRating = async (args: RatingArgs): Promise<RatingResponse> => {
  const { serverId, query, signal } = args;

  if (!serverId) {
    throw new Error('No server id');
  }

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

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

  return null;
};

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

  if (!serverId || !server) {
    throw new Error('No server id');
  }

  const res = await ssApiClient({ serverId, signal }).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, server, '')),
    startIndex: 0,
    totalRecordCount: res.body.topSongs.song.length || 0,
  };
};

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

  if (!serverId) {
    throw new Error('No server id');
  }

  const res = await ssApiClient({ serverId, signal }).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 { signal, serverId, query } = args;

  if (!serverId) {
    throw new Error('No server id');
  }

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

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

  return null;
};

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