live-hub-py/web/frontend/src/layouts/MainLayout.tsx

import { useEffect, useState } from 'react';
import { Layout, Menu, Avatar, Space, Typography, Button, Modal, Dropdown } from 'antd';
import {
  DashboardOutlined, UserOutlined, LogoutOutlined,
  CloudServerOutlined, TeamOutlined, ApiOutlined, KeyOutlined,
  MenuFoldOutlined, MenuUnfoldOutlined, SwapOutlined,
  SunOutlined, MoonOutlined, DesktopOutlined, FileTextOutlined,
} from '@ant-design/icons';
import { useNavigate, useLocation, Outlet } from 'react-router-dom';
import { getUser, clearAuth, hasPerm, type AuthUser } from '../store/auth';
import { authApi } from '../api/modules';
import { useTheme, type ThemeMode } from '../store/theme';

const { Sider, Content } = Layout;
const { Text } = Typography;

const ROLE_LABELS: Record<string, string> = {
  super_admin: '超级管理员',
  operation: '运营',
  support: '客服',
};

const THEME_OPTIONS: { key: ThemeMode; label: string; icon: React.ReactNode }[] = [
  { key: 'light', label: '亮色', icon: <SunOutlined /> },
  { key: 'dark', label: '暗色', icon: <MoonOutlined /> },
  { key: 'system', label: '跟随系统', icon: <DesktopOutlined /> },
];

export default function MainLayout({ onLogout }: { onLogout?: () => void }) {
  const navigate = useNavigate();
  const location = useLocation();
  const [user] = useState<AuthUser | null>(getUser());
  const [collapsed, setCollapsed] = useState(false);
  const { mode, isDark, setMode } = useTheme();

  useEffect(() => {
    if (!user) navigate('/login');
  }, [user]);

  if (!user) return null;

  const menuItems: { key: string; label: string; icon: React.ReactNode }[] = [];

  // Dashboard - 所有人可见
  menuItems.push({ key: '/', label: '概览', icon: <DashboardOutlined /> });

  // 账号管理
  if (hasPerm(user, 'account:view_all') || hasPerm(user, 'account:view_assigned')) {
    menuItems.push({ key: '/accounts', label: '账号管理', icon: <UserOutlined /> });
  }

  // 分配管理
  if (hasPerm(user, 'account:assign')) {
    menuItems.push({ key: '/assignments', label: '分配管理', icon: <SwapOutlined /> });
  }

  // 登录任务
  if (hasPerm(user, 'login:batch') || hasPerm(user, 'login:view_all')) {
    menuItems.push({ key: '/login-tasks', label: '登录任务', icon: <ApiOutlined /> });
  }

  // Cookie 管理
  if (hasPerm(user, 'cookie:view')) {
    menuItems.push({ key: '/cookies', label: 'Cookie 管理', icon: <KeyOutlined /> });
  }

  // 代理配置
  if (hasPerm(user, 'proxy:manage')) {
    menuItems.push({ key: '/proxy', label: '代理配置', icon: <CloudServerOutlined /> });
  }

  // 请求日志（运营和管理员可见）
  if (hasPerm(user, 'audit:view') || hasPerm(user, 'login:batch')) {
    menuItems.push({ key: '/http-logs', label: '请求日志', icon: <FileTextOutlined /> });
  }

  // 用户管理
  if (hasPerm(user, 'user:view')) {
    menuItems.push({ key: '/users', label: '用户管理', icon: <TeamOutlined /> });
  }

  const handleLogout = () => {
    Modal.confirm({
      title: '确认退出',
      content: '确定要退出登录吗？',
      okText: '退出',
      cancelText: '取消',
      okButtonProps: { type: 'primary' },
      centered: true,
      onOk: async () => {
        try {
          await authApi.logout();
        } catch {}
        clearAuth();
        onLogout?.();
        navigate('/login', { replace: true });
      },
    });
  };

  const siderTheme = isDark ? 'dark' : 'light';
  const siderTextColor = isDark ? '#fff' : undefined;

  return (
    <Layout style={{ height: '100vh', overflow: 'hidden' }}>
      <Sider trigger={null} collapsed={collapsed} onCollapse={setCollapsed} theme={siderTheme}>
        <div style={{
          display: 'flex', flexDirection: 'column', height: '100%',
        }}>
          <div style={{
            height: 48, margin: '12px 12px 0', fontSize: 16,
            textAlign: 'center', lineHeight: '48px', fontWeight: 'bold',
            whiteSpace: 'nowrap', overflow: 'hidden', flexShrink: 0,
            display: 'flex', alignItems: 'center', justifyContent: 'center', gap: 8,
            color: siderTextColor,
          }}>
            <span style={{ overflow: 'hidden', textOverflow: 'ellipsis' }}>
              {collapsed ? '鱼' : '斗鱼登录后台'}
            </span>
            <Button
              type="text"
              size="small"
              icon={collapsed ? <MenuUnfoldOutlined /> : <MenuFoldOutlined />}
              onClick={() => setCollapsed(!collapsed)}
              style={{ color: siderTextColor, flexShrink: 0 }}
            />
          </div>
          <Menu
            theme={siderTheme}
            mode="inline"
            selectedKeys={[location.pathname]}
            items={menuItems}
            onClick={({ key }) => navigate(key)}
            style={{ flex: 1, overflow: 'auto', marginTop: 8 }}
          />
          <div style={{
            borderTop: isDark ? '1px solid rgba(255,255,255,0.1)' : '1px solid rgba(0,0,0,0.06)',
            padding: '12px',
            flexShrink: 0,
          }}>
            <Space style={{ color: siderTextColor, width: '100%', marginBottom: collapsed ? 0 : 8 }}>
              <Avatar icon={<UserOutlined />} size="small" />
              {!collapsed && (
                <>
                  <Text style={{ color: siderTextColor }}>{user.username}</Text>
                  <Text style={{ color: isDark ? 'rgba(255,255,255,0.65)' : 'rgba(0,0,0,0.45)', fontSize: 12 }}>
                    ({ROLE_LABELS[user.role] || user.role})
                  </Text>
                </>
              )}
            </Space>
            {collapsed ? (
              <Space direction="vertical" style={{ width: '100%' }}>
                <Dropdown
                  menu={{ items: THEME_OPTIONS.map(o => ({ key: o.key, label: o.label, icon: o.icon, onClick: () => setMode(o.key) })) }}
                  trigger={['click']}
                >
                  <Button
                    block
                    type="text"
                    size="small"
                    icon={isDark ? <MoonOutlined /> : <SunOutlined />}
                    style={{ color: isDark ? 'rgba(255,255,255,0.65)' : undefined }}
                  />
                </Dropdown>
                <Button
                  block
                  type="text"
                  size="small"
                  icon={<LogoutOutlined />}
                  onClick={handleLogout}
                  style={{ color: isDark ? 'rgba(255,255,255,0.65)' : undefined }}
                />
              </Space>
            ) : (
              <Space style={{ width: '100%', justifyContent: 'space-between' }}>
                <Dropdown
                  menu={{ items: THEME_OPTIONS.map(o => ({ key: o.key, label: o.label, icon: o.icon, onClick: () => setMode(o.key) })) }}
                  trigger={['click']}
                >
                  <Button
                    type="text"
                    size="small"
                    icon={isDark ? <MoonOutlined /> : <SunOutlined />}
                    style={{ color: isDark ? 'rgba(255,255,255,0.65)' : undefined }}
                  >
                    {THEME_OPTIONS.find(o => o.key === mode)?.label}
                  </Button>
                </Dropdown>
                <Button
                  type="text"
                  size="small"
                  icon={<LogoutOutlined />}
                  onClick={handleLogout}
                  style={{ color: isDark ? 'rgba(255,255,255,0.65)' : undefined }}
                >
                  退出
                </Button>
              </Space>
            )}
          </div>
        </div>
      </Sider>
      <Layout>
        <Content style={{ margin: 16, padding: 24, borderRadius: 8, overflow: 'auto' }}>
          <Outlet />
        </Content>
      </Layout>
    </Layout>
  );
}