xiaoxia-saas/apps/web/src/hooks/useAuth.ts

/**
 * 认证相关 Hooks
 */
import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { useNavigate } from "react-router-dom";
import * as authApi from "@/api/auth";
import { useAuthStore } from "@/store/authStore";

// 登录 Hook
export const useLogin = () => {
  const navigate = useNavigate();
  const setAuth = useAuthStore((state) => state.setAuth);

  const mutation = useMutation({
    mutationFn: authApi.login,
  });

  // 手动处理成功后的逻辑
  const login = async (credentials: Parameters<typeof authApi.login>[0]) => {
    const data = await mutation.mutateAsync(credentials);
    const refreshToken = data.refresh_token ?? null;

    // 先保存 token
    localStorage.setItem("access_token", data.access_token);
    if (refreshToken) {
      localStorage.setItem("refresh_token", refreshToken);
    } else {
      localStorage.removeItem("refresh_token");
    }

    // 获取用户信息
    const user = await authApi.getCurrentUser();
    setAuth(user, data.access_token, refreshToken);
    navigate("/");
    return data;
  };

  return { ...mutation, mutateAsync: login };
};

// 注册 Hook
export const useRegister = () => {
  const navigate = useNavigate();

  const mutation = useMutation({
    mutationFn: authApi.register,
  });

  const register = async (data: Parameters<typeof authApi.register>[0]) => {
    const result = await mutation.mutateAsync(data);
    navigate("/login", {
      state: { message: "注册成功！请查收验证邮件。" },
    });
    return result;
  };

  return { ...mutation, mutateAsync: register };
};

// 登出 Hook
export const useLogout = () => {
  const navigate = useNavigate();
  const clearAuth = useAuthStore((state) => state.clearAuth);
  const queryClient = useQueryClient();

  const mutation = useMutation({
    mutationFn: authApi.logout,
  });

  const logout = async () => {
    try {
      await mutation.mutateAsync();
    } catch (error) {
      // 即使登出失败也清除本地状态
    } finally {
      clearAuth();
      queryClient.clear();
      navigate("/");
    }
  };

  return { ...mutation, mutateAsync: logout };
};

// 获取当前用户 Hook
export const useCurrentUser = () => {
  const isAuthenticated = useAuthStore((state) => state.isAuthenticated);

  return useQuery({
    queryKey: ["currentUser"],
    queryFn: authApi.getCurrentUser,
    enabled: isAuthenticated,
  });
};