Refresh QuickPickProvider on user collaboration events

microsoft · Jan 29, 2024 · 718b293 · 718b293
1 parent c87b341
commit 718b293
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/web/client/extension.ts b/src/web/client/extension.ts
@@ -368,6 +368,7 @@ export function createWebWorkerInstance(
                                 data.userId
                             );
                             WebExtensionContext.userCollaborationProvider.refresh();
+                            WebExtensionContext.quickPickProvider.refresh();
                         }
                         if (data.type === Constants.workerEventMessages.UPDATE_CONNECTED_USERS) {
                             WebExtensionContext.updateConnectedUsersInContext(
@@ -377,6 +378,7 @@ export function createWebWorkerInstance(
                                 data.entityId
                             );
                             WebExtensionContext.userCollaborationProvider.refresh();
+                            WebExtensionContext.quickPickProvider.refresh();
                         }
                     };
                 }

diff --git a/src/web/client/webViews/QuickPickProvider.ts b/src/web/client/webViews/QuickPickProvider.ts
@@ -7,6 +7,7 @@ import * as vscode from "vscode";
 import WebExtensionContext from "../WebExtensionContext";
 import { IEntityInfo } from "../common/interfaces";
 import * as Constants from "../common/constants";
+import { getFileEntityId, getFileEntityName } from "../utilities/fileAndEntityUtil";
 
 interface IQuickPickItem extends vscode.QuickPickItem {
     label: string;
@@ -21,6 +22,18 @@ export class QuickPickProvider {
         this.items = new Array<IQuickPickItem>();
     }
 
+    public refresh() {
+        if (vscode.window.activeTextEditor) {
+            const fileFsPath = vscode.window.activeTextEditor.document.fileName;
+            const entityInfo: IEntityInfo = {
+                entityId: getFileEntityId(fileFsPath),
+                entityName: getFileEntityName(fileFsPath),
+                rootWebPageId: WebExtensionContext.entityDataMap.getEntityMap.get(getFileEntityId(fileFsPath))?.rootWebPageId as string
+            };
+            this.updateQuickPickItems(entityInfo);
+        }
+    }
+
     public async updateQuickPickItems(entityInfo: IEntityInfo) {
         const connectedUsersMap = WebExtensionContext.connectedUsers.getUserMap;