SurfSense/surfsense_web/atoms/chats/chat-mutations.atom.ts

import { atomWithMutation } from "jotai-tanstack-query";
import { deleteChat } from "@/lib/apis/chats.api";
import { activeSearchSpaceIdAtom } from "../seach-spaces/seach-space-queries.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);
				}
			);
		},
	};
});