refactor(logic): ♻️ update ReferenceScanner to support excludedSlot parameter

Author: esnya

src/ReferenceReplacement/Logic/ReferenceScanner.cs

@@ -10,11 +10,11 @@ namespace ReferenceReplacement.Logic;
 
 internal static class ReferenceScanner
 {
-    public static ReferenceScanResult Scan(Slot root, IWorldElement source, IWorldElement target)
+    public static ReferenceScanResult Scan(Slot root, IWorldElement source, IWorldElement target, Slot? excludedSlot = null)
     {
         ArgumentNullException.ThrowIfNull(root);
 
-        ReferenceScanSession session = new(source, target);
+        ReferenceScanSession session = new(source, target, excludedSlot);
         session.VisitSlot(root, TraversalPath.FromSlot(root));
         return session.BuildResult();
     }
@@ -23,7 +23,7 @@ internal static ReferenceScanResult Scan(HierarchyBlueprint blueprintRoot, IWorl
     {
         ArgumentNullException.ThrowIfNull(blueprintRoot);
 
-        ReferenceScanSession session = new(source, target);
+        ReferenceScanSession session = new(source, target, excludedSlot: null);
         session.VisitBlueprint(blueprintRoot, new(blueprintRoot.Label));
         return session.BuildResult();
     }
@@ -32,6 +32,7 @@ private sealed class ReferenceScanSession
     {
         private readonly IWorldElement _source;
         private readonly IWorldElement _target;
+        private readonly Slot? _excludedSlot;
         private readonly List<SyncReferenceMatch> _matches = new();
         private readonly HashSet<ISyncRef> _visitedRefs = new();
         private readonly HashSet<object> _visitedEnumerables = new(ReferenceEqualityComparer.Instance);
@@ -40,15 +41,16 @@ private sealed class ReferenceScanSession
         private int _incompatibleCount;
         private string? _lastPath;
 
-        internal ReferenceScanSession(IWorldElement source, IWorldElement target)
+        internal ReferenceScanSession(IWorldElement source, IWorldElement target, Slot? excludedSlot)
         {
             _source = source ?? throw new ArgumentNullException(nameof(source));
             _target = target ?? throw new ArgumentNullException(nameof(target));
+            _excludedSlot = excludedSlot;
         }
 
         internal void VisitSlot(Slot? slot, TraversalPath path)
         {
-            if (slot == null)
+            if (slot == null || ReferenceEquals(slot, _excludedSlot))
             {
                 return;
             }
@@ -387,4 +389,4 @@ internal sealed class ReferenceEqualityComparer : IEqualityComparer<object>
     public new bool Equals(object? x, object? y) => ReferenceEquals(x, y);
 
     public int GetHashCode(object obj) => RuntimeHelpers.GetHashCode(obj);
-}
+}

src/ReferenceReplacement/UI/ReferenceReplacementDialog.cs

@@ -46,8 +46,6 @@ public static ReferenceReplacementDialog Create(User owner, Slot dialogSlot)
         return new ReferenceReplacementDialog(owner, dialogSlot);
     }
 
-    public bool HasProcessRoot => GetProcessRootSlot() != null;
-
     public bool IsAlive => !_disposed && !_rootSlot.IsDestroyed && !_rootSlot.IsRemoved;
 
     public void Focus()
@@ -74,16 +72,6 @@ public void RepositionFor(User? user)
         _rootSlot.PointAtUserHead(float3.Backward, verticalAxisOnly: true);
     }
 
-    public void TrySetProcessRoot(Slot? slot)
-    {
-        if (slot == null || _processRootRef.Target != null)
-        {
-            return;
-        }
-
-        _processRootRef.Target = slot;
-    }
-
     public void Close()
     {
         if (_disposed)
@@ -227,7 +215,7 @@ private void Analyze(bool applyChanges)
 
         if (!applyChanges)
         {
-            ReferenceScanResult scanResult = ReferenceScanner.Scan(root, source, target);
+            ReferenceScanResult scanResult = ReferenceScanner.Scan(root, source, target, _rootSlot);
             if (scanResult.Matches.Count == 0)
             {
                 UpdateStatus("No references found in the selected root.");
@@ -302,7 +290,7 @@ private bool TryResolveInputs(out Slot root, out IWorldElement source, out IWorl
 
     private void ApplyReplacement(Slot root, IWorldElement source, IWorldElement target)
     {
-        ReferenceScanResult scanResult = ReferenceScanner.Scan(root, source, target);
+        ReferenceScanResult scanResult = ReferenceScanner.Scan(root, source, target, _rootSlot);
         if (scanResult.Matches.Count == 0)
         {
             UpdateStatus("No references found in the selected root.");
@@ -388,4 +376,4 @@ private void PrepareRootSlot(User owner)
         _rootSlot.LocalPosition = float3.Zero;
         _rootSlot.LocalRotation = floatQ.Identity;
     }
-}
+}