From ac4fb1efd96c7922a6a902a05e5ceb2860c5a7cb Mon Sep 17 00:00:00 2001
From: Thor Kramer Brigsted <thor1996@VID.net.local>
Date: Thu, 26 Oct 2017 12:40:34 +0200
Subject: [PATCH] Added editor onUpdateNode callback on Connect/Disconnect

---
 Scripts/Editor/NodeEditorAction.cs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Scripts/Editor/NodeEditorAction.cs b/Scripts/Editor/NodeEditorAction.cs
index 92a93c3..d6beb8c 100644
--- a/Scripts/Editor/NodeEditorAction.cs
+++ b/Scripts/Editor/NodeEditorAction.cs
@@ -68,10 +68,12 @@ public partial class NodeEditorWindow {
                     } else {
                         hoveredPort.VerifyConnections();
                         if (hoveredPort.IsConnected) {
+                            Node node = hoveredPort.node;
                             NodePort output = hoveredPort.Connection;
                             hoveredPort.Disconnect(output);
                             draggedOutput = output;
                             draggedOutputTarget = hoveredPort;
+                            NodeEditor.onUpdateNode(node);
                         }
                     }
                 } else if (IsHoveringNode && IsHoveringTitle(hoveredNode)) {
@@ -85,7 +87,9 @@ public partial class NodeEditorWindow {
                     if (IsDraggingPort) {
                         //If connection is valid, save it
                         if (draggedOutputTarget != null) {
+                            Node node = draggedOutputTarget.node;
                             if (graph.nodes.Count != 0) draggedOutput.Connect(draggedOutputTarget);
+                            NodeEditor.onUpdateNode(node);
                             EditorUtility.SetDirty(graph);
                         }
                         //Release dragged connection