import { useEffect, useState } from 'react';
import { Layout, Menu, Avatar, Space, Typography, Button } from 'antd';
import {
  DashboardOutlined, UserOutlined, LogoutOutlined,
  CloudServerOutlined, TeamOutlined, ApiOutlined, KeyOutlined,
  MenuFoldOutlined, MenuUnfoldOutlined,
} 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';

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

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

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

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

  if (!user) return null;

  const menuItems: any[] = [];

  // 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, 'login:batch') || hasPerm(user, 'login:view_assigned')) {
    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, 'user:view')) {
    menuItems.push({ key: '/users', label: '用户管理', icon: <TeamOutlined /> });
  }

  const handleLogout = async () => {
    try {
      await authApi.logout();
    } catch {}
    clearAuth();
    onLogout?.();
    navigate('/login', { replace: true });
  };

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