Merge pull request #104 from neuroglia-io/fix-dagre-child-added-bubbling

Bubbling "child added" event so parents can update their own nodes/clusters in Blazor.Dagre
neuroglia-io · Aug 14, 2024 · 2cb5789 · 2cb5789
2 parents 2003539 + 1f62b4f
commit 2cb5789
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/Neuroglia.Blazor.Dagre/Models/ClusterViewModel.cs b/src/Neuroglia.Blazor.Dagre/Models/ClusterViewModel.cs
@@ -66,6 +66,7 @@ public ClusterViewModel(
             child.Changed += OnChildChanged;
             if (child is IClusterViewModel cluster)
             {
+                cluster.ChildAdded += this.OnChildAdded;
                 this._allClusters.Add(cluster.Id, cluster);
                 this.Flatten(cluster);
             }
@@ -106,6 +107,7 @@ public virtual void AddChild(INodeViewModel node)
         this.ChildAdded?.Invoke(this, node);
         if (node is IClusterViewModel cluster)
         {
+            cluster.ChildAdded += this.OnChildAdded;
             this._allClusters.Add(cluster.Id, cluster);
             this.Flatten(cluster);
             return;
@@ -114,6 +116,22 @@ public virtual void AddChild(INodeViewModel node)
         this.OnChange();
     }
 
+    /// <inheritdoc/>
+    public virtual void OnChildAdded(object? sender, INodeViewModel child)
+    {
+        if (child is IClusterViewModel cluster)
+        {
+            this._allClusters.Add(cluster.Id, cluster);
+            this.Flatten(cluster);
+        }
+        else if (child is INodeViewModel node)
+        {
+            this._allNodes.Add(node.Id, node);
+        }
+        this.OnChange();
+        this.ChildAdded?.Invoke(this, child);
+    }
+
     /// <summary>
     /// Adds nested nodes/clusters to allNodes/Clusters
     /// </summary>

diff --git a/src/Neuroglia.Blazor.Dagre/Models/GraphViewModel.cs b/src/Neuroglia.Blazor.Dagre/Models/GraphViewModel.cs
@@ -429,7 +429,6 @@ public virtual async Task OnWheelAsync(ElementReference sender, WheelEventArgs e
         if (this.Wheel != null) await this.Wheel.Invoke(new(e, sender, element));
     }
 
-    /// <inheritdoc/>
     public virtual void OnChildAdded(object? sender, INodeViewModel child)
     {
         if (child is IClusterViewModel cluster)