General-AI-Platform-Web/src/app.tsx

import { AvatarDropdown, AvatarName, Footer, Question, SelectLang } from '@/components';
import { LinkOutlined } from '@ant-design/icons';
import type { Settings as LayoutSettings } from '@ant-design/pro-components';
import { SettingDrawer } from '@ant-design/pro-components';
import { Link, RunTimeLayoutConfig, getLocale, history, useIntl } from '@umijs/max';
import { RequestConfig } from 'umi';
import defaultSettings from '../config/defaultSettings';
import { errorConfig } from './requestErrorConfig';
const isDev = process.env.NODE_ENV === 'development';
const loginPath = '/user/login';
// @ts-ignore
import { SelectRole } from '@/components/RightContent';
import zhCN from '@/locales/zh-CN';
import { postMenuGetMenu } from '@/services/system/Menu';
import { getUserGetUserInfo } from '@/services/system/User';
import fixMenuItemIcon from '@/utils/FixMenuItemIcon';
import { getAllRouteNameTile } from '@/utils/common';
import { addLocale } from '@@/plugin-locale';
import { MenuDataItem } from '@ant-design/pro-layout';
import cookie from 'react-cookies';
/**
 * @see  https://umijs.org/zh-CN/plugins/plugin-initial-state
 * */
// 自定义路由路径集合
const isCustomPageTitlePaths : string[]= ['/resource/algorithm-model-detail']


export async function getInitialState(): Promise<{
  settings?: Partial<LayoutSettings>;
  currentUser?: API.UserView;
  loading?: boolean;
  fetchUserInfo?: any;
  menuData?: MenuDataItem[];
}> {
  const fetchUserInfo = async () => {
    try {
      const msg = await getUserGetUserInfo();
      return msg.data.userInfo;
    } catch (error) {
      history.push(loginPath);
    }
    return undefined;
  };
  // 如果不是登录页面，执行
  const { location } = history;
  if (location.pathname !== loginPath && localStorage.getItem('access')) {
    const currentUser = await fetchUserInfo();
    const menus = await postMenuGetMenu();
    if (getLocale() === 'zh-CN') {
      let localData = getAllRouteNameTile(menus.data.routes, '');
      let localRes: any = {};
      localData.forEach((v) => {
        // console.log(v.title,'localData_v')
        localRes[`menu${v.name}`] = v.title;
      });
      console.log(localRes, 'getAllRouteNameTile');
      addLocale('zh-CN', localRes, {
        momentLocale: 'zh-CN',
        antd: zhCN,
      });
    }

    return {
      fetchUserInfo,
      currentUser,
      settings: defaultSettings as Partial<LayoutSettings>,
      menuData: menus.data.routes,
    };
  }

  return {
    fetchUserInfo,
    settings: defaultSettings as Partial<LayoutSettings>,
  };
}

// ProLayout 支持的api https://procomponents.ant.design/components/layout
export const layout: RunTimeLayoutConfig = ({ initialState, setInitialState }) => {

  return {
    actionsRender: () => [
      <SelectRole key="selectRole" />,
      <Question key="doc" />,
      <SelectLang key="SelectLang" />,
    ],
    avatarProps: {
      src: SERVER_HOST + initialState?.currentUser?.avatarUrl,
      title: <AvatarName />,
      render: (_, avatarChildren) => {
        // console.log(avatarChildren,'')
        return <AvatarDropdown>{avatarChildren}</AvatarDropdown>;
      },
    },
    pageTitleRender: () => {
      console.log(history, 'pageTitleRender');
      // TODO 目前使用路由path全匹配，后续改成从接口字段读取
      let isCustom: boolean = false
      for(let i=0; i<isCustomPageTitlePaths.length; i++){
        const item = isCustomPageTitlePaths[i]
        if(history.location.pathname.includes(item)){
          isCustom = true
          break;
        } else {
          isCustom = false
        }
      }
      return isCustom ? ("") : (undefined)
      // 默认不进行重定义 使用null即可
    },
    waterMarkProps: {
      // content: initialState?.currentUser?.name,
    },
    menu: {
      loading: initialState?.loading || false,
      locale: true,
      params: {
        userId: initialState?.currentUser?.id,
      },
      request: async () => {
        // console.log(initialState?.menuData,'initialState_menuData')
        return initialState?.menuData || [];
      },
    },
    footerRender: () => <Footer />,
    onPageChange: () => {
      const { location } = history;
      // 如果没有登录，重定向到 login
      if (!initialState?.currentUser && location.pathname !== loginPath) {
        history.push(loginPath);
      }
    },
    menuDataRender: () => {
      if (initialState?.menuData !== undefined) {
        return fixMenuItemIcon(initialState?.menuData || []);
      }
    },
    layoutBgImgList: [
      {
        src: 'https://mdn.alipayobjects.com/yuyan_qk0oxh/afts/img/D2LWSqNny4sAAAAAAAAAAAAAFl94AQBr',
        left: 85,
        bottom: 100,
        height: '303px',
      },
      {
        src: 'https://mdn.alipayobjects.com/yuyan_qk0oxh/afts/img/C2TWRpJpiC0AAAAAAAAAAAAAFl94AQBr',
        bottom: -68,
        right: -45,
        height: '303px',
      },
      {
        src: 'https://mdn.alipayobjects.com/yuyan_qk0oxh/afts/img/F6vSTbj8KpYAAAAAAAAAAAAAFl94AQBr',
        bottom: 0,
        left: 0,
        width: '331px',
      },
    ],
    links: isDev
      ? [
          <Link key="openapi" to="/umi/plugin/openapi" target="_blank">
            <LinkOutlined />
            <span>OpenAPI 文档</span>
          </Link>,
        ]
      : [],
    menuHeaderRender: undefined,
    // 自定义 403 页面
    // unAccessible: <div>unAccessible</div>,
    // 增加一个 loading 的状态
    childrenRender: (children) => {
      // if (initialState?.loading) return <PageLoading />;
      return (
        <>
          {children}
          {isDev && (
            <SettingDrawer
              disableUrlParams
              enableDarkTheme
              settings={initialState?.settings}
              onSettingChange={(settings) => {
                setInitialState((preInitialState) => ({
                  ...preInitialState,
                  settings,
                }));
              }}
            />
          )}
        </>
      );
    },
    ...initialState?.settings,
  };
};

const csrfAndJwtHeaderInterceptor = (url: string, options: RequestConfig) => {
  const csrfHeader = { 'X-CSRFToken': cookie.load('csrftoken') };
  let bearerToken = {};
  const access = localStorage.getItem('access');
  if (access) {
    bearerToken = { 'X-Token': `${access}` };
  }
  let locale = getLocale();
  return {
    url: `${url}`,
    options: {
      ...options,
      interceptors: true,
      headers: { ...csrfHeader, ...bearerToken, 'Accept-Language': locale },
    },
  };
};

const responseHeaderInterceptor = (response: Response) => {
  if ('new-token' in response.headers) {
    localStorage.setItem('access', (response.headers['new-token'] || '') as string);
    window.location.reload();
  }
  return response;
};

/**
 * @name request 配置，可以配置错误处理
 * 它基于 axios 和 ahooks 的 useRequest 提供了一套统一的网络请求和错误处理方案。
 * @doc https://umijs.org/docs/max/request#配置
 */
export const request = {
  ...errorConfig,
  requestInterceptors: [csrfAndJwtHeaderInterceptor],
  responseInterceptors: [responseHeaderInterceptor],
};