From acc9840830d146ba24a783dae59a63999b36d4b0 Mon Sep 17 00:00:00 2001
From: leo <longshuang@msn.cn>
Date: Mon, 15 Nov 2021 12:20:33 +0800
Subject: [PATCH] feature<RevisionFiles>: enable to search file in revision
 files

---
 src/Views/Widgets/RevisionFiles.xaml    | 102 +++++++++++-----
 src/Views/Widgets/RevisionFiles.xaml.cs | 151 ++++++++++++++----------
 2 files changed, 159 insertions(+), 94 deletions(-)

diff --git a/src/Views/Widgets/RevisionFiles.xaml b/src/Views/Widgets/RevisionFiles.xaml
index 02da03bd..b28742ea 100644
--- a/src/Views/Widgets/RevisionFiles.xaml
+++ b/src/Views/Widgets/RevisionFiles.xaml
@@ -21,39 +21,77 @@
             <ColumnDefinition Width="*"/>
         </Grid.ColumnDefinitions>
 
-        <Border Grid.Column="0" Background="{DynamicResource Brush.Contents}" BorderBrush="{DynamicResource Brush.Border2}" BorderThickness="1">
-            <controls:Tree x:Name="treeFiles" SelectionChanged="OnFilesSelectionChanged" ContextMenuOpening="OnFilesContextMenuOpening">
-                <controls:Tree.ItemTemplate>
-                    <HierarchicalDataTemplate ItemsSource="{Binding Children}">
-                        <StackPanel Orientation="Horizontal" Height="24">
-                            <Path x:Name="Icon" Width="14" Height="14" Data="{StaticResource Icon.File}"/>
-                            <TextBlock Margin="6,0,0,0" Text="{Binding Path, Converter={StaticResource PureFileName}}"/>
-                        </StackPanel>
+        <Grid Grid.Column="0">
+            <Grid.RowDefinitions>
+                <RowDefinition Height="Auto"/>
+                <RowDefinition Height="*"/>
+            </Grid.RowDefinitions>
 
-                        <HierarchicalDataTemplate.Triggers>
-                            <DataTrigger Binding="{Binding IsFolder}" Value="True">
-                                <Setter TargetName="Icon" Property="Fill" Value="Goldenrod"/>
-                                <Setter TargetName="Icon" Property="Opacity" Value="1"/>
-                            </DataTrigger>
-                            <MultiDataTrigger>
-                                <MultiDataTrigger.Conditions>
-                                    <Condition Binding="{Binding IsFolder}" Value="True"/>
-                                    <Condition Binding="{Binding RelativeSource={RelativeSource AncestorType={x:Type controls:TreeItem}}, Path=IsExpanded}" Value="False"/>
-                                </MultiDataTrigger.Conditions>
-                                <Setter TargetName="Icon" Property="Data" Value="{StaticResource Icon.Folder.Fill}"/>
-                            </MultiDataTrigger>
-                            <MultiDataTrigger>
-                                <MultiDataTrigger.Conditions>
-                                    <Condition Binding="{Binding IsFolder}" Value="True"/>
-                                    <Condition Binding="{Binding RelativeSource={RelativeSource AncestorType={x:Type controls:TreeItem}}, Path=IsExpanded}" Value="True"/>
-                                </MultiDataTrigger.Conditions>
-                                <Setter TargetName="Icon" Property="Data" Value="{StaticResource Icon.Folder.Open}"/>
-                            </MultiDataTrigger>
-                        </HierarchicalDataTemplate.Triggers>
-                    </HierarchicalDataTemplate>
-                </controls:Tree.ItemTemplate>
-            </controls:Tree>
-        </Border>
+            <Grid Grid.Row="0">
+                <Grid.ColumnDefinitions>
+                    <ColumnDefinition Width="24"/>
+                    <ColumnDefinition Width="*"/>
+                </Grid.ColumnDefinitions>
+
+                <Border
+                    Grid.Column="0" Grid.ColumnSpan="2"
+                    BorderBrush="{DynamicResource Brush.Border2}"
+                    BorderThickness="1"/>
+                <Path
+                    Grid.Column="0"
+                    Width="14" Height="14"
+                    Fill="{DynamicResource Brush.FG2}"
+                    Data="{StaticResource Icon.Search}"
+                    IsHitTestVisible="False"/>
+                <controls:TextEdit
+                    Grid.Column="1"
+                    Height="24"
+                    Margin="0"
+                    Placeholder="{DynamicResource Text.CommitViewer.Changes.Search}"
+                    BorderThickness="0"
+                    TextChanged="OnSearchFilterChanged"/>
+            </Grid>
+
+            <Border Grid.Row="1" Margin="0,4,0,0" Background="{DynamicResource Brush.Contents}" BorderBrush="{DynamicResource Brush.Border2}" BorderThickness="1">
+                <controls:Tree x:Name="treeFiles" SelectionChanged="OnFilesSelectionChanged">
+                    <controls:Tree.ItemContainerStyle>
+                        <Style TargetType="{x:Type controls:TreeItem}" BasedOn="{StaticResource Style.TreeItem}">
+                            <Setter Property="IsExpanded" Value="{Binding IsExpanded, Mode=TwoWay}"/>
+                            <EventSetter Event="ContextMenuOpening" Handler="OnFilesContextMenuOpening"/>
+                        </Style>
+                    </controls:Tree.ItemContainerStyle>
+                    <controls:Tree.ItemTemplate>
+                        <HierarchicalDataTemplate ItemsSource="{Binding Children}">
+                            <StackPanel Orientation="Horizontal" Height="24">
+                                <Path x:Name="Icon" Width="14" Height="14" Data="{StaticResource Icon.File}"/>
+                                <TextBlock Margin="6,0,0,0" Text="{Binding Path, Converter={StaticResource PureFileName}}"/>
+                            </StackPanel>
+
+                            <HierarchicalDataTemplate.Triggers>
+                                <DataTrigger Binding="{Binding IsFolder}" Value="True">
+                                    <Setter TargetName="Icon" Property="Fill" Value="Goldenrod"/>
+                                    <Setter TargetName="Icon" Property="Opacity" Value="1"/>
+                                </DataTrigger>
+                                <MultiDataTrigger>
+                                    <MultiDataTrigger.Conditions>
+                                        <Condition Binding="{Binding IsFolder}" Value="True"/>
+                                        <Condition Binding="{Binding RelativeSource={RelativeSource AncestorType={x:Type controls:TreeItem}}, Path=IsExpanded}" Value="False"/>
+                                    </MultiDataTrigger.Conditions>
+                                    <Setter TargetName="Icon" Property="Data" Value="{StaticResource Icon.Folder.Fill}"/>
+                                </MultiDataTrigger>
+                                <MultiDataTrigger>
+                                    <MultiDataTrigger.Conditions>
+                                        <Condition Binding="{Binding IsFolder}" Value="True"/>
+                                        <Condition Binding="{Binding RelativeSource={RelativeSource AncestorType={x:Type controls:TreeItem}}, Path=IsExpanded}" Value="True"/>
+                                    </MultiDataTrigger.Conditions>
+                                    <Setter TargetName="Icon" Property="Data" Value="{StaticResource Icon.Folder.Open}"/>
+                                </MultiDataTrigger>
+                            </HierarchicalDataTemplate.Triggers>
+                        </HierarchicalDataTemplate>
+                    </controls:Tree.ItemTemplate>
+                </controls:Tree>
+            </Border>
+        </Grid>
 
         <GridSplitter Grid.Column="1" Width="1" HorizontalAlignment="Center" VerticalAlignment="Stretch" Background="Transparent"/>
 
diff --git a/src/Views/Widgets/RevisionFiles.xaml.cs b/src/Views/Widgets/RevisionFiles.xaml.cs
index db1593bf..103df7db 100644
--- a/src/Views/Widgets/RevisionFiles.xaml.cs
+++ b/src/Views/Widgets/RevisionFiles.xaml.cs
@@ -18,6 +18,8 @@ namespace SourceGit.Views.Widgets {
         private string repo = null;
         private string sha = null;
         private bool isLFSEnabled = false;
+        private List<Models.Object> cached = new List<Models.Object>();
+        private string filter = null;
 
         /// <summary>
         ///     文件列表树节点
@@ -26,6 +28,7 @@ namespace SourceGit.Views.Widgets {
             public Models.ObjectType Type { get; set; } = Models.ObjectType.None;
             public string Path { get; set; } = "";
             public string SHA { get; set; } = null;
+            public bool IsExpanded { get; set; } = false;
             public bool IsFolder => Type == Models.ObjectType.None;
             public List<FileNode> Children { get; set; } = new List<FileNode>();
         }
@@ -44,72 +47,89 @@ namespace SourceGit.Views.Widgets {
                 var objects = cmd.Result();
                 if (cmd.Ctx.IsCancelRequested) return;
 
-                var nodes = new List<FileNode>();
-                var folders = new Dictionary<string, FileNode>();
-
-                foreach (var obj in objects) {
-                    var sepIdx = obj.Path.IndexOf('/');
-                    if (sepIdx == -1) {
-                        nodes.Add(new FileNode() {
-                            Type = obj.Type,
-                            Path = obj.Path,
-                            SHA = obj.SHA,
-                        });
-                    } else {
-                        FileNode lastFolder = null;
-                        var start = 0;
-
-                        while (sepIdx != -1) {
-                            var folder = obj.Path.Substring(0, sepIdx);
-                            if (folders.ContainsKey(folder)) {
-                                lastFolder = folders[folder];
-                            } else if (lastFolder == null) {
-                                lastFolder = new FileNode() {
-                                    Type = Models.ObjectType.None,
-                                    Path = folder,
-                                    SHA = null,
-                                };
-                                nodes.Add(lastFolder);
-                                folders.Add(folder, lastFolder);
-                            } else {
-                                var cur = new FileNode() {
-                                    Type = Models.ObjectType.None,
-                                    Path = folder,
-                                    SHA = null,
-                                };
-                                folders.Add(folder, cur);
-                                lastFolder.Children.Add(cur);
-                                lastFolder = cur;
-                            }
-
-                            start = sepIdx + 1;
-                            sepIdx = obj.Path.IndexOf('/', start);
-                        }
-
-                        lastFolder.Children.Add(new FileNode() {
-                            Type = obj.Type,
-                            Path = obj.Path,
-                            SHA = obj.SHA,
-                        });
-                    }
-
-                    obj.Path = null;
-                }
-
-                folders.Clear();
-                objects.Clear();
-
-                SortFileNodes(nodes);
-
-                Dispatcher.Invoke(() => {
-                    treeFiles.ItemsSource = nodes;
-                    GC.Collect();
-                });
+                cached = objects;
+                ShowVisibles();        
             });
         }
 
         public void Cleanup() {
             treeFiles.ItemsSource = new List<FileNode>();
+            cached = new List<Models.Object>();
+        }
+
+        private void ShowVisibles() {
+            var nodes = new List<FileNode>();
+            var folders = new Dictionary<string, FileNode>();
+            var visibles = new List<Models.Object>();
+
+            if (string.IsNullOrEmpty(filter)) {
+                visibles.AddRange(cached);
+            } else {
+                foreach (var obj in cached) {
+                    if (obj.Path.ToUpper().Contains(filter)) visibles.Add(obj);
+                }
+            }
+
+            var expanded = visibles.Count <= 50;
+
+            foreach (var obj in visibles) {
+                var sepIdx = obj.Path.IndexOf('/');
+                if (sepIdx == -1) {
+                    nodes.Add(new FileNode() {
+                        Type = obj.Type,
+                        Path = obj.Path,
+                        SHA = obj.SHA,
+                    });
+                } else {
+                    FileNode lastFolder = null;
+                    var start = 0;
+
+                    while (sepIdx != -1) {
+                        var folder = obj.Path.Substring(0, sepIdx);
+                        if (folders.ContainsKey(folder)) {
+                            lastFolder = folders[folder];
+                        } else if (lastFolder == null) {
+                            lastFolder = new FileNode() {
+                                Type = Models.ObjectType.None,
+                                Path = folder,
+                                SHA = null,
+                                IsExpanded = expanded,
+                            };
+                            nodes.Add(lastFolder);
+                            folders.Add(folder, lastFolder);
+                        } else {
+                            var cur = new FileNode() {
+                                Type = Models.ObjectType.None,
+                                Path = folder,
+                                SHA = null,
+                                IsExpanded = expanded,
+                            };
+                            folders.Add(folder, cur);
+                            lastFolder.Children.Add(cur);
+                            lastFolder = cur;
+                        }
+
+                        start = sepIdx + 1;
+                        sepIdx = obj.Path.IndexOf('/', start);
+                    }
+
+                    lastFolder.Children.Add(new FileNode() {
+                        Type = obj.Type,
+                        Path = obj.Path,
+                        SHA = obj.SHA,
+                    });
+                }
+            }
+
+            folders.Clear();
+            visibles.Clear();
+
+            SortFileNodes(nodes);
+
+            Dispatcher.Invoke(() => {
+                treeFiles.ItemsSource = nodes;
+                GC.Collect();
+            });
         }
 
         private void SortFileNodes(List<FileNode> nodes) {
@@ -259,7 +279,7 @@ namespace SourceGit.Views.Widgets {
         }
 
         private void OnFilesContextMenuOpening(object sender, ContextMenuEventArgs e) {
-            var item = treeFiles.FindItem(e.OriginalSource as DependencyObject);
+            var item = sender as Controls.TreeItem;
             if (item == null) return;
 
             var node = item.DataContext as FileNode;
@@ -322,6 +342,13 @@ namespace SourceGit.Views.Widgets {
         private void OnRequestBringIntoView(object sender, RequestBringIntoViewEventArgs e) {
             e.Handled = true;
         }
+
+        private void OnSearchFilterChanged(object sender, TextChangedEventArgs e) {
+            var edit = sender as Controls.TextEdit;
+            filter = edit.Text.ToUpper();
+            Task.Run(() => ShowVisibles());
+            e.Handled = true;
+        }
         #endregion
     }
 }