From 9a2cfbb08e8578db671b7c3dbd2aa19b37086808 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=98=D0=B2=D0=B0=D0=BD=20=D0=98=D0=B2=D0=B0=D1=88=D0=BA?=
 =?UTF-8?q?=D0=B8=D0=BD?= <ivashkinia@yandex.ru>
Date: Thu, 6 Mar 2025 13:35:48 +0500
Subject: [PATCH] Tree MenuItem hot fix

---
 .../Windows/TreeGraph/TreeGraphView.xaml        | 17 ++++++++++++++++-
 .../Windows/TreeGraph/TreeGraphViewModel.cs     | 12 +++++++++---
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/StructureHelper/Windows/TreeGraph/TreeGraphView.xaml b/StructureHelper/Windows/TreeGraph/TreeGraphView.xaml
index 2d74ec2..4aa05b7 100644
--- a/StructureHelper/Windows/TreeGraph/TreeGraphView.xaml
+++ b/StructureHelper/Windows/TreeGraph/TreeGraphView.xaml
@@ -85,6 +85,20 @@
                             <Image Width="16" Height="16" Source="/Windows/MainWindow/Delete.png" />
                         </MenuItem.Icon>
                     </MenuItem>
+                    <MenuItem Header="Tree" 
+                              Command="{Binding TreeCommand}"
+                              CommandParameter="{Binding ElementName=FunctionTreeView}">
+                        <MenuItem.Icon>
+                            <Image Width="16" Height="16" Source="/Windows/MainWindow/Tree.png" />
+                        </MenuItem.Icon>
+                    </MenuItem>
+                    <MenuItem Header="Rename" 
+                              Command="{Binding RenameCommand}"
+                              CommandParameter="{Binding ElementName=FunctionTreeView}">
+                        <MenuItem.Icon>
+                            <Image Width="16" Height="16" Source="/Windows/MainWindow/Edit.png" />
+                        </MenuItem.Icon>
+                    </MenuItem>
                 </ContextMenu>
             </TreeView.ContextMenu>
         </TreeView>
@@ -136,7 +150,8 @@
             </Grid.RowDefinitions>
             <Button Grid.Column="0" Margin="5" Content="Tree"
                     Background="AntiqueWhite"
-                    Command="{Binding TreeCommand}"/>
+                    Command="{Binding TreeCommand}"
+                    CommandParameter="{Binding ElementName=FunctionTreeView}"/>
             <Button Grid.Column="1" Margin="5" Content="Rename"
                     FontSize="10"
                     Background="HotPink"
diff --git a/StructureHelper/Windows/TreeGraph/TreeGraphViewModel.cs b/StructureHelper/Windows/TreeGraph/TreeGraphViewModel.cs
index b7373cd..77bf2be 100644
--- a/StructureHelper/Windows/TreeGraph/TreeGraphViewModel.cs
+++ b/StructureHelper/Windows/TreeGraph/TreeGraphViewModel.cs
@@ -121,7 +121,7 @@ namespace StructureHelper.Windows.TreeGraph
         }
         public ICommand TreeCommand
         {
-            get => _treeCommand ??= new RelayCommand(o => NewTree());
+            get => _treeCommand ??= new RelayCommand(o => NewTree(o));
         }
         public ICommand RenameCommand
         {
@@ -276,9 +276,15 @@ namespace StructureHelper.Windows.TreeGraph
             }
             Save();
         }
-        private void NewTree()
+        private void NewTree(object parameter)
         {
-            if (SelectedFuntion is null)
+            var selectedTreeViewItem = TreeGraphView_win.FunctionTreeView.SelectedItem as TreeViewItemViewModel;
+            if (selectedTreeViewItem is null)
+            {
+                return;
+            }
+            var selectedTreeViewItemParent = selectedTreeViewItem.Parent;
+            if (selectedTreeViewItemParent is null)
             {
                 return;
             }