SurfSense/surfsense_web/atoms/workspaces/workspace-mutation.atoms.ts

import { atomWithMutation } from "jotai-tanstack-query";
import { toast } from "sonner";
import type {
	CreateSearchSpaceRequest,
	DeleteSearchSpaceRequest,
	UpdateSearchSpaceApiAccessRequest,
	UpdateSearchSpaceRequest,
} from "@/contracts/types/workspace.types";
import { searchSpacesApiService } from "@/lib/apis/workspaces-api.service";
import { cacheKeys } from "@/lib/query-client/cache-keys";
import { queryClient } from "@/lib/query-client/client";
import { activeWorkspaceIdAtom } from "./workspace-query.atoms";

export const createSearchSpaceMutationAtom = atomWithMutation(() => {
	return {
		mutationKey: ["create-search-space"],
		mutationFn: async (request: CreateSearchSpaceRequest) => {
			return searchSpacesApiService.createSearchSpace(request);
		},

		onSuccess: () => {
			toast.success("Search space created successfully");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.searchSpaces.all,
			});
		},
	};
});

export const updateSearchSpaceMutationAtom = atomWithMutation((get) => {
	const activeSearchSpaceId = get(activeWorkspaceIdAtom);

	return {
		mutationKey: ["update-search-space", activeSearchSpaceId],
		enabled: !!activeSearchSpaceId,
		mutationFn: async (request: UpdateSearchSpaceRequest) => {
			return searchSpacesApiService.updateSearchSpace(request);
		},

		onSuccess: (_, request: UpdateSearchSpaceRequest) => {
			toast.success("Search space updated successfully");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.searchSpaces.all,
			});
			if (request.id) {
				queryClient.invalidateQueries({
					queryKey: cacheKeys.searchSpaces.detail(String(request.id)),
				});
			}
		},
	};
});

export const updateSearchSpaceApiAccessMutationAtom = atomWithMutation((get) => {
	const activeSearchSpaceId = get(activeWorkspaceIdAtom);

	return {
		mutationKey: ["update-search-space-api-access", activeSearchSpaceId],
		enabled: !!activeSearchSpaceId,
		mutationFn: async (request: UpdateSearchSpaceApiAccessRequest) => {
			return searchSpacesApiService.updateSearchSpaceApiAccess(request);
		},

		onSuccess: (_, request: UpdateSearchSpaceApiAccessRequest) => {
			toast.success("API access updated successfully");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.searchSpaces.all,
			});
			queryClient.invalidateQueries({
				queryKey: cacheKeys.searchSpaces.detail(String(request.id)),
			});
		},
	};
});

export const deleteSearchSpaceMutationAtom = atomWithMutation((get) => {
	const activeSearchSpaceId = get(activeWorkspaceIdAtom);

	return {
		mutationKey: ["delete-search-space", activeSearchSpaceId],
		enabled: !!activeSearchSpaceId,
		mutationFn: async (request: DeleteSearchSpaceRequest) => {
			return searchSpacesApiService.deleteSearchSpace(request);
		},

		onSuccess: (_, request: DeleteSearchSpaceRequest) => {
			toast.success("Search space deleted successfully");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.searchSpaces.all,
			});
			if (request.id) {
				queryClient.removeQueries({
					queryKey: cacheKeys.searchSpaces.detail(String(request.id)),
				});
			}
		},
	};
});