// lib/hooks/useAuth.ts
import { useState, useEffect, useCallback } from 'react';
import { auth } from '@/lib/firebase';
import { onAuthStateChanged, signOut as firebaseSignOut, User } from 'firebase/auth';
import { getUserProfile } from '@/lib/firestore'; // your function

type AugmentedUser = User & { role?: string; plan?: string; [key: string]: any };

interface AuthState {
  user: AugmentedUser | null;
  loading: boolean;
  error: Error | null;
  signOut: () => Promise<void>;
}

export function useAuth(): AuthState {
  const [user, setUser] = useState<AugmentedUser | null>(null);
  const [loading, setLoading] = useState(true);
  const [error, setError] = useState<Error | null>(null);

  useEffect(() => {
    const unsubscribe = onAuthStateChanged(auth, async (firebaseUser) => {
      try {
        if (firebaseUser) {
          const profile = await getUserProfile(firebaseUser.uid);
          setUser({ ...firebaseUser, ...profile } as AugmentedUser);
        } else {
          setUser(null);
        }
        setError(null);
      } catch (err) {
        console.error('Auth state error:', err);
        setError(err instanceof Error ? err : new Error('Auth failed'));
        setUser(firebaseUser || null); // fallback
      } finally {
        setLoading(false);
      }
    });

    return () => unsubscribe();
  }, []);

  const signOut = useCallback(async () => {
    await firebaseSignOut(auth);
  }, []);

  return { user, loading, error, signOut };
}