52 lines
2.0 KiB
TypeScript
52 lines
2.0 KiB
TypeScript
import { App } from "antd";
|
|
import { createContext, useCallback, useContext, useEffect, useMemo, useState, type ReactNode } from "react";
|
|
|
|
import { getAdminReferences, type AdminReferenceData, type LocalUser } from "@/services/api/auth";
|
|
|
|
type AdminContextValue = {
|
|
references: AdminReferenceData;
|
|
referencesLoading: boolean;
|
|
reloadReferences: () => Promise<void>;
|
|
updateUserReference: (user: LocalUser) => void;
|
|
};
|
|
|
|
const emptyReferences: AdminReferenceData = { users: [], channels: [] };
|
|
const AdminContext = createContext<AdminContextValue | null>(null);
|
|
|
|
export function AdminProvider({ children }: { children: ReactNode }) {
|
|
const { message } = App.useApp();
|
|
const [references, setReferences] = useState<AdminReferenceData>(emptyReferences);
|
|
const [referencesLoading, setReferencesLoading] = useState(true);
|
|
|
|
const reloadReferences = useCallback(async () => {
|
|
setReferencesLoading(true);
|
|
try {
|
|
setReferences(await getAdminReferences());
|
|
} catch (error) {
|
|
message.error(error instanceof Error ? error.message : "读取后台基础数据失败");
|
|
} finally {
|
|
setReferencesLoading(false);
|
|
}
|
|
}, [message]);
|
|
|
|
useEffect(() => {
|
|
void reloadReferences();
|
|
}, [reloadReferences]);
|
|
|
|
const updateUserReference = useCallback((user: LocalUser) => {
|
|
setReferences((current) => ({
|
|
...current,
|
|
users: current.users.map((item) => item.id === user.id ? { id: user.id, username: user.username, displayName: user.displayName } : item),
|
|
}));
|
|
}, []);
|
|
|
|
const value = useMemo(() => ({ references, referencesLoading, reloadReferences, updateUserReference }), [references, referencesLoading, reloadReferences, updateUserReference]);
|
|
return <AdminContext.Provider value={value}>{children}</AdminContext.Provider>;
|
|
}
|
|
|
|
export function useAdminContext() {
|
|
const value = useContext(AdminContext);
|
|
if (!value) throw new Error("useAdminContext 必须在 AdminProvider 内使用");
|
|
return value;
|
|
}
|