feat: 优化项目目录结构
This commit is contained in:
41
apps/frontend/hooks/user/use-user-me.ts
Normal file
41
apps/frontend/hooks/user/use-user-me.ts
Normal file
@@ -0,0 +1,41 @@
|
||||
import { UserApi } from "@/lib/api";
|
||||
import useSWR from "swr";
|
||||
|
||||
export function useUserMe({ onError }: { onError?: (e: any) => void } = {}) {
|
||||
const isClientSide = typeof window !== 'undefined';
|
||||
|
||||
const { data: user, isLoading, error } = useSWR(
|
||||
'/api/user/me',
|
||||
async () => {
|
||||
if (isClientSide && !localStorage.getItem('token')) {
|
||||
throw Object.assign(new Error('未登录'), { statusCode: -1 });
|
||||
}
|
||||
return await UserApi.me();
|
||||
},
|
||||
{
|
||||
onError: (error) => {
|
||||
if (error.statusCode === 401) {
|
||||
if (isClientSide) {
|
||||
localStorage.removeItem('token');
|
||||
}
|
||||
}
|
||||
|
||||
onError?.(error);
|
||||
},
|
||||
revalidateIfStale: false,
|
||||
revalidateOnFocus: false,
|
||||
shouldRetryOnError: (err) => {
|
||||
if ([-1, 401].includes(err.statusCode)) {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
},
|
||||
}
|
||||
);
|
||||
|
||||
return {
|
||||
user,
|
||||
isLoading,
|
||||
error
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user