[v0.0.2] Comment node; Cleaning

Depra-Inc · May 12, 2024 · 58604d5 · 58604d5
1 parent f0d0c9b
commit 58604d5
Show file tree

Hide file tree

Showing 5 changed files with 50 additions and 31 deletions.
diff --git a/Editor/View/CodeGraphView.cs b/Editor/View/CodeGraphView.cs
@@ -87,6 +87,16 @@ public void Create(CodeGraphNode node)
 			Bind();
 		}
 
+		internal void RemoveNode(CodeGraphEditorNode editorNode)
+		{
+			Undo.RecordObject(_serializedObject.targetObject, "Removed Node");
+
+			_nodes.Remove(editorNode);
+			_tree.Nodes.Remove(editorNode.Node);
+			_nodeLookup.Remove(editorNode.Node.Guid);
+			_serializedObject.Update();
+		}
+
 		private void PopulateNodes(IEnumerable<CodeGraphNode> graphNodes)
 		{
 			foreach (var node in graphNodes)
@@ -95,6 +105,26 @@ private void PopulateNodes(IEnumerable<CodeGraphNode> graphNodes)
 			}
 		}
 
+		internal void RemoveConnection(Edge edge)
+		{
+			if (_edgeLookup.Remove(edge, out var connection))
+			{
+				_tree.Connections.Remove(connection);
+			}
+		}
+
+		internal void CreateEdge(Edge edge)
+		{
+			var inputNode = (CodeGraphEditorNode) edge.input.node;
+			var outputNode = (CodeGraphEditorNode) edge.output.node;
+
+			var inputIndex = inputNode.Ports.IndexOf(edge.input);
+			var outputIndex = outputNode.Ports.IndexOf(edge.output);
+
+			var connection = new CodeGraphConnection(inputNode.Id, inputIndex, outputNode.Id, outputIndex);
+			_tree.Connections.Add(connection);
+		}
+
 		private void PopulateConnections(IEnumerable<CodeGraphConnection> connections)
 		{
 			foreach (var connection in connections)
@@ -133,24 +163,6 @@ private void AddNode(CodeGraphNode node)
 			AddElement(editorNode);
 		}
 
-		internal void RemoveNode(CodeGraphEditorNode editorNode)
-		{
-			Undo.RecordObject(_serializedObject.targetObject, "Removed Node");
-
-			_nodes.Remove(editorNode);
-			_tree.Nodes.Remove(editorNode.Node);
-			_nodeLookup.Remove(editorNode.Node.Guid);
-			_serializedObject.Update();
-		}
-
-		internal void RemoveConnection(Edge edge)
-		{
-			if (_edgeLookup.Remove(edge, out var connection))
-			{
-				_tree.Connections.Remove(connection);
-			}
-		}
-
 		private void AddGridBackground()
 		{
 			var grid = new GridBackground();
@@ -167,18 +179,6 @@ private void AddStyles()
 			styleSheets.Add(styleSheet);
 		}
 
-		internal void CreateEdge(Edge edge)
-		{
-			var inputNode = (CodeGraphEditorNode) edge.input.node;
-			var outputNode = (CodeGraphEditorNode) edge.output.node;
-
-			var inputIndex = inputNode.Ports.IndexOf(edge.input);
-			var outputIndex = outputNode.Ports.IndexOf(edge.output);
-
-			var connection = new CodeGraphConnection(inputNode.Id, inputIndex, outputNode.Id, outputIndex);
-			_tree.Connections.Add(connection);
-		}
-
 		private void Bind()
 		{
 			_serializedObject.Update();

diff --git a/Runtime/Nodes/CommentNode.cs b/Runtime/Nodes/CommentNode.cs
@@ -0,0 +1,14 @@
+using System;
+using Depra.CodeGraph.Attributes;
+using UnityEngine;
+
+namespace Depra.CodeGraph
+{
+	[Serializable]
+	[NodeTitle("Comment")]
+	[NodeMenuPath("Comment")]
+	internal sealed class CommentNode : CodeGraphNode
+	{
+		[SerializeField] private string _text;
+	}
+}
diff --git a/Runtime/Nodes/CommentNode.cs.meta b/Runtime/Nodes/CommentNode.cs.meta
diff --git a/Runtime/Nodes/StartNode.cs b/Runtime/Nodes/StartNode.cs
@@ -1,10 +1,12 @@
 // SPDX-License-Identifier: Apache-2.0
 // © 2024 Nikolay Melnikov <n.melnikov@depra.org>
 
+using System;
 using Depra.CodeGraph.Attributes;
 
 namespace Depra.CodeGraph
 {
+	[Serializable]
 	[HasFlowOutput]
 	[NodeTitle("Start")]
 	[NodeMenuPath("Process/Start")]

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "com.depra.code-graph",
-	"version": "0.0.1",
+	"version": "0.0.2",
 	"displayName": "Depra.CodeGraph",
 	"description": "",
 	"unity": "2022.3",