import { atomWithMutation } from "jotai-tanstack-query";
import { deleteChat } from "@/lib/apis/chat-apis";
import { activeSearchSpaceIdAtom } from "../../seach-spaces/active-seach-space.atom";
import { queryClient } from "@/lib/query-client/client";
import { cacheKeys } from "@/lib/query-client/cache-keys";
import { toast } from "sonner";
import { Chat } from "@/app/dashboard/[search_space_id]/chats/chats-client";

export const deleteChatMutationAtom = atomWithMutation((get) => {
  const searchSpaceId = get(activeSearchSpaceIdAtom);
  const authToken = localStorage.getItem("surfsense_bearer_token");

  return {
    mutationKey: cacheKeys.activeSearchSpace.chats(searchSpaceId ?? ""),
    enabled: !!searchSpaceId && !!authToken,
    mutationFn: async (chatId: number) => {
      if (!authToken) {
        throw new Error("No authentication token found");
      }
      if (!searchSpaceId) {
        throw new Error("No search space id found");
      }

      return deleteChat(chatId, authToken);
    },

    onSuccess: (_, chatId) => {
      toast.success("Chat deleted successfully");
      queryClient.setQueryData(cacheKeys.activeSearchSpace.chats(searchSpaceId!), (oldData: Chat[]) => {
        return oldData.filter((chat) => chat.id !== chatId);
      });
    },
  };
});