checking for updates against current spblitz

Author: dzsquared

.gitignore

@@ -2,3 +2,5 @@ out
 node_modules
 .vscode-test/
 *.vsix
+secrets.txt
+apiconfig.json

package.json

@@ -16,6 +16,7 @@
         "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",
@@ -30,6 +31,10 @@
     "main": "./out/extension",
     "contributes": {
         "commands": [
+            {
+                "command": "extension.sp_blitzversion",
+                "title": "First Responder Kit: Check sp_blitz Version"
+            },
             {
                 "command": "extension.sp_blitz",
                 "title": "First Responder Kit: Import sp_blitz"
@@ -75,6 +80,16 @@
         ],
         "menus": {
             "objectExplorer/item/context": [
+                {
+                  "command": "extension.sp_blitzversion",
+                  "when": "connectionProvider == MSSQL && nodeType && nodeType == Database",
+                  "group": "FRK"
+                },
+                {
+                  "command": "extension.sp_blitzversion",
+                  "when": "connectionProvider == MSSQL && nodeType && nodeType == Connection",
+                  "group": "FRK"
+                },
               {
                 "command": "extension.run_sp_blitzindex",
                 "when": "connectionProvider == MSSQL && nodeType && nodeType == Database",

src/apiconfig.ts

@@ -0,0 +1,4 @@
+export interface apiconfig
+{
+    token: string;   
+}

src/extension.ts

@@ -4,11 +4,20 @@ import * as request from 'request-promise-native';
 import * as sqlops from 'sqlops';
 import { error } from 'util';
 import {placeScript} from './placescript';
-import {BlitzIndex} from './blitzindex';
+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 getblitzversion = async (context: sqlops.ObjectExplorerContext) => {
+        new updatecheck().checkForUpdates(context);
+    };
+    var disposable_spblitzversion = vscode.commands.registerCommand('extension.sp_blitzversion', getblitzversion);
+    context.subscriptions.push(disposable_spblitzversion);
+
     //importing the full spblitz script
     var getblitz = async () => {
         let fileName = "sp_Blitz.sql";
@@ -155,6 +164,7 @@ export function activate(context: vscode.ExtensionContext) {
     var disposable_runspblitzindex = vscode.commands.registerCommand('extension.run_sp_blitzindex', runspblitzindex);
     context.subscriptions.push(disposable_runspblitzindex);
 
+
     //creating the quickrun script for blitzcache
     var runspblitzcache = async () => {
         let fileName = "exec_sp_blitzcache.sql";

src/updateCheck.ts

@@ -0,0 +1,60 @@
+'use strict';
+
+import * as sqlops from 'sqlops';
+import * as vscode from 'vscode';
+import * as request from 'request-promise-native';
+import * as apiConfig from  './apiconfig';
+let apiconfig: apiConfig.apiconfig = require('../apiconfig.json');
+
+export class updatecheck {
+    public async checkForUpdates(context: sqlops.ObjectExplorerContext) {
+        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;
+        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 query = `declare @versionno datetime
+            exec sp_blitz @help = 1, @versiondate = @versionno output
+            select convert(varchar(10),@versionno,112) as versionno`;
+            let connectionUri = await sqlops.connection.getUriForConnection(connection.connectionId);
+            if (connection) {
+                let results = await queryProvider.runQueryAndReturn(connectionUri, query);
+                //vscode.window.showInformationMessage(results.rowCount.toString());
+                let cell = results.rows[0][0];
+                let currentVersion = cell.displayValue;
+
+                //get live most recent version
+                var options = {
+                    uri: baseUrl,
+                    headers: {
+                        'Authorization': apitoken,
+                        'User-Agent': 'Request-Promise'
+                    },
+                    json: true,
+                    simple: false
+                };
+                var scriptText = await request.get(options);
+                //vscode.window.showInformationMessage("newest version: " + scriptText.tag_name);
+
+                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
+                    
+                } else {
+                    vscode.window.showInformationMessage("You're up to date!", {modal:false}, "Close")
+                }
+                
+                //confirm if user wants to update scripts
+            }
+        } catch (e) {
+            vscode.window.showErrorMessage(e);
+        }
+    }
+}