2022-03-24 15:49:33 +00:00
|
|
|
import { useState } from './useState';
|
2022-04-05 14:30:53 +00:00
|
|
|
import { useRole } from './useRole';
|
2022-03-24 15:49:33 +00:00
|
|
|
|
|
|
|
export function useSession() {
|
|
|
|
function getToken() {
|
|
|
|
const localToken = localStorage.getItem('token');
|
|
|
|
const sessionToken = sessionStorage.getItem('token');
|
|
|
|
|
|
|
|
return localToken || sessionToken || '';
|
|
|
|
}
|
|
|
|
|
|
|
|
function setToken(data) {
|
|
|
|
if (data.keepLogin) {
|
|
|
|
localStorage.setItem('token', data.token);
|
|
|
|
} else {
|
|
|
|
sessionStorage.setItem('token', data.token);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function destroy() {
|
|
|
|
localStorage.removeItem('token');
|
|
|
|
sessionStorage.getItem('token');
|
|
|
|
|
|
|
|
const { setUser } = useState();
|
|
|
|
|
|
|
|
setUser({
|
|
|
|
id: 0,
|
|
|
|
username: '',
|
|
|
|
nickname: '',
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2022-04-05 14:30:53 +00:00
|
|
|
async function login(token, keepLogin) {
|
|
|
|
const { fetch } = useRole();
|
|
|
|
|
|
|
|
setToken({ token, keepLogin });
|
|
|
|
|
|
|
|
await fetch();
|
|
|
|
}
|
|
|
|
|
2022-03-24 15:49:33 +00:00
|
|
|
function isLoggedIn() {
|
|
|
|
const localToken = localStorage.getItem('token');
|
|
|
|
const sessionToken = sessionStorage.getItem('token');
|
|
|
|
|
|
|
|
return !!(localToken || sessionToken);
|
|
|
|
}
|
|
|
|
|
|
|
|
return {
|
|
|
|
getToken,
|
|
|
|
setToken,
|
|
|
|
destroy,
|
2022-04-05 14:30:53 +00:00
|
|
|
login,
|
2022-03-24 15:49:33 +00:00
|
|
|
isLoggedIn,
|
|
|
|
};
|
|
|
|
}
|