feishin/src/renderer/features/shared/mutations/create-favorite-mutation.ts

import { useMutation, useQueryClient } from '@tanstack/react-query';
import { HTTPError } from 'ky';
import { api } from '/@/renderer/api';
import { queryKeys } from '/@/renderer/api/query-keys';
import {
  AlbumArtistDetailResponse,
  AlbumDetailResponse,
  FavoriteArgs,
  FavoriteResponse,
  LibraryItem,
} from '/@/renderer/api/types';
import { MutationHookArgs } from '/@/renderer/lib/react-query';
import { getServerById, useSetAlbumListItemDataById, useSetQueueFavorite } from '/@/renderer/store';

export const useCreateFavorite = (args: MutationHookArgs) => {
  const { options } = args || {};
  const queryClient = useQueryClient();
  const setAlbumListData = useSetAlbumListItemDataById();
  const setQueueFavorite = useSetQueueFavorite();

  return useMutation<
    FavoriteResponse,
    HTTPError,
    Omit<FavoriteArgs, 'server' | 'apiClientProps'>,
    null
  >({
    mutationFn: (args) => {
      const server = getServerById(args.serverId);
      if (!server) throw new Error('Server not found');
      return api.controller.createFavorite({ ...args, apiClientProps: { server } });
    },
    onSuccess: (_data, variables) => {
      const { serverId } = variables;

      if (!serverId) return;

      for (const id of variables.query.id) {
        // Set the userFavorite property to true for the album in the album list data store
        if (variables.query.type === LibraryItem.ALBUM) {
          setAlbumListData(id, { userFavorite: true });
        }
      }

      if (variables.query.type === LibraryItem.SONG) {
        setQueueFavorite(variables.query.id, true);
      }

      // We only need to set if we're already on the album detail page
      if (variables.query.type === LibraryItem.ALBUM && variables.query.id.length === 1) {
        const queryKey = queryKeys.albums.detail(serverId, { id: variables.query.id[0] });
        const previous = queryClient.getQueryData<AlbumDetailResponse>(queryKey);

        if (previous) {
          queryClient.setQueryData<AlbumDetailResponse>(queryKey, {
            ...previous,
            userFavorite: true,
          });
        }
      }

      // We only need to set if we're already on the album detail page
      if (variables.query.type === LibraryItem.ALBUM_ARTIST && variables.query.id.length === 1) {
        const queryKey = queryKeys.albumArtists.detail(serverId, {
          id: variables.query.id[0],
        });

        const previous = queryClient.getQueryData<AlbumArtistDetailResponse>(queryKey);

        if (previous) {
          queryClient.setQueryData<AlbumArtistDetailResponse>(queryKey, {
            ...previous,
            userFavorite: true,
          });
        }
      }
    },
    ...options,
  });
};