This commit is contained in:
2025-05-21 08:46:12 +03:00
parent 9531bff01a
commit 079ce23363
94 changed files with 4630 additions and 2704 deletions

View File

@@ -1,92 +1,78 @@
import { useQuery } from "@tanstack/react-query"
import { create as batshitCreate, keyResolver } from "@yornaath/batshit"
import { create } from "zustand"
import { immer } from "zustand/middleware/immer"
import { getUser } from "~/lib/api/client/user"
import type { FullUser, PartialUser, UserId } from "~/lib/api/types"
import { create as batshitCreate, keyResolver } from "@yornaath/batshit";
import { create } from "zustand";
import { immer } from "zustand/middleware/immer";
import { getUser } from "~/lib/api/client/user";
import type { FullUser, PartialUser, UserId } from "~/lib/api/types";
type UsersStore = {
users: Record<UserId, PartialUser>
currentUserId: UserId | undefined
fetchUsersIfNotPresent: (userIds: UserId[]) => Promise<void>
addUser: (user: PartialUser) => void
removeUser: (userId: UserId) => void
setCurrentUserId: (userId: UserId) => void
getCurrentUser: () => FullUser | undefined
}
users: Record<UserId, PartialUser>;
currentUserId: UserId | undefined;
fetchUsersIfNotPresent: (userIds: UserId[]) => Promise<void>;
addUser: (user: PartialUser) => void;
removeUser: (userId: UserId) => void;
setCurrentUserId: (userId: UserId) => void;
getCurrentUser: () => FullUser | undefined;
};
const usersFetcher = batshitCreate({
fetcher: async (userIds: UserId[]) => {
let users = []
let users = [];
for (const userId of userIds) {
users.push(getUser(userId))
users.push(getUser(userId));
}
return await Promise.all(users)
return await Promise.all(users);
},
resolver: keyResolver("id")
})
export const useUserQuery = (userId: UserId) => useQuery(
{
queryKey: ["users", userId],
queryFn: async () => {
const user = await getUser(userId)
return user
},
select: (data) => {
useUsersStore.getState().addUser(data)
return data
}
}
)
resolver: keyResolver("id"),
});
export const useUsersStore = create<UsersStore>()(
immer(
(set, get) => ({
users: {},
currentUserId: undefined,
fetchUsersIfNotPresent: async (userIds) => {
let userPromises: Promise<PartialUser>[] = []
for (const userId of userIds) {
const user = get().users[userId]
if (!user) {
userPromises.push(usersFetcher.fetch(userId))
}
immer((set, get) => ({
users: {},
currentUserId: undefined,
fetchUsersIfNotPresent: async (userIds) => {
let userPromises: Promise<PartialUser>[] = [];
for (const userId of userIds) {
const user = get().users[userId];
if (!user) {
userPromises.push(usersFetcher.fetch(userId));
}
}
const users = await Promise.all(userPromises)
const activeUsers = users.filter(Boolean)
const users = await Promise.all(userPromises);
const activeUsers = users.filter(Boolean);
set((state) => {
for (const user of activeUsers) {
if (user?.id)
state.users[user.id] = user
}
})
},
addUser: (user) => set((state) => {
set((state) => {
for (const user of activeUsers) {
if (user?.id) state.users[user.id] = user;
}
});
},
addUser: (user) =>
set((state) => {
if (user.id !== get().currentUserId)
state.users[user.id] = user
state.users[user.id] = user;
else {
const currentUser = get().users[user.id]
const currentUser = get().users[user.id];
if (currentUser)
state.users[user.id] = { ...currentUser, ...user }
else
state.users[user.id] = user
state.users[user.id] = { ...currentUser, ...user };
else state.users[user.id] = user;
}
}),
removeUser: (userId) => set((state) => {
delete state.users[userId]
removeUser: (userId) =>
set((state) => {
delete state.users[userId];
}),
setCurrentUserId: (userId) => set((state) => {
state.currentUserId = userId
setCurrentUserId: (userId) =>
set((state) => {
state.currentUserId = userId;
}),
getCurrentUser: () => !!get().currentUserId ? get().users[get().currentUserId!] as FullUser : undefined
}),
)
)
getCurrentUser: () =>
!!get().currentUserId
? (get().users[get().currentUserId!] as FullUser)
: undefined,
})),
);