Config for auto-refresh SELF view

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

package.json

@@ -53,6 +53,11 @@
         "title": "General",
         "order": 1,
         "properties": {
+          "vscode-db2i.autoRefreshSelfCodesView": {
+            "type": "boolean",
+            "description": "Automatically refresh the SELF codes view every 5 seconds",
+            "default": false
+          },
           "vscode-db2i.pageSize": {
             "type": "number",
             "description": "Page size for Schema browser",

src/extension.ts

@@ -18,6 +18,7 @@ import { SQLJobManager } from "./connection/manager";
 import { JDBCOptions } from "./connection/types";
 import { SQLJob } from "./connection/sqlJob";
 import { selfCodesResultsView } from "./views/jobManager/selfCodes/selfCodesResultsView";
+import Configuration from "./configuration";
 
 export interface Db2i {
   sqlJobManager: SQLJobManager,
@@ -36,6 +37,7 @@ export function activate(context: vscode.ExtensionContext): Db2i {
   loadBase();
 
   const exampleBrowser = new ExampleBrowser(context);
+  const selfCodesView = new selfCodesResultsView(context);
 
   context.subscriptions.push(
     ...languageInit(),
@@ -58,7 +60,7 @@ export function activate(context: vscode.ExtensionContext): Db2i {
     ),
     vscode.window.registerTreeDataProvider(
       'vscode-db2i.self.nodes',
-      new selfCodesResultsView(context)
+      selfCodesView
     )
   );
 
@@ -81,6 +83,7 @@ export function activate(context: vscode.ExtensionContext): Db2i {
       determineFeatures();
       // Refresh the examples when we have it, so we only display certain examples
       exampleBrowser.refresh();
+      selfCodesView.setRefreshEnabled(Configuration.get(`autoRefreshSelfCodesView`) || false)
     })
   });
 

src/views/jobManager/selfCodes/selfCodesResultsView.ts

@@ -36,14 +36,10 @@ export class selfCodesResultsView implements TreeDataProvider<any> {
         }
       }),
       vscode.commands.registerCommand(`vscode-db2i.self.enableAutoRefresh`, async () => {
-        this.autoRefresh = true;
-        vscode.commands.executeCommand(`setContext`, `vscode-db2i.self.autoRefresh`, true);
-        vscode.window.showInformationMessage("SELF Code Auto Refresh Enabled");
+        this.setRefreshEnabled(true, true);
       }),
       vscode.commands.registerCommand(`vscode-db2i.self.disableAutoRefresh`, async () => {
-        this.autoRefresh = false;
-        vscode.commands.executeCommand(`setContext`, `vscode-db2i.self.autoRefresh`, false);
-        vscode.window.showInformationMessage("SELF Code Auto Refresh Disabled");
+        this.setRefreshEnabled(false, true);
       })
     );
     setInterval(async () => {
@@ -54,6 +50,15 @@ export class selfCodesResultsView implements TreeDataProvider<any> {
 
   }
 
+  setRefreshEnabled(enabled: boolean, withMessage = false): void {
+    this.autoRefresh = enabled;
+    vscode.commands.executeCommand(`setContext`, `vscode-db2i.self.autoRefresh`, enabled);
+
+    if (withMessage) {
+      vscode.window.showInformationMessage(`SELF Code Auto Refresh ${enabled ? 'Enabled' : 'Disabled'}`);
+    }
+  }
+
   async getSelfCodes(): Promise<SelfCodeNode[]> {
     const selected = JobManager.getRunningJobs();
     if (selected) {
@@ -81,22 +86,27 @@ export class selfCodesResultsView implements TreeDataProvider<any> {
       return [];
     } else {
       const selfCodes = await this.getSelfCodes();
-      return selfCodes.map((error) => {
-        const hitsTxt = error.MATCHES.toString().padStart(10, ' ');
-        const label = `${error.LOGGED_SQLSTATE} (${error.LOGGED_SQLCODE})`;
-        const details = `${error.MESSAGE_TEXT} ${error.MATCHES < 100 ? hitsTxt : '💯'.padStart(10, ' ')} 🔥`;
-        const hoverMessage = new vscode.MarkdownString(
-          `**SQL Statement💻:** ${error.STMTTEXT}\n\n---\n\n**SQL Job🛠️:** ${error.JOB_NAME}\n\n---\n\n**Occurrences🔥:** ${error.MATCHES}\n\n---\n\n**Details✏️:** ${error.MESSAGE_SECOND_LEVEL_TEXT}`
-        );
-        hoverMessage.isTrusted = true;
-        const treeItem = new SelfCodeTreeItem(
-          label,
-          details,
-          hoverMessage,
-          vscode.TreeItemCollapsibleState.None
-        );
-        return treeItem;
-      });
+
+      if (selfCodes) {
+        return selfCodes.map((error) => {
+          const hitsTxt = error.MATCHES.toString().padStart(10, ' ');
+          const label = `${error.LOGGED_SQLSTATE} (${error.LOGGED_SQLCODE})`;
+          const details = `${error.MESSAGE_TEXT} ${error.MATCHES < 100 ? hitsTxt : '💯'.padStart(10, ' ')} 🔥`;
+          const hoverMessage = new vscode.MarkdownString(
+            `**SQL Statement💻:** ${error.STMTTEXT}\n\n---\n\n**SQL Job🛠️:** ${error.JOB_NAME}\n\n---\n\n**Occurrences🔥:** ${error.MATCHES}\n\n---\n\n**Details✏️:** ${error.MESSAGE_SECOND_LEVEL_TEXT}`
+          );
+          hoverMessage.isTrusted = true;
+          const treeItem = new SelfCodeTreeItem(
+            label,
+            details,
+            hoverMessage,
+            vscode.TreeItemCollapsibleState.None
+          );
+          return treeItem;
+        });
+      }
+
+      return [];
     }
   }
   getParent?(element: any) {
@@ -128,7 +138,6 @@ export class SelfCodeTreeItem extends TreeItem {
     public readonly collapsibleState: vscode.TreeItemCollapsibleState
   ) {
     super(errorMessage, collapsibleState);
-    console.log(hoverMessage);
     this.tooltip = hoverMessage; // Hover text
     this.description = details; // Additional details shown in the tree view
   }