Add service info correctly

Author: worksofliam

src/config.ts

@@ -6,6 +6,7 @@ import { ServerComponent, UpdateStatus } from "./connection/serverComponent";
 import { JobManagerView } from "./views/jobManager/jobManagerView";
 import Configuration from "./configuration";
 import { ConfigManager } from "./views/jobManager/ConfigManager";
+import { Examples, ServiceInfoLabel } from "./views/examples";
 
 interface IBMiLevels {
   version: number;
@@ -67,6 +68,9 @@ export function setupConfig(context: ExtensionContext) {
   getInstance().onEvent(`disconnected`, async () => {
     JobManagerView.setVisible(false);
     await JobManager.endAll();
+
+    // Remove old service examples
+    delete Examples[ServiceInfoLabel];
   });
 }
 

src/database/serviceInfo.ts

@@ -0,0 +1,17 @@
+import { SQLExample } from "../views/examples";
+import { JobManager } from "../config";
+import Statement from "./statement";
+
+export async function getServiceInfo(): Promise<SQLExample[]|undefined> {
+  // The reason we check for a selection is because we don't want it to prompt the user to start one here
+  if (JobManager.getSelection()) {
+    const resultSet = await JobManager.runSQL<{SERVICE_NAME: string, EXAMPLE: string}>(`select SERVICE_NAME, EXAMPLE from qsys2.services_info`);
+
+    return resultSet.map(r => ({
+      name: Statement.prettyName(r.SERVICE_NAME),
+      content: [r.EXAMPLE],
+    }))
+  } else {
+    return undefined;
+  }
+}

src/views/examples/exampleBrowser.ts

@@ -1,9 +1,10 @@
 import { EventEmitter, MarkdownString, workspace } from "vscode";
 import { window } from "vscode";
 import { CancellationToken, Event, ExtensionContext, ProviderResult, ThemeIcon, TreeDataProvider, TreeItem, TreeItemCollapsibleState, commands } from "vscode";
-import { SQLExample, Examples } from ".";
+import { SQLExample, Examples, ServiceInfoLabel } from ".";
 import { OSData, fetchSystemInfo } from "../../config";
 import { getInstance } from "../../base";
+import { getServiceInfo } from "../../database/serviceInfo";
 
 const openExampleCommand = `vscode-db2i.examples.open`;
 
@@ -59,7 +60,14 @@ export class ExampleBrowser implements TreeDataProvider<any> {
     return element;
   }
 
-  getChildren(element?: ExampleGroupItem): ProviderResult<any[]> {
+  async getChildren(element?: ExampleGroupItem): Promise<any[]> {
+    if (!Examples[ServiceInfoLabel]) {
+      getServiceInfo().then(serviceExamples => {
+        Examples[ServiceInfoLabel] = serviceExamples;
+        this.refresh();
+      })
+    }
+
     if (this.currentFilter) {
       // If there is a filter, then show all examples that include this criteria
       let items: SQLExampleItem[] = [];

src/views/examples/index.ts

@@ -13,7 +13,9 @@ export interface SQLExample {
   requirements?: ExampleSystemRequirements;
 };
 
-export const Examples: SQLExamplesList = {
+export const ServiceInfoLabel = `IBM i (SQL) Services`;
+
+export let Examples: SQLExamplesList = {
   "Data Definition Language (DDL)": [
     {
       "name": "Create Schema",