Disable treeview caching

Author: shirsho-12

src/commands/showPanel.tsx

@@ -14,7 +14,7 @@ import { Editor } from "../utils/editor";
 import { FRONTEND_ELEMENT_ID } from "../constants";
 import { client, SOURCE_ACADEMY_ICON_URI } from "../extension";
 import _ from "lodash";
-import { treeDataProvider } from "../treeview";
+import { enableAssessmentsTreeData } from "../treeview";
 
 let panel: vscode.WebviewPanel | null = null;
 // This needs to be a reference to active
@@ -98,7 +98,7 @@ async function handleMessage(
         const { assessmentOverviews, courseId } = message;
         context.globalState.update("assessmentOverviews", assessmentOverviews);
         context.globalState.update("courseId", courseId);
-        treeDataProvider.refresh();
+        enableAssessmentsTreeData();
         break;
     }
     console.log(`${Date.now()} Finish handleMessage: ${message.type}`);

src/treeview/index.ts

@@ -3,6 +3,9 @@ import { VscAssessmentOverview } from "../utils/messages";
 import { SOURCE_ACADEMY_ICON_URI } from "../extension";
 
 export let treeDataProvider: AssessmentsSidebarProvider;
+export function enableAssessmentsTreeData() {
+  treeDataProvider?.enableData();
+}
 
 // This will be a source of bug on first extension loads!!
 let courseId: number;
@@ -19,7 +22,12 @@ export function setupTreeView(context: vscode.ExtensionContext) {
 export class AssessmentsSidebarProvider
   implements vscode.TreeDataProvider<BaseTreeItem>
 {
-  constructor(private context: vscode.ExtensionContext) {}
+  constructor(private context: vscode.ExtensionContext, private dataEnabled: boolean = false) {}
+
+  enableData() {
+    this.dataEnabled = true;
+    this.refresh();
+  }
 
   private _onDidChangeTreeData: vscode.EventEmitter<
     BaseTreeItem | undefined | null | void
@@ -33,6 +41,9 @@ export class AssessmentsSidebarProvider
   }
 
   getChildren(element?: BaseTreeItem): Thenable<BaseTreeItem[]> {
+    if (!this.dataEnabled) {
+      return Promise.resolve([]);
+    }
     // @ts-ignore
     const assessmentOverviews: VscAssessmentOverview[] =
       this.context.globalState.get("assessmentOverviews");