chore: remove patch client routes

src/app.tsx

@@ -1,4 +1,4 @@
-import { Navigate, RequestConfig, history } from '@umijs/max';
+import { RequestConfig, history } from '@umijs/max';
 import { requestConfig } from './request-config';
 import { queryCurrentUserState } from './services/profile/apis';
 
@@ -40,21 +40,22 @@ export async function getInitialState() {
   };
 }
 
-export const patchClientRoutes = async (params: { routes: any[] }) => {
-  const { routes } = params;
-  const data = await queryCurrentUserState({
-    skipErrorHandler: true
-  });
+// export const patchClientRoutes = async (params: { routes: any[] }) => {
+//   const { routes } = params;
+//   console.log('routes============999', routes);
+//   const data = await queryCurrentUserState({
+//     skipErrorHandler: true
+//   });
 
-  routes.unshift({
-    path: '/',
-    element: data?.is_admin ? (
-      <Navigate to="/dashboard" replace />
-    ) : (
-      <Navigate to="/playground" replace />
-    )
-  });
-};
+//   routes.unshift({
+//     path: '/',
+//     element: data?.is_admin ? (
+//       <Navigate to="/dashboard" replace />
+//     ) : (
+//       <Navigate to="/playground" replace />
+//     )
+//   });
+// };
 
 export const request: RequestConfig = {
   baseURL: ' /v1',

src/assets/styles/common.less

@@ -10,6 +10,10 @@
   margin-left: 5px;
 }
 
+.m-l-8 {
+  margin-left: 8px;
+}
+
 .flex {
   display: flex;
 }

src/layouts/index.tsx

@@ -52,23 +52,23 @@ const filterRoutes = (
 };
 
 // 格式化路由 处理因 wrapper 导致的 菜单 path 不一致
-const mapRoutes = (routes: IRoute[]) => {
+const mapRoutes = (routes: IRoute[], role: string) => {
   if (routes.length === 0) {
     return [];
   }
   return routes.map((route) => {
     // 需要 copy 一份, 否则会污染原始数据
-    const newRoute = { ...route };
+    const newRoute = { ...route, role };
     if (route.originPath) {
       newRoute.path = route.originPath;
     }
 
     if (Array.isArray(route.routes)) {
-      newRoute.routes = mapRoutes(route.routes);
+      newRoute.routes = mapRoutes(route.routes, role);
     }
 
     if (Array.isArray(route.children)) {
-      newRoute.children = mapRoutes(route.children);
+      newRoute.children = mapRoutes(route.children, role);
     }
 
     return newRoute;
@@ -117,8 +117,9 @@ export default (props: any) => {
     }
   );
 
-  console.log('clientRoutes===========', clientRoutes, newRoutes);
-  const [route] = useAccessMarkedRoutes(mapRoutes(newRoutes));
+  const role = initialState?.currentUser?.is_admin ? 'admin' : 'user';
+  const [route] = useAccessMarkedRoutes(mapRoutes(newRoutes, role));
+  console.log('clientRoutes===========', route, clientRoutes, newRoutes);
 
   patchRoutes({
     routes: route.children,
@@ -150,7 +151,7 @@ export default (props: any) => {
           // 如果没有登录，重定向到 login
           if (!initialState?.currentUser && location.pathname !== loginPath) {
             history.push(loginPath);
-          } else if (location.path === '/') {
+          } else if (location.pathname === '/') {
             const pathname = initialState?.currentUser?.is_admin
               ? '/dashboard'
               : '/playground';

src/layouts/rightRender.tsx

@@ -6,7 +6,7 @@ import {
   LogoutOutlined,
   SettingOutlined
 } from '@ant-design/icons';
-import { history } from '@umijs/max';
+import { SelectLang, history } from '@umijs/max';
 import { Avatar, Dropdown, Menu, Spin, version } from 'antd';
 
 export function getRightRenderContent(opts: {
@@ -56,6 +56,21 @@ export function getRightRenderContent(opts: {
     );
   }
 
+  const renderExtraActions = () => {
+    return (
+      <SelectLang
+        style={{ padding: '10px 18px', marginBottom: '20px' }}
+        reload={false}
+        icon={
+          <span style={{ fontSize: '12px' }}>
+            <GlobalOutlined />
+            <span className="m-l-8">语言</span>
+          </span>
+        }
+      ></SelectLang>
+    );
+  };
+
   // 如果没有打开Locale，并且头像为空就取消掉这个返回的内容
   if (!avatar) return null;
 
@@ -87,18 +102,6 @@ export function getRightRenderContent(opts: {
       //     console.log('theme');
       //   }
       // },
-      {
-        key: 'lang',
-        label: (
-          <>
-            <GlobalOutlined />
-            语言
-          </>
-        ),
-        onClick: () => {
-          console.log('lang');
-        }
-      },
       {
         key: 'logout',
         label: (
@@ -136,20 +139,23 @@ export function getRightRenderContent(opts: {
   }
 
   return (
-    <div className="umi-plugin-layout-right anticon">
-      {opts.runtimeConfig.logout ? (
-        <>
-          <Dropdown
-            {...dropdownProps}
-            overlayClassName="umi-plugin-layout-container"
-          >
-            {avatar}
-          </Dropdown>
-          <span></span>
-        </>
-      ) : (
-        avatar
-      )}
+    <div style={{ width: '100%' }}>
+      {renderExtraActions()}
+      <div className="umi-plugin-layout-right anticon">
+        {opts.runtimeConfig.logout ? (
+          <>
+            <Dropdown
+              {...dropdownProps}
+              overlayClassName="umi-plugin-layout-container"
+            >
+              {avatar}
+            </Dropdown>
+            <span></span>
+          </>
+        ) : (
+          avatar
+        )}
+      </div>
     </div>
   );
 }

src/pages/login/index.tsx

@@ -27,6 +27,10 @@ const Login = () => {
 
   const [form] = Form.useForm();
 
+  const gotoDefaultPage = (userInfo: any) => {
+    const pathname = userInfo?.is_admin ? '/dashboard' : '/playground';
+    history.push(pathname);
+  };
   const fetchUserInfo = async () => {
     const userInfo = await initialState?.fetchUserInfo?.();
 
@@ -38,6 +42,7 @@ const Login = () => {
         }));
       });
     }
+    return userInfo;
   };
 
   const handleLogin = async (values: any) => {
@@ -47,8 +52,9 @@ const Login = () => {
         username: values.username,
         password: values.password
       });
-      await fetchUserInfo();
-      history.push('/');
+      const userInfo = await fetchUserInfo();
+
+      gotoDefaultPage(userInfo);
     } catch (error) {
       console.log('error====', error);
     }