client/src/_core/hooks/useAuth.ts

import { getLoginUrl } from "@/const";
import { trpc } from "@/lib/trpc";
import { TRPCClientError } from "@trpc/client";
import { useCallback, useEffect, useMemo } from "react";

type UseAuthOptions = {
  redirectOnUnauthenticated?: boolean;
  redirectPath?: string;
};

export function useAuth(options?: UseAuthOptions) {
  const { redirectOnUnauthenticated = false, redirectPath = getLoginUrl() } =
    options ?? {};
  const utils = trpc.useUtils();

  const meQuery = trpc.auth.me.useQuery(undefined, {
    retry: false,
    refetchOnWindowFocus: false,
  });

  const logoutMutation = trpc.auth.logout.useMutation({
    onSuccess: () => {
      utils.auth.me.setData(undefined, null);
    },
  });

  const logout = useCallback(async () => {
    try {
      await logoutMutation.mutateAsync();
    } catch (error: unknown) {
      if (
        error instanceof TRPCClientError &&
        error.data?.code === "UNAUTHORIZED"
      ) {
        return;
      }
      throw error;
    } finally {
      utils.auth.me.setData(undefined, null);
      await utils.auth.me.invalidate();
    }
  }, [logoutMutation, utils]);

  const state = useMemo(() => {
    // 本地模式: 移除localStorage存储
    // localStorage.setItem("manus-runtime-user-info", JSON.stringify(meQuery.data));
    return {
      user: meQuery.data ?? null,
      loading: meQuery.isLoading || logoutMutation.isPending,
      error: meQuery.error ?? logoutMutation.error ?? null,
      // 本地模式: 始终返回true,允许所有用户访问
      isAuthenticated: true,
    };
  }, [
    meQuery.data,
    meQuery.error,
    meQuery.isLoading,
    logoutMutation.error,
    logoutMutation.isPending,
  ]);

  // 本地模式: 禁用重定向逻辑
  // useEffect(() => {
  //   if (!redirectOnUnauthenticated) return;
  //   if (meQuery.isLoading || logoutMutation.isPending) return;
  //   if (state.user) return;
  //   if (typeof window === "undefined") return;
  //   if (window.location.pathname === redirectPath) return;
  //   window.location.href = redirectPath
  // }, [
  //   redirectOnUnauthenticated,
  //   redirectPath,
  //   logoutMutation.isPending,
  //   meQuery.isLoading,
  //   state.user,
  // ]);

  return {
    ...state,
    refresh: () => meQuery.refetch(),
    logout,
  };
}