import { useState, useRef, useEffect } from 'react';
import { UIRegistry, MessageHub, EditorPluginManager } from '@esengine/editor-core';
import type { MenuItem as PluginMenuItem } from '@esengine/editor-core';
import * as LucideIcons from 'lucide-react';
import '../styles/MenuBar.css';

interface MenuItem {
  label?: string;
  shortcut?: string;
  icon?: string;
  disabled?: boolean;
  separator?: boolean;
  submenu?: MenuItem[];
  onClick?: () => void;
}

interface MenuBarProps {
  locale?: string;
  uiRegistry?: UIRegistry;
  messageHub?: MessageHub;
  pluginManager?: EditorPluginManager;
  onNewScene?: () => void;
  onOpenScene?: () => void;
  onSaveScene?: () => void;
  onSaveSceneAs?: () => void;
  onOpenProject?: () => void;
  onCloseProject?: () => void;
  onExit?: () => void;
  onOpenPluginManager?: () => void;
  onOpenProfiler?: () => void;
  onOpenPortManager?: () => void;
  onOpenSettings?: () => void;
  onToggleDevtools?: () => void;
  onOpenAbout?: () => void;
  onCreatePlugin?: () => void;
  onReloadPlugins?: () => void;
}

export function MenuBar({
    locale = 'en',
    uiRegistry,
    messageHub,
    pluginManager,
    onNewScene,
    onOpenScene,
    onSaveScene,
    onSaveSceneAs,
    onOpenProject,
    onCloseProject,
    onExit,
    onOpenPluginManager,
    onOpenProfiler: _onOpenProfiler,
    onOpenPortManager,
    onOpenSettings,
    onToggleDevtools,
    onOpenAbout,
    onCreatePlugin,
    onReloadPlugins
}: MenuBarProps) {
    const [openMenu, setOpenMenu] = useState<string | null>(null);
    const [pluginMenuItems, setPluginMenuItems] = useState<PluginMenuItem[]>([]);
    const menuRef = useRef<HTMLDivElement>(null);

    const updateMenuItems = () => {
        if (uiRegistry && pluginManager) {
            const items = uiRegistry.getChildMenus('window');
            // 过滤掉被禁用插件的菜单项
            const enabledPlugins = pluginManager.getAllPluginMetadata()
                .filter((p) => p.enabled)
                .map((p) => p.name);

            // 只显示启用插件的菜单项
            const filteredItems = items.filter((item) => {
                // 检查菜单项是否属于某个插件
                return enabledPlugins.some((pluginName) => {
                    const plugin = pluginManager.getEditorPlugin(pluginName);
                    if (plugin && plugin.registerMenuItems) {
                        const pluginMenus = plugin.registerMenuItems();
                        return pluginMenus.some((m) => m.id === item.id);
                    }
                    return false;
                });
            });

            setPluginMenuItems(filteredItems);
        } else if (uiRegistry) {
            // 如果没有 pluginManager，显示所有菜单项
            const items = uiRegistry.getChildMenus('window');
            setPluginMenuItems(items);
        }
    };

    useEffect(() => {
        updateMenuItems();
    }, [uiRegistry, pluginManager]);

    useEffect(() => {
        if (messageHub) {
            const unsubscribeInstalled = messageHub.subscribe('plugin:installed', () => {
                updateMenuItems();
            });

            const unsubscribeEnabled = messageHub.subscribe('plugin:enabled', () => {
                updateMenuItems();
            });

            const unsubscribeDisabled = messageHub.subscribe('plugin:disabled', () => {
                updateMenuItems();
            });

            return () => {
                unsubscribeInstalled();
                unsubscribeEnabled();
                unsubscribeDisabled();
            };
        }
    }, [messageHub, uiRegistry, pluginManager]);

    const t = (key: string) => {
        const translations: Record<string, Record<string, string>> = {
            en: {
                file: 'File',
                newScene: 'New Scene',
                openScene: 'Open Scene',
                saveScene: 'Save Scene',
                saveSceneAs: 'Save Scene As...',
                openProject: 'Open Project',
                closeProject: 'Close Project',
                exit: 'Exit',
                edit: 'Edit',
                undo: 'Undo',
                redo: 'Redo',
                cut: 'Cut',
                copy: 'Copy',
                paste: 'Paste',
                delete: 'Delete',
                selectAll: 'Select All',
                window: 'Window',
                sceneHierarchy: 'Scene Hierarchy',
                inspector: 'Inspector',
                assets: 'Assets',
                console: 'Console',
                viewport: 'Viewport',
                pluginManager: 'Plugin Manager',
                tools: 'Tools',
                createPlugin: 'Create Plugin',
                reloadPlugins: 'Reload Plugins',
                portManager: 'Port Manager',
                settings: 'Settings',
                help: 'Help',
                documentation: 'Documentation',
                checkForUpdates: 'Check for Updates',
                about: 'About',
                devtools: 'Developer Tools'
            },
            zh: {
                file: '文件',
                newScene: '新建场景',
                openScene: '打开场景',
                saveScene: '保存场景',
                saveSceneAs: '场景另存为...',
                openProject: '打开项目',
                closeProject: '关闭项目',
                exit: '退出',
                edit: '编辑',
                undo: '撤销',
                redo: '重做',
                cut: '剪切',
                copy: '复制',
                paste: '粘贴',
                delete: '删除',
                selectAll: '全选',
                window: '窗口',
                sceneHierarchy: '场景层级',
                inspector: '检视器',
                assets: '资产',
                console: '控制台',
                viewport: '视口',
                pluginManager: '插件管理器',
                tools: '工具',
                createPlugin: '创建插件',
                reloadPlugins: '重新加载插件',
                portManager: '端口管理器',
                settings: '设置',
                help: '帮助',
                documentation: '文档',
                checkForUpdates: '检查更新',
                about: '关于',
                devtools: '开发者工具'
            }
        };
        return translations[locale]?.[key] || key;
    };

    const menus: Record<string, MenuItem[]> = {
        file: [
            { label: t('newScene'), shortcut: 'Ctrl+N', onClick: onNewScene },
            { label: t('openScene'), shortcut: 'Ctrl+O', onClick: onOpenScene },
            { separator: true },
            { label: t('saveScene'), shortcut: 'Ctrl+S', onClick: onSaveScene },
            { label: t('saveSceneAs'), shortcut: 'Ctrl+Shift+S', onClick: onSaveSceneAs },
            { separator: true },
            { label: t('openProject'), onClick: onOpenProject },
            { label: t('closeProject'), onClick: onCloseProject },
            { separator: true },
            { label: t('exit'), onClick: onExit }
        ],
        edit: [
            { label: t('undo'), shortcut: 'Ctrl+Z', disabled: true },
            { label: t('redo'), shortcut: 'Ctrl+Y', disabled: true },
            { separator: true },
            { label: t('cut'), shortcut: 'Ctrl+X', disabled: true },
            { label: t('copy'), shortcut: 'Ctrl+C', disabled: true },
            { label: t('paste'), shortcut: 'Ctrl+V', disabled: true },
            { label: t('delete'), shortcut: 'Delete', disabled: true },
            { separator: true },
            { label: t('selectAll'), shortcut: 'Ctrl+A', disabled: true }
        ],
        window: [
            ...pluginMenuItems.map((item) => ({
                label: item.label || '',
                icon: item.icon,
                disabled: item.disabled,
                onClick: item.onClick
            })),
            ...(pluginMenuItems.length > 0 ? [{ separator: true } as MenuItem] : []),
            { label: t('pluginManager'), onClick: onOpenPluginManager },
            { separator: true },
            { label: t('devtools'), onClick: onToggleDevtools }
        ],
        tools: [
            { label: t('createPlugin'), onClick: onCreatePlugin },
            { label: t('reloadPlugins'), shortcut: 'Ctrl+R', onClick: onReloadPlugins },
            { separator: true },
            { label: t('portManager'), onClick: onOpenPortManager },
            { separator: true },
            { label: t('settings'), onClick: onOpenSettings }
        ],
        help: [
            { label: t('documentation'), disabled: true },
            { separator: true },
            { label: t('about'), onClick: onOpenAbout }
        ]
    };

    useEffect(() => {
        const handleClickOutside = (event: MouseEvent) => {
            if (menuRef.current && !menuRef.current.contains(event.target as Node)) {
                setOpenMenu(null);
            }
        };

        document.addEventListener('mousedown', handleClickOutside);
        return () => {
            document.removeEventListener('mousedown', handleClickOutside);
        };
    }, []);

    const handleMenuClick = (menuKey: string) => {
        setOpenMenu(openMenu === menuKey ? null : menuKey);
    };

    const handleMenuItemClick = (item: MenuItem) => {
        if (!item.disabled && !item.separator && item.onClick && item.label) {
            item.onClick();
            setOpenMenu(null);
        }
    };

    return (
        <div className="menu-bar" ref={menuRef}>
            {Object.keys(menus).map((menuKey) => (
                <div key={menuKey} className="menu-item">
                    <button
                        className={`menu-button ${openMenu === menuKey ? 'active' : ''}`}
                        onClick={() => handleMenuClick(menuKey)}
                    >
                        {t(menuKey)}
                    </button>
                    {openMenu === menuKey && menus[menuKey] && (
                        <div className="menu-dropdown">
                            {menus[menuKey].map((item, index) => {
                                if (item.separator) {
                                    return <div key={index} className="menu-separator" />;
                                }
                                const IconComponent = item.icon ? (LucideIcons as any)[item.icon] : null;
                                return (
                                    <button
                                        key={index}
                                        className={`menu-dropdown-item ${item.disabled ? 'disabled' : ''}`}
                                        onClick={() => handleMenuItemClick(item)}
                                        disabled={item.disabled}
                                    >
                                        <span className="menu-item-content">
                                            {IconComponent && <IconComponent size={16} />}
                                            <span>{item.label || ''}</span>
                                        </span>
                                        {item.shortcut && <span className="menu-shortcut">{item.shortcut}</span>}
                                    </button>
                                );
                            })}
                        </div>
                    )}
                </div>
            ))}
        </div>
    );
}