mapeditor Bug fixes

MapEditor/Definition.json

@@ -4,6 +4,7 @@
   "name": "Map Editor",
   "version": "$(AssemblyVersion)",
   "assemblies": [ "MapEditor" ],
+  "updateUrl": "https://railroader.alinanova.dev/update.json",
   "requires": [
     "Zamu.StrangeCustoms"
   ],

MapEditor/Dialogs/TrackNodeDialog.cs

@@ -76,7 +76,7 @@ private static void BuildNodeEditor(UIPanelBuilder builder)
       builder.AddField("Flip Switch Stand", builder.AddToggle(() => NodeManager.GetFlipSwitchStand(), val => NodeManager.FlipSwitchStand(val))!);
       builder.HStack(stack =>
       {
-        stack.AddButtonCompact("Add", NodeManager.AddNode);
+        stack.AddButtonCompact("Add", () => NodeManager.AddNode());
         stack.AddButtonCompact("Split", () => NodeManager.SplitNode());
         stack.AddButtonCompact("Remove", () => NodeManager.RemoveNode(Input.GetKey(KeyCode.LeftShift)));
         stack.AddButtonCompact("Show", EditorContext.MoveCameraToSelectedNode);

MapEditor/EditorContext.cs

@@ -250,22 +250,23 @@ public static void MoveCameraToSelectedNode()
       }
     }
 
+    public static void UpdateUiHelpers()
+    {
+      CreateUiHelpers();
+    }
+
     #region UI helpers
 
     private static void CreateUiHelpers()
     {
       foreach (var trackNode in Graph.Shared.Nodes!)
       {
-        var gameObject = new GameObject("TrackNodeHelper");
-        gameObject.transform.SetParent(trackNode.transform);
-        gameObject.AddComponent<TrackNodeHelper>();
+        AttachUiHelper(trackNode);
       }
 
       foreach (var trackSegment in Graph.Shared.Segments!)
       {
-        var gameObject = new GameObject("TrackSegmentHelper");
-        gameObject.transform.SetParent(trackSegment.transform);
-        gameObject.AddComponent<TrackSegmentHelper>();
+        AttachUiHelper(trackSegment);
       }
     }
 
@@ -290,6 +291,28 @@ private static void DestroyUiHelpers()
       }
 
     }
+
+    internal static void AttachUiHelper(TrackNode node)
+    {
+      if (node.transform.Find("TrackNodeHelper") != null)
+      {
+        return;
+      }
+      var gameObject = new GameObject("TrackNodeHelper");
+      gameObject.transform.SetParent(node.transform);
+      gameObject.AddComponent<TrackNodeHelper>();
+    }
+
+    internal static void AttachUiHelper(TrackSegment segment)
+    {
+      if (segment.transform.Find("TrackSegmentHelper") != null)
+      {
+        return;
+      }
+      var gameObject = new GameObject("TrackSegmentHelper");
+      gameObject.transform.SetParent(segment.transform);
+      gameObject.AddComponent<TrackSegmentHelper>();
+    }
     #endregion
 
   }

MapEditor/Managers/NodeManager.cs

@@ -84,9 +84,9 @@ public static void DivideScaling()
 
     #endregion
 
-    public static void AddNode()
+    public static void AddNode(TrackNode? node = null)
     {
-      var node = EditorContext.SelectedNode!;
+      node ??= EditorContext.SelectedNode!;
       var nid = EditorContext.TrackNodeIdGenerator.Next()!;
       var sid = EditorContext.TrackSegmentIdGenerator.Next()!;
       EditorContext.ChangeManager.AddChange(new CompoundChange(

MapEditor/StateTracker/Node/TrackNodeGhost.cs

@@ -44,6 +44,7 @@ public void CreateNode()
       gameObject.SetActive(true);
       Graph.Shared.AddNode(node);
       EditorContext.PatchEditor!.AddOrUpdateNode(node);
+      EditorContext.AttachUiHelper(node);
     }
 
     public void DestroyNode()

MapEditor/StateTracker/Segment/TrackSegmentGhost.cs

@@ -53,19 +53,15 @@ public void UpdateSegment(TrackSegment segment)
     public void CreateSegment()
     {
       var gameObject = new GameObject($"Segment {id}");
-      gameObject.SetActive(true);
+      gameObject.SetActive(false);
       var segment = gameObject.AddComponent<TrackSegment>();
       segment.id = id;
       segment.transform.SetParent(Graph.Shared.transform);
       UpdateSegment(segment);
-      gameObject.SetActive(false);
+      gameObject.SetActive(true);
       Graph.Shared.AddSegment(segment);
-      // var helper = new GameObject("TrackSegmentHelper");
-      // helper.transform.SetParent(segment.transform);
-      // var sh = helper.AddComponent<TrackSegmentHelper>();
-      // sh.Rebuild();
-
       EditorContext.PatchEditor!.AddOrUpdateSegment(segment);
+      EditorContext.AttachUiHelper(segment);
     }
 
     public void DestroySegment()

MapEditor/mod.yaml

@@ -4,6 +4,7 @@ name: Map Editor
 desc: |-
   A Map Editor for manipulating track mods.
   WARNING: Extremely WIP. I will not be providing support. Use at your own risk.
+downloadUrl: https://github.com/AlinaNova21/Railroader-Mods/releases/download/v1.4.24204.650/MapEditor_1.1.24204.650.zip
 version: '1.1.24204.650'
 changelog:
 - version: '1.1.24204.650'