badoHair_fe/contexts/AuthContext.tsx

"use client";

import React, { createContext, useContext, useState, useEffect, startTransition, ReactNode } from "react";
import { getToken } from "@/lib/api";
import * as authApi from "@/lib/api/auth";

export type UserProfile = authApi.UserProfile;

interface AuthContextType {
  user: UserProfile | null;
  isLoading: boolean;
  isAdmin: boolean;
  login: (email: string, password: string) => Promise<UserProfile>;
  register: (email: string, password: string, name: string) => Promise<UserProfile | null>;
  logout: () => void;
  refreshUser: () => Promise<void>;
}

const AuthContext = createContext<AuthContextType | undefined>(undefined);

export const AuthProvider = ({ children }: { children: ReactNode }) => {
  const [user, setUser] = useState<UserProfile | null>(null);
  const [isLoading, setIsLoading] = useState(true);

  useEffect(() => {
    const token = getToken();
    if (!token) {
      startTransition(() => setIsLoading(false));
      return;
    }
    authApi.getMe()
      .then(setUser)
      .catch(() => authApi.logout())
      .finally(() => setIsLoading(false));
  }, []);

  const login = async (email: string, password: string): Promise<UserProfile> => {
    const profile = await authApi.login(email, password);
    setUser(profile);
    return profile;
  };

  const register = async (email: string, password: string, name: string): Promise<UserProfile | null> => {
    const profile = await authApi.register(email, password, name);
    if (profile) setUser(profile);
    return profile;
  };

  const logout = () => {
    authApi.logout();
    setUser(null);
  };

  const refreshUser = async () => {
    const profile = await authApi.getMe();
    setUser(profile);
  };

  return (
    <AuthContext.Provider
      value={{
        user,
        isLoading,
        isAdmin: user?.role === "admin",
        login,
        register,
        logout,
        refreshUser,
      }}
    >
      {children}
    </AuthContext.Provider>
  );
};

export const useAuth = () => {
  const ctx = useContext(AuthContext);
  if (!ctx) throw new Error("useAuth must be used within AuthProvider");
  return ctx;
};