From 122f89fc57fbae778e3f17354b6e5d8659b02611 Mon Sep 17 00:00:00 2001
From: rabbitism <dongbin@irihi.tech>
Date: Wed, 14 Feb 2024 11:40:19 +0800
Subject: [PATCH] feat: add function to set selected item from outside and sync
 to UI.

---
 demo/Ursa.Demo/Pages/NavMenuDemo.axaml        | 19 +++----
 .../ViewModels/NavMenuDemoViewModel.cs        | 50 +++++++++++++++++
 src/Ursa/Controls/NavMenu/NavMenu.cs          | 54 ++++++++++++++++++-
 src/Ursa/Controls/NavMenu/NavMenuItem.cs      | 22 +++++++-
 4 files changed, 133 insertions(+), 12 deletions(-)

diff --git a/demo/Ursa.Demo/Pages/NavMenuDemo.axaml b/demo/Ursa.Demo/Pages/NavMenuDemo.axaml
index 72a2f89..f1ac1cd 100644
--- a/demo/Ursa.Demo/Pages/NavMenuDemo.axaml
+++ b/demo/Ursa.Demo/Pages/NavMenuDemo.axaml
@@ -26,10 +26,10 @@
             Grid.ColumnSpan="2">
             Collapse
         </ToggleButton>
-        <TextBlock
-            Grid.Row="1"
-            Grid.Column="0"
-            Text="{ReflectionBinding #menu.SelectedItem.Header}" />
+        <StackPanel Grid.Row="1" Grid.Column="0">
+            <TextBlock Text="{Binding SelectedMenuItem.Header}" />
+            <Button Command="{Binding RandomCommand}">Random a selection</Button>
+        </StackPanel>
         <Border
             Grid.Row="2"
             Grid.Column="0"
@@ -42,6 +42,7 @@
                 IconBinding="{Binding IconIndex}"
                 IsHorizontalCollapsed="{Binding #collapse.IsChecked, Mode=OneWay}"
                 ItemsSource="{Binding MenuItems}"
+                SelectedItem="{Binding SelectedMenuItem}"
                 SubMenuBinding="{Binding Children}">
                 <u:NavMenu.Styles>
                     <Style x:DataType="vm:MenuItem" Selector="u|NavMenuItem">
@@ -62,20 +63,20 @@
                     </DataTemplate>
                 </u:NavMenu.IconTemplate>
                 <u:NavMenu.Header>
-                    <Grid ColumnDefinitions="Auto, Auto" HorizontalAlignment="Center">
+                    <Grid HorizontalAlignment="Center" ColumnDefinitions="Auto, Auto">
                         <Image
-                            u:NavMenu.CanToggle="True"
                             Width="48"
                             Height="48"
-                            Margin="4 12"
+                            Margin="4,12"
+                            u:NavMenu.CanToggle="True"
                             RenderOptions.BitmapInterpolationMode="HighQuality"
                             Source="../Assets/Ursa.ico" />
                         <TextBlock
                             Grid.Column="1"
-                            Classes="H5"
                             VerticalAlignment="Center"
-                            Text="Ursa Avalonia"
+                            Classes="H5"
                             IsVisible="{Binding !#menu.IsHorizontalCollapsed}"
+                            Text="Ursa Avalonia"
                             Theme="{DynamicResource TitleTextBlock}" />
                     </Grid>
                 </u:NavMenu.Header>
diff --git a/demo/Ursa.Demo/ViewModels/NavMenuDemoViewModel.cs b/demo/Ursa.Demo/ViewModels/NavMenuDemoViewModel.cs
index ed44191..66bec4c 100644
--- a/demo/Ursa.Demo/ViewModels/NavMenuDemoViewModel.cs
+++ b/demo/Ursa.Demo/ViewModels/NavMenuDemoViewModel.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Collections.ObjectModel;
 using System.Threading.Tasks;
 using System.Windows.Input;
@@ -9,6 +10,13 @@ namespace Ursa.Demo.ViewModels;
 
 public class NavMenuDemoViewModel: ObservableObject
 {
+    private MenuItem? _selectedMenuItem;
+
+    public MenuItem? SelectedMenuItem
+    {
+        get=>_selectedMenuItem;
+        set => SetProperty(ref _selectedMenuItem, value);
+    }
     public ObservableCollection<MenuItem> MenuItems { get; set; } = new ObservableCollection<MenuItem>
     {
         new MenuItem { Header = "Introduction" , Children =
@@ -54,6 +62,30 @@ public class NavMenuDemoViewModel: ObservableObject
         new MenuItem { Header = "TwoTonePathIcon" },
         new MenuItem { Header = "ThemeToggler" }
     };
+
+    public ICommand RandomCommand { get; set; }
+    public NavMenuDemoViewModel()
+    {
+        RandomCommand = new RelayCommand(OnRandom);
+    }
+
+    private void OnRandom()
+    {
+        var items = GetLeaves();
+        var index = new Random().Next(items.Count);
+        SelectedMenuItem = items[index];
+    }
+    
+    private List<MenuItem> GetLeaves()
+    {
+        List<MenuItem> items = new();
+        foreach (var item in MenuItems)
+        {
+            items.AddRange(item.GetLeaves());
+        }
+
+        return items;
+    }
 }
 
 public class MenuItem
@@ -77,4 +109,22 @@ public class MenuItem
     }
 
     public ObservableCollection<MenuItem> Children { get; set; } = new ObservableCollection<MenuItem>();
+
+    public IEnumerable<MenuItem> GetLeaves()
+    {
+        if (this.Children.Count == 0)
+        {
+            yield return this;
+            yield break;
+        }
+
+        foreach (var child in Children)
+        {
+            var items = child.GetLeaves();
+            foreach (var item in items)
+            {
+                yield return item;
+            }
+        }
+    }
 }
\ No newline at end of file
diff --git a/src/Ursa/Controls/NavMenu/NavMenu.cs b/src/Ursa/Controls/NavMenu/NavMenu.cs
index e1b629e..fed4dfd 100644
--- a/src/Ursa/Controls/NavMenu/NavMenu.cs
+++ b/src/Ursa/Controls/NavMenu/NavMenu.cs
@@ -1,4 +1,5 @@
 using System.Diagnostics;
+using System.Reflection;
 using Avalonia;
 using Avalonia.Controls;
 using Avalonia.Controls.Metadata;
@@ -164,7 +165,25 @@ public class NavMenu: ItemsControl
 
     private void OnSelectedItemChange(AvaloniaPropertyChangedEventArgs<object?> args)
     {
-        Debug.WriteLine(args.NewValue.Value);
+        if (_updateFromUI) return;
+        var newValue = args.NewValue.Value;
+        if (newValue is null)
+        {
+            ClearAll();
+            return;
+        }
+        var leaves = GetLeafMenus();
+        bool found = false;
+        foreach (var leaf in leaves)
+        {
+            if (leaf == newValue || leaf.DataContext == newValue)
+            {
+                leaf.SelectItem(leaf);
+                found = true;
+            }
+        }
+        if (found) return;
+        ClearAll();
     }
 
     protected override bool NeedsContainerOverride(object? item, int index, out object? recycleKey)
@@ -177,8 +196,11 @@ public class NavMenu: ItemsControl
         return new NavMenuItem();
     }
 
+    private bool _updateFromUI;
+
     internal void SelectItem(NavMenuItem item, NavMenuItem parent)
     {
+        _updateFromUI = true;
         // if (item.IsSelected) return;
         foreach (var child in LogicalChildren)
         {
@@ -199,6 +221,34 @@ public class NavMenu: ItemsControl
         {
             SelectedItem = item;
         }
-        item.IsSelected = true;
+        item.BringIntoView();
+        // item.IsSelected = true;
+        _updateFromUI = false;
+    }
+
+    private IEnumerable<NavMenuItem> GetLeafMenus()
+    {
+        foreach (var child in LogicalChildren)   
+        {
+            if (child is NavMenuItem item)
+            {
+                var leafs = item.GetLeafMenus();
+                foreach (var leaf in leafs)
+                {
+                    yield return leaf;
+                }
+            }
+        }
+    }
+
+    private void ClearAll()
+    {
+        foreach (var child in LogicalChildren)   
+        {
+            if (child is NavMenuItem item)
+            {
+                item.ClearSelection();
+            }
+        }
     }
 }
\ No newline at end of file
diff --git a/src/Ursa/Controls/NavMenu/NavMenuItem.cs b/src/Ursa/Controls/NavMenu/NavMenuItem.cs
index f2550bf..10568d5 100644
--- a/src/Ursa/Controls/NavMenu/NavMenuItem.cs
+++ b/src/Ursa/Controls/NavMenu/NavMenuItem.cs
@@ -265,7 +265,7 @@ public class NavMenuItem: HeaderedSelectingItemsControl
         e.Handled = true;
     }
     
-    protected void SelectItem(NavMenuItem item)
+    internal void SelectItem(NavMenuItem item)
     {
         if (item == this)
         {
@@ -329,4 +329,24 @@ public class NavMenuItem: HeaderedSelectingItemsControl
 
         return logical != null ? result : @default;
     }
+    
+    internal IEnumerable<NavMenuItem> GetLeafMenus()
+    {
+        if (this.ItemCount == 0)
+        {
+            yield return this;
+            yield break;
+        }
+        foreach (var child in LogicalChildren)   
+        {
+            if (child is NavMenuItem item)
+            {
+                var items = item.GetLeafMenus();
+                foreach (var i in items)
+                {
+                    yield return i;
+                }
+            }
+        }
+    }
 }
\ No newline at end of file