Spaces:
Running
Running
File size: 2,580 Bytes
ba95018 | 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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | import { createContext, useContext, useState, useEffect, useCallback } from 'react';
const AuthContext = createContext();
const API_BASE = import.meta.env.VITE_API_URL || '';
export function AuthProvider({ children }) {
const [user, setUser] = useState(null);
const [token, setToken] = useState(null);
const [loading, setLoading] = useState(true);
// On mount, check if we have a stored token and verify it
useEffect(() => {
const storedToken = sessionStorage.getItem('ot-token');
const storedUser = sessionStorage.getItem('ot-user');
if (storedToken && storedUser) {
// Verify the token is still valid
fetch(`${API_BASE}/api/verify`, {
headers: { Authorization: `Bearer ${storedToken}` },
})
.then(res => {
if (res.ok) {
setToken(storedToken);
setUser(storedUser);
} else {
// Token invalid/expired — clear storage
sessionStorage.removeItem('ot-token');
sessionStorage.removeItem('ot-user');
}
})
.catch(() => {
sessionStorage.removeItem('ot-token');
sessionStorage.removeItem('ot-user');
})
.finally(() => setLoading(false));
} else {
setLoading(false);
}
}, []);
const login = useCallback(async (username, password) => {
const res = await fetch(`${API_BASE}/api/login`, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ username, password }),
});
if (!res.ok) {
const err = await res.json().catch(() => ({}));
throw new Error(err.detail || 'Login failed');
}
const data = await res.json();
setToken(data.token);
setUser(data.username);
sessionStorage.setItem('ot-token', data.token);
sessionStorage.setItem('ot-user', data.username);
return data;
}, []);
const logout = useCallback(() => {
setToken(null);
setUser(null);
sessionStorage.removeItem('ot-token');
sessionStorage.removeItem('ot-user');
}, []);
const getAuthHeaders = useCallback(() => {
if (!token) return {};
return { Authorization: `Bearer ${token}` };
}, [token]);
const isAuthenticated = !!token && !!user;
return (
<AuthContext.Provider value={{ user, token, loading, isAuthenticated, login, logout, getAuthHeaders }}>
{children}
</AuthContext.Provider>
);
}
export function useAuth() {
const ctx = useContext(AuthContext);
if (!ctx) throw new Error('useAuth must be used within AuthProvider');
return ctx;
}
|