Creating a context menu for local and remote branch nodes

Author: StanleyGoldman

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

@@ -305,9 +305,9 @@ private void OnTreeGUI(Rect rect)
                             }).Start();
                     }
                 },
-                node =>
-                {
-                    Debug.Log("Right Click");
+                node => {
+                    GenericMenu menu = CreateContextMenuForLocalBranchNode(node);
+                    menu.ShowAsContext();
                 });
 
             if (treeHadFocus && treeLocals.SelectedNode == null)
@@ -364,9 +364,9 @@ private void OnTreeGUI(Rect rect)
                         }
                     }
                 },
-                node =>
-                {
-                    Debug.Log("Right Click");
+                node => {
+                    GenericMenu menu = CreateContextMenuForRemoteBranchNode(node);
+                    menu.ShowAsContext();
                 });
 
             if (treeHadFocus && treeRemotes.SelectedNode == null)
@@ -385,6 +385,39 @@ private void OnTreeGUI(Rect rect)
             GUILayout.Space(rect.y - initialRect.y);
         }
 
+        private GenericMenu CreateContextMenuForLocalBranchNode(TreeNode node)
+        {
+            var genericMenu = new GenericMenu();
+
+            var deleteGuiContent = new GUIContent("Delete");
+            var switchGuiContent = new GUIContent("Switch");
+
+            if (node.IsActive)
+            {
+                genericMenu.AddDisabledItem(deleteGuiContent);
+                genericMenu.AddDisabledItem(switchGuiContent);
+            }
+            else
+            {
+                genericMenu.AddItem(deleteGuiContent, false, (userData) => {
+                    Debug.Log("Delete Branch");
+                }, node);
+
+                genericMenu.AddItem(switchGuiContent, false, (userData) => {
+                    Debug.Log("Switch Branch");
+                }, node);
+            }
+
+            return genericMenu;
+        }
+
+        private GenericMenu CreateContextMenuForRemoteBranchNode(TreeNode node)
+        {
+            var genericMenu = new GenericMenu();
+
+            return genericMenu;
+        }
+
         private int CompareBranches(GitBranch a, GitBranch b)
         {
             //if (IsFavorite(a.Name))