P2P Export - Nodes - Patch

leenkx/blender/lnx/logicnode/custom/LN_leenkx_close_connection.py

@@ -5,63 +5,40 @@ class LeenkxCloseConnectionNode(LnxLogicTreeNode):
     """Close connection on the network"""
     bl_idname = 'LNLeenkxCloseConnectionNode'
     bl_label = 'Close Leenkx Connection'
+    bl_description = ('Tear down a Leenkx connection, or drop a single peer from it')
     lnx_version = 1
     lnx_category = 'Leenkx'
-    ind = 2
 
-    @staticmethod
-    def get_enum_id_value(obj, prop_name, value):
-        return obj.bl_rna.properties[prop_name].enum_items[value].identifier
+    def update_sockets(self, context):
+        if not self.inputs:
+            return
+        links = {}
+        for socket in self.inputs:
+            for link in socket.links:
+                links.setdefault(socket.name, []).append(
+                    (link.from_node, link.from_socket))
+        for socket in self.inputs[2:]:
+            self.inputs.remove(socket)
+        if self.property0 == 'peer':
+            self.add_input('LnxStringSocket', 'ID')
+        tree = self.id_data
+        for socket in self.inputs:
+            if socket.name in links:
+                for from_node, from_socket in links[socket.name]:
+                    tree.links.new(from_socket, socket)
 
-    @staticmethod
-    def get_count_in(type_name):
-        return {
-            'client': 0,
-            'peer': 1
-        }.get(type_name, 0)
-
-    def get_enum(self):
-        return self.get('property0', 0)
-
-    def set_enum(self, value):
-        # Checking the selection of each type
-        select_current = self.get_enum_id_value(self, 'property0', value)
-        select_prev = self.property0
-
-        #Check if type changed
-        if select_prev != select_current:
-
-            for i in self.inputs:
-                self.inputs.remove(i)
-
-            # Arguements for type Host
-            if (self.get_count_in(select_current) == 0):
-                    self.add_input('LnxNodeSocketAction', 'In')
-                    self.add_input('LnxDynamicSocket', 'Connection')
-                    self.add_input('LnxStringSocket', 'Url')
-
-            # Arguements for type Client
-            if (self.get_count_in(select_current) == 1):
-                    self.add_input('LnxNodeSocketAction', 'In')
-                    self.add_input('LnxDynamicSocket', 'Connection')
-                    self.add_input('LnxStringSocket', 'ID')
-
-        self['property0'] = value
-
-
-    property0: HaxeEnumProperty(
-        'property1',
+    property0: HaxeEnumProperty('property0',
         items = [('client', 'Leenkx Client', 'Close the Leenkx Client connection on the network'),
-                ('peer', 'Single Peer', 'Close a Peer connection on network')],
+                ('peer', 'Single Peer', 'Drop a peer from the connection (forgets its address; it can rejoin on a new handshake)')],
         name='', default='client',
-        set=set_enum, 
-        get=get_enum)
+        update=update_sockets)
 
 
     def lnx_init(self, context):
         self.add_input('LnxNodeSocketAction', 'In')
         self.add_input('LnxDynamicSocket', 'Connection')
-        self.add_input('LnxStringSocket', 'Url')
+        if self.property0 == 'peer':
+            self.add_input('LnxStringSocket', 'ID')
 
         self.add_output('LnxNodeSocketAction', 'Out')