From df77f18bae5bae5b657ff33e66f11bd91b88adcf Mon Sep 17 00:00:00 2001
From: Tako <53405089+Takoooooo@users.noreply.github.com>
Date: Sat, 23 Oct 2021 01:29:20 +0300
Subject: [PATCH] Merge pull request #6702 from
 YohDeadfall/fix-dev-tools-events

Fixed handled event misses in dev tools
---
 .../Diagnostics/Models/EventChainLink.cs      |  2 +-
 .../Diagnostics/ViewModels/EventTreeNode.cs   | 26 +++++++++++++++++++
 .../Diagnostics/ViewModels/FiredEvent.cs      |  5 ++--
 3 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/src/Avalonia.Diagnostics/Diagnostics/Models/EventChainLink.cs b/src/Avalonia.Diagnostics/Diagnostics/Models/EventChainLink.cs
index 4f493bdcc2..d986a11c45 100644
--- a/src/Avalonia.Diagnostics/Diagnostics/Models/EventChainLink.cs
+++ b/src/Avalonia.Diagnostics/Diagnostics/Models/EventChainLink.cs
@@ -29,7 +29,7 @@ namespace Avalonia.Diagnostics.Models
             }
         }
 
-        public bool Handled { get; }
+        public bool Handled { get; set; }
 
         public RoutingStrategies Route { get; }
     }
diff --git a/src/Avalonia.Diagnostics/Diagnostics/ViewModels/EventTreeNode.cs b/src/Avalonia.Diagnostics/Diagnostics/ViewModels/EventTreeNode.cs
index 65fd81cc78..a79816390d 100644
--- a/src/Avalonia.Diagnostics/Diagnostics/ViewModels/EventTreeNode.cs
+++ b/src/Avalonia.Diagnostics/Diagnostics/ViewModels/EventTreeNode.cs
@@ -55,6 +55,8 @@ namespace Avalonia.Diagnostics.ViewModels
 
                 // FIXME: This leaks event handlers.
                 Event.AddClassHandler(typeof(object), HandleEvent, allRoutes, handledEventsToo: true);
+                Event.RouteFinished.Subscribe(HandleRouteFinished);
+                
                 _isRegistered = true;
             }
         }
@@ -92,6 +94,30 @@ namespace Avalonia.Diagnostics.ViewModels
             else
                 handler();
         }
+        
+        private void HandleRouteFinished(RoutedEventArgs e)
+        {
+            if (!_isRegistered || IsEnabled == false)
+                return;
+            if (e.Source is IVisual v && BelongsToDevTool(v))
+                return;
+
+            var s = e.Source;
+            var handled = e.Handled;
+            var route = e.Route;
+
+            void handler()
+            {
+                if (_currentEvent != null && handled)
+                {
+                    var linkIndex = _currentEvent.EventChain.Count - 1;
+                    var link = _currentEvent.EventChain[linkIndex];
+
+                    link.Handled = true;
+                    _currentEvent.HandledBy = link;
+                }
+            }
+        }
 
         private static bool BelongsToDevTool(IVisual v)
         {
diff --git a/src/Avalonia.Diagnostics/Diagnostics/ViewModels/FiredEvent.cs b/src/Avalonia.Diagnostics/Diagnostics/ViewModels/FiredEvent.cs
index 32df2f8745..8069300922 100644
--- a/src/Avalonia.Diagnostics/Diagnostics/ViewModels/FiredEvent.cs
+++ b/src/Avalonia.Diagnostics/Diagnostics/ViewModels/FiredEvent.cs
@@ -63,8 +63,8 @@ namespace Avalonia.Diagnostics.ViewModels
         {
             if (EventChain.Count > 0)
             {
-                var prevLink = EventChain[EventChain.Count-1];
-
+                var prevLink = EventChain[EventChain.Count - 1];
+                
                 if (prevLink.Route != link.Route)
                 {
                     link.BeginsNewRoute = true;
@@ -72,6 +72,7 @@ namespace Avalonia.Diagnostics.ViewModels
             }
 
             EventChain.Add(link);
+
             if (HandledBy == null && link.Handled)
                 HandledBy = link;
         }