"use client";
import {useSession} from "next-auth/react";
import {ROLES} from "@/constants/roles";
export function useRole() {
    const {data: session} = useSession();
    const role = String(session?.user?.role);

    const isTrainer = role === ROLES.TRAINER;
    const isLearner = role === ROLES.LEARNER;

    const hasRole = (targetRole: string | string[]) => {
        if (Array.isArray(targetRole)) return targetRole.includes(role);
        return role === targetRole;
    };

    return {role, isTrainer, isLearner, hasRole};
}