Reset count on refresh

Author: tanchekwei

WorkspaceLauncherForVSCode/Classes/CountTracker.cs

@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Linq;
 using WorkspaceLauncherForVSCode.Enums;
 
 namespace WorkspaceLauncherForVSCode.Classes;
@@ -35,4 +36,22 @@ public void Update(WorkspaceType localType, int value)
       _vscodeLocalCounts[localType] = value;
     }
   }
+
+  public void Reset()
+  {
+    foreach (var key in _counts.Keys.ToList())
+    {
+      _counts[key] = 0;
+    }
+
+    foreach (var key in _vscodeRemoteCounts.Keys.ToList())
+    {
+      _vscodeRemoteCounts[key] = 0;
+    }
+
+    foreach (var key in _vscodeLocalCounts.Keys.ToList())
+    {
+      _vscodeLocalCounts[key] = 0;
+    }
+  }
 }

WorkspaceLauncherForVSCode/Pages/VisualStudioCodePage.cs

@@ -186,6 +186,8 @@ private async Task RefreshWorkspacesAsync(bool isUserInitiated)
 
             IsLoading = true;
 
+            _countTracker.Reset();
+
             if (isUserInitiated)
             {
                 _vscodeService.LoadInstances(_settingsManager.EnabledEditions);