matrix_fe/src/app.tsx

import { AvatarDropdown, AvatarName, Question, SelectLang } from '@/components';
import { current } from '@/services/matrix/admin';
import { LinkOutlined } from '@ant-design/icons';
import { MenuDataItem, SettingDrawer } from '@ant-design/pro-components';
import { Link, history } from '@umijs/max';

import { appList } from '@/services/matrix/admin';
import type { Settings as LayoutSettings } from '@ant-design/pro-components';
import type { RunTimeLayoutConfig } from '@umijs/max';
import { JSXElementConstructor, ReactElement, ReactNode, ReactPortal } from 'react';
import defaultSettings from '../config/defaultSettings';
import access from './access';
import { errorConfig } from './requestErrorConfig';
const isDev = process.env.NODE_ENV === 'development';
const loginPath = '/user/login';

/**
 * @see  https://umijs.org/zh-CN/plugins/plugin-initial-state
 * */
export async function getInitialState(): Promise<{
  settings?: Partial<LayoutSettings>;
  currentUser?: API.MatrixAdmin;
  loading?: boolean;
  fetchUserInfo?: () => Promise<API.MatrixAdmin | undefined>;
}> {
  const fetchUserInfo = async () => {
    try {
      const msg = await current({
        skipErrorHandler: true,
      });
      return msg.data;
    } catch (error) {
      history.push(loginPath);
    }
    return undefined;
  };
  // 如果不是登录页面，执行
  const { location } = history;
  if (location.pathname !== loginPath) {
    const currentUser = await fetchUserInfo();
    return {
      fetchUserInfo,
      currentUser,
      settings: defaultSettings as Partial<LayoutSettings>,
    };
  }
  return {
    fetchUserInfo,
    settings: defaultSettings as Partial<LayoutSettings>,
  };
}

const matrixMenu = (app: API.MatrixAppBo): MenuDataItem => {
  if (!app.enableCash) {
    return {
      path: '/advList/' + app.code,
      name: app.name,
      access: 'canAdmin',
      component: './AdvRecordList',
    };
  }
  return {
    name: app.name,
    access: 'canAdmin',
    children: [
      {
        name: '广告数据',
        path: '/advListV2/' + app.code,
        access: 'canAdmin',
        component: './AdvRecordListV2',
      },
      {
        name: '用户管理',
        path: '/user/' + app.code,
        access: 'canAdmin',
        component: './UserManagement',
      },
      {
        name: '提现管理',
        path: '/money/' + app.code,
        access: 'canAdmin',
        component: './MoneyManagement',
      },
    ],
  };
};

// ProLayout 支持的api https://procomponents.ant.design/components/layout
export const layout: RunTimeLayoutConfig = ({ initialState, setInitialState }) => {
  return {
    menu: {
      // 每当 initialState?.currentUser?.userid 发生修改时重新执行 request
      params: {
        userId: initialState?.currentUser?.id,
      },
      request: async () => {
        const apps = await appList();
        const menuData: MenuDataItem[] = [];
        const role = initialState?.currentUser?.role || 100;
        apps.data?.forEach((x: any) => {
          if (x.hide === 1 && role <= 2) {
          } else {
            menuData.push(matrixMenu(x));
          }
          if (role === 4) {
            menuData.push({
              path: '/app/' + x.code,
              name: x.name,
              access: 'canDeviceOwner',
              component: './DeviceOwnerApp',
            });
          }
        });

        menuData.push({
          path: '/adminList',
          name: '人员管理',
          access: 'canAdmin',
        });

        menuData.push({
          path: '/bind',
          name: '绑定设备',
          access: 'canDeviceOwner',
        });

        menuData.push({
          path: '/appList',
          name: '应用列表',
          access: 'canAdmin',
        });

        menuData.push({
          path: '/super',
          name: '魔法之地',
          access: 'superAdmin',
        });

        return menuData;
      },
    },
    // 添加权限校验逻辑
    menuDataRender: (menuData: MenuDataItem[]) =>
      menuData.map((item) => {
        const val = item.access;
        if (!val) {
          return item;
        }

        const accessData: { [key: string]: boolean | 0 | undefined } = access(initialState);
        if (!accessData[val]) {
          return {
            ...item,
            hideInMenu: true,
          };
        }

        return item;
      }),
    actionsRender: () => [<Question key="doc" />, <SelectLang key="SelectLang" />],
    avatarProps: {
      // src: initialState?.currentUser?.avatar,
      title: <AvatarName />,
      render: (
        _: any,
        avatarChildren:
          | string
          | number
          | boolean
          | ReactElement<any, string | JSXElementConstructor<any>>
          | Iterable<ReactNode>
          | ReactPortal
          | null
          | undefined,
      ) => {
        return <AvatarDropdown>{avatarChildren}</AvatarDropdown>;
      },
    },
    // waterMarkProps: {
    //   content: initialState?.currentUser?.name,
    // },
    // footerRender: () => <Footer />,
    onPageChange: () => {
      const { location } = history;
      // 如果没有登录，重定向到 login
      if (!initialState?.currentUser && location.pathname !== loginPath) {
        history.push(loginPath);
      }
    },
    bgLayoutImgList: [
      {
        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 onPointerOverCapture={undefined} onPointerMoveCapture={undefined} />
            <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,
  };
};

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