Restoring functionality to work with trees

Adding functionality to check/uncheck all
Restoring intergations with ChangesView

Author: StanleyGoldman

src/GitHub.Api/UI/TreeBase.cs

@@ -102,6 +102,24 @@ public void Load(IEnumerable<TData> treeDatas)
             }
         }
 
+        public void CheckAll(bool isChecked)
+        {
+            var nodeCheckState = isChecked ? CheckState.Checked : CheckState.Empty;
+            foreach (var node in Nodes)
+            {
+                var wasChecked = node.CheckState == CheckState.Checked;
+                node.CheckState = nodeCheckState;
+                if (isChecked && !wasChecked)
+                {
+                    AddCheckedNode(node);
+                }
+                else if (!isChecked && wasChecked)
+                {
+                    RemoveCheckedNode(node);
+                }
+            }
+        }
+
         protected abstract IEnumerable<string> GetCollapsedFolders();
 
         protected void AddNode(string path, string label, int level, bool isFolder, bool isActive, bool isHidden,

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

@@ -57,7 +57,7 @@ public override void OnGUI()
         {
             GUILayout.BeginHorizontal();
             {
-                EditorGUI.BeginDisabledGroup(false);
+                EditorGUI.BeginDisabledGroup(gitStatusEntries == null || !gitStatusEntries.Any());
                 {
                     if (GUILayout.Button(SelectAllButton, EditorStyles.miniButtonLeft))
                     {
@@ -258,48 +258,38 @@ private void OnCommitDetailsAreaGUI()
 
         private void SelectAll()
         {
-//            for (var index = 0; index < tree.CommitTargets.Count; ++index)
-//            {
-//                tree.CommitTargets[index].All = true;
-//            }
+            this.treeChanges.CheckAll(true);
         }
 
         private void SelectNone()
         {
-//            for (var index = 0; index < tree.CommitTargets.Count; ++index)
-//            {
-//                tree.CommitTargets[index].All = false;
-//            }
+            this.treeChanges.CheckAll(false);
         }
 
         private void Commit()
         {
             // Do not allow new commits before we have received one successful update
-//            isBusy = true;
-//
-//            var files = Enumerable.Range(0, tree.Entries.Count)
-//                .Where(i => tree.CommitTargets[i].All)
-//                .Select(i => tree.Entries[i].Path)
-//                .ToList();
-//
-//            ITask addTask;
-//
-//            if (files.Count == tree.Entries.Count)
-//            {
-//                addTask = Repository.CommitAllFiles(commitMessage, commitBody);
-//            }
-//            else
-//            {
-//                addTask = Repository.CommitFiles(files, commitMessage, commitBody);
-//            }
-//
-//            addTask
-//                .FinallyInUI((b, exception) => 
-//                    {
-//                        commitMessage = "";
-//                        commitBody = "";
-//                        isBusy = false;
-//                    }).Start();
+            isBusy = true;
+
+            var files = treeChanges.GetCheckedFiles().ToList();
+            ITask addTask;
+
+            if (files.Count == gitStatusEntries.Count)
+            {
+                addTask = Repository.CommitAllFiles(commitMessage, commitBody);
+            }
+            else
+            {
+                addTask = Repository.CommitFiles(files, commitMessage, commitBody);
+            }
+
+            addTask
+                .FinallyInUI((b, exception) => 
+                    {
+                        commitMessage = "";
+                        commitBody = "";
+                        isBusy = false;
+                    }).Start();
         }
 
         public override bool IsBusy