Add Execute functionality to File Context Menu

Save temporary opened SQL Script

Author: joaoribe

src/SQLScriptsExplorer.Addin/Controls/FileExplorerTreeView.xaml

@@ -46,7 +46,12 @@
  <Image Source="../Resources/Images/OpenFile.png"/>
  </MenuItem.Icon>
  </MenuItem>
- <MenuItem Header="Edit" Click="mnuEditFile_Click"/>
+ <MenuItem Header="Edit" Click="mnuEditFile_Click" />
+ <MenuItem Header="Execute" Click="mnuExecuteFile_Click">
+ <MenuItem.Icon>
+ <Image Source="../Resources/Images/Run.png"/>
+ </MenuItem.Icon>
+ </MenuItem>
  <Separator/>
  <MenuItem Header="Open Containing Folder" Click="mnuOpenFolder_Click">
  <MenuItem.Icon>

src/SQLScriptsExplorer.Addin/Controls/FileExplorerTreeView.xaml.cs

@@ -169,6 +169,23 @@ private void mnuEditFile_Click(object sender, RoutedEventArgs e)
  }
  }
 
+ private void mnuExecuteFile_Click(object sender, RoutedEventArgs e)
+ {
+ var treeNode = TreeViewMain.SelectedItem as TreeNode;
+
+ if (treeNode != null && treeNode.Type == TreeNodeType.File && !isEditMode)
+ {
+ try
+ {
+ DocumentManager.ExecuteTemplate(treeNode.FileName, treeNode.FileFullPath);
+ }
+ catch (Exception ex)
+ {
+ MessageBox.Show(ex.Message);
+ }
+ }
+ }
+
  private void mnuRename_Click(object sender, RoutedEventArgs e)
  {
  // Get the TextBox control as defined in TreeView.ItemTemplate

src/SQLScriptsExplorer.Addin/Infrastructure/DocumentManager.cs

@@ -24,12 +24,14 @@ public static void OpenTemplate(string fileName, string fileFullPath)
 
  DTE dte = Package.GetGlobalService(typeof(DTE)) as DTE;
  var fileDocument = dte.ItemOperations.NewFile(@"General\Text File", fileName).Document;
-
+ 
  TextSelection textSelection = fileDocument.Selection as TextSelection;
  textSelection.SelectAll();
  textSelection.Text = string.Empty;
  textSelection.Insert(fileContent);
  textSelection.StartOfDocument();
+
+ fileDocument.Save();
  }
  else
  {
@@ -64,6 +66,30 @@ public static void EditTemplate(string fileName, string fileFullPath)
  }
  }
 
+ public static void ExecuteTemplate(string fileName, string fileFullPath)
+ {
+ string CMD_QUERY_EXECUTE = "Query.Execute";
+
+ try
+ {
+ ThreadHelper.ThrowIfNotOnUIThread();
+
+ OpenTemplate(fileName, fileFullPath);
+
+ DTE dte = Package.GetGlobalService(typeof(DTE)) as DTE;
+
+ // Ensure the document we are executing is the document we have opened by checking its name
+ if (dte.ActiveDocument != null && dte.ActiveDocument.ProjectItem.Name.Equals(fileName))
+ {
+ dte.ExecuteCommand(CMD_QUERY_EXECUTE);
+ }
+ }
+ catch (Exception ex)
+ {
+ throw ex;
+ }
+ }
+
  public static void FormatSelection()
  {
  try

src/SQLScriptsExplorer.Addin/Resources/Images/Run.png

@@ -241,6 +241,7 @@
  <Resource Include="Resources\Images\Settings.png" />
  <Resource Include="Resources\Images\Success.png" />
  <Resource Include="Resources\Images\SQLScriptsManager.png" />
+ <Resource Include="Resources\Images\Run.png" />
  <Content Include="Resources\Images\ToolbarCommand.png" />
  <Content Include="Resources\Images\ToolbarCommandPackage.ico" />
  <Content Include="Resources\Images\ToolWindow1Package.ico" />