import type { AxiosResponse } from 'axios'; import { BACKEND_ERROR_CODE, createFlatRequest } from '@sa/axios'; import { useAuthStore } from '@/store/modules/auth'; import { $t } from '@/locales'; import { localStg } from '@/utils/storage'; import { getServiceBaseURL } from '@/utils/service'; import { handleRefreshToken, showErrorMsg } from './shared'; import type { RequestInstanceState } from './type'; const isHttpProxy = import.meta.env.DEV && import.meta.env.VITE_HTTP_PROXY === 'Y'; const { baseURL } = getServiceBaseURL(import.meta.env, isHttpProxy); export const request = createFlatRequest( { baseURL, 'axios-retry': { retries: 0 }, headers: { timeout: 6000 } }, { async onRequest(config) { const { headers } = config; // set token const token = localStg.get('token'); const namespaceId = localStg.get('namespaceId'); // const Authorization = token ? `Bearer ${token}` : null; headers['SNAIL-JOB-AUTH'] = token; headers['SNAIL-JOB-NAMESPACE-ID'] = namespaceId; Object.assign(headers, { 'SNAIL-JOB-AUTH': token, 'SNAIL-JOB-NAMESPACE-ID': namespaceId }); return config; }, isBackendSuccess(response) { // when the backend response code is "0000"(default), it means the request is success // to change this logic by yourself, you can modify the `VITE_SERVICE_SUCCESS_CODE` in `.env` file return String(response.data.status) === import.meta.env.VITE_SERVICE_SUCCESS_CODE; }, async onBackendFail(response, instance) { const authStore = useAuthStore(); function handleLogout() { authStore.resetStore(); } function logoutAndCleanup() { handleLogout(); window.removeEventListener('beforeunload', handleLogout); request.state.errMsgStack = request.state.errMsgStack.filter(msg => msg !== response.data.message); } // when the backend response code is in `logoutCodes`, it means the user will be logged out and redirected to login page const logoutCodes = import.meta.env.VITE_SERVICE_LOGOUT_CODES?.split(',') || []; if (logoutCodes.includes(response.data.status?.toString())) { handleLogout(); return null; } // when the backend response code is in `modalLogoutCodes`, it means the user will be logged out by displaying a modal const modalLogoutCodes = import.meta.env.VITE_SERVICE_MODAL_LOGOUT_CODES?.split(',') || []; if (modalLogoutCodes.includes(response.data.status?.toString())) { request.state.errMsgStack = [...(request.state.errMsgStack || []), response.data.message]; // prevent the user from refreshing the page window.addEventListener('beforeunload', handleLogout); // prevent repeated pop-ups if (!request.state.isLogout) { request.state.isLogout = true; window.$dialog?.error({ title: $t('common.error'), content: $t('request.logoutWithModalMsg'), positiveText: $t('common.confirm'), maskClosable: false, closeOnEsc: false, onPositiveClick() { request.state.isLogout = false; logoutAndCleanup(); }, onClose() { request.state.isLogout = false; logoutAndCleanup(); } }); } return null; } // when the backend response code is in `expiredTokenCodes`, it means the token is expired, and refresh token // the api `refreshToken` can not return error code in `expiredTokenCodes`, otherwise it will be a dead loop, should return `logoutCodes` or `modalLogoutCodes` const expiredTokenCodes = import.meta.env.VITE_SERVICE_EXPIRED_TOKEN_CODES?.split(',') || []; if (expiredTokenCodes.includes(response.data.status) && !request.state.isRefreshingToken) { request.state.isRefreshingToken = true; const refreshConfig = await handleRefreshToken(response.config); request.state.isRefreshingToken = false; if (refreshConfig) { return instance.request(refreshConfig) as Promise; } } return null; }, transformBackendResponse(response) { return response.data.total ? response.data : response.data.data; }, onError(error) { // when the request is fail, you can show error message let message = error.message; let backendErrorCode = ''; // get backend error message and code if (String(error.code) === BACKEND_ERROR_CODE) { message = error.response?.data?.message || message; backendErrorCode = error.response?.data?.status || ''; } // the error message is displayed in the modal const modalLogoutCodes = import.meta.env.VITE_SERVICE_MODAL_LOGOUT_CODES?.split(',') || []; if (modalLogoutCodes.includes(backendErrorCode)) { return; } // when the token is expired, refresh token and retry request, so no need to show error message const expiredTokenCodes = import.meta.env.VITE_SERVICE_EXPIRED_TOKEN_CODES?.split(',') || []; if (expiredTokenCodes.includes(backendErrorCode)) { return; } showErrorMsg(request.state, message); } } );