from .src.utils.uitls import AlwaysEqualProxy,AlwaysTupleZero
import json
class MultiParamFormula:

    DESCRIPTION = """
    输入参数：p0,p1,p2... 支持任意类型，可根据需要输入和选择输入参数
    输出参数：p0,p1,p2... 支持任意类型，可根据需要输出
    表达式：python的正确代码语法，多个参数输出需如下形式：tuple([p0,p1,p2,p3]) 对应出参：p0,p1,p2,p3
    示例：
    入参：p0,p1 表达式：p0>p1  出参：p0 比较大小返回bool类型用于判断选节点
    入参：p0,p1 (p0类型IMAGE,p1类型INT) 表达式：p0[p1:p1+1]  出参：p0 返回指定下标的图片
    入参：p0  (p0数组)  表达式：len(p0)  出参：p0 数组长度
    入参：p0,p1 (p0类型LIST,p1类型INT) 表达式：p0[p1]  出参：p0 返回指定下标的元素
    入参：p0 (p0类型字符串数组) 表达式：','.join(p0)  出参：p0 返回字符串数组‘,’分割合并的字符串
    入参：p0 表达式：print('调试输出',p0)  出参：无 用于调试输出，查看数据结构
    高级模式启用：可执行多行代码，返回值会必须赋值给“result” 否则会报错
    """
    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "expression": ("STRING",{"multiline": True}),
                "advanced": (["enable", "disable"],{"default": "disable"}), 
                "errorEnd": (["enable", "disable"],{"default": "disable"}), 
            },
            "optional": {
                "p0": (AlwaysEqualProxy("*"), ),
                "p1": (AlwaysEqualProxy("*"), ),
            },
            "hidden": {
                "unique_id": "UNIQUE_ID",           #节点编号
                "dynprompt": "DYNPROMPT",
                "prompt": "PROMPT",                 #流程节点信息
                "extra_pnginfo": "EXTRA_PNGINFO",    #前端流程图信息
                "isView":("BOOLEAN",{"default": True}),
            }
        }

    RETURN_TYPES = AlwaysTupleZero(AlwaysEqualProxy("*"),)
    RETURN_NAMES = ('r0',)
    FUNCTION = "evaluate"
    CATEGORY = "lam"
    OUTPUT_NODE = False

    def evaluate(self,advanced,expression,errorEnd='disable',isView=True, **kwargs):
        lookup = {'json':json}
        for arg in kwargs:
            lookup[arg] = kwargs[arg]
        
        msg='完成'
        r=""
        expand=None
        ui={}
        try:
            if advanced=='enable':
                exec(expression, lookup)
                if 'result' not in lookup:
                    msg='表达式没有返回值'
                else:
                    r = lookup['result']
                if "expand" in lookup:
                    expand = lookup['expand']
                if "ui" in lookup:
                    ui=lookup['ui']
            else:
                r = eval(expression, lookup)
        except Exception as e:
            if errorEnd=='enable':
                raise Exception(e)
            print(e)
            msg='表达式错误'

        if not isinstance(r, tuple):
            r = (r,)

        ui['value']=[msg]
        return {"ui": ui, "result": r,"expand":expand} if isView else r
    
NODE_CLASS_MAPPINGS = {
    "MultiParamFormula": MultiParamFormula
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "MultiParamFormula": "多参代码表达式"
}