web/src/layouts/index.tsx

import logo from '@/assets/logo.png';
import { Layout, Space, theme } from 'antd';
import classnames from 'classnames';
import React, { useEffect, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { Outlet, useLocation, useNavigate } from 'umi';
import '../locales/config';
import User from './components/user';
import styles from './index.less';

const { Header, Content } = Layout;

const App: React.FC = (props) => {
  const { t } = useTranslation();
  const navigate = useNavigate();
  const {
    token: { colorBgContainer, borderRadiusLG },
  } = theme.useToken();
  const [current, setCurrent] = useState('knowledge');

  const location = useLocation();
  useEffect(() => {
    if (location.pathname !== '/') {
      const path = location.pathname.split('/');
      // setCurrent(path[1]);
    }
    console.log(location.pathname.split('/'));
  }, [location.pathname]);

  const handleChange = (path: string) => {
    // setCurrent(path)
    navigate(path);
  };
  const tagsData = [
    { path: '/knowledge', name: 'knowledge' },
    { path: '/chat', name: 'chat' },
    { path: '/file', name: 'file' },
  ];

  return (
    <Layout className={styles.layout}>
      <Layout>
        <Header
          style={{
            padding: '0 8px',
            background: colorBgContainer,
            display: 'flex',
            justifyContent: 'space-between',
            alignItems: 'center',
          }}
        >
          <img src={logo} alt="" style={{ height: 30, width: 30 }} />
          <Space size={[0, 8]} wrap>
            {tagsData.map((item) => (
              <span
                key={item.name}
                className={classnames(styles['tag'], {
                  [styles['checked']]: current === item.name,
                })}
                onClick={() => handleChange(item.path)}
              >
                {item.name}
              </span>
            ))}
          </Space>
          <User></User>
        </Header>
        <Content
          style={{
            margin: '24px 16px',

            minHeight: 280,
            background: colorBgContainer,
            borderRadius: borderRadiusLG,
            overflow: 'auto',
          }}
        >
          <Outlet />
        </Content>
      </Layout>
    </Layout>
  );
};

export default App;