tianpu-ems/frontend/src/layouts/MainLayout.tsx

import { useState, useEffect, useCallback } from 'react';
import { Layout, Menu, Avatar, Dropdown, Typography, Badge, Popover, List, Tag, Empty } from 'antd';
import {
  DashboardOutlined, MonitorOutlined, BarChartOutlined, AlertOutlined,
  FileTextOutlined, CloudOutlined, SettingOutlined, UserOutlined,
  MenuFoldOutlined, MenuUnfoldOutlined, LogoutOutlined, BellOutlined,
  ThunderboltOutlined, AppstoreOutlined, WarningOutlined, CloseCircleOutlined,
  InfoCircleOutlined, FundProjectionScreenOutlined, GlobalOutlined,
} from '@ant-design/icons';
import { Outlet, useNavigate, useLocation } from 'react-router-dom';
import { getUser, removeToken } from '../utils/auth';
import { getAlarmStats, getAlarmEvents } from '../services/api';

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

const menuItems = [
  { key: '/', icon: <DashboardOutlined />, label: '能源总览' },
  { key: '/monitoring', icon: <MonitorOutlined />, label: '实时监控' },
  { key: '/devices', icon: <AppstoreOutlined />, label: '设备管理' },
  { key: '/analysis', icon: <BarChartOutlined />, label: '能耗分析' },
  { key: '/alarms', icon: <AlertOutlined />, label: '告警管理' },
  { key: '/carbon', icon: <CloudOutlined />, label: '碳排放管理' },
  { key: '/reports', icon: <FileTextOutlined />, label: '报表管理' },
  { key: 'bigscreen-group', icon: <FundProjectionScreenOutlined />, label: '可视化大屏',
    children: [
      { key: '/bigscreen', icon: <FundProjectionScreenOutlined />, label: '2D 能源大屏' },
      { key: '/bigscreen-3d', icon: <GlobalOutlined />, label: '3D 园区大屏' },
    ],
  },
  { key: '/system', icon: <SettingOutlined />, label: '系统管理',
    children: [
      { key: '/system/users', label: '用户管理' },
      { key: '/system/roles', label: '角色权限' },
      { key: '/system/settings', label: '系统设置' },
    ],
  },
];

const SEVERITY_CONFIG: Record<string, { icon: React.ReactNode; color: string }> = {
  critical: { icon: <CloseCircleOutlined style={{ color: '#f5222d' }} />, color: 'red' },
  warning: { icon: <WarningOutlined style={{ color: '#faad14' }} />, color: 'orange' },
  info: { icon: <InfoCircleOutlined style={{ color: '#1890ff' }} />, color: 'blue' },
};

export default function MainLayout() {
  const [collapsed, setCollapsed] = useState(false);
  const [alarmCount, setAlarmCount] = useState(0);
  const [recentAlarms, setRecentAlarms] = useState<any[]>([]);
  const navigate = useNavigate();
  const location = useLocation();
  const user = getUser();

  const fetchAlarms = useCallback(async () => {
    try {
      const [stats, events] = await Promise.all([
        getAlarmStats(),
        getAlarmEvents({ status: 'active', page_size: 5 }),
      ]);
      // Stats shape: { severity: { status: count } } — sum all "active" counts
      const statsData = (stats as any) || {};
      let activeTotal = 0;
      for (const severity of Object.values(statsData)) {
        if (severity && typeof severity === 'object') {
          activeTotal += (severity as any).active || 0;
        }
      }
      setAlarmCount(activeTotal);
      const items = (events as any)?.items || (events as any) || [];
      setRecentAlarms(Array.isArray(items) ? items : []);
    } catch {
      // silently ignore - notifications are non-critical
    }
  }, []);

  useEffect(() => {
    fetchAlarms();
    const timer = setInterval(fetchAlarms, 30000);
    return () => clearInterval(timer);
  }, [fetchAlarms]);

  const handleLogout = () => {
    removeToken();
    localStorage.removeItem('user');
    navigate('/login');
  };

  const userMenu = {
    items: [
      { key: 'profile', icon: <UserOutlined />, label: '个人信息' },
      { type: 'divider' as const },
      { key: 'logout', icon: <LogoutOutlined />, label: '退出登录', onClick: handleLogout },
    ],
  };

  return (
    <Layout style={{ minHeight: '100vh' }}>
      <Sider trigger={null} collapsible collapsed={collapsed} width={220}
        style={{ background: '#001529' }}>
        <div style={{
          height: 64, display: 'flex', alignItems: 'center', justifyContent: 'center',
          borderBottom: '1px solid rgba(255,255,255,0.1)',
        }}>
          <ThunderboltOutlined style={{ fontSize: 24, color: '#1890ff', marginRight: collapsed ? 0 : 8 }} />
          {!collapsed && <Text strong style={{ color: '#fff', fontSize: 16 }}>天普EMS</Text>}
        </div>
        <Menu
          theme="dark" mode="inline"
          selectedKeys={[location.pathname]}
          defaultOpenKeys={['/system']}
          items={menuItems}
          onClick={({ key }) => {
            if (key === '/bigscreen' || key === '/bigscreen-3d') {
              window.open(key, '_blank');
            } else {
              navigate(key);
            }
          }}
        />
      </Sider>
      <Layout>
        <Header style={{
          padding: '0 24px', background: '#fff', display: 'flex',
          alignItems: 'center', justifyContent: 'space-between',
          boxShadow: '0 1px 4px rgba(0,0,0,0.08)',
        }}>
          <div style={{ display: 'flex', alignItems: 'center', cursor: 'pointer' }}
            onClick={() => setCollapsed(!collapsed)}>
            {collapsed ? <MenuUnfoldOutlined style={{ fontSize: 18 }} /> :
              <MenuFoldOutlined style={{ fontSize: 18 }} />}
          </div>
          <div style={{ display: 'flex', alignItems: 'center', gap: 16 }}>
            <Popover
              trigger="click"
              placement="bottomRight"
              title={<div style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'center' }}>
                <span>告警通知</span>
                {alarmCount > 0 && <Tag color="red">{alarmCount} 条活跃</Tag>}
              </div>}
              content={
                <div style={{ width: 320, maxHeight: 360, overflow: 'auto' }}>
                  {recentAlarms.length === 0 ? (
                    <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} description="暂无活跃告警" />
                  ) : (
                    <List size="small" dataSource={recentAlarms} renderItem={(alarm: any) => {
                      const sev = SEVERITY_CONFIG[alarm.severity] || SEVERITY_CONFIG.info;
                      return (
                        <List.Item
                          style={{ cursor: 'pointer', padding: '8px 0' }}
                          onClick={() => navigate('/alarms')}
                        >
                          <List.Item.Meta
                            avatar={sev.icon}
                            title={<span style={{ fontSize: 13 }}>{alarm.device_name || alarm.title || '未知设备'}</span>}
                            description={<>
                              <div style={{ fontSize: 12 }}>{alarm.message || alarm.title}</div>
                              <div style={{ fontSize: 11, color: '#999' }}>{alarm.triggered_at}</div>
                            </>}
                          />
                        </List.Item>
                      );
                    }} />
                  )}
                  <div style={{ textAlign: 'center', padding: '8px 0', borderTop: '1px solid #f0f0f0' }}>
                    <a onClick={() => navigate('/alarms')} style={{ fontSize: 13 }}>查看全部告警</a>
                  </div>
                </div>
              }
            >
              <Badge count={alarmCount} size="small" overflowCount={99}>
                <BellOutlined style={{ fontSize: 18, cursor: 'pointer' }} />
              </Badge>
            </Popover>
            <Dropdown menu={userMenu} placement="bottomRight">
              <div style={{ cursor: 'pointer', display: 'flex', alignItems: 'center', gap: 8 }}>
                <Avatar size="small" icon={<UserOutlined />} style={{ background: '#1890ff' }} />
                <Text>{user?.full_name || user?.username || '用户'}</Text>
              </div>
            </Dropdown>
          </div>
        </Header>
        <Content style={{ margin: 16, padding: 24, background: '#f5f5f5', minHeight: 280 }}>
          <Outlet />
        </Content>
      </Layout>
    </Layout>
  );
}