extracted connection drag state logic

Author: alexej

connection_drag_state.py

@@ -0,0 +1,21 @@
+class ConnectionDragState:
+    def __init__(self):
+        self.active = False
+        self.start_node = None  # Typically a Node or node id
+        self.end_pos = None     # (x, y) tuple
+
+    def start(self, start_node, end_pos=None):
+        self.active = True
+        self.start_node = start_node
+        self.end_pos = end_pos
+
+    def update_end(self, end_pos):
+        self.end_pos = end_pos
+
+    def stop(self):
+        self.active = False
+        self.start_node = None
+        self.end_pos = None
+
+    def is_active(self):
+        return self.active

editor.py

@@ -17,6 +17,7 @@
 from fps_counter import FPSCounter
 from renderer import NodeEditorRenderer  # <-- new import
 from canvas_panning import CanvasPanning
+from connection_drag_state import ConnectionDragState
 
 class NodeEditor:
     def __init__(self, toolbar=None, undo_depth=10):
@@ -29,9 +30,7 @@ def __init__(self, toolbar=None, undo_depth=10):
         self.connections = ConnectionList()
         self.undo_stack = UndoStack(max_depth=undo_depth)
         self.selection = NodeSelection() # multiple selection of nodes
-        self.dragging_connection: bool = False
-        self.connection_start_node: tuple[int, int] | None = None
-        self.connection_end_pos: tuple[int,int] | None = None
+        self.connection_drag = ConnectionDragState()
         self.next_node_id = 1
         self.zoom: float = 1.0  # 1.0 = 100%, min 0.1 (1:10), max e.g. 2.0
         self.toolbar = toolbar if toolbar else Toolbar()
@@ -182,6 +181,9 @@ def handle_mouse_motion(self, event):
             self.panning_state.update_panning(
                 (x, y), self.zoom, PANNING_FOLLOWS_MOUSE
             )
+        # Handle connection dragging
+        if self.connection_drag.is_active():
+            self.connection_drag.update_end((x, y))
 
     def handle_mouse_wheel(self, event):
         # Get mouse position