afrov2/components/UserProvider.tsx

"use client";

import React, { createContext, useContext, useState, ReactNode } from 'react';
import { User } from '../types';
import { useRouter, usePathname } from 'next/navigation';

interface UserContextType {
    user: User | null;
    login: (user: User) => void;
    logout: () => void;
}

const UserContext = createContext<UserContextType | undefined>(undefined);

export function UserProvider({ children }: { children: ReactNode }) {
    const [user, setUser] = useState<User | null>(null);
    const router = useRouter();
    const pathname = usePathname();

    React.useEffect(() => {
        const savedUser = localStorage.getItem('afro_user');
        if (savedUser) {
            setUser(JSON.parse(savedUser));
        }
    }, []);

    // Global suspension check
    React.useEffect(() => {
        if (user?.isSuspended && pathname !== '/suspended') {
            router.push('/suspended');
        }
    }, [user, pathname, router]);

    const login = (userData: User) => {
        setUser(userData);
        localStorage.setItem('afro_user', JSON.stringify(userData));
    };

    const logout = () => {
        setUser(null);
        localStorage.removeItem('afro_user');
    };

    return (
        <UserContext.Provider value={{ user, login, logout }}>
            {children}
        </UserContext.Provider>
    );
}

export function useUser() {
    const context = useContext(UserContext);
    if (context === undefined) {
        throw new Error('useUser must be used within a UserProvider');
    }
    return context;
}