Merge pull request #508 from github-for-unity/fixes/last-few-fixes

Last few fixes

Author: StanleyGoldman

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

@@ -323,6 +323,9 @@ private void OnTreeGUI(Rect rect)
                     if (node.IsFolder)
                         return;
 
+                    if(node.IsActive)
+                        return;
+
                     SwitchBranch(node.Name);
                 },
                 node => {

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

@@ -56,6 +56,12 @@ public void Load(IEnumerable<ITreeData> data, string title)
         {
             var collapsedFoldersEnumerable = folders.Where(pair => pair.Value.IsCollapsed).Select(pair => pair.Key);
             var collapsedFolders = new HashSet<string>(collapsedFoldersEnumerable);
+            string selectedNodeName = null;
+            if (SelectedNode != null)
+            {
+                selectedNodeName = SelectedNode.Name;
+                SelectedNode = null;
+            }
 
             folders.Clear();
             nodes.Clear();
@@ -94,6 +100,11 @@ public void Load(IEnumerable<ITreeData> data, string title)
                             IsFolder = isFolder
                         };
 
+                        if (selectedNodeName != null && name == selectedNodeName)
+                        {
+                            SelectedNode = node;
+                        }
+
                         if (node.IsActive)
                         {
                             activeNode = node;