buster/web/src/api/buster_rest/users/queryRequests.ts

import {
  PREFETCH_STALE_TIME,
  useCreateReactMutation,
  useCreateReactQuery
} from '@/api/createReactQuery';
import {
  getUser,
  getUser_server,
  updateOrganizationUser,
  getMyUserInfo,
  getMyUserInfo_server
} from './requests';
import { useMemoizedFn } from 'ahooks';
import { QueryClient, useQueryClient } from '@tanstack/react-query';
import type { OrganizationUser } from '@/api/asset_interfaces';
import { queryKeys } from '@/api/query_keys';

export const useGetMyUserInfo = () => {
  const queryFn = useMemoizedFn(async () => getMyUserInfo());

  return useCreateReactQuery({
    ...queryKeys['/users/response:getUserMyself'],
    queryFn
    // staleTime: PREFETCH_STALE_TIME,
    // enabled: false
  });
};

export const prefetchGetMyUserInfo = async (queryClientProp?: QueryClient) => {
  const queryClient = queryClientProp || new QueryClient();
  await queryClient.prefetchQuery({
    ...queryKeys['/users/response:getUserMyself'],
    queryFn: () => getMyUserInfo()
  });
};

export const useGetUser = (params: Parameters<typeof getUser>[0]) => {
  const queryFn = useMemoizedFn(() => {
    return getUser(params);
  });

  return useCreateReactQuery({
    queryKey: config.USER_QUERY_KEY_ID(params.userId),
    queryFn,
    staleTime: PREFETCH_STALE_TIME
  });
};

export const useUpdateUser = () => {
  const queryClient = useQueryClient();
  const mutationFn = useMemoizedFn(async (params: Parameters<typeof updateOrganizationUser>[0]) => {
    queryClient.setQueryData(
      config.USER_QUERY_KEY_ID(params.userId),
      (oldData: OrganizationUser) => {
        return {
          ...oldData,
          ...params
        };
      }
    );
    const res = await updateOrganizationUser(params);
    return res;
  });

  return useCreateReactMutation({
    mutationFn: mutationFn
  });
};

export const prefetchGetUser = async (userId: string, queryClientProp?: QueryClient) => {
  const queryClient = queryClientProp || new QueryClient();
  await queryClient.prefetchQuery({
    ...queryKeys['/users/response:getUser'](userId),
    queryFn: () => getUser_server({ userId })
  });
  return queryClient;
};