umami/src/components/hooks/useRequireLogin.ts

29 lines
670 B
TypeScript

import { useEffect } from 'react';
import useApi from 'components/hooks/useApi';
import useUser from 'components/hooks/useUser';
export function useRequireLogin(handler?: (data?: object) => void) {
const { get } = useApi();
const { user, setUser } = useUser();
useEffect(() => {
async function loadUser() {
try {
const data = await get('/auth/verify');
setUser(typeof handler === 'function' ? handler(data) : (data as any)?.user);
} catch {
location.href = `${process.env.basePath || ''}/login`;
}
}
if (!user) {
loadUser();
}
}, [user]);
return { user };
}
export default useRequireLogin;