Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
File size: 3,733 Bytes
61d29fc ddbd63e 61d29fc ddbd63e 61d29fc dc964b3 ddbd63e dc964b3 61d29fc 9a9548a 61d29fc 9a9548a 61d29fc ddbd63e 61d29fc ddbd63e 61d29fc | 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 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 | import React, { createContext, useContext, useState, useEffect, ReactNode } from 'react';
interface User {
id: number;
email: string;
username?: string;
full_name?: string;
avatar_url?: string;
oauth_provider?: string;
state?: string;
county?: string;
city?: string;
school_board?: string;
profile_completed?: boolean;
}
interface AuthContextType {
user: User | null;
token: string | null;
login: (provider: string) => void;
logout: () => void;
isAuthenticated: boolean;
isLoading: boolean;
authError: string | null;
clearAuthError: () => void;
}
const AuthContext = createContext<AuthContextType | undefined>(undefined);
export const AuthProvider: React.FC<{ children: ReactNode }> = ({ children }) => {
const [user, setUser] = useState<User | null>(null);
const [token, setToken] = useState<string | null>(null);
const [isLoading, setIsLoading] = useState(true);
const [authError, setAuthError] = useState<string | null>(null);
const API_URL = import.meta.env.PROD
? '/api'
: 'http://localhost:8000';
// Load user from localStorage on mount
useEffect(() => {
// Check for token in URL FIRST (OAuth callback)
const urlParams = new URLSearchParams(window.location.search);
const urlToken = urlParams.get('token');
const urlError = urlParams.get('error');
if (urlError) {
// Show OAuth error to user
setAuthError(urlError);
// Clean URL (remove error from address bar)
window.history.replaceState({}, document.title, window.location.pathname);
setIsLoading(false);
return;
}
if (urlToken) {
localStorage.setItem('auth_token', urlToken);
setToken(urlToken);
fetchUser(urlToken);
// Clean URL (remove token from address bar)
window.history.replaceState({}, document.title, window.location.pathname);
return; // Exit early, fetchUser will handle loading state
}
// Check for stored token
const storedToken = localStorage.getItem('auth_token');
if (storedToken) {
setToken(storedToken);
fetchUser(storedToken);
} else {
setIsLoading(false);
}
}, []);
const fetchUser = async (authToken: string) => {
try {
const response = await fetch(`${API_URL}/auth/me`, {
headers: {
'Authorization': `Bearer ${authToken}`,
},
});
if (response.ok) {
const userData = await response.json();
setUser(userData);
} else {
// Token is invalid
localStorage.removeItem('auth_token');
setToken(null);
}
} catch (error) {
console.error('Error fetching user:', error);
localStorage.removeItem('auth_token');
setToken(null);
} finally {
setIsLoading(false);
}
};
const login = (provider: string) => {
// Redirect to OAuth endpoint
const redirectUri = encodeURIComponent(window.location.origin);
const authUrl = `${API_URL}/auth/login/${provider}?redirect_uri=${redirectUri}`;
window.location.href = authUrl;
};
const logout = () => {
localStorage.removeItem('auth_token');
setToken(null);
setUser(null);
};
const clearAuthError = () => {
setAuthError(null);
};
return (
<AuthContext.Provider
value={{
user,
token,
login,
logout,
isAuthenticated: !!user,
isLoading,
authError,
clearAuthError,
}}
>
{children}
</AuthContext.Provider>
);
};
export const useAuth = () => {
const context = useContext(AuthContext);
if (context === undefined) {
throw new Error('useAuth must be used within an AuthProvider');
}
return context;
};
|