from enum import Enum
from typing import List


class Feature(Enum):
    """功能权限枚举，带序号和描述"""
    ROBOT = "🔧 群机器人 [总开关]"

    def __new__(cls, *args):
        if len(args) == 2:
            value, description = args
        elif len(args) == 1 and isinstance(args[0], tuple):
            value, description = args[0]
        else:
            raise TypeError(f"Invalid arguments for Feature.__new__: {args}")

        obj = object.__new__(cls)
        obj._value_ = value
        obj.description = description
        return obj

    def __str__(self):
        return self.description

    @classmethod
    def get_max_value(cls) -> int:
        return max(member.value for member in cls)

    _dynamic_features = {}

    @classmethod
    def register_feature(cls, key: str, description: str) -> 'Feature':
        if key in cls._dynamic_features:
            return cls._dynamic_features[key]

        new_value = cls.get_max_value() + 1
        new_feature = object.__new__(cls)
        new_feature._value_ = new_value
        new_feature.description = description
        cls._dynamic_features[key] = new_feature
        return new_feature

    @classmethod
    def get_feature(cls, key: str) -> 'Feature':
        return cls._dynamic_features.get(key)

    @classmethod
    def get_all_features(cls) -> List['Feature']:
        return list(cls) + list(cls._dynamic_features.values())

    @classmethod
    def _missing_(cls, value):
        if isinstance(value, int):
            for member in cls:
                if member.value == value:
                    return member
            for feature in cls._dynamic_features.values():
                if feature.value == value:
                    return feature
        return None



if __name__ == '__main__':
    # 测试新增功能
    print("当前功能列表：")
    for feature in Feature.get_all_features():
        print(f"{feature.value} - {feature.description}")

    # 测试注册新功能
    print("\n注册新功能...")
    new_feature = Feature.register_feature("TEST_FEATURE", "🧪 测试功能 [测试]")
    print(f"新功能: {new_feature.value} - {new_feature.description}")

    # 验证新功能是否添加成功
    print("\n更新后的功能列表：")
    for feature in Feature.get_all_features():
        print(f"{feature.value} - {feature.description}")

    # 测试重复注册
    print("\n测试重复注册...")
    same_feature = Feature.register_feature("TEST_FEATURE", "🧪 测试功能 [测试]")
    print(f"重复注册结果: {same_feature.value} - {same_feature.description}")

    # 测试获取功能
    print("\n测试获取功能...")
    retrieved_feature = Feature.get_feature("TEST_FEATURE")
    print(f"获取到的功能: {retrieved_feature.value} - {retrieved_feature.description}")