Adding functionality to track checked files

Author: StanleyGoldman

src/GitHub.Api/UI/TreeLoader.cs

@@ -8,7 +8,8 @@ public interface ITree
     {
         void AddNode(string path, string label, int level, bool isFolder, bool isActive, bool isHidden, bool isCollapsed, bool isSelected);
         void Clear();
-        HashSet<string> GetCollapsedFolders();
+        IEnumerable<string> GetCollapsedFolders();
+        IEnumerable<string> GetCheckedFiles();
         string Title { get; }
         bool DisplayRootNode { get; }
         bool IsCheckable { get; }
@@ -20,7 +21,7 @@ public static class TreeLoader
     {
         public static void Load(ITree tree, IEnumerable<ITreeData> treeDatas)
         {
-            var collapsedFolders = tree.GetCollapsedFolders();
+            var collapsedFolders = new HashSet<string>(tree.GetCollapsedFolders());
             var selectedNodePath = tree.SelectedNodePath;
 
             tree.Clear();

src/UnityExtension/Assets/Editor/GitHub.Unity/UI/TreeControl.cs

@@ -30,6 +30,7 @@ public abstract class Tree: ITree
         [SerializeField] private List<TreeNode> nodes = new List<TreeNode>();
         [SerializeField] private TreeNode selectedNode = null;
         [SerializeField] private TreeNodeDictionary folders = new TreeNodeDictionary();
+        [SerializeField] private TreeNodeDictionary checkedFileNodes = new TreeNodeDictionary();
 
         [NonSerialized] private Stack<bool> indents = new Stack<bool>();
         [NonSerialized] private Action<TreeNode> rightClickNextRender;
@@ -112,14 +113,19 @@ public void AddNode(string path, string label, int level, bool isFolder, bool is
         public void Clear()
         {
             folders.Clear();
+            checkedFileNodes.Clear();
             nodes.Clear();
             SelectedNode = null;
         }
 
-        public HashSet<string> GetCollapsedFolders()
+        public IEnumerable<string> GetCollapsedFolders()
         {
-            var collapsedFoldersEnumerable = folders.Where(pair => pair.Value.IsCollapsed).Select(pair => pair.Key);
-            return new HashSet<string>(collapsedFoldersEnumerable);
+            return folders.Where(pair => pair.Value.IsCollapsed).Select(pair => pair.Key);
+        }
+
+        public IEnumerable<string> GetCheckedFiles()
+        {
+            return checkedFileNodes.Where(pair => pair.Value.CheckState == CheckState.Checked).Select(pair => pair.Key);
         }
 
         public Rect Render(Rect containingRect, Rect rect, Vector2 scroll, Action<TreeNode> singleClick = null, Action<TreeNode> doubleClick = null, Action<TreeNode> rightClick = null)
@@ -270,6 +276,13 @@ private void ToggleNodeChecked(int idx, TreeNode node)
             {
                 ToggleChildrenChecked(idx, node, isChecked);
             }
+            else
+            {
+                if (isChecked)
+                {
+                    checkedFileNodes.Add(node.Path, node);
+                }
+            }
 
             ToggleParentFoldersChecked(idx, node, isChecked);
         }
@@ -290,6 +303,13 @@ private void ToggleChildrenChecked(int idx, TreeNode node, bool isChecked)
                 {
                     ToggleChildrenChecked(i, childNode, isChecked);
                 }
+                else
+                {
+                    if (isChecked)
+                    {
+                        checkedFileNodes.Add(node.Path, node);
+                    }
+                }
             }
         }