import { useMemo } from 'react';
import { useAuthContext } from '../providers/AuthProvider';

export function useDepartment(department: string | string[]) {
  const { user, loading } = useAuthContext();

  const hasDepartment = useMemo(() => {
    if (!user) return false;

    const departments = Array.isArray(department) ? department : [department];
    
    // Check single department
    if (user.department && departments.includes(user.department)) {
      return true;
    }
    
    // Check multiple departments
    if (user.departments) {
      return departments.some(dept => user.departments?.includes(dept));
    }
    
    return false;
  }, [user, department]);

  const isInDepartment = useMemo(() => {
    if (!user) return false;
    
    const departments = Array.isArray(department) ? department : [department];
    
    // Check if user is in ALL specified departments
    if (user.departments) {
      return departments.every(dept => user.departments?.includes(dept));
    }
    
    // Fallback to single department
    return user.department ? departments.includes(user.department) : false;
  }, [user, department]);

  return {
    hasDepartment,
    isInDepartment,
    loading,
    userDepartment: user?.department,
    userDepartments: user?.departments || [],
  };
}