Update NodeUI.cs

Author: internetfishy

game/addons/tools/Code/NodeGraph/NodeUI.cs

@@ -298,7 +298,7 @@ private void UpdatePlugs( bool firstTime )
 
 			if ( !match.IsValid() )
 			{
-				Inputs.Insert( index, new PlugIn( this, plug ) );
+				Inputs.Insert( index, CreatePlugIn( this, plug ) );
 			}
 			else if ( Inputs.IndexOf( match ) != index )
 			{
@@ -318,7 +318,7 @@ private void UpdatePlugs( bool firstTime )
 
 			if ( !match.IsValid() )
 			{
-				Outputs.Insert( index, new PlugOut( this, plug ) );
+				Outputs.Insert( index, CreatePlugOut( this, plug ) );
 			}
 			else if ( Outputs.IndexOf( match ) != index )
 			{
@@ -335,6 +335,16 @@ private void UpdatePlugs( bool firstTime )
 		Graph?.NodePositionChanged( this );
 	}
 
+	protected virtual PlugIn CreatePlugIn( NodeUI ui, IPlugIn plugIn )
+	{
+		return new PlugIn( ui, plugIn );
+	}
+	
+	protected virtual PlugOut CreatePlugOut( NodeUI ui, IPlugOut plugOut )
+	{
+		return new PlugOut( ui, plugOut );
+	}
+
 	protected virtual void Layout()
 	{
 		var hasThumb = !Node.HasTitleBar && Node.DisplayInfo.Icon is not null || Node.Thumbnail is not null;