Adding functionality to checkout a remote branch via right click

Author: StanleyGoldman

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

@@ -301,45 +301,8 @@ private void OnTreeGUI(Rect rect)
 
             rect = treeRemotes.Render(rect, scroll,
                 node => { },
-                selectedNode =>
-                {
-                    var indexOfFirstSlash = selectedNode.Name.IndexOf('/');
-                    var originName = selectedNode.Name.Substring(0, indexOfFirstSlash);
-                    var branchName = selectedNode.Name.Substring(indexOfFirstSlash + 1);
-
-                    if (Repository.LocalBranches.Any(localBranch => localBranch.Name == branchName))
-                    {
-                        EditorUtility.DisplayDialog(WarningCheckoutBranchExistsTitle,
-                            String.Format(WarningCheckoutBranchExistsMessage, branchName),
-                            WarningCheckoutBranchExistsOK);
-                    }
-                    else
-                    {
-                        var confirmCheckout = EditorUtility.DisplayDialog(ConfirmCheckoutBranchTitle,
-                            String.Format(ConfirmCheckoutBranchMessage, selectedNode.Name, originName),
-                            ConfirmCheckoutBranchOK,
-                            ConfirmCheckoutBranchCancel);
-
-                        if (confirmCheckout)
-                        {
-                            GitClient
-                                .CreateBranch(branchName, selectedNode.Name)
-                                .FinallyInUI((success, e) =>
-                                {
-                                    if (success)
-                                    {
-                                        Redraw();
-                                    }
-                                    else
-                                    {
-                                        EditorUtility.DisplayDialog(Localization.SwitchBranchTitle,
-                                            String.Format(Localization.SwitchBranchFailedDescription, selectedNode.Name),
-                                            Localization.Ok);
-                                    }
-                                })
-                                .Start();
-                        }
-                    }
+                selectedNode => {
+                    CheckoutRemoteBranch(selectedNode.Name);
                 },
                 node => {
                     GenericMenu menu = CreateContextMenuForRemoteBranchNode(node);
@@ -362,6 +325,40 @@ private void OnTreeGUI(Rect rect)
             GUILayout.Space(rect.y - initialRect.y);
         }
 
+        private void CheckoutRemoteBranch(string selectedNodeName)
+        {
+            var indexOfFirstSlash = selectedNodeName.IndexOf('/');
+            var originName = selectedNodeName.Substring(0, indexOfFirstSlash);
+            var branchName = selectedNodeName.Substring(indexOfFirstSlash + 1);
+
+            if (Repository.LocalBranches.Any(localBranch => localBranch.Name == branchName))
+            {
+                EditorUtility.DisplayDialog(WarningCheckoutBranchExistsTitle,
+                    String.Format(WarningCheckoutBranchExistsMessage, branchName), WarningCheckoutBranchExistsOK);
+            }
+            else
+            {
+                var confirmCheckout = EditorUtility.DisplayDialog(ConfirmCheckoutBranchTitle,
+                    String.Format(ConfirmCheckoutBranchMessage, selectedNodeName, originName), ConfirmCheckoutBranchOK,
+                    ConfirmCheckoutBranchCancel);
+
+                if (confirmCheckout)
+                {
+                    GitClient.CreateBranch(branchName, selectedNodeName).FinallyInUI((success, e) => {
+                        if (success)
+                        {
+                            Redraw();
+                        }
+                        else
+                        {
+                            EditorUtility.DisplayDialog(Localization.SwitchBranchTitle,
+                                String.Format(Localization.SwitchBranchFailedDescription, selectedNodeName), Localization.Ok);
+                        }
+                    }).Start();
+                }
+            }
+        }
+
         private GenericMenu CreateContextMenuForLocalBranchNode(TreeNode node)
         {
             var genericMenu = new GenericMenu();
@@ -420,6 +417,12 @@ private GenericMenu CreateContextMenuForRemoteBranchNode(TreeNode node)
         {
             var genericMenu = new GenericMenu();
 
+            var checkoutGuiContent = new GUIContent("Checkout");
+            
+            genericMenu.AddItem(checkoutGuiContent, false, () => {
+                CheckoutRemoteBranch(node.Name);
+            });
+            
             return genericMenu;
         }