option to get release notes or download new FRK

Author: dzsquared

package.json

@@ -31,10 +31,18 @@
     "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"
             },
+            {
+                "command": "extension.sp_blitzall",
+                "title": "First Responder Kit: Import sp_blitz and all its friends"
+            },
             {
                 "command": "extension.sp_blitz",
                 "title": "First Responder Kit: Import sp_blitz"
@@ -81,13 +89,13 @@
         "menus": {
             "objectExplorer/item/context": [
                 {
-                  "command": "extension.sp_blitzversion",
-                  "when": "connectionProvider == MSSQL && nodeType && nodeType == Database",
+                  "command": "extension.OEdiagnostics",
+                  "when": "connectionProvider == MSSQL",
                   "group": "FRK"
                 },
                 {
                   "command": "extension.sp_blitzversion",
-                  "when": "connectionProvider == MSSQL && nodeType && nodeType == Connection",
+                  "when": "connectionProvider == MSSQL",
                   "group": "FRK"
                 },
               {

src/extension.ts

@@ -11,14 +11,54 @@ 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());
+        vscode.window.showInformationMessage(context.nodeInfo.label);
+        vscode.window.showInformationMessage(context.nodeInfo.nodePath);
+    };
+    var disposable_oediag = vscode.commands.registerCommand('extension.OEdiagnostics', oediag);
+    context.subscriptions.push(disposable_oediag);
+    
     // checking spblitz versioning
     var getblitzversion = async (context: sqlops.ObjectExplorerContext) => {
-        new updatecheck().checkForUpdates(context);
+        var amIUPD = new updatecheck();
+        let updateReturn = await amIUPD.checkForUpdates(context);
+
+        // updateChoice.then(function(updateReturn) {
+        // if (updateReturn) {
+            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.showInformationMessage("nothing came back");
+            }
+        // }
+        // });
     };
     var disposable_spblitzversion = vscode.commands.registerCommand('extension.sp_blitzversion', getblitzversion);
     context.subscriptions.push(disposable_spblitzversion);
 
-    //importing the full spblitz script
+    
+    //importing all first responder kit scripts
+    var getblitzall = async () => {
+        let fileName = "Install-All-Scripts.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_spblitzall = vscode.commands.registerCommand('extension.sp_blitzall', getblitzall);
+    context.subscriptions.push(disposable_spblitzall);
+
+    //importing the original spblitz script
     var getblitz = async () => {
         let fileName = "sp_Blitz.sql";
         var options = {

src/updateCheck.ts

@@ -4,10 +4,11 @@ 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 {
-    public async checkForUpdates(context: sqlops.ObjectExplorerContext) {
+    public async checkForUpdates(context: sqlops.ObjectExplorerContext): Promise<string> {
         let baseUrl = "https://api.github.com/repos/BrentOzarULTD/SQL-Server-First-Responder-Kit/releases/latest";
         // /repos/:owner/:repo/releases/latest
         let apitoken = 'token ' + apiconfig.token;
@@ -41,17 +42,31 @@ export class updatecheck {
                 let recentVersion = scriptText.tag_name;
                 //compare against db version
                 if (recentVersion >  currentVersion) {
-                    vscode.window.showInformationMessage("New Version of First Responder Kit available.", {modal:false}, "Get It", "Tell Me More");
-                
-                    //tell me more : https://github.com/BrentOzarULTD/SQL-Server-First-Responder-Kit/releases/tag/<tag>
-                
-                    //get it : download all command
-                    
+                    // var whatClicked =
+                    var buttonName = await vscode.window.showInformationMessage("New Version of First Responder Kit available.", {modal:false}, "Get It", "Tell Me More");
+
+                    // whatClicked.then(function(buttonName) {
+                        vscode.window.showInformationMessage(buttonName);
+                        if (buttonName){
+                            if (buttonName == "Get It") {
+                                return 'update';
+                            } else if (buttonName == "Tell Me More") {
+                                return recentVersion;
+                            }
+                        } else {
+                            return '';
+                        }
+                    // });
+
                 } else {
-                    vscode.window.showInformationMessage("You're up to date!", {modal:false}, "Close")
+                    vscode.window.showInformationMessage("You're up to date!", {modal:false}, "Close");
+                    return '';
                 }
 
                 //confirm if user wants to update scripts
+            } else {
+                vscode.window.showErrorMessage("Please connect to SQL server to check First Responder Kit version.");
+                return '';
             }
         } catch (e) {
             vscode.window.showErrorMessage(e);