/** * 该文件可自行根据业务逻辑进行调整 */ import type { HttpResponse } from '@vben/request'; import { useAppConfig } from '@vben/hooks'; import { preferences } from '@vben/preferences'; import { authenticateResponseInterceptor, errorMessageResponseInterceptor, RequestClient, } from '@vben/request'; import { useAccessStore } from '@vben/stores'; import { message } from 'ant-design-vue'; // eslint-disable-next-line n/no-extraneous-import import { Base64 } from 'js-base64'; import { useAuthStore } from '#/store'; import { refreshTokenApi } from './core'; const { apiURL } = useAppConfig(import.meta.env, import.meta.env.PROD); const arr = new Set([ 'accept_tel', 'doctor', 'express_mobile', 'id_card', 'idcard', 'mobile', 'password', 'patient_mobile', 'phone', ]); // 敏感数据 const sensitiveData = new Set(['id_card', 'idcard']); function createRequestClient(baseURL: string) { const client = new RequestClient({ baseURL, }); /** * 重新认证逻辑 */ async function doReAuthenticate() { const accessStore = useAccessStore(); const authStore = useAuthStore(); accessStore.setAccessToken(null); if ( preferences.app.loginExpiredMode === 'modal' && accessStore.isAccessChecked ) { accessStore.setLoginExpired(true); } else { await authStore.logout(true); } } /** * 刷新token逻辑 */ async function doRefreshToken() { const accessStore = useAccessStore(); const resp = await refreshTokenApi(); const newToken = resp.data; accessStore.setAccessToken(newToken); return newToken; } function formatToken(token: null | string) { return token ? `Bearer ${token}` : null; } // 请求头处理 client.addRequestInterceptor({ fulfilled: async (config) => { const accessStore = useAccessStore(); config.headers.Authorization = formatToken(accessStore.accessToken); config.headers['Accept-Language'] = preferences.app.locale; // 加密 // eslint-disable-next-line eqeqeq if (config.data != null) { config.data = getRes(config.data, false); } // eslint-disable-next-line eqeqeq if (config.params != null) { config.params = getRes(config.params, false); } return config; }, }); // response数据解构 client.addResponseInterceptor({ fulfilled: (response) => { const { data: responseData, status } = response; // 这里吧 data 换成了 result const { code, result } = responseData; // 判断返回状态码 if (status >= 200 && status <= 500) { switch (code) { case 0: { // 解密 // eslint-disable-next-line eqeqeq if (result != null) { return getRes(result); } return result; } case 401: { response.status = 401; // 401 错误,需要重新登录 authenticateResponseInterceptor({ client, doReAuthenticate, doRefreshToken, enableRefreshToken: preferences.app.enableRefreshToken, formatToken, }); break; } case 403: case 404: case 500: { throw Object.assign({}, response, { response }); } } } throw Object.assign({}, response, { response }); }, }); // token过期的处理 client.addResponseInterceptor( authenticateResponseInterceptor({ client, doReAuthenticate, doRefreshToken, enableRefreshToken: preferences.app.enableRefreshToken, formatToken, }), ); // 通用的错误处理,如果没有进入上面的错误处理逻辑,就会进入这里 client.addResponseInterceptor( errorMessageResponseInterceptor((msg: string, error) => { // 这里可以根据业务进行定制,你可以拿到 error 内的信息进行定制化处理,根据不同的 code 做不同的提示,而不是直接使用 message.error 提示 msg // 当前mock接口返回的错误字段是 error 或者 message const responseData = error?.response?.data ?? {}; const errorMessage = responseData?.error ?? responseData?.message ?? ''; // 如果没有错误信息,则会根据状态码进行提示 message.error(errorMessage || msg); }), ); return client; } export const requestClient = createRequestClient(apiURL); export const baseRequestClient = new RequestClient({ baseURL: apiURL }); /** * 加解密方法 * @param obj * @param isDecode */ function getRes(obj: any, isDecode = true) { try { for (const key in obj) { if (Array.isArray(obj[key])) { obj[key] = getRes(obj[key]); } else if (typeof obj[key] === 'object') { obj[key] = getRes(obj[key]); } else { // 判断obj[key]是否在arr中 if (arr.has(key)) { let aseFile = ''; aseFile = isDecode ? customBase64Decode(obj[key], key) : customBase64Encode(obj[key]); const isGarbled = // eslint-disable-next-line no-control-regex /[\u0000-\u0008\v\f\u000E-\u001F\u007F-\u00FF\u{E000}-\u{F8FF}]/u.test( aseFile, ); if (isGarbled) { aseFile = obj[key]; } if (isDecode && sensitiveData.has(key)) { // 数据脱敏,自动匹配姓名、手机号、身份证 switch (key) { // case 'accept_name': // case 'express_name': // case 'express_region': // case 'patient': { // // 保留前两个字符,其余用星号代替 // aseFile = aseFile.slice(0, 1).padEnd(aseFile.length, '*'); // break; // } // case 'express_mobile': // case 'mobile': { // // 保留前三位和后四位,中间用星号代替 // aseFile = // aseFile.slice(0, 3).padEnd(aseFile.length - 4, '*') + // aseFile.slice(-4); // break; // } case 'id_card': case 'idcard': { // 保留前六位和后四位,中间用星号代替 aseFile = aseFile.slice(0, 6).padEnd(aseFile.length - 4, '*') + aseFile.slice(-4); break; } default: { // 默认情况下,全部用星号代替 break; } } } obj[key] = aseFile; } } } return obj; } catch (error) { console.error('错误', error); return obj; } } /** * 解密 * @param data * @param key */ function customBase64Decode(data: string, key = '') { // return data try { // 第一次Base64解码 const decodedFirst = Base64.decode(data); // 截取从第30个字符开始往后的内容 const subStr = decodedFirst.slice(30); // 第二次Base64解码 const base64 = Base64.decode(subStr); return base64.length > 0 ? base64 : data; } catch (error) { console.error( `解码Base64字符串【${data}】时发生错误,KEY:【${key}】`, error, ); return data; } } /** * 加密 * @param data */ function customBase64Encode(data: string) { // eslint-disable-next-line eqeqeq if (data == null || data === '') return data; // return data try { // 和加密方法差不多但是是要生成30个随机字符 let randomString = ''; for (let i = 0; i < 30; i++) { const randomChar = String.fromCodePoint( Math.floor(Math.random() * (126 - 32 + 1)) + 32, ); randomString += randomChar; } const encoded = Base64.encode(data); return Base64.encode(randomString + encoded); } catch (error) { console.error(`加密字符串【${data}】时发生错误`, error); return data; } }