improve checks for current job and current schema

codefori · Jan 14, 2025 · 5ba9b37 · 5ba9b37
1 parent 14846f7
commit 5ba9b37
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 6 deletions.
diff --git a/src/aiProviders/continue/listTablesContextProvider.ts b/src/aiProviders/continue/listTablesContextProvider.ts
@@ -112,7 +112,10 @@ class ListDb2iTables implements IContextProvider {
   }
 }
 
-export async function registerDb2iTablesProvider(schema: string) {
+export async function registerDb2iTablesProvider(schema?: string) {
+  if (!schema) {
+    return;
+  }
   const continueID = `Continue.continue`;
   const continueEx = vscode.extensions.getExtension(continueID);
   if (continueEx) {

diff --git a/src/extension.ts b/src/extension.ts
@@ -102,7 +102,9 @@ export function activate(context: vscode.ExtensionContext): Db2i {
       exampleBrowser.refresh();
       selfCodesView.setRefreshEnabled(Configuration.get(`jobSelfViewAutoRefresh`) || false);
       // register list tables
-      registerDb2iTablesProvider(JobManager.getSelection().job.options.libraries[0]);
+      const currentJob = JobManager.getSelection();
+      const currentSchema = currentJob?.job.options.libraries[0];
+      registerDb2iTablesProvider(currentSchema);
       if (devMode && runTests) {
         runTests();
       }

diff --git a/src/views/jobManager/jobManagerView.ts b/src/views/jobManager/jobManagerView.ts
@@ -305,15 +305,28 @@ export class JobManagerView implements TreeDataProvider<any> {
 
     const selectedJob = JobManager.getSelection();
     const selectedSchema = selectedJob.job.options.libraries[0];
-    
+
     // re-register db2i tables context provider with current schema
-    if (provider && provider.getCurrentSchema().toLowerCase() !== selectedSchema.toLowerCase()) {
+    if (
+      provider &&
+      selectedJob !== undefined &&
+      provider.getCurrentSchema() &&
+      selectedSchema &&
+      provider.getCurrentSchema().trim().toLowerCase() !==
+        selectedSchema.trim().toLowerCase()
+    ) {
       registerDb2iTablesProvider(selectedSchema);
     }
 
-    setCancelButtonVisibility(selectedJob && selectedJob.job.getStatus() === "busy");
+    setCancelButtonVisibility(
+      selectedJob && selectedJob.job.getStatus() === "busy"
+    );
     sqlLanguageStatus.setState(selectedJob !== undefined);
-    commands.executeCommand(`setContext`, `vscode-db2i:jobManager.hasJob`, selectedJob !== undefined);
+    commands.executeCommand(
+      `setContext`,
+      `vscode-db2i:jobManager.hasJob`,
+      selectedJob !== undefined
+    );
   }
 
   getTreeItem(element: vscode.TreeItem) {