smart-crop-ui/src/components/auth/AuthContext.tsx

import { createContext, useContext, useState, useEffect, ReactNode } from 'react';
import { User, AuthState } from '../../types/auth';

interface AuthContextType {
  authState: AuthState;
  login: (user: User) => void;
  logout: () => void;
  updateUser: (user: User) => void;
  checkAuth: () => boolean;
}

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

export function AuthProvider({ children }: { children: ReactNode }) {
  const [authState, setAuthState] = useState<AuthState>({
    isAuthenticated: false,
    user: null,
    token: null,
    refreshToken: null,
  });
  
  const [isInitialized, setIsInitialized] = useState(false);

  // 初始化时检查登录状态
  useEffect(() => {
    // 确保在浏览器环境中执行
    if (typeof window === 'undefined') {
      setIsInitialized(true);
      return;
    }

    // 延迟执行以避免构造函数错误
    const timer = setTimeout(() => {
      initAuth();
    }, 0);

    return () => clearTimeout(timer);
  }, []);

  const initAuth = async () => {
    try {
      // 动态导入以避免模块初始化时的错误
      const authStorage = await import('../../lib/authStorage');
      
      const token = authStorage.getToken();
      const user = authStorage.getUser();

      if (token && user) {
        // 检查token是否过期
        if (authStorage.isTokenExpired()) {
          // 尝试刷新token
          const refreshed = await authStorage.refreshAuthToken();
          if (refreshed) {
            setAuthState({
              isAuthenticated: true,
              user,
              token: refreshed.token,
              refreshToken: refreshed.refreshToken,
            });
          } else {
            // 刷新失败，自动使用默认账号登录
            await autoLoginWithDefaultAccount();
          }
        } else {
          setAuthState({
            isAuthenticated: true,
            user,
            token,
            refreshToken: authStorage.getRefreshToken() || token,
          });
        }
      } else {
        // 没有登录信息，自动使用默认账号登录
        await autoLoginWithDefaultAccount();
      }
    } catch (error) {
      console.error('Auth initialization error:', error);
      // 初始化失败时设置为未登录状态
      setAuthState({
        isAuthenticated: false,
        user: null,
        token: null,
        refreshToken: null,
      });
    } finally {
      setIsInitialized(true);
    }
  };

  // 自动登录默认账号
  const autoLoginWithDefaultAccount = async () => {
    try {
      // 动态导入避免循环依赖和模块初始化错误
      const authStorage = await import('../../lib/authStorage');
      
      // 使用默认管理员账号自动登录
      const result = await authStorage.validatePasswordLogin('admin', 'admin123', 'AUTO');
      
      if (result.success && result.user) {
        const newToken = authStorage.generateToken();
        const newRefreshToken = authStorage.generateToken();
        
        authStorage.saveToken(newToken, newRefreshToken);
        authStorage.saveUser(result.user);
        
        setAuthState({
          isAuthenticated: true,
          user: result.user,
          token: newToken,
          refreshToken: newRefreshToken,
        });
      } else {
        // 自动登录失败，显示登录页面
        setAuthState({
          isAuthenticated: false,
          user: null,
          token: null,
          refreshToken: null,
        });
      }
    } catch (error) {
      console.error('Auto login error:', error);
      setAuthState({
        isAuthenticated: false,
        user: null,
        token: null,
        refreshToken: null,
      });
    }
  };

  // 定期检查token有效性
  useEffect(() => {
    if (!authState.isAuthenticated || !isInitialized) return;

    const interval = setInterval(async () => {
      try {
        const authStorage = await import('../../lib/authStorage');
        
        if (authStorage.isTokenExpired()) {
          const refreshed = await authStorage.refreshAuthToken();
          if (refreshed) {
            setAuthState(prev => ({
              ...prev,
              token: refreshed.token,
              refreshToken: refreshed.refreshToken,
            }));
          } else {
            logout();
          }
        }
      } catch (error) {
        console.error('Token refresh error:', error);
      }
    }, 5 * 60 * 1000); // 每5分钟检查一次

    return () => clearInterval(interval);
  }, [authState.isAuthenticated, isInitialized]);

  const login = async (user: User) => {
    try {
      const authStorage = await import('../../lib/authStorage');
      const token = authStorage.generateToken();
      const refreshToken = authStorage.generateToken();

      authStorage.saveToken(token, refreshToken);
      authStorage.saveUser(user);

      setAuthState({
        isAuthenticated: true,
        user,
        token,
        refreshToken,
      });
    } catch (error) {
      console.error('Login error:', error);
    }
  };

  const logout = async () => {
    try {
      const authStorage = await import('../../lib/authStorage');
      authStorage.clearAuth();
      
      setAuthState({
        isAuthenticated: false,
        user: null,
        token: null,
        refreshToken: null,
      });
    } catch (error) {
      console.error('Logout error:', error);
    }
  };

  const updateUser = async (user: User) => {
    try {
      const authStorage = await import('../../lib/authStorage');
      authStorage.saveUser(user);
      
      setAuthState(prev => ({
        ...prev,
        user,
      }));
    } catch (error) {
      console.error('Update user error:', error);
    }
  };

  const checkAuth = (): boolean => {
    return authState.isAuthenticated;
  };

  // 在初始化完成前显示加载状态
  if (!isInitialized) {
    return (
      <div className="h-screen flex items-center justify-center bg-gray-50">
        <div className="text-center">
          <div className="inline-block h-12 w-12 animate-spin rounded-full border-4 border-solid border-green-600 border-r-transparent mb-4"></div>
          <p className="text-gray-600">正在加载...</p>
        </div>
      </div>
    );
  }

  return (
    <AuthContext.Provider value={{ authState, login, logout, updateUser, checkAuth }}>
      {children}
    </AuthContext.Provider>
  );
}

export function useAuth() {
  const context = useContext(AuthContext);
  if (context === undefined) {
    throw new Error('useAuth must be used within an AuthProvider');
  }
  return context;
}