Add code kill button

Very useful during troubleshooting, plus helpful for LV teams.

Author: ThadHouse

FRC-Extension/Buttons/KillButton.cs

@@ -0,0 +1,92 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using EnvDTE;
+using Microsoft.VisualStudio.Shell;
+using RobotDotNet.FRC_Extension.RoboRIO_Code;
+using Task = System.Threading.Tasks.Task;
+
+namespace RobotDotNet.FRC_Extension.Buttons
+{
+    public class KillButton : ButtonBase
+    {
+        private bool m_killing = false;
+
+        public KillButton(Frc_ExtensionPackage package)
+            : base(package, false, GuidList.guidFRC_ExtensionCmdSet, (int)PkgCmdIDList.cmdidKillButton)
+        {
+
+        }
+
+        public override async void ButtonCallback(object sender, EventArgs e)
+        {
+            var menuCommand = sender as OleMenuCommand;
+            if (menuCommand == null)
+            {
+                return;
+            }
+            if (!m_killing)
+            {
+                try
+                {
+                    SettingsPageGrid page;
+                    string teamNumber = m_package.GetTeamNumber(out page);
+
+                    if (teamNumber == null) return;
+
+                    DeployManager m = new DeployManager(m_package.PublicGetService(typeof (DTE)) as DTE);
+
+                    var writer = OutputWriter.Instance;
+
+                    menuCommand.Visible = false;
+                    m_killing = true;
+
+                    //Connect to RoboRIO
+                    writer.WriteLine("Attempting to Connect to RoboRIO");
+
+                    Task<bool> rioConnectionTask = m.StartConnectionTask(teamNumber);
+                    Task delayTask = Task.Delay(10000);
+
+                    //Successfully extracted files.
+
+                    writer.WriteLine("Waiting for Connection to Finish");
+                    if (await Task.WhenAny(rioConnectionTask, delayTask) == rioConnectionTask)
+                    {
+                        //Connected
+                        if (rioConnectionTask.Result == true)
+                        {
+                            writer.WriteLine("Killing currently running robot code.");
+                            await RoboRIOConnection.RunCommand(DeployProperties.KillOnlyCommand, ConnectionUser.LvUser);
+                            writer.WriteLine("Done.");
+                            m_killing = false;
+                            menuCommand.Visible = true;
+                        }
+                        else
+                        {
+                            //Did not successfully connect
+                            writer.WriteLine("Failed to Connect to RoboRIO. Exiting.");
+                            m_killing = false;
+                            menuCommand.Visible = true;
+                        }
+                    }
+                    else
+                    {
+                        //Timedout
+                        writer.WriteLine("RoboRIO connection timedout. Exiting.");
+                        m_killing = false;
+                        menuCommand.Visible = true;
+                    }
+                }
+                catch (Exception ex)
+                {
+                    m_output.WriteLine(ex.ToString());
+                    m_killing = false;
+                    menuCommand.Visible = true;
+                    OutputWriter.Instance.WriteLine("Code Kill Failed");
+                }
+            }
+        }
+    }
+}

FRC-Extension/FRC-Extension.csproj

@@ -165,6 +165,7 @@
     <Compile Include="Buttons\DeployDebugButton.cs" />
     <Compile Include="Buttons\DownloadMonoButton.cs" />
     <Compile Include="Buttons\InstallMonoButton.cs" />
+    <Compile Include="Buttons\KillButton.cs" />
     <Compile Include="Buttons\NetConsoleButton.cs" />
     <Compile Include="Buttons\SettingsButton.cs" />
     <Compile Include="MonoCode\MonoDeploy.cs" />

FRC-Extension/FRC-Extension.vsct

@@ -98,7 +98,15 @@
         </Strings>
       </Button>
 
-      <Button guid="guidFRC_ExtensionCmdSet" id="cmdidNetconsole" priority="0x0103" type="Button">
+      <Button guid="guidFRC_ExtensionCmdSet" id="cmdidKillCode" priority="0x0102" type="Button">
+        <Parent guid="guidFRC_ExtensionCmdSet" id="FRCMainMenuGroup"/>
+        <CommandFlag>DynamicVisibility</CommandFlag>
+        <Strings>
+          <ButtonText>Kill Robot Code</ButtonText>
+        </Strings>
+      </Button>
+
+      <Button guid="guidFRC_ExtensionCmdSet" id="cmdidNetconsole" priority="0x0104" type="Button">
         <Parent guid="guidFRC_ExtensionCmdSet" id="FRCMainMenuGroup" />
         <CommandFlag>DynamicVisibility</CommandFlag>
         <CommandFlag>DefaultInvisible</CommandFlag>
@@ -108,7 +116,7 @@
 
       </Button>
 
-      <Button guid="guidFRC_ExtensionCmdSet" id="cmdidSettings" priority="0x0102" type="Button">
+      <Button guid="guidFRC_ExtensionCmdSet" id="cmdidSettings" priority="0x0103" type="Button">
         <Parent guid="guidFRC_ExtensionCmdSet" id="FRCMainMenuGroup" />
         <Strings>
           <ButtonText>Settings</ButtonText>
@@ -181,6 +189,7 @@
       <IDSymbol name="cmdidNetconsole" value ="0x0103"/>
       <IDSymbol name="cmdidSettings" value="0x0102" />
       <IDSymbol name="cmdidDebugCode" value="0x0106"/>
+      <IDSymbol name="cmdidKillCode" value="0x0107"/>
       <IDSymbol name="cmdidAboutButton" value="0x0220"/>
       <IDSymbol name="MonoSubMenu" value="0x1100"/>
       <IDSymbol name="MonoSubMenuGroup" value="0x1150"/>

FRC-Extension/FRC-ExtensionPackage.cs

@@ -74,6 +74,8 @@ internal DialogPage PublicGetDialogPage(Type dialogPageType)
         private DeployDebugButton m_deployButton;
         private DeployDebugButton m_debugButton;
 
+        private KillButton m_killButton;
+
         //private MonoButtons m_monoButtons;
         private DownloadMonoButton m_downloadMonoButton;
         private InstallMonoButton m_installMonoButton;
@@ -100,6 +102,8 @@ protected override void Initialize()
 
             m_deployButton = new DeployDebugButton(this, (int)PkgCmdIDList.cmdidDeployCode, false);
             m_debugButton = new DeployDebugButton(this, (int)PkgCmdIDList.cmdidDebugCode, true);
+
+            m_killButton = new KillButton(this);
 
 
             string monoFolder = WPILibFolderStructure.CreateMonoFolder();

FRC-Extension/PkgCmdID.cs

@@ -21,5 +21,7 @@ static class PkgCmdIDList
 
         public const uint cmdidAboutButton = 0x0220;
 
+        public const uint cmdidKillButton = 0x107;
+
     };
 }

FRC-Extension/RoboRIOCode/DeployProperties.cs

@@ -16,6 +16,8 @@ public static class DeployProperties
             ". /etc/profile.d/natinst-path.sh; /usr/local/frc/bin/frcKillRobot.sh -t -r",
         };
 
+        public const string KillOnlyCommand = "/usr/local/frc/bin/frcKillRobot.sh";
+
         public const string DebugFlagDir = "/tmp/";
 
         public static readonly string[] DebugFlagCommand =