File size: 2,152 Bytes
24e6f5b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | import { createContext, useContext, useState, useEffect } from 'react';
import api from '../api/axios';
import { useNavigate } from 'react-router-dom';
const AuthContext = createContext(null);
export const AuthProvider = ({ children }) => {
const [user, setUser] = useState(null);
const [loading, setLoading] = useState(true);
const navigate = useNavigate();
useEffect(() => {
checkUser();
}, []);
const checkUser = async () => {
const token = localStorage.getItem('access_token');
if (token) {
try {
const response = await api.get('auth/profile/');
setUser(response.data);
} catch (error) {
// Token might be invalid or expired (and refresh failed)
localStorage.removeItem('access_token');
localStorage.removeItem('refresh_token');
setUser(null);
}
}
setLoading(false);
};
const login = async (username, password) => {
const response = await api.post('auth/login/', { username, password });
localStorage.setItem('access_token', response.data.access);
localStorage.setItem('refresh_token', response.data.refresh);
await checkUser();
navigate('/dashboard');
};
const register = async (username, email, password) => {
const response = await api.post('auth/register/', { username, email, password });
// Auto-login with the tokens returned from registration
localStorage.setItem('access_token', response.data.access);
localStorage.setItem('refresh_token', response.data.refresh);
setUser(response.data.user);
navigate('/dashboard');
};
const logout = () => {
localStorage.removeItem('access_token');
localStorage.removeItem('refresh_token');
setUser(null);
navigate('/');
};
return (
<AuthContext.Provider value={{ user, login, register, logout, loading }}>
{!loading && children}
</AuthContext.Provider>
);
};
export const useAuth = () => useContext(AuthContext);
|