negareh-admin/src/App.tsx

import { type FC } from 'react'
import '@/assets/fonts/irancell/style.css'
import 'swiper/swiper-bundle.css';
import 'react-loading-skeleton/dist/skeleton.css'
import 'react-toastify/dist/ReactToastify.css'
import { BrowserRouter } from 'react-router-dom'
import { QueryCache, QueryClient, QueryClientProvider } from '@tanstack/react-query'
import { ToastContainer } from 'react-toastify'
import Toast from '@/components/Toast'
import { getRefreshToken, getToken, removeRefreshToken, removeToken, setRefreshToken, setToken } from './config/func'
import MainRouter from './router/MainRouter'
import AuthRouter from './router/AuthRouter'
import { Paths } from './config/Paths';
import type { IApiErrorRepsonse } from './shared/types/Types';
import { refreshToken } from './pages/auth/service/AuthService';

declare global {
  interface Window {
    isRefreshingToken?: boolean;
    isRefreshTokenExpired?: boolean;
  }
}

const queryClient = new QueryClient({
  queryCache: new QueryCache({
    onError: async (error: IApiErrorRepsonse) => {
      if (error?.response?.status === 401) {
        // اگر refresh token منقضی شده باشد، دیگر تلاش نکن
        if (window.isRefreshTokenExpired) {
          return;
        }

        // بررسی کن که آیا خطا از refresh endpoint است
        const isRefreshEndpoint = error?.config?.url?.includes('/public/auth/refresh');

        // اگر خطا از refresh endpoint است، refresh token منقضی شده
        if (isRefreshEndpoint) {
          window.isRefreshTokenExpired = true;
          await removeToken();
          await removeRefreshToken();
          window.location.href = Paths.auth.login;
          return;
        }

        try {
          // Use a flag to track if refresh is in progress
          if (window.isRefreshingToken) {
            // Wait for the refresh to complete
            await new Promise(resolve => {
              const checkComplete = setInterval(() => {
                if (!window.isRefreshingToken) {
                  clearInterval(checkComplete);
                  resolve(true);
                }
              }, 100);
            });
            return;
          }

          // Set the flag to indicate refresh is in progress
          window.isRefreshingToken = true;

          try {
            const refreshTokenValue = await getRefreshToken();

            if (!refreshTokenValue) {
              window.isRefreshTokenExpired = true;
              await removeToken();
              await removeRefreshToken();
              window.location.href = Paths.auth.login;
              return;
            }

            const { data } = await refreshToken({ refreshToken: refreshTokenValue || '' });

            if (data?.accessToken?.token) {
              // Save the new token
              await setToken(data?.accessToken?.token);

              // If refresh token also returned, update it
              if (data.refreshToken?.token) {
                await setRefreshToken(data.refreshToken?.token);
              }

              // Reset expired flag if refresh was successful
              window.isRefreshTokenExpired = false;

              // Retry the failed request
              queryClient.invalidateQueries();
              return;
            } else {
              throw new Error('Invalid token response');
            }
          } finally {
            window.isRefreshingToken = false;
          }
        } catch (refreshError: unknown) {
          console.error('Token refresh failed:', refreshError);

          // بررسی کن که آیا خطا از refresh endpoint است
          const errorWithConfig = refreshError as IApiErrorRepsonse;
          const isRefreshError = errorWithConfig?.config?.url?.includes('/admin/auth/refresh');

          // اگر refresh token خودش 401 خورده، آن را منقضی شده علامت بزن
          if (isRefreshError || errorWithConfig?.response?.statusCode === 401) {
            window.isRefreshTokenExpired = true;
          }

          // Clear tokens and redirect to login
          await removeToken();
          await removeRefreshToken();
          window.location.href = Paths.auth.login;
        }
      }
    }
  }),
  defaultOptions: {
    queries: {
      refetchOnWindowFocus: false,
      retry: false
      // staleTime: 86400000 // 1 day in milliseconds
    }
  }
});

const App: FC = () => {

  const isLoggedIn = getToken()

  return (
    <QueryClientProvider client={queryClient}>
      <BrowserRouter>
        {isLoggedIn ? <MainRouter /> : <AuthRouter />}
      </BrowserRouter>
      <Toast />
      <ToastContainer
        position="top-left"
        autoClose={3000}
        hideProgressBar={false}
        newestOnTop={false}
        closeOnClick
        rtl
        pauseOnFocusLoss
        draggable
        pauseOnHover
      />
    </QueryClientProvider>
  )
}

export default App