from arm.logicnode.arm_nodes import *


def remove_extra_inputs(self, context):
    if not any(p == self.property0 for p in ['Or', 'And']):
        while len(self.inputs) > self.min_inputs:
            self.inputs.remove(self.inputs[-1])

class GateNode(ArmLogicTreeNode):
    """Logic nodes way to do "if" statements. When activated, it
    compares if its two inputs are being Equal, Greater Equal,
    Less Equal, or Not Equal, regardless of variable type, and passes
    through its active input to the output that matches the result of
    the comparison.

    "And" and "Or" are being used for booleans only, and pass through
    the input when both booleans are true (And) or at least one (Or)."""
    bl_idname = 'LNGateNode'
    bl_label = 'Gate'
    arm_version = 1

    min_inputs = 3
    property0: EnumProperty(
        items = [('Equal', 'Equal', 'Equal'),
                 ('Almost Equal', 'Almost Equal', 'Almost Equal'),
                 ('Greater', 'Greater', 'Greater'),
                 ('Greater Equal', 'Greater Equal', 'Greater Equal'),
                 ('Less', 'Less', 'Less'),
                 ('Less Equal', 'Less Equal', 'Less Equal'),
                 ('Or', 'Or', 'Or'),
                 ('And', 'And', 'And')],
        name='', default='Equal',
        update=remove_extra_inputs)
    property1: FloatProperty(name='Tolerance', description='Precision for float compare', default=0.0001)

    def __init__(self):
        super(GateNode, self).__init__()
        array_nodes[str(id(self))] = self

    def init(self, context):
        super(GateNode, self).init(context)
        self.add_input('ArmNodeSocketAction', 'In')
        self.add_input('NodeSocketShader', 'Input 1')
        self.add_input('NodeSocketShader', 'Input 2')
        self.add_output('ArmNodeSocketAction', 'True')
        self.add_output('ArmNodeSocketAction', 'False')

    def draw_buttons(self, context, layout):
        layout.prop(self, 'property0')

        if self.property0 == 'Almost Equal':
            layout.prop(self, 'property1')

        if any(p == self.property0 for p in ['Or', 'And']):
            row = layout.row(align=True)
            op = row.operator('arm.node_add_input', text='New', icon='PLUS', emboss=True)
            op.node_index = str(id(self))
            op.socket_type = 'NodeSocketShader'
            op2 = row.operator('arm.node_remove_input', text='', icon='X', emboss=True)
            op2.node_index = str(id(self))
