Merge pull request #499 from github-for-unity/fixes/create-branch-mode

Create branch fixes

Author: StanleyGoldman

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

@@ -45,6 +45,7 @@ class BranchesView : Subview
         [SerializeField] private string newBranchName;
         [SerializeField] private Vector2 scroll;
         [SerializeField] private bool disableDelete;
+        [SerializeField] private bool disableCreate;
 
         [SerializeField] private CacheUpdateEvent lastLocalAndRemoteBranchListChangedEvent;
         [NonSerialized] private bool localAndRemoteBranchListHasUpdate;
@@ -102,6 +103,7 @@ private void MaybeUpdateData()
             }
 
             disableDelete = treeLocals.SelectedNode == null || treeLocals.SelectedNode.IsFolder || treeLocals.SelectedNode.IsActive;
+            disableCreate = treeLocals.SelectedNode == null || treeLocals.SelectedNode.IsFolder || treeLocals.SelectedNode.Level == 0;
         }
 
         public override void OnGUI()
@@ -134,6 +136,16 @@ private void Render()
                 OnTreeGUI(new Rect(0f, 0f, Position.width, Position.height - rect.height + Styles.CommitAreaPadding)); 
             }
             GUILayout.EndScrollView();
+
+            if (Event.current.type == EventType.Repaint)
+            {
+                // Effectuating mode switch
+                if (mode != targetMode)
+                {
+                    mode = targetMode;
+                    Redraw();
+                }
+            }
         }
 
         private void BuildTree()
@@ -186,10 +198,14 @@ private void OnButtonBarGUI()
 
                 // Create button
                 GUILayout.FlexibleSpace();
-                if (GUILayout.Button(CreateBranchButton, EditorStyles.miniButton, GUILayout.ExpandWidth(false)))
+                EditorGUI.BeginDisabledGroup(disableCreate);
                 {
-                    targetMode = BranchesMode.Create;
+                    if (GUILayout.Button(CreateBranchButton, EditorStyles.miniButton, GUILayout.ExpandWidth(false)))
+                    {
+                        targetMode = BranchesMode.Create;
+                    }
                 }
+                EditorGUI.EndDisabledGroup();
             }
             // Branch name + cancel + create
             else if (mode == BranchesMode.Create)