82 lines
2.5 KiB
TypeScript
82 lines
2.5 KiB
TypeScript
import type { AxiosError, AxiosInstance, InternalAxiosRequestConfig } from "axios";
|
||
|
||
import { ApiCodeEnum } from "@/enums/api";
|
||
import { useUserStoreHook } from "@/stores/user";
|
||
import { AuthStorage } from "@/utils/auth";
|
||
import type { ApiResult } from "@/api/common";
|
||
|
||
// 携带续期重试标记，mergeConfig 会把它带进重试后的新 config
|
||
type RetryRequestConfig = InternalAxiosRequestConfig & { isRetried?: boolean };
|
||
|
||
/**
|
||
* 注册令牌续期拦截器：过期自动续期并重试请求
|
||
*/
|
||
export function installTokenRefresh(http: AxiosInstance) {
|
||
// 续期进行中时先等待完成，避免带旧令牌发出后再 401 重来；续期失败则继续用现有令牌
|
||
http.interceptors.request.use(async (config: InternalAxiosRequestConfig) => {
|
||
if (config.anonymous) {
|
||
return config;
|
||
}
|
||
|
||
await useUserStoreHook()
|
||
.waitRefresh()
|
||
.catch(() => {});
|
||
|
||
const token = AuthStorage.getAccessToken();
|
||
if (token) {
|
||
config.headers.Authorization = `Bearer ${token}`;
|
||
}
|
||
|
||
return config;
|
||
});
|
||
|
||
http.interceptors.response.use(undefined, async (error: AxiosError<ApiResult>) => {
|
||
const { config, response } = error;
|
||
|
||
// 网络错误交回基础拦截器提示
|
||
if (!response) {
|
||
return Promise.reject(error);
|
||
}
|
||
|
||
const { code } = response.data ?? {};
|
||
|
||
// Token 过期
|
||
if (code === ApiCodeEnum.ACCESS_TOKEN_INVALID) {
|
||
const retryConfig = config as RetryRequestConfig | undefined;
|
||
const userStore = useUserStoreHook();
|
||
|
||
// 匿名请求已重试过、或本就不参与续期：直接退出登录
|
||
if (!retryConfig || retryConfig.isRetried || retryConfig.anonymous) {
|
||
await userStore.redirectToLogin("expired");
|
||
return Promise.reject(new Error("Token Invalid"));
|
||
}
|
||
|
||
retryConfig.isRetried = true;
|
||
|
||
try {
|
||
await userStore.refreshTokenOnce();
|
||
|
||
const token = AuthStorage.getAccessToken();
|
||
if (!token) {
|
||
throw new Error("续期未返回访问令牌");
|
||
}
|
||
|
||
retryConfig.headers.set("Authorization", `Bearer ${token}`);
|
||
return http(retryConfig);
|
||
} catch {
|
||
await userStore.redirectToLogin("expired");
|
||
return Promise.reject(new Error("Token refresh failed"));
|
||
}
|
||
}
|
||
|
||
// Refresh token 失效
|
||
if (code === ApiCodeEnum.REFRESH_TOKEN_INVALID) {
|
||
await useUserStoreHook().redirectToLogin("expired");
|
||
return Promise.reject(new Error("Token Invalid"));
|
||
}
|
||
|
||
// 其它错误交回基础拦截器处理
|
||
return Promise.reject(error);
|
||
});
|
||
}
|