Function useProfileMutations

apps/web/src/hooks/useCloudAgentProfiles.ts:173–409

(options: UseProfileMutationsOptions = {})

 * Hook returning all profile mutation functions
 */
export function useProfileMutations(options: UseProfileMutationsOptions = {}) {
  const { organizationId } = options;
  const queryClient = useQueryClient();
  const trpc = useTRPC();

  const invalidateProfiles = async () => {
    await queryClient.invalidateQueries({
      queryKey: trpc.agentProfiles.list.queryKey({ organizationId }),
    });
  };

  const invalidateProfile = async (profileId: string) => {
    await queryClient.invalidateQueries({
      queryKey: trpc.agentProfiles.get.queryKey({ profileId, organizationId }),
    });
    await invalidateProfiles();
  };

  const createProfile = useMutation(
    trpc.agentProfiles.create.mutationOptions({
      onSuccess: async () => {
        await invalidateProfiles();
      },
    })
  );

  const updateProfile = useMutation(
    trpc.agentProfiles.update.mutationOptions({
      onSuccess: async (_data, variables) => {
        await invalidateProfile(variables.profileId);
      },
    })
  );

  const deleteProfile = useMutation(
    trpc.agentProfiles.delete.mutationOptions({
      onSuccess: async (_data, variables) => {
        await invalidateProfile(variables.profileId);
      },
    })
  );

  const setAsDefault = useMutation(
    trpc.agentProfiles.setAsDefault.mutationOptions({
      onSuccess: async () => {
        // Setting a default flips isDefault on both the new and previous default,
        // so invalidate every single-profile get query in addition to the list.
        await queryClient.invalidateQueries({
          queryKey: trpc.agentProfiles.get.queryKey(),
        });
        await invalidateProfiles();
      },
    })
  );

  const clearDefault = useMutation(
    trpc.agentProfiles.clearDefault.mutationOptions({
      onSuccess: async () => {