"""
组件配置文件
包含组件发现、注册和分类的相关配置
"""
from typing import Dict, List, Any
from data.models.component import ComponentCategoryEnum

class ComponentDiscovery:
    """组件自动发现配置"""
    DEFAULT_PACKAGE = "components"  # 默认组件包名
    AUTO_REGISTER = True  # 是否在启动时自动注册所有组件

class ComponentCategory:
    """组件类别配置"""
    # 组件类别定义
    HTTP_REQUEST = "HTTP请求"
    SCRIPT = "脚本"
    FLOW = "流程"
    ROBOT = "机器人调度"
    SITE = "库位"
    DEVICE = "设备"
    SUBTASK = "子任务"
    TASK = "任务"
    BASE = "基础"
    
    # 组件类别映射配置
    # 用于将组件类型映射到对应的类别，便于前端展示
    @classmethod
    def get_mapping(cls) -> Dict[str, str]:
        """获取组件类型到类别的映射"""
        return {
            # HTTP请求组件
            "http": cls.HTTP_REQUEST,
            
            # 脚本组件
            "script": cls.SCRIPT,
            
            # 流程控制组件
            "if": cls.FLOW,
            "if_else": cls.FLOW,
            "if_else_if": cls.FLOW,
            "for_each": cls.FLOW,
            "while": cls.FLOW,
            "break": cls.FLOW,
            "return": cls.FLOW,
            "delay": cls.FLOW,
            
            # 机器人调度组件
            "robot": cls.ROBOT,
            
            # 库位组件
            "site": cls.SITE,
            
            # 设备组件
            "modbus": cls.DEVICE,
            
            # 子任务组件
            "subtask": cls.SUBTASK,
        }
    
    # 组件分类优先级
    # 控制前端组件面板中各分类的显示顺序
    @classmethod
    def get_priority(cls) -> List[str]:
        """获取组件类别优先级列表"""
        return [
            cls.SUBTASK,
            cls.SCRIPT,
            cls.HTTP_REQUEST,
            cls.TASK,
            cls.FLOW,
            cls.BASE,
            cls.SITE,
            cls.ROBOT,
            cls.DEVICE
        ]

class ComponentCategoryConfig:
    """组件分类配置"""
    
    @classmethod
    def get_category_name(cls, category_enum):
        """获取分类名称"""
        names = {
            ComponentCategoryEnum.SUBTASK: "子任务",
            ComponentCategoryEnum.SCRIPT: "脚本",
            ComponentCategoryEnum.HTTP: "HTTP请求",
            ComponentCategoryEnum.TASK: "任务",
            ComponentCategoryEnum.FLOW: "流程",
            ComponentCategoryEnum.BASIC: "基础",
            ComponentCategoryEnum.STORAGE: "库位",
            ComponentCategoryEnum.ROBOT: "机器人调度",
            ComponentCategoryEnum.DEVICE: "设备"
        }
        return names.get(category_enum, category_enum.value)
    
    @classmethod
    def get_category_description(cls, category_enum):
        """获取分类描述"""
        descriptions = {
            ComponentCategoryEnum.SUBTASK: "可重用的子任务组件",
            ComponentCategoryEnum.SCRIPT: "执行脚本代码的组件",
            ComponentCategoryEnum.HTTP: "发送HTTP请求的组件",
            ComponentCategoryEnum.TASK: "任务管理相关组件",
            ComponentCategoryEnum.FLOW: "流程控制相关组件",
            ComponentCategoryEnum.BASIC: "基础功能组件",
            ComponentCategoryEnum.STORAGE: "库位管理相关组件",
            ComponentCategoryEnum.ROBOT: "机器人调度相关组件",
            ComponentCategoryEnum.DEVICE: "设备控制相关组件"
        }
        return descriptions.get(category_enum, f"{category_enum.value}类组件")
    
    @classmethod
    def get_category_order(cls, category_enum):
        """获取分类排序"""
        orders = {
            ComponentCategoryEnum.BASIC: 1,
            ComponentCategoryEnum.FLOW: 2,
            ComponentCategoryEnum.TASK: 3,
            ComponentCategoryEnum.SUBTASK: 4,
            ComponentCategoryEnum.SCRIPT: 5,
            ComponentCategoryEnum.HTTP: 6,
            ComponentCategoryEnum.ROBOT: 7,
            ComponentCategoryEnum.STORAGE: 8,
            ComponentCategoryEnum.DEVICE: 9
        }
        return orders.get(category_enum, 99)