/**
 * 组件菜单工厂
 * 提供分类的组件菜单，参考测试计划工具的设计理念
 */

import {
    THREAD_GROUP,
    SETUP_THREAD_GROUP,
    TEARDOWN_THREAD_GROUP,
    HTTP_REQUEST,
    PALLETIZE_TASK,
    INBOUND_TASK,
    OUTBOUND_TASK,
    VIEW_RESULTS_TREE,
    SUMMARY_REPORT,
    AGGREGATE_REPORT,
    RESPONSE_ASSERTION,
    JSON_ASSERTION,
    HTTP_REQUEST_DEFAULT_CONFIG,
    HTTP_COOKIE_MANAGER,
    HTTP_HEADER_MANAGER,
    USER_DEFINED_VARIABLES,
    CSV_DATA_SET_CONFIG,
    REGULAR_EXPRESSION_EXTRACTOR,
    JSON_EXTRACTOR,
    CONSTANT_TIMER,
    LOOP_CONTROLLER,
    IF_CONTROLLER,
    SIMPLE_CONTROLLER,
    WHILE_CONTROLLER,
    SWITCH_CONTROLLER,
    FOR_EACH_CONTROLLER,
} from './JmeterComponents';

import { COMPONENT_CONFIGS } from './JmeterComponents';

/**
 * 菜单分类
 */
export const MENU_CATEGORIES = {
    THREADS: 'threads',           // 线程组
    SAMPLERS: 'samplers',         // 采样器
    LOGIC_CONTROLLERS: 'logic_controllers', // 逻辑控制器
    CONFIG_ELEMENTS: 'config_elements', // 配置元件
    PRE_PROCESSORS: 'pre_processors', // 前置处理器
    POST_PROCESSORS: 'post_processors', // 后置处理器
    ASSERTIONS: 'assertions',      // 断言
    LISTENERS: 'listeners',        // 监听器
    TIMERS: 'timers',              // 定时器
};

/**
 * 组件菜单映射
 * 按照测试计划工具的分类方式组织组件
 */
export const COMPONENT_MENU_MAP = {
    [MENU_CATEGORIES.THREADS]: [
        { type: THREAD_GROUP, label: '线程组', icon: '👥' },
        { type: SETUP_THREAD_GROUP, label: 'setUp线程组', icon: '⬆️' },
        { type: TEARDOWN_THREAD_GROUP, label: 'tearDown线程组', icon: '⬇️' },
    ],
    
    [MENU_CATEGORIES.SAMPLERS]: [
        { type: HTTP_REQUEST, label: 'HTTP请求', icon: '🌐' },
        { type: PALLETIZE_TASK, label: '组托任务', icon: '📦' },
        { type: INBOUND_TASK, label: '入库任务', icon: '📥' },
        { type: OUTBOUND_TASK, label: '出库任务', icon: '📤' },
    ],
    
    [MENU_CATEGORIES.LOGIC_CONTROLLERS]: [
        { type: SIMPLE_CONTROLLER, label: '简单控制器', icon: '📁' },
        { type: LOOP_CONTROLLER, label: '循环控制器', icon: '🔄' },
        { type: IF_CONTROLLER, label: '如果（If）控制器', icon: '❓' },
        { type: WHILE_CONTROLLER, label: 'While控制器', icon: '🔁' },
        { type: SWITCH_CONTROLLER, label: 'Switch控制器', icon: '🔀' },
        { type: FOR_EACH_CONTROLLER, label: 'ForEach控制器', icon: '🔂' },
    ],
    
    [MENU_CATEGORIES.CONFIG_ELEMENTS]: [
        { type: HTTP_REQUEST_DEFAULT_CONFIG, label: 'HTTP请求默认值', icon: '⚙️' },
        { type: HTTP_COOKIE_MANAGER, label: 'HTTP Cookie管理器', icon: '🍪' },
        { type: HTTP_HEADER_MANAGER, label: 'HTTP头管理器', icon: '📋' },
        { type: USER_DEFINED_VARIABLES, label: '用户定义的变量', icon: '📝' },
        { type: CSV_DATA_SET_CONFIG, label: 'CSV数据集配置', icon: '📊' },
    ],
    
    [MENU_CATEGORIES.PRE_PROCESSORS]: [
        // 前置处理器组件
    ],
    
    [MENU_CATEGORIES.POST_PROCESSORS]: [
        { type: REGULAR_EXPRESSION_EXTRACTOR, label: '正则表达式提取器', icon: '🔍' },
        { type: JSON_EXTRACTOR, label: 'JSON提取器', icon: '📄' },
    ],
    
    [MENU_CATEGORIES.ASSERTIONS]: [
        { type: RESPONSE_ASSERTION, label: '响应断言', icon: '✓' },
        { type: JSON_ASSERTION, label: 'JSON断言', icon: '📋' },
        { type: 'duration_assertion', label: '持续时间断言', icon: '⏱️' },
        { type: 'size_assertion', label: '大小断言', icon: '📏' },
    ],
    
    [MENU_CATEGORIES.LISTENERS]: [
        { type: VIEW_RESULTS_TREE, label: '查看结果树', icon: '🌳' },
        { type: SUMMARY_REPORT, label: '汇总报告', icon: '📊' },
        { type: AGGREGATE_REPORT, label: '聚合报告', icon: '📈' },
    ],
    
    [MENU_CATEGORIES.TIMERS]: [
        { type: CONSTANT_TIMER, label: '固定定时器', icon: '⏱️' },
    ],
};

/**
 * 获取分类的中文名称
 */
export const getCategoryLabel = (category) => {
    const labels = {
        [MENU_CATEGORIES.THREADS]: '线程组',
        [MENU_CATEGORIES.SAMPLERS]: '采样器',
        [MENU_CATEGORIES.LOGIC_CONTROLLERS]: '逻辑控制器',
        [MENU_CATEGORIES.CONFIG_ELEMENTS]: '配置元件',
        [MENU_CATEGORIES.PRE_PROCESSORS]: '前置处理器',
        [MENU_CATEGORIES.POST_PROCESSORS]: '后置处理器',
        [MENU_CATEGORIES.ASSERTIONS]: '断言',
        [MENU_CATEGORIES.LISTENERS]: '监听器',
        [MENU_CATEGORIES.TIMERS]: '定时器',
    };
    return labels[category] || category;
};

/**
 * 根据节点类型判断可以添加哪些类型的子组件
 */
export const getAllowedChildTypes = (nodeType) => {
    // 测试计划可以添加所有类型
    if (nodeType === 'test_plan') {
        return Object.values(MENU_CATEGORIES);
    }
    
    // 线程组可以添加采样器、逻辑控制器、配置元件、前置处理器、后置处理器、断言、监听器、定时器
    if (nodeType === THREAD_GROUP || nodeType === SETUP_THREAD_GROUP || nodeType === TEARDOWN_THREAD_GROUP) {
        return [
            MENU_CATEGORIES.SAMPLERS,
            MENU_CATEGORIES.LOGIC_CONTROLLERS,
            MENU_CATEGORIES.CONFIG_ELEMENTS,
            MENU_CATEGORIES.PRE_PROCESSORS,
            MENU_CATEGORIES.POST_PROCESSORS,
            MENU_CATEGORIES.ASSERTIONS,
            MENU_CATEGORIES.LISTENERS,
            MENU_CATEGORIES.TIMERS,
        ];
    }
    
    // 逻辑控制器可以添加采样器、逻辑控制器、配置元件、前置处理器、后置处理器、断言、监听器、定时器
    if ([SIMPLE_CONTROLLER, LOOP_CONTROLLER, IF_CONTROLLER].includes(nodeType)) {
        return [
            MENU_CATEGORIES.SAMPLERS,
            MENU_CATEGORIES.LOGIC_CONTROLLERS,
            MENU_CATEGORIES.CONFIG_ELEMENTS,
            MENU_CATEGORIES.PRE_PROCESSORS,
            MENU_CATEGORIES.POST_PROCESSORS,
            MENU_CATEGORIES.ASSERTIONS,
            MENU_CATEGORIES.LISTENERS,
            MENU_CATEGORIES.TIMERS,
        ];
    }
    
    // 采样器可以添加前置处理器、后置处理器、断言、监听器、定时器
    if ([HTTP_REQUEST, PALLETIZE_TASK, INBOUND_TASK, OUTBOUND_TASK].includes(nodeType)) {
        return [
            MENU_CATEGORIES.PRE_PROCESSORS,
            MENU_CATEGORIES.POST_PROCESSORS,
            MENU_CATEGORIES.ASSERTIONS,
            MENU_CATEGORIES.LISTENERS,
            MENU_CATEGORIES.TIMERS,
        ];
    }
    
    // 默认：可以添加所有类型
    return Object.values(MENU_CATEGORIES);
};

/**
 * 获取组件配置信息
 */
export const getComponentInfo = (componentType) => {
    return COMPONENT_CONFIGS[componentType] || null;
};

/**
 * 检查组件类型是否存在于菜单中
 */
export const isComponentInMenu = (componentType) => {
    for (const components of Object.values(COMPONENT_MENU_MAP)) {
        if (components.some(comp => comp.type === componentType)) {
            return true;
        }
    }
    return false;
};