from plugins_func.register import register_function,ToolType, ActionResponse, Action
from config.logger import setup_logging

TAG = __name__
logger = setup_logging()

plugin_loader_function_desc = {
                "type": "function",
                "function": {
                    "name": "plugin_loader",
                    "description": "当用户想加载或卸载插件/function时，调用此函数：支持的插件列表为[plugins]",
                    "parameters": {
                        "type": "object",
                        "properties": {
                            "oper": {
                                "type": "string",
                                "description": "load or unload"
                            },
                            "name":{
                                "type": "string",
                                "description": "要加载或卸载的插件名字"
                            }
                        },
                        "required": ["oper","name"]
                    }
                }
            }

@register_function('plugin_loader', plugin_loader_function_desc, ToolType.SYSTEM_CTL)
def plugin_loader(conn, oper: str, name: str):
    """插件加载"""
    if oper not in ["load", "unload"]:
        return ActionResponse(action=Action.RESPONSE, result="插件操作失败", response="不支持的操作")
    
    cur_support = conn.func_handler.current_support_functions()
    if oper == "load":
        if name in cur_support:
            return ActionResponse(action=Action.RESPONSE, result="插件加载失败", response=f"{name}插件已加载,无需重复加载")
        func = conn.func_handler.function_registry.register_function(name)
        if not func:
            return ActionResponse(action=Action.RESPONSE, result="插件加载失败", response="插件未找到")
        res = f"{name}插件加载成功"
    else:
        if name not in cur_support:
            return ActionResponse(action=Action.RESPONSE, result="插件卸载失败", response=f"{name}插件未加载")
        bOK = conn.func_handler.function_registry.unregister_function(name)
        if not bOK:
            return ActionResponse(action=Action.RESPONSE, result="插件卸载失败", response="插件未找到")
        res = f"{name}插件卸载成功"
    conn.func_handler.upload_functions_desc()
    return ActionResponse(action=Action.RESPONSE, result="插件操作成功", response=res)