Reset metadata watchers when refreshing workspaces

Author: code-asher

src/workspacesProvider.ts

@@ -14,6 +14,7 @@ export enum WorkspaceQuery {
 export class WorkspaceProvider implements vscode.TreeDataProvider<vscode.TreeItem> {
   private workspaces: WorkspaceTreeItem[] = []
   private agentMetadata: Record<WorkspaceAgent["id"], AgentMetadataEvent[]> = {}
+  private watcherDisposers: (() => void)[] = []
 
   constructor(private readonly getWorkspacesQuery: WorkspaceQuery, private readonly storage: Storage) {
     this.fetchAndRefresh()
@@ -22,6 +23,8 @@ export class WorkspaceProvider implements vscode.TreeDataProvider<vscode.TreeIte
   // fetchAndRefrehsh fetches new workspaces then re-renders the entire tree.
   async fetchAndRefresh() {
     const workspacesTreeItem: WorkspaceTreeItem[] = []
+    this.watcherDisposers.forEach((dispose) => dispose())
+    this.watcherDisposers = []
     // If the URL is set then we are logged in.
     if (this.storage.getURL()) {
       const resp = await getWorkspaces({ q: this.getWorkspacesQuery })
@@ -71,6 +74,9 @@ export class WorkspaceProvider implements vscode.TreeDataProvider<vscode.TreeIte
     return Promise.resolve(this.workspaces)
   }
 
+  // monitorMetadata opens a web socket to monitor metadata on the specified
+  // agent and registers a disposer on this.watcherDisposers that can be used to
+  // stop the watch.
   async monitorMetadata(agentId: WorkspaceAgent["id"]): Promise<void> {
     const agentMetadataURL = new URL(`${this.storage.getURL()}/api/v2/workspaceagents/${agentId}/watch-metadata`)
     const agentMetadataEventSource = new EventSource(agentMetadataURL.toString(), {
@@ -97,6 +103,9 @@ export class WorkspaceProvider implements vscode.TreeDataProvider<vscode.TreeIte
         agentMetadataEventSource.close()
       }
     })
+    this.watcherDisposers.push(() => {
+      agentMetadataEventSource.close()
+    })
   }
 }