测试脚本_6

Bin

2025-12-17 2b99d77d73ba568beff0a549534017caaad8a6de

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import type { APIUser } from "@humansignal/core/types/user";
import { getApiInstance } from "../../lib/api-provider/api-instance";
import { useAtomValue } from "jotai";
import { atomWithMutation, atomWithQuery, queryClientAtom } from "jotai-tanstack-query";
import { useCallback } from "react";
 
const currentUserAtom = atomWithQuery(() => ({
  queryKey: ["current-user"],
  async queryFn() {
    const api = getApiInstance();
    return await api.invoke<APIUser>("me");
  },
}));
 
const currentUserUpdateAtom = atomWithMutation((get) => ({
  mutationKey: ["update-current-user"],
  async mutationFn({ pk, user }: { pk: number; user: Partial<APIUser> }) {
    const api = getApiInstance();
    return await api.invoke<APIUser>("updateUser", { pk }, { body: user });
  },
 
  onSettled() {
    const queryClient = get(queryClientAtom);
    queryClient.invalidateQueries({ queryKey: ["current-user"] });
  },
}));
 
export function useCurrentUserAtom() {
  const user = useAtomValue(currentUserAtom);
  const updateUser = useAtomValue(currentUserUpdateAtom);
  const queryClient = useAtomValue(queryClientAtom);
  const refetch = useCallback(() => queryClient.invalidateQueries({ queryKey: ["current-user"] }), []);
  const update = useCallback(
    (userUpdate: Partial<APIUser>) => {
      if (!user.data) {
        console.error("User is not loaded. Try fetching first.");
        return;
      }
      updateUser.mutate({ pk: user.data.id, user: userUpdate });
    },
    [user.data?.id, updateUser.mutate],
  );
 
  const updateAsync = useCallback(
    (userUpdate: Partial<APIUser>) => {
      if (!user.data) {
        console.error("User is not loaded. Try fetching first.");
        return;
      }
      return updateUser.mutateAsync({ pk: user.data.id, user: userUpdate });
    },
    [user.data?.id, updateUser.mutate],
  );
 
  const commonResponse = {
    isInProgress: user.isFetching || updateUser.isPending,
    isUpdating: updateUser.isPending,
    loaded: user.isSuccess,
    fetch: refetch,
    update,
    updateAsync,
  } as const;
 
  return user.isSuccess
    ? ({
        user: user.data,
        error: null,
        ...commonResponse,
      } as const)
    : ({
        user: null,
        error: user.error,
        ...commonResponse,
      } as const);
}