Adds support for a button to explicitly set the robot project

Author: ThadHouse

FRC-Extension/Buttons/DeployDebugButton.cs

@@ -125,6 +125,10 @@ public override void QueryCallback(object sender, EventArgs e)
                     {
                         if (project.Globals.VariableExists["RobotProject"])
                         {
+                            if (project.Globals["RobotProject"].ToString() != "yes")
+                            {
+                                continue;
+                            }
                             var vsproject = project.Object as VSLangProj.VSProject;
 
                             if (vsproject != null)

FRC-Extension/Buttons/SetMainRobotButton.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+using System.Runtime.InteropServices;
 using System.Text;
 using System.Threading.Tasks;
 using EnvDTE;
@@ -24,14 +25,128 @@ public override void QueryCallback(object sender, EventArgs e)
             var menuCommand = sender as OleMenuCommand;
             if (menuCommand != null)
             {
-                menuCommand.Visible = false;
-                menuCommand.Enabled = false;
+                if (GetSelectedProject() != null)
+                {
+                    menuCommand.Visible = true;
+                    menuCommand.Enabled = true;
+                }
+                else
+                {
+                    menuCommand.Visible = false;
+                    menuCommand.Enabled = false;
+                }               
             }
         }
 
+        public Project GetSelectedProject()
+        {
+            IntPtr hierarchyPointer = IntPtr.Zero;
+            IntPtr selectionContainerPointer = IntPtr.Zero;
+
+            try
+            {
+
+
+                Object selectedObject = null;
+                IVsMultiItemSelect multiItemSelect;
+                uint projectItemId;
+
+                IVsMonitorSelection monitorSelection =
+                    (IVsMonitorSelection)Package.GetGlobalService(
+                        typeof(SVsShellMonitorSelection));
+
+                monitorSelection.GetCurrentSelection(out hierarchyPointer,
+                    out projectItemId,
+                    out multiItemSelect,
+                    out selectionContainerPointer);
+
+                IVsHierarchy selectedHierarchy = Marshal.GetTypedObjectForIUnknown(
+                    hierarchyPointer,
+                    typeof(IVsHierarchy)) as IVsHierarchy;
+
+                if (selectedHierarchy != null)
+                {
+                    if (ErrorHandler.Failed(selectedHierarchy.GetProperty(
+                        projectItemId,
+                        (int)__VSHPROPID.VSHPROPID_ExtObject,
+                        out selectedObject)))
+                    {
+                        return null;
+                    }
+                }
+
+                Project project = selectedObject as Project;
+
+                var vsproject = project?.Object as VSProject;
+
+                if (vsproject != null)
+                {
+                    //If we are an assembly, and its named WPILib, enable the deploy
+                    if (
+                        (from Reference reference in vsproject.References
+                         where reference.SourceProject == null
+                         select reference.Name).Any(name => name.Contains("WPILib")))
+                    {
+                        return project;
+                    }
+                }
+
+                return null;
+            }
+            finally
+            {
+                if (selectionContainerPointer != IntPtr.Zero)
+                {
+                    Marshal.Release(selectionContainerPointer);
+                }
+
+                if (hierarchyPointer != IntPtr.Zero)
+                {
+                    Marshal.Release(hierarchyPointer);
+                }
+            }
+        }
+
+
         public override void ButtonCallback(object sender, EventArgs e)
         {
-            OutputWriter.Instance.WriteLine("Set Pressed");
+            Project project = GetSelectedProject();
+
+            if (project == null)
+            {
+                return;
+            }
+
+            ClearAllProjectsInSolution();
+
+            project.Globals["RobotProject"] = "yes";
+            project.Globals.VariablePersists["RobotProject"] = true;
+            project.Save();
+        }
+
+        private void ClearAllProjectsInSolution()
+        {
+            var dte = m_package.PublicGetService(typeof(DTE)) as DTE;
+            var sb = dte?.Solution;
+            if (sb == null)
+            {
+                return;
+            }
+
+
+            foreach (Project p in sb.Projects)
+            {
+                var globals = p.Globals;
+                if (globals == null)
+                {
+                    continue;
+                }
+                if (globals.VariableExists["RobotProject"])
+                {
+                    globals["RobotProject"] = "no";
+                }
+                p.Save();
+            }
         }
     }
 }

FRC-Extension/FRC-Extension.vsct

@@ -89,7 +89,7 @@
         <CommandFlag>DynamicVisibility</CommandFlag>
         <CommandFlag>DefaultInvisible</CommandFlag>
         <Strings>
-          <ButtonText>Set Main Robot Project</ButtonText>
+          <ButtonText>Set As Main Robot Project</ButtonText>
           <CommandName>cmdidSetRobotProject</CommandName>
         </Strings>
       </Button>