import { | |
ProfileSettingBaseKey, | |
ProfileSettingRouteKey, | |
} from '@/constants/setting'; | |
import { useLogout } from '@/hooks/login-hooks'; | |
import { useCallback } from 'react'; | |
import { useNavigate } from 'umi'; | |
export const useHandleMenuClick = () => { | |
const navigate = useNavigate(); | |
const { logout } = useLogout(); | |
const handleMenuClick = useCallback( | |
(key: ProfileSettingRouteKey) => () => { | |
if (key === ProfileSettingRouteKey.Logout) { | |
logout(); | |
} else { | |
navigate(`/${ProfileSettingBaseKey}/${key}`); | |
} | |
}, | |
[logout, navigate], | |
); | |
return { handleMenuClick }; | |
}; | |