Change views as features change

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

Author: worksofliam

src/IBMiDetail.ts

@@ -1,3 +1,4 @@
+import { commands } from "vscode";
 import { getInstance } from "./base";
 import { ServerComponent } from "./connection/serverComponent";
 
@@ -19,6 +20,7 @@ export class IBMiDetail {
 
   setFeatureSupport(featureId: Db2FeatureIds, supported: boolean) {
     this.features[featureId] = supported;
+    commands.executeCommand(`setContext`, `vscode-db2i:${featureId}Supported`, supported);
   }
 
   getVersion() {
@@ -30,6 +32,12 @@ export class IBMiDetail {
   }
 
   async fetchSystemInfo() {
+    // Disable all features
+    const features = Object.keys(featureRequirements) as Db2FeatureIds[];
+    for (const featureId of features) {
+      this.setFeatureSupport(featureId, false);
+    }
+
     const instance = getInstance();
     const content = instance.getContent();
 
@@ -52,7 +60,6 @@ export class IBMiDetail {
       levelCheckFailed = true;
     }
 
-    const features = Object.keys(featureRequirements) as Db2FeatureIds[];
     for (const featureId of features) {
       const requiredLevelForFeature = featureRequirements[featureId][String(this.version)];
       const supported = requiredLevelForFeature && this.db2Level >= requiredLevelForFeature;

src/config.ts

@@ -35,7 +35,6 @@ export async function onConnectOrServerInstall(): Promise<boolean> {
   await ServerComponent.checkForUpdate();
 
   updateStatusBar();
-  toggleViews();
 
   if (ServerComponent.isInstalled()) {
     JobManagerView.setVisible(true);
@@ -83,15 +82,6 @@ export function initConfig(context: ExtensionContext) {
   });
 }
 
-export function toggleViews() {
-  const features = osDetail.getFeatures();
-
-  const featureIds = Object.keys(features) as (keyof typeof features)[];
-  for (const featureId of featureIds) {
-    commands.executeCommand(`setContext`, `vscode-db2i:${featureId}Supported`, features[featureId]);
-  }
-}
-
 export async function askAboutNewJob(startup?: boolean): Promise<boolean> {
   const instance = getInstance();
   const connection = instance.getConnection();

src/views/jobManager/selfCodes/selfCodesResultsView.ts

@@ -9,7 +9,7 @@ import {
   Uri,
   Disposable
 } from "vscode";
-import { JobManager } from "../../../config";
+import { JobManager, osDetail } from "../../../config";
 import { SelfCodeNode, SelfIleStackFrame } from "./nodes";
 import { openExampleCommand } from "../../examples/exampleBrowser";
 import { SQLExample } from "../../examples";
@@ -94,15 +94,21 @@ export class selfCodesResultsView implements TreeDataProvider<any> {
                     where user_name = current_user
                     order by logged_time desc`;
 
-    const result = await selected.job.query<SelfCodeNode>(content).run(10000);
-    if (result.success) {
-      const data: SelfCodeNode[] = result.data.map((row) => ({
-        ...row,
-        INITIAL_STACK: JSON.parse(row.INITIAL_STACK as unknown as string)
-      }));
+    try {
+      const result = await selected.job.query<SelfCodeNode>(content).run(10000);
+      if (result.success) {
+        const data: SelfCodeNode[] = result.data.map((row) => ({
+          ...row,
+          INITIAL_STACK: JSON.parse(row.INITIAL_STACK as unknown as string)
+        }));
 
-    
-      return data;
+      
+        return data;
+      }
+    } catch (e) {
+      this.setRefreshEnabled(false);
+      osDetail.setFeatureSupport(`SELF`, false);
+      vscode.window.showErrorMessage(`An error occured fetching SELF code errors, and therefore will be disabled.`);
     }
   }