144 lines
3.6 KiB
TypeScript
144 lines
3.6 KiB
TypeScript
/**
|
|
* Auth Context
|
|
*
|
|
* Provides authentication state and methods to the application
|
|
*/
|
|
|
|
'use client';
|
|
|
|
import React, { createContext, useContext, useState, useEffect, useCallback } from 'react';
|
|
import { useRouter } from 'next/navigation';
|
|
import { authApi, User } from '../api';
|
|
|
|
interface AuthContextType {
|
|
user: User | null;
|
|
loading: boolean;
|
|
login: (email: string, password: string) => Promise<void>;
|
|
register: (data: {
|
|
email: string;
|
|
password: string;
|
|
firstName: string;
|
|
lastName: string;
|
|
organizationId: string;
|
|
}) => Promise<void>;
|
|
logout: () => Promise<void>;
|
|
refreshUser: () => Promise<void>;
|
|
isAuthenticated: boolean;
|
|
}
|
|
|
|
const AuthContext = createContext<AuthContextType | undefined>(undefined);
|
|
|
|
export function AuthProvider({ children }: { children: React.ReactNode }) {
|
|
const [user, setUser] = useState<User | null>(null);
|
|
const [loading, setLoading] = useState(true);
|
|
const router = useRouter();
|
|
|
|
// Memoized function to refresh user data
|
|
const refreshUser = useCallback(async () => {
|
|
try {
|
|
if (authApi.isAuthenticated()) {
|
|
const currentUser = await authApi.me();
|
|
setUser(currentUser);
|
|
}
|
|
} catch (error) {
|
|
console.error('Failed to refresh user:', error);
|
|
// Token invalid, clear storage
|
|
if (typeof window !== 'undefined') {
|
|
localStorage.removeItem('accessToken');
|
|
localStorage.removeItem('refreshToken');
|
|
localStorage.removeItem('user');
|
|
}
|
|
setUser(null);
|
|
}
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
// Check if user is already logged in
|
|
const checkAuth = async () => {
|
|
try {
|
|
if (authApi.isAuthenticated()) {
|
|
const storedUser = authApi.getStoredUser();
|
|
if (storedUser) {
|
|
// Verify token is still valid by fetching current user
|
|
const currentUser = await authApi.me();
|
|
setUser(currentUser);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
console.error('Auth check failed:', error);
|
|
// Token invalid, clear storage
|
|
if (typeof window !== 'undefined') {
|
|
localStorage.removeItem('accessToken');
|
|
localStorage.removeItem('refreshToken');
|
|
localStorage.removeItem('user');
|
|
}
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
};
|
|
|
|
checkAuth();
|
|
}, []);
|
|
|
|
const login = async (email: string, password: string) => {
|
|
try {
|
|
await authApi.login({ email, password });
|
|
// Fetch complete user data after login and update state
|
|
const currentUser = await authApi.me();
|
|
setUser(currentUser);
|
|
// Navigate to dashboard
|
|
router.push('/dashboard');
|
|
} catch (error) {
|
|
throw error;
|
|
}
|
|
};
|
|
|
|
const register = async (data: {
|
|
email: string;
|
|
password: string;
|
|
firstName: string;
|
|
lastName: string;
|
|
organizationId: string;
|
|
}) => {
|
|
try {
|
|
await authApi.register(data);
|
|
// Fetch complete user data after registration and update state
|
|
const currentUser = await authApi.me();
|
|
setUser(currentUser);
|
|
// Navigate to dashboard
|
|
router.push('/dashboard');
|
|
} catch (error) {
|
|
throw error;
|
|
}
|
|
};
|
|
|
|
const logout = async () => {
|
|
try {
|
|
await authApi.logout();
|
|
} finally {
|
|
setUser(null);
|
|
router.push('/login');
|
|
}
|
|
};
|
|
|
|
const value = {
|
|
user,
|
|
loading,
|
|
login,
|
|
register,
|
|
logout,
|
|
refreshUser,
|
|
isAuthenticated: !!user,
|
|
};
|
|
|
|
return <AuthContext.Provider value={value}>{children}</AuthContext.Provider>;
|
|
}
|
|
|
|
export function useAuth() {
|
|
const context = useContext(AuthContext);
|
|
if (context === undefined) {
|
|
throw new Error('useAuth must be used within an AuthProvider');
|
|
}
|
|
return context;
|
|
}
|