SurfSense/surfsense_web/atoms/members/members-mutation.atoms.ts

import { atomWithMutation } from "jotai-tanstack-query";
import { toast } from "sonner";
import type {
	DeleteMembershipRequest,
	DeleteMembershipResponse,
	LeaveWorkspaceRequest,
	LeaveWorkspaceResponse,
	UpdateMembershipRequest,
	UpdateMembershipResponse,
} from "@/contracts/types/members.types";
import { membersApiService } from "@/lib/apis/members-api.service";
import { cacheKeys } from "@/lib/query-client/cache-keys";
import { queryClient } from "@/lib/query-client/client";

export const updateMemberMutationAtom = atomWithMutation(() => {
	return {
		meta: { suppressGlobalErrorToast: true },
		mutationFn: async (request: UpdateMembershipRequest) => {
			return membersApiService.updateMember(request);
		},
		onSuccess: (_: UpdateMembershipResponse, request: UpdateMembershipRequest) => {
			toast.success("Member updated successfully");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.members.all(request.workspace_id.toString()),
			});
		},
		onError: () => {
			toast.error("Failed to update member");
		},
	};
});

export const deleteMemberMutationAtom = atomWithMutation(() => {
	return {
		meta: { suppressGlobalErrorToast: true },
		mutationFn: async (request: DeleteMembershipRequest) => {
			return membersApiService.deleteMember(request);
		},
		onSuccess: (_: DeleteMembershipResponse, request: DeleteMembershipRequest) => {
			toast.success("Member removed successfully");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.members.all(request.workspace_id.toString()),
			});
		},
		onError: () => {
			toast.error("Failed to remove member");
		},
	};
});

export const leaveWorkspaceMutationAtom = atomWithMutation(() => {
	return {
		meta: { suppressGlobalErrorToast: true },
		mutationFn: async (request: LeaveWorkspaceRequest) => {
			return membersApiService.leaveWorkspace(request);
		},
		onSuccess: (_: LeaveWorkspaceResponse, request: LeaveWorkspaceRequest) => {
			toast.success("Successfully left the workspace");
			queryClient.invalidateQueries({
				queryKey: cacheKeys.members.all(request.workspace_id.toString()),
			});
		},
		onError: () => {
			toast.error("Failed to leave workspace");
		},
	};
});