create execution script for blitzindex against correct connection from object explorer

Author: dzsquared

src/extension.ts

@@ -183,7 +183,7 @@ export function activate(context: vscode.ExtensionContext) {
         scriptText += `
         -- for more info: https://www.brentozar.com/blitzindex/
         `;
-        new placeScript().placescript(fileName,scriptText);
+        new placeScript().placescript(fileName,scriptText,context);
 
     };
     var disposable_runspblitzindex = vscode.commands.registerCommand('extension.run_sp_blitzindex', runspblitzindex);

src/placescript.ts

@@ -5,17 +5,24 @@ import * as vscode from 'vscode';
 
 export class placeScript {
       // places scriptText into fileName editor with current connection
-      public async placescript(fileName, scriptText) {
+      public async placescript(fileName, scriptText, context?: sqlops.ObjectExplorerContext) {
         var setting: vscode.Uri = vscode.Uri.parse("untitled:" + fileName);
         try {
-            let connection = await sqlops.connection.getCurrentConnection();
+            var connectId;
+            if (context) {
+                let connection = context.connectionProfile;
+                connectId = connection.id;
+            } else {
+                let connection = await sqlops.connection.getCurrentConnection();
+                connectId = connection.connectionId;
+            }
             let doc = await vscode.workspace.openTextDocument(setting);
             let editor = await vscode.window.showTextDocument(doc, 1, false);
             editor.edit(edit => {
                 edit.insert(new vscode.Position(0, 0), scriptText);
             });
-            if (connection) {
-                await sqlops.queryeditor.connect(doc.uri.toString(), connection.connectionId);
+            if (connectId) {
+                await sqlops.queryeditor.connect(doc.uri.toString(), connectId);
             }
         } catch (err) {
             vscode.window.showErrorMessage(err);