refactor: add log and clear apis on dispose

vitest-dev · Nov 19, 2024 · 59e2ad6 · 59e2ad6
1 parent d7182da
commit 59e2ad6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/watcher.ts b/src/watcher.ts
@@ -8,7 +8,7 @@ import { log } from './log'
 
 export class ExtensionWatcher extends vscode.Disposable {
   private watcherByFolder = new Map<vscode.WorkspaceFolder, vscode.FileSystemWatcher>()
-  private apisByFolder = new Map<vscode.WorkspaceFolder, VitestFolderAPI[]>()
+  private apisByFolder = new WeakMap<vscode.WorkspaceFolder, VitestFolderAPI[]>()
 
   constructor(private readonly testTree: TestTree) {
     super(() => {
@@ -26,6 +26,7 @@ export class ExtensionWatcher extends vscode.Disposable {
     const folder = api.workspaceFolder
     const apis = this.apisByFolder.get(folder) ?? []
     if (!apis.includes(api)) {
+      log.info('[API] Watching', api.id)
       apis.push(api)
     }
     this.apisByFolder.set(folder, apis)