Updates the download button to offer to install immediately.

Author: ThadHouse

FRC-Extension/Buttons/DownloadMonoButton.cs

@@ -6,19 +6,20 @@
 using Microsoft.VisualStudio.Shell;
 using Microsoft.VisualStudio.Shell.Interop;
 using RobotDotNet.FRC_Extension.MonoCode;
-using RobotDotNet.FRC_Extension.WPILibFolder;
 
 namespace RobotDotNet.FRC_Extension.Buttons
 {
     public class DownloadMonoButton : ButtonBase
     {
         private readonly MonoFile m_monoFile;
         private bool m_downloading = false;
+        private readonly InstallMonoButton m_installButton;
 
-        public DownloadMonoButton(Frc_ExtensionPackage package, MonoFile monoFile)
-            : base(package, false, GuidList.guidFRC_ExtensionCmdSet, (int) PkgCmdIDList.cmdidDownloadMono)
+        public DownloadMonoButton(Frc_ExtensionPackage package, MonoFile monoFile, InstallMonoButton installButton)
+            : base(package, false, GuidList.guidFRC_ExtensionCmdSet, (int)PkgCmdIDList.cmdidDownloadMono)
         {
             m_monoFile = monoFile;
+            m_installButton = installButton;
         }
 
 
@@ -62,18 +63,23 @@ public override async void ButtonCallback(object sender, EventArgs e)
                             IVsUIShell uiShell = (IVsUIShell)m_package.PublicGetService(typeof(SVsUIShell));
                             Guid clsid = Guid.Empty;
                             int result;
-                            Microsoft.VisualStudio.ErrorHandler.ThrowOnFailure(uiShell.ShowMessageBox(
+                            uiShell.ShowMessageBox(
                                 0,
                                 ref clsid,
-                                "Mono Successfully Downloaded",
+                                "Mono Successfully Downloaded. Would you like to install it to the RoboRIO?",
                                 string.Format(CultureInfo.CurrentCulture, "", this.ToString()),
                                 string.Empty,
                                 0,
-                                OLEMSGBUTTON.OLEMSGBUTTON_OK,
+                                OLEMSGBUTTON.OLEMSGBUTTON_YESNO,
                                 OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_FIRST,
                                 OLEMSGICON.OLEMSGICON_INFO,
                                 0, // false
-                                out result));
+                                out result);
+                            if (result == 6)
+                            {
+                                //Install Mono.
+                                await m_installButton.DeployMono(m_installButton.GetMenuCommand());
+                            }
                         }
                         else
                         {

FRC-Extension/Buttons/InstallMonoButton.cs

@@ -19,6 +19,11 @@ public InstallMonoButton(Frc_ExtensionPackage package, MonoFile monoFile)
             m_monoFile = monoFile;
         }
 
+        internal OleMenuCommand GetMenuCommand()
+        {
+            return m_oleMenuItem;
+        }
+
         public override async void ButtonCallback(object sender, EventArgs e)
         {
             var menuCommand = sender as OleMenuCommand;
@@ -59,11 +64,7 @@ public override async void ButtonCallback(object sender, EventArgs e)
                             if (properFileExists)
                             {
                                 //We can deploy
-                                m_installing = true;
-                                menuCommand.Visible = false;
                                 await DeployMono(menuCommand);
-                                m_installing = false;
-                                menuCommand.Visible = true;
                             }
                             else
                             {
@@ -86,7 +87,7 @@ public override async void ButtonCallback(object sender, EventArgs e)
                 }
             }
         }
-        private async Task DeployMono(OleMenuCommand menuCommand)
+        internal async Task DeployMono(OleMenuCommand menuCommand)
         {
             try
             {

FRC-Extension/FRC-ExtensionPackage.cs

@@ -100,16 +100,17 @@ protected override void Initialize()
 
             m_deployButton = new DeployDebugButton(this, (int)PkgCmdIDList.cmdidDeployCode, false);
             m_debugButton = new DeployDebugButton(this, (int)PkgCmdIDList.cmdidDebugCode, true);
+            
 
             string monoFolder = WPILibFolderStructure.CreateMonoFolder();
 
             string monoFile = monoFolder + Path.DirectorySeparatorChar + DeployProperties.MonoVersion;
 
             m_monoFile = new MonoFile(monoFile);
 
-            m_downloadMonoButton = new DownloadMonoButton(this, m_monoFile);
             m_installMonoButton = new InstallMonoButton(this, m_monoFile);
-
+            m_downloadMonoButton = new DownloadMonoButton(this, m_monoFile, m_installMonoButton);
+            
 
             m_aboutButton = new AboutButton(this);
             m_netConsoleButton = new NetConsoleButton(this);