detecting correct connection from object explorer context

Author: dzsquared

package.json

@@ -16,25 +16,11 @@
         "url": "https://github.com/dzsquared/sqlops-firstresponderkit.git"
     },
     "activationEvents": [
-        "onCommand:extension.sp_blitzversion",
-        "onCommand:extension.sp_blitz",
-        "onCommand:extension.run_sp_blitz",
-        "onCommand:extension.sp_blitzwho",
-        "onCommand:extension.sp_blitzindex",
-        "onCommand:extension.run_sp_blitzindex",
-        "onCommand:extension.sp_blitzcache",
-        "onCommand:extension.run_sp_blitzcache",
-        "onCommand:extension.sp_blitzfirst",
-        "onCommand:extension.run_sp_blitzfirst",
         "*"
     ],
     "main": "./out/extension",
     "contributes": {
         "commands": [
-            {
-                "command": "extension.OEdiagnostics",
-                "title": "First Responder Kit: Node Info"
-            },
             {
                 "command": "extension.sp_blitzversion",
                 "title": "First Responder Kit: Check sp_blitz Version"
@@ -88,11 +74,6 @@
         ],
         "menus": {
             "objectExplorer/item/context": [
-                {
-                  "command": "extension.OEdiagnostics",
-                  "when": "connectionProvider == MSSQL",
-                  "group": "FRK"
-                },
                 {
                   "command": "extension.sp_blitzversion",
                   "when": "connectionProvider == MSSQL && nodeType && nodeType == Server",

snippets/snippets.json

@@ -10,7 +10,8 @@
             "@CheckProcedureCacheFilter = NULL,",
             "@CheckServerInfo = 1",
             "-- uncomment the following line to write results to an output table",
-            "--, @OutputDatabaseName = 'DBAtools', @OutputSchemaName = 'dbo', @OutputTableName = 'BlitzResults'"
+            "--, @OutputDatabaseName = 'DBAtools', @OutputSchemaName = 'dbo', @OutputTableName = 'BlitzResults'",
+            "-- for more info: https://www.brentozar.com/blitz/"
         ],
         "description": "sp_blitz execution script"
     },
@@ -21,7 +22,8 @@
            "@DatabaseName = '${1:DatabaseName}',",
            "--@TableName = '${2:TableName}'",
             "--@Mode = 4",
-            "--0=Diagnose, 1=Summarize, 2=Index Usage Detail, 3=Missing Index Detail, 4=Diagnose Details"
+            "--0=Diagnose, 1=Summarize, 2=Index Usage Detail, 3=Missing Index Detail, 4=Diagnose Details",
+            "-- for more info: https://www.brentozar.com/blitzindex/"
         ],
         "description": "sp_blitzindex execution script"
     },
@@ -31,7 +33,8 @@
             "EXEC [dbo].[sp_BlitzCache]",
             "@SortOrder = 'reads',",
             "    -- CPU, executions, xpm, recent compilations, memory grant, writes, all",
-            "@Top = 10"
+            "@Top = 10",
+            "-- for more info: https://www.brentozar.com/blitzcache/"
         ],
         "description": "sp_blitzcache execution script"
     },
@@ -41,7 +44,8 @@
             "EXEC [dbo].[sp_BlitzFirst]",
             "@Seconds = 5,",
             "@ShowSleepingSPIDs = 0,",
-            "@ExpertMode = 0 --1 will also run sp_BlitzWho"
+            "@ExpertMode = 0 --1 will also run sp_BlitzWho",
+            "-- for more info: https://www.brentozar.com/askbrent/"
         ],
         "description": "sp_blitzfirst execution script"
     }

src/blitzindex.ts

@@ -5,12 +5,10 @@ import * as sqlops from 'sqlops';
 import { error } from 'util';
 import {placeScript} from './placescript';
 import {updatecheck} from './updateCheck';
-// import {BlitzIndex} from './blitzindex';
 
 export function activate(context: vscode.ExtensionContext) {
     const baseUrl = "https://raw.githubusercontent.com/BrentOzarULTD/SQL-Server-First-Responder-Kit/master/";
 
-    
     // checking spblitz versioning
     var oediag = async (context: sqlops.ObjectExplorerContext) => {
         vscode.window.showInformationMessage(context.isConnectionNode.toString());
@@ -25,15 +23,13 @@ export function activate(context: vscode.ExtensionContext) {
         var amIUPD = new updatecheck();
         let updateReturn = await amIUPD.checkForUpdates(context);
 
-        //vscode.window.showInformationMessage(updateReturn);
         if (updateReturn == 'update') {
             getblitzall();
         } else if (updateReturn != '') {
             let versionURL = 'https://github.com/BrentOzarULTD/SQL-Server-First-Responder-Kit/releases/tag/' + updateReturn;
             vscode.commands.executeCommand('vscode.open', vscode.Uri.parse(versionURL));
         } else {
             // do nothing
-            vscode.window.showErrorMessage("Sorry about this. Please report the issue on Github.");
         }
     };
     var disposable_spblitzversion = vscode.commands.registerCommand('extension.sp_blitzversion', getblitzversion);
@@ -48,63 +44,58 @@ export function activate(context: vscode.ExtensionContext) {
         };
         console.log('Bringing in the first responder kit from the mothership.');
         const scriptText = await request.get(options);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_spblitzall = vscode.commands.registerCommand('extension.sp_blitzall', getblitzall);
     context.subscriptions.push(disposable_spblitzall);
 
-    //importing the original spblitz script
+    //importing spblitz script
     var getblitz = async () => {
         let fileName = "sp_Blitz.sql";
         var options = {
             uri: baseUrl + fileName,
         };
         console.log('Bringing in the first responder kit from the mothership.');
         const scriptText = await request.get(options);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_spblitz = vscode.commands.registerCommand('extension.sp_blitz', getblitz);
     context.subscriptions.push(disposable_spblitz);
 
-    //importing the full spblitzcache script
+    //importing spblitzcache script
     var getblitzcache = async () => {
         let fileName = "sp_BlitzCache.sql";
         var options = {
             uri: baseUrl + fileName,
         };
         console.log('Bringing in the first responder kit from the mothership.');
         const scriptText = await request.get(options);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_spblitzcache = vscode.commands.registerCommand('extension.sp_blitzcache', getblitzcache);
     context.subscriptions.push(disposable_spblitzcache);
 
-    //importing the full spblitzfirst script
+    //importing spblitzfirst script
     var getblitzfirst = async () => {
         let fileName = "sp_BlitzFirst.sql";
         var options = {
             uri: baseUrl + fileName,
         };
         console.log('Bringing in the first responder kit from the mothership.');
         const scriptText = await request.get(options);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);   
     };
     var disposable_spblitzfirst = vscode.commands.registerCommand('extension.sp_blitzfirst', getblitzfirst);
     context.subscriptions.push(disposable_spblitzfirst);
 
-    //importing the spblitzwho script
+    //importing spblitzwho script
     var getblitzwho = async () => {
         let fileName = "sp_BlitzWho.sql";
         var options = {
             uri: baseUrl + fileName,
         };
         console.log('Bringing in the first responder kit from the mothership.');
         const scriptText = await request.get(options);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_spblitzwho = vscode.commands.registerCommand('extension.sp_blitzwho', getblitzwho);
@@ -127,27 +118,25 @@ export function activate(context: vscode.ExtensionContext) {
         -- for more info: https://www.brentozar.com/blitz/
         `;
         var setting: vscode.Uri = vscode.Uri.parse("untitled:" + fileName);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_runspblitz = vscode.commands.registerCommand('extension.run_sp_blitz', runspblitz);
     context.subscriptions.push(disposable_runspblitz);
 
-    //importing the full spblitz script
+    //importing spblitzindex script
     var getblitzindex = async () => {
         let fileName = "sp_BlitzIndex.sql";
         var options = {
             uri: baseUrl + fileName,
         };
         console.log('Bringing in the first responder kit from the mothership.');
         const scriptText = await request.get(options);
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_spblitzindex = vscode.commands.registerCommand('extension.sp_blitzindex', getblitzindex);
     context.subscriptions.push(disposable_spblitzindex);
 
-    //creating the quickrun script
+    //creating the quickrun script for blitzindex
     var runspblitzindex = async (context?: sqlops.ObjectExplorerContext) => {
         console.log('Preparing sample run script.');
         let fileName = "exec_sp_blitzindex.sql";
@@ -212,13 +201,12 @@ export function activate(context: vscode.ExtensionContext) {
 
         -- for more info: https://www.brentozar.com/blitzcache/
         `;
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_runspblitzcache = vscode.commands.registerCommand('extension.run_sp_blitzcache',runspblitzcache)
     context.subscriptions.push(disposable_runspblitzcache);
 
-    //creating the quickrun script
+    //creating the quickrun script for blitzfirst
     var runspblitzfirst = async () => {
         let fileName = "exec_sp_blitzfirst.sql";
         console.log('Preparing sample run script.');
@@ -229,16 +217,13 @@ export function activate(context: vscode.ExtensionContext) {
 
         -- for more info: https://www.brentozar.com/askbrent/
         `;
-        //await placeScript.placescript(fileName, scriptText);
         new placeScript().placescript(fileName,scriptText);
     };
     var disposable_runspblitzfirst = vscode.commands.registerCommand('extension.run_sp_blitzfirst',runspblitzfirst);
     context.subscriptions.push(disposable_runspblitzfirst);
 
 }
 
-
-// this method is called when your extension is deactivated
 export function deactivate() {
 }
 

src/extension.ts

@@ -4,7 +4,6 @@ import * as sqlops from 'sqlops';
 import * as vscode from 'vscode';
 import * as request from 'request-promise-native';
 import * as apiConfig from  './apiconfig';
-import { STATUS_CODES } from 'http';
 let apiconfig: apiConfig.apiconfig = require('../apiconfig.json');
 
 export class updatecheck {
@@ -15,21 +14,21 @@ export class updatecheck {
         let queryProvider = sqlops.dataprotocol.getProvider<sqlops.QueryProvider>(context.connectionProfile.providerName, sqlops.DataProviderType.QueryProvider);
         vscode.window.showInformationMessage("Checking First Responder Kit for Updates");
         try {
-            let connection = await sqlops.connection.getCurrentConnection();
+            let connection = context.connectionProfile;
             let query = `declare @versionno datetime
                 IF OBJECT_ID('dbo.sp_Blitz') IS NULL
                 set @versionno = '1/1/1900'
                 ELSE
                 exec sp_blitz @help = 1, @versiondate = @versionno output
                 select convert(varchar(10),@versionno,112) as versionno`;
 
-            let connectionUri = await sqlops.connection.getUriForConnection(connection.connectionId);
+            let connectionUri = await sqlops.connection.getUriForConnection(connection.id);
             if (connection) {
                 let results = await queryProvider.runQueryAndReturn(connectionUri, query);
                 let cell = results.rows[0][0];
                 let currentVersion = cell.displayValue;
 
-                //get live most recent version
+                //get live most recent version from github
                 var options = {
                     uri: baseUrl,
                     headers: {
@@ -40,14 +39,27 @@ export class updatecheck {
                     simple: false
                 };
                 var scriptText = await request.get(options);
-
                 let recentVersion = scriptText.tag_name;
+
                 //compare against db version
-                if (recentVersion >  currentVersion) {
+                if (currentVersion == '19000101') {
+                    let updateMsg = 'First Responder Kit not detected on this server. Current version of First Responder Kit is ' + recentVersion + '.'
+                    var buttonName = await vscode.window.showInformationMessage(updateMsg, {modal:false}, "Get It Now", "Tell Me More");
+                    if (buttonName){
+                        if (buttonName == "Get It Now") {
+                            return 'update';
+                        } else if (buttonName == "Tell Me More") {
+                            return recentVersion;
+                        }
+                    } else {
+                        return '';
+                    }
+                }
+                else if (recentVersion >  currentVersion) {
                     let updateMsg = 'New Version of First Responder Kit available (' + recentVersion + ').  You have version ' + currentVersion +'.'
-                    var buttonName = await vscode.window.showInformationMessage(updateMsg, {modal:false}, "Get It", "Tell Me More");
+                    var buttonName = await vscode.window.showInformationMessage(updateMsg, {modal:false}, "Get It Now", "Tell Me More");
                     if (buttonName){
-                        if (buttonName == "Get It") {
+                        if (buttonName == "Get It Now") {
                             return 'update';
                         } else if (buttonName == "Tell Me More") {
                             return recentVersion;